From 2e9568da8d18940c61f50e285ce4432107c43726 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 8 Apr 2020 15:04:33 +0200
Subject: [PATCH] Revert "Add counter for frames to save instead of just flag"

This reverts commit 16612a2d
---
 core-writer/src/module/ZmqRecvModule.cpp | 18 +++++-------------
 1 file changed, 5 insertions(+), 13 deletions(-)

diff --git a/core-writer/src/module/ZmqRecvModule.cpp b/core-writer/src/module/ZmqRecvModule.cpp
index 9f15287..600dcb1 100644
--- a/core-writer/src/module/ZmqRecvModule.cpp
+++ b/core-writer/src/module/ZmqRecvModule.cpp
@@ -84,18 +84,17 @@ void ZmqRecvModule::stop_recv()
     receiving_threads_.clear();
 }
 
-void ZmqRecvModule::start_saving(const int n_frames_to_save)
+void ZmqRecvModule::start_saving()
 {
     #ifdef DEBUG_OUTPUT
         using namespace date;
         using namespace chrono;
         cout << "[" << system_clock::now() << "]";
         cout << "[ZmqRecvModule::start_saving]";
-        cout << " n_frames_to_save " << n_frames_to_save;
-        cout << endl;
+        cout << " Enable saving." << endl;
     #endif
 
-    n_frames_to_save_ = n_frames_to_save;
+    is_saving_ = true;
 }
 
 void ZmqRecvModule::stop_saving()
@@ -108,12 +107,7 @@ void ZmqRecvModule::stop_saving()
         cout << " Disable saving." << endl;
     #endif
 
-    n_frames_to_save_ = 0;
-}
-
-bool ZmqRecvModule::is_saving()
-{
-    return n_frames_to_save_ != 0;
+    is_saving_ = false;
 }
 
 void ZmqRecvModule::receive_thread(const string& connect_address)
@@ -134,12 +128,10 @@ void ZmqRecvModule::receive_thread(const string& connect_address)
                 continue;
             }
 
-            if (!is_saving()) {
+            if (!is_saving_.load(memory_order_relaxed)) {
                 continue;
             }
 
-            n_frames_to_save_--;
-
             auto frame_metadata = frame.first;
             auto frame_data = frame.second;