fix: correctly receive all chunks in recorder

src/recorder.py

@@ -28,6 +28,7 @@ class RecorderClient(QObject):
             socket.AF_INET, socket.SOCK_STREAM)
         self.timer: Optional[QTimer] = None
         self.connected: bool = False
+        self.buffer: bytes = b""
 
     @pyqtSlot()
     def start(self):
@@ -40,26 +41,26 @@ class RecorderClient(QObject):
         print("Connected to server")
 
     def poll_socket(self):
-        buffer: bytes = b""
         if not self.connected:
             return
 
         try:
+            while True:
                 chunk: bytes = self.socket.recv(self.DATA_CHUNK_SIZE)
                 if not chunk:
                     return
-            buffer += chunk
+                self.buffer += chunk
 
                 while True:
-                if len(buffer) < 4:
+                    if len(self.buffer) < 4:
                         break
-                msg_len: int = struct.unpack(">I", buffer[:4])[0]
+                    msg_len: int = struct.unpack(">I", self.buffer[:4])[0]
                     msg_end: int = 4 + msg_len
-                if len(buffer) < msg_end:
+                    if len(self.buffer) < msg_end:
                         break
 
-                message: bytes = buffer[4:msg_end]
+                    message: bytes = self.buffer[4:msg_end]
-                buffer = buffer[msg_end:]
+                    self.buffer = self.buffer[msg_end:]
                     self.on_message(message)
         except BlockingIOError:
             pass

src/snapshot.py

@@ -63,7 +63,7 @@ class Snapshot:
         (nbr_raycasts,), data = iter_unpack(">B", data)
         raycast_distances, data = iter_unpack(f">{nbr_raycasts}f", data)
 
-        (h, w), data = iter_unpack(">ii", data)
+        (h, w), data = iter_unpack(">II", data)
 
         if h * w > 0:
             image = np.frombuffer(data, np.uint8).reshape(h, w, 3)