Once records have been collected set the flag to terminate

misko · Mar 8, 2024 · 9a0b281 · 9a0b281
1 parent 53bcb2d
commit 9a0b281
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spf/data_collector.py b/spf/data_collector.py
@@ -156,6 +156,7 @@ def __init__(self, yaml_config, filename_npy, position_controller, tag=""):
         self.filename_npy = filename_npy
         self.record_matrix = None
         self.position_controller = position_controller
+        self.finished_collecting = False
 
     def radios_to_online(self):
         # record matrix
@@ -281,7 +282,7 @@ def run_collector_thread(self):
         raise NotImplementedError
 
     def is_collecting(self):
-        return True
+        return not self.finished_collecting
 
 
 class DroneDataCollector(DataCollector):
@@ -310,6 +311,8 @@ def run_collector_thread(self):
 
                 read_thread.read_lock.release()
 
+        self.finished_collecting = True
+
 
 class GrblDataCollector(DataCollector):
     def __init__(self, *args, **kwargs):
@@ -339,3 +342,4 @@ def run_collector_thread(self):
                 )
 
                 read_thread.read_lock.release()
+        self.finished_collecting = True