X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fstoresession.cpp;h=46dddcd64411fb85ac998c19d6b493977c52df5d;hp=f6fef4635c630fadcb772c14e1013b96c2d61a17;hb=1df18f6b6d20abd3f1ede94325b6781bf5038ea9;hpb=2b81ae4682ade4109ffa442794de36ceb32045eb

diff --git a/pv/storesession.cpp b/pv/storesession.cpp
index f6fef463..46dddcd6 100644
--- a/pv/storesession.cpp
+++ b/pv/storesession.cpp
@@ -22,7 +22,7 @@
 
 #include "storesession.hpp"
 
-#include <pv/sigsession.hpp>
+#include <pv/session.hpp>
 #include <pv/data/logic.hpp>
 #include <pv/data/logicsnapshot.hpp>
 #include <pv/view/signal.hpp>
@@ -108,7 +108,7 @@ bool StoreSession::start()
 
 	// Get the snapshot
 	const deque< shared_ptr<data::LogicSnapshot> > &snapshots =
-		data->get_snapshots();
+		data->logic_snapshots();
 
 	if (snapshots.empty()) {
 		error_ = tr("No snapshots to save.");
@@ -198,6 +198,9 @@ void StoreSession::store_proc(shared_ptr<data::LogicSnapshot> snapshot)
 		units_stored_ = start_sample >> progress_scale;
 	}
 
+	// Zeroing the progress variables indicates completion
+	units_stored_ = unit_count_ = 0;
+
 	progress_updated();
 
 	output_.reset();