Serial Reader half complete

Author: 5herlocked

SerialReader.py

@@ -1,2 +1,51 @@
-import Frame
-import SerialInterface
+# Serial Reader Threaded class
+#
+#
+#
+import struct
+from multiprocessing import Queue
+from threading import Thread
+from Frame import ShortRangeRadarFrameHeader, Frame, FrameError
+from SerialInterface import SerialInterface
+
+
+class SerialReader:
+    def __init__(self, file, control, data):
+        self.interface = SerialInterface(control, data, frame_type=ShortRangeRadarFrameHeader)
+        self.file = file
+
+        self.running = False
+
+        self.received_queue = Queue()
+
+        self.receiving_process = Thread(target=self.process_receiving)
+        self.storing_process = Thread(target=self.process_storing)
+
+    def process_receiving(self):
+        while self.running:
+            frame = self.interface.recv_item(self.interface.data_rx_queue)
+
+            if frame:
+                try:
+                    frame = Frame(frame, frame_type=self.interface.frame_type)
+                    pass
+                except(KeyError, struct.error, IndexError, FrameError, OverflowError) as e:
+                    print('Exception occurred: ', e)
+                    print('Skipping frame')
+
+    def process_storing(self):
+        while self.running:
+            pass
+
+    def send_profile(self, profile_file):
+        with open(profile_file) as f:
+            for line in f.readlines():
+                if line.startswith('%'):
+                    continue
+                else:
+                    self.interface.send_item(self.interface.control_tx_queue, line)
+
+    def stop(self):
+        self.running = False
+
+        self.receiving_process.join()

UartSaver.py

@@ -145,11 +145,10 @@ def process_frame():
                 # Some data got in the wrong place, just skip the frame
                 print('Exception occurred: ', e)
                 print("Skipping frame due to error...")
+    # Sleep to allow other threads to run
+    time.sleep(0.001)
 
 
-# Sleep to allow other threads to run
-time.sleep(0.001)
-
 if __name__ == "__main__":
     import argparse
 

collect_data.py

@@ -21,7 +21,44 @@
 #           save_frame()
 
 
-import threading
+from threading import Thread
 import SerialReader
 import CameraReader
 
+
+def open_files():
+    pass
+
+
+def watch_camera():
+    with open('' + args.output_name + '.mp4', mode='wb+') as vid:
+        # Create an instance of CameraReader and pass the file to it
+        pass
+
+    pass
+
+
+def watch_serial():
+    with open('' + args.output_name + '.csv', mode='w+') as csv:
+        # Create an instance of SerialReader and pass the file to it
+        pass
+    pass
+
+
+async def watch():
+    camera_thread = Thread(target=watch_camera)
+    serial_thread = Thread(target=watch_serial)
+
+
+if __name__ == '__main__':
+    import argparse
+
+    parser = argparse.ArgumentParser(description='Run the data collection pipeline')
+    parser.add_help = True
+    parser.add_argument('--control_port', help='COM port for configuration, control')
+    parser.add_argument('--data_port', help='COM port for radar data transfer')
+    parser.add_argument('--output_name', help='Name of the output collection of files')
+    parser.add_argument('--prof', help='Radar Chirp profile to use')
+    args = parser.parse_args()
+
+    await watch()

requirements.txt

@@ -1,4 +1,6 @@
 matplotlib~=3.4.3
 numpy~=1.21.4
 pyserial~=3.5
-plotly~=5.6.0
+plotly~=5.6.0
+opencv-python~=4.5.5
+aiofiles~=0.8.0