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

diff --git a/pv/storesession.cpp b/pv/storesession.cpp
index 058b46a9..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>
@@ -53,7 +53,7 @@ namespace pv {
 const size_t StoreSession::BlockSize = 1024 * 1024;
 
 StoreSession::StoreSession(const std::string &file_name,
-	const SigSession &session) :
+	const Session &session) :
 	file_name_(file_name),
 	session_(session),
 	interrupt_(false),
@@ -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();