chore: qtm multicam csv processing

Mouwrice · Dec 14, 2023 · cedd9ca · cedd9ca
1 parent c8df2ec
commit cedd9ca
Show file tree

Hide file tree

Showing 6 changed files with 222,529 additions and 5 deletions.
diff --git a/data/multicam_asil_01/qtm_multicam_asil_01.csv b/data/multicam_asil_01/qtm_multicam_asil_01.csv
diff --git a/data/multicam_asil_02/qtm_multicam_asil_02.csv b/data/multicam_asil_02/qtm_multicam_asil_02.csv
diff --git a/data/multicam_asil_03/qtm_multicam_asil_03.csv b/data/multicam_asil_03/qtm_multicam_asil_03.csv
diff --git a/data/multicam_ms_01/qtm_multicam_ms_01.csv b/data/multicam_ms_01/qtm_multicam_ms_01.csv
diff --git a/data/multicam_ms_02/qtm_multicam_ms_02.csv b/data/multicam_ms_02/qtm_multicam_ms_02.csv
diff --git a/src/csv_objects/qtm_writer.py b/src/csv_objects/qtm_writer.py
@@ -1,5 +1,3 @@
-import time
-
 import qtm_rt
 
 from csv_objects.csv_object import CSVWriter
@@ -10,14 +8,15 @@ class QTMWriter:
     Class to write the captured data to a CSV file.
     """
 
-    def __init__(self, path: str):
+    def __init__(self, path: str, time_per_frame: float = 0.01):
         self.csv_writer = CSVWriter(path)
+        self.time_per_frame = time_per_frame
 
     def on_packet(self, packet: qtm_rt.QRTPacket):
         """ Callback function that is called everytime a data packet arrives from QTM """
         _, markers = packet.get_3d_markers()
         # packets at 100 Hz so time is frame number * 10 ms
-        packet_time = packet.framenumber * 10
+        packet_time = packet.framenumber * self.time_per_frame
         if self.csv_writer is not None:
             for i, marker in enumerate(markers):
                 self.csv_writer.write(packet.framenumber, packet_time, i, marker.x, marker.y, marker.z)
@@ -28,7 +27,7 @@ async def main():
     if connection is None:
         return
 
-    writer = QTMWriter("qtm_multicam_7.csv_objects")
+    writer = QTMWriter("qtm_multicam_ms_02.csv", time_per_frame=1 / 120)
 
     await connection.stream_frames(components=["3d"], on_packet=writer.on_packet)