fictrac: also log version

murthylab · May 14, 2021 · f7ee05c · f7ee05c
1 parent bc27940
commit f7ee05c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flyvr/fictrac/fictrac_driver.py b/flyvr/fictrac/fictrac_driver.py
@@ -14,6 +14,9 @@
     SHMEMFicTracState, fictrac_state_to_vec, NUM_FICTRAC_FIELDS
 
 
+H5_DATA_VERSION = 1
+
+
 class FicTracDriver(object):
     """
     This class drives the tracking of the fly via a separate software called FicTrac. It invokes this process and
@@ -88,6 +91,9 @@ def run(self, options):
         flyvr_shared_state.logger.create("/fictrac/output", shape=[2048, NUM_FICTRAC_FIELDS],
                                          maxshape=[None, NUM_FICTRAC_FIELDS], dtype=np.float64,
                                          chunks=(2048, NUM_FICTRAC_FIELDS))
+        flyvr_shared_state.logger.log("/fictrac/output",
+                                      H5_DATA_VERSION,
+                                      attribute_name='__version')
 
         # Start FicTrac
         with open(self.console_output_file, "wb") as out: