Merge pull request #1733 from danielparthier/WinWCP_rec_datetime

WinWcpRawIO: Add rec_datetime to header dict and block annotation from file

Author: zm711

neo/rawio/winwcprawio.py

@@ -9,6 +9,7 @@
 """
 
 import struct
+import datetime
 
 import numpy as np
 
@@ -61,6 +62,7 @@ def _parse_header(self):
                     continue
                 key, val = line.split("=")
                 if key in [
+                    "VER",
                     "NC",
                     "NR",
                     "NBH",
@@ -80,6 +82,12 @@ def _parse_header(self):
                 header[key] = val
 
             nb_segment = header["NR"]
+
+            # get rec_datetime when WCP data file version is later than 8
+            if header["VER"] > 8:
+                rec_datetime = datetime.datetime.strptime(header["RTIME"], "%d/%m/%Y %H:%M:%S")
+            else:
+                rec_datetime = None
             all_sampling_interval = []
             # loop for record number
             for seg_index in range(header["NR"]):
@@ -167,7 +175,8 @@ def _parse_header(self):
 
         # insert some annotation at some place
         self._generate_minimal_annotations()
-
+        bl_annotations = self.raw_annotations["blocks"][0]
+        bl_annotations["rec_datetime"] = rec_datetime
     def _segment_t_start(self, block_index, seg_index):
         return 0.0
 

neo/test/iotest/test_winwcpio.py

@@ -14,7 +14,10 @@ class TestRawBinarySignalIO(
 ):
     ioclass = WinWcpIO
     entities_to_download = ["winwcp"]
-    entities_to_test = ["winwcp/File_winwcp_1.wcp"]
+    entities_to_test = [
+        "winwcp/File_winwcp_1.wcp",
+        "winwcp/file_with_recording_time/File_winwcp_2.wcp",
+    ]
 
 
 if __name__ == "__main__":

neo/test/rawiotest/test_winwcprawio.py

@@ -10,7 +10,10 @@ class TestWinWcpRawIO(
 ):
     rawioclass = WinWcpRawIO
     entities_to_download = ["winwcp"]
-    entities_to_test = ["winwcp/File_winwcp_1.wcp"]
+    entities_to_test = [
+        "winwcp/File_winwcp_1.wcp",
+        "winwcp/file_with_recording_time/File_winwcp_2.wcp",
+    ]
 
 
 if __name__ == "__main__":