unlogger.py: fix YUV replay (#22299)

4 years ago · 1e494a5224
parent c01e622e82
commit 1e494a5224
1 changed files with 4 additions and 1 deletions
--- a/tools/replay/unlogger.py
+++ b/tools/replay/unlogger.py
@ -173,7 +173,10 @@ def _get_address_send_func(address):
  return sock.send

 def _get_vipc_server(length):
-  w, h = {3 * w * h: (w, h) for (w, h) in [tici_f_frame_size, eon_f_frame_size]}[length]
+  sizes = {3 * w * h: (w, h) for (w, h) in [tici_f_frame_size, eon_f_frame_size]} # RGB
+  sizes.update({(3 * w * h) / 2: (w, h) for (w, h) in [tici_f_frame_size, eon_f_frame_size]}) # YUV
+
+  w, h = sizes[length]

  vipc_server = VisionIpcServer("camerad")
  vipc_server.create_buffers(VisionStreamType.VISION_STREAM_RGB_BACK, 4, True, w, h)