feat: wire ReplayStatsTracker into SessionController + complete GraphView.updateTrend

This commit is contained in:
jack
2026-06-20 18:28:55 -07:00
parent 7206334a27
commit 20a34e22ee
3 changed files with 73 additions and 0 deletions
@@ -11,6 +11,7 @@ from pygui.backend.cluster_average import average_clusters, group_sensors_by_rad
from pygui.backend.data.csv_recorder import CsvRecorder
from pygui.backend.models.frame import Frame
from pygui.backend.models.frame_player import FramePlayer, frames_from_wafer_data
from pygui.backend.models.replay_stats import ReplayStatsTracker
from pygui.backend.models.sensor_editor import SensorEditor
from pygui.backend.models.session_model import SessionModel
from pygui.backend.models.threshold_classifier import ThresholdConfig
@@ -34,6 +35,8 @@ class SessionController(QObject):
sensorsChanged = Signal()
loadedFileChanged = Signal()
clusterAveragingEnabledChanged = Signal()
# trend: per-frame avg for live graph
trendData = Signal(str) # JSON array of floats
# private: marshal a worker-thread frame onto the main thread
_liveFrame = Signal(object) # Frame
@@ -65,6 +68,7 @@ class SessionController(QObject):
self._loaded_file: str = ""
self._cluster_averaging_enabled = False
self._active_clusters: list[list[int]] = []
self._stats_tracker = ReplayStatsTracker()
# ---- properties QML binds to ----
@Property(str, notify=modeChanged)
@@ -248,6 +252,7 @@ class SessionController(QObject):
self._active_clusters = group_sensors_by_radius(self._sensors)
self._player.load(frames)
self._model.reset()
self._stats_tracker.reset()
self._loaded_file = file_path
self.loadedFileChanged.emit()
self.sensorsChanged.emit()
@@ -378,6 +383,7 @@ class SessionController(QObject):
# Clear out any old data from the prev sessions
self._model.reset()
self._stats_tracker.reset()
self._last = None
self._last_raw_frame = None
@@ -424,6 +430,8 @@ class SessionController(QObject):
@Slot(object)
@slot_error_boundary
def _on_live_frame(self, frame: Frame) -> None:
import json
self._stats_tracker.track(frame.seq, frame.values)
# Main thread (queued). Record raw, process edited; mark dirty for repaint.
self._last_raw_frame = frame
edited = Frame(seq=frame.seq, time=frame.time,
@@ -432,6 +440,7 @@ class SessionController(QObject):
if self._recorder.is_recording:
self._recorder.write(frame) # always record raw values, never edited
self._dirty = True
self.trendData.emit(json.dumps(self._stats_tracker.get_trend()))
def _flush_repaint(self) -> None:
if not self._dirty: