logs

Author: adeebshihadeh

tools/replay/consoleui.py

@@ -88,6 +88,13 @@ def __init__(self, replay: Replay):
     # Set up signal handler for clean exit
     signal.signal(signal.SIGINT, lambda s, f: setattr(self, '_exit', True))
 
+    # Set up log callback
+    self.replay._seg_mgr.set_log_callback(self._on_log)
+
+  def _on_log(self, level: int, msg: str) -> None:
+    with self._lock:
+      self._logs.append((level, msg))
+
   def _init_curses(self, stdscr) -> None:
     self._stdscr = stdscr
     curses.curs_set(0)

tools/replay/seg_mgr.py

@@ -69,6 +69,7 @@ def __init__(self, route_name: str, flags: int = 0, data_dir: str = "", auto_sou
 
     self.segment_cache_limit = MIN_SEGMENTS_CACHE
     self._on_segment_merged_callback: Optional[Callable[[], None]] = None
+    self._log_callback: Optional[Callable[[int, str], None]] = None  # (level, msg)
 
   def __del__(self):
     if hasattr(self, '_cv'):
@@ -115,6 +116,13 @@ def set_current_segment(self, seg_num: int) -> None:
   def set_callback(self, callback: Callable[[], None]) -> None:
     self._on_segment_merged_callback = callback
 
+  def set_log_callback(self, callback: Callable[[int, str], None]) -> None:
+    self._log_callback = callback
+
+  def _log(self, level: int, msg: str) -> None:
+    if self._log_callback:
+      self._log_callback(level, msg)
+
   def set_filters(self, filters: list[bool]) -> None:
     self._filters = filters
 
@@ -180,12 +188,14 @@ def _load_segments_in_range(self, seg_nums: list[int], cur_seg_num: int) -> bool
           continue
 
       # Load segment (blocking - downloads and parses)
+      self._log(0, f"loading segment {seg_num}...")
       seg_data = self._load_segment(seg_num)
       with self._cv:
         self._segments[seg_num] = seg_data
         self._needs_update = True
         self._cv.notify_all()
       loaded_any = True
+      self._log(0, f"segment {seg_num} loaded with {len(seg_data.events)} events")
 
       # Only load one segment at a time to be responsive
       return loaded_any
@@ -279,4 +289,5 @@ def _merge_segments(self, seg_nums: list[int]) -> bool:
       self._event_data = merged_event_data
       self._merged_segments = segments_to_merge
 
+    self._log(0, f"merged segments: {sorted(segments_to_merge)}")
     return True