callbacks

PlotCallback(trace, dims=None, *, trace_index=0, ax=None, show=True, pause=0.001)

Bases: HyDRACallback

Live matplotlib visualization for HyDRA trace analysis progress.

Source code in src/flowcean/hydra/callbacks.py

def __init__(
    self,
    trace: Trace,
    dims: Sequence[int] | None = None,
    *,
    trace_index: int = 0,
    ax: Axes | None = None,
    show: bool = True,
    pause: float = 0.001,
) -> None:
    self.trace = trace
    self.dims = list(dims) if dims is not None else [0]
    self.trace_index = trace_index
    self.ax = ax
    self.show = show
    self.pause = pause
    self._finalized_segments: list[tuple[TraceSegment, int]] = []
    self._grouping_segments: list[tuple[TraceSegment, int]] = []
    self._active_segment: TraceSegment | None = None
    self._overlay_artists: list[Artist] = []
    self._base_plotted = False
    self._mode_colors: dict[int, str] = {}

    self._validate_trace()