minor tweaks

src/navigate/model/data_sources/bdv_data_source.py

@@ -189,7 +189,7 @@ def write(self, data: npt.ArrayLike, **kw) -> None:
                     if e.errno == 28:
                         logger.error("No disk space left on device. Closing the file.")
                         self.close()
-                        raise "No disk space left on device."
+                        raise Exception("No disk space left on device.")
 
         self._current_frame += 1
 

src/navigate/model/features/image_writer.py

@@ -153,6 +153,9 @@ def __init__(
         #: int: Minimum disk space required in bytes.
         self.min_disk_space = 10 * 1024 * 1024 * 1024 # 10 GB
 
+        #: float: Time of last disk space check
+        self.last_disk_space_check = 0
+
         # initialize saving
         self.initialize_saving(sub_dir, image_name)
 
@@ -165,12 +168,14 @@ def save_image(self, frame_ids):
             Index into self.model.data_buffer.
         """
 
-        last_disk_space_check = time.time()
-
         for idx in frame_ids:
+            if (idx < 0) or (idx > (self.number_of_frames - 1)):
+                msg = f"Received invalid index {idx}. Skipping this frame."
+                logger.debug(f"Received invalid index: {msg}.")
+                continue
 
             # Check disk space at regular intervals to prevent running out of space
-            if time.time() - last_disk_space_check > self.disk_space_check_interval:
+            if time.time() - self.last_disk_space_check > self.disk_space_check_interval:
                 _, _, free = shutil.disk_usage(self.save_directory)
                 logger.info(f"Free Disk Space: {free / 1024 / 1024 / 1024} GB")
                 if free < self.min_disk_space:
@@ -181,12 +186,7 @@ def save_image(self, frame_ids):
                         ("warning", "Insufficient Disk Space. Acquisition Terminated")
                     )
                     return
-                last_disk_space_check = time.time()
-
-            if (idx < 0) or (idx > (self.number_of_frames - 1)):
-                msg = f"Received invalid index {idx}. Skipping this frame."
-                logger.debug(f"Received invalid index: {msg}.")
-                continue
+                self.last_disk_space_check = time.time()
 
             # check the saving flag
             if self.saving_flags:
@@ -308,6 +308,7 @@ def calculate_and_check_disk_space(self):
 
         Assumes 16-bit image type, without compression."""
 
+        self.last_disk_space_check = time.time()
         # Return disk usage statistics in bytes
         _, _, free = shutil.disk_usage(self.save_directory)
         logger.info(f"Free Disk Space: {free}")