X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=pv%2Fsigsession.cpp;h=5ab30ab1299e2e7bb81475053049eb46f189c3e1;hb=84b1b7d28acc52214940ac94a84943cf9e9685fa;hp=354082c7262c351a3fe09423761823e7beab302d;hpb=5ac961e325a9d2cbafdd8fae3a6704d7348cf19a;p=pulseview.git

diff --git a/pv/sigsession.cpp b/pv/sigsession.cpp
index 354082c7..5ab30ab1 100644
--- a/pv/sigsession.cpp
+++ b/pv/sigsession.cpp
@@ -20,6 +20,7 @@
 
 #include "sigsession.h"
 
+#include "devicemanager.h"
 #include "data/analog.h"
 #include "data/analogsnapshot.h"
 #include "data/logic.h"
@@ -39,7 +40,8 @@ namespace pv {
 // TODO: This should not be necessary
 SigSession* SigSession::_session = NULL;
 
-SigSession::SigSession() :
+SigSession::SigSession(DeviceManager &device_manager) :
+	_device_manager(device_manager),
 	_sdi(NULL),
 	_capture_state(Stopped)
 {
@@ -59,11 +61,28 @@ SigSession::~SigSession()
 	_session = NULL;
 }
 
+struct sr_dev_inst* SigSession::get_device() const
+{
+	return _sdi;
+}
+
 void SigSession::set_device(struct sr_dev_inst *sdi)
 {
+	if (_sdi)
+		_device_manager.release_device(_sdi);
+	if (sdi)
+		_device_manager.use_device(sdi, this);
 	_sdi = sdi;
 }
 
+void SigSession::release_device(struct sr_dev_inst *sdi)
+{
+	(void)sdi;
+
+	assert(_capture_state == Stopped);
+	_sdi = NULL;
+}
+
 void SigSession::load_file(const string &name,
 	function<void (const QString)> error_handler)
 {
@@ -162,6 +181,10 @@ void SigSession::load_thread_proc(const string name,
 	sr_session_destroy();
 
 	set_capture_state(Stopped);
+
+	// Confirm that SR_DF_END was received
+	assert(!_cur_logic_snapshot);
+	assert(!_cur_analog_snapshot);
 }
 
 void SigSession::sample_thread_proc(struct sr_dev_inst *sdi,
@@ -200,6 +223,10 @@ void SigSession::sample_thread_proc(struct sr_dev_inst *sdi,
 	sr_session_destroy();
 
 	set_capture_state(Stopped);
+
+	// Confirm that SR_DF_END was received
+	assert(!_cur_logic_snapshot);
+	assert(!_cur_analog_snapshot);
 }
 
 void SigSession::feed_in_header(const sr_dev_inst *sdi)