StoreSession: Fixed completion mechanism

[pulseview.git] / pv / storesession.cpp

diff --git a/pv/storesession.cpp b/pv/storesession.cpp
index 058b46a99ee8d1d5b6434cd33e6da93c91d080fd..221804c40baaf07721f7c229704f5cae072f1ca4 100644 (file)
--- a/pv/storesession.cpp
+++ b/pv/storesession.cpp
@@ -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),
@@ -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();