working the player
This commit is contained in:
273
cht/ui/timeline.py
Normal file
273
cht/ui/timeline.py
Normal file
@@ -0,0 +1,273 @@
|
||||
"""
|
||||
Timeline: state machine + shared slider for the recording.
|
||||
|
||||
State is a plain ViewState dataclass — two orthogonal flags replace the old
|
||||
4-state enum. A single "changed" GObject signal is emitted on any mutation;
|
||||
consumers read timeline.state directly.
|
||||
|
||||
live=True — watching UDP relay; play/pause and seek are disabled
|
||||
live=False — scrub mode; paused controls play/pause
|
||||
"""
|
||||
|
||||
import logging
|
||||
import time
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
import gi
|
||||
gi.require_version("Gtk", "4.0")
|
||||
from gi.repository import Gtk, GLib, GObject
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@dataclass
|
||||
class ViewState:
|
||||
live: bool = False # True = watching live UDP relay
|
||||
paused: bool = False # scrub mode: is playback paused?
|
||||
cursor: float = 0.0 # current position in recording (seconds)
|
||||
duration: float = 0.0 # recording duration (seconds)
|
||||
scene_markers: list[float] = field(default_factory=list)
|
||||
|
||||
|
||||
class Timeline(GObject.Object):
|
||||
"""State machine for recording playback.
|
||||
|
||||
Single signal: "changed" — emitted on any state mutation.
|
||||
Consumers read timeline.state directly (no signal arguments).
|
||||
"""
|
||||
|
||||
__gsignals__ = {
|
||||
"changed": (GObject.SignalFlags.RUN_FIRST, None, ()),
|
||||
}
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.state = ViewState()
|
||||
log.info("Timeline created")
|
||||
|
||||
def _emit(self):
|
||||
self.emit("changed")
|
||||
|
||||
# -- Duration / cursor (called from background threads via GLib.idle_add) --
|
||||
|
||||
def set_duration(self, duration):
|
||||
"""Update recording duration. In live mode, cursor follows end."""
|
||||
if duration <= self.state.duration:
|
||||
return
|
||||
self.state.duration = duration
|
||||
if self.state.live:
|
||||
self.state.cursor = duration
|
||||
self._emit()
|
||||
|
||||
def set_cursor(self, position):
|
||||
"""Set cursor position (from slider drag or playback sync)."""
|
||||
position = max(0.0, min(position, self.state.duration))
|
||||
if abs(position - self.state.cursor) < 0.05:
|
||||
return
|
||||
self.state.cursor = position
|
||||
self._emit()
|
||||
|
||||
def tick_live(self):
|
||||
"""Advance cursor by 1s in live mode (smooth progression between probes)."""
|
||||
if not self.state.live:
|
||||
return
|
||||
self.state.cursor += 1.0
|
||||
# Only cap once duration is known — avoids clamping to 0 before first probe
|
||||
if self.state.duration > 0:
|
||||
self.state.cursor = min(self.state.cursor, self.state.duration)
|
||||
self._emit()
|
||||
|
||||
def add_scene_marker(self, timestamp):
|
||||
"""Add a scene change marker (does not emit — no UI update needed)."""
|
||||
self.state.scene_markers.append(timestamp)
|
||||
self.state.scene_markers.sort()
|
||||
|
||||
# -- User actions --
|
||||
|
||||
def go_live(self):
|
||||
"""Go to live mode at the recording end."""
|
||||
self.state.live = True
|
||||
self.state.paused = False
|
||||
self.state.cursor = self.state.duration
|
||||
self._emit()
|
||||
|
||||
def toggle_live(self, live_player_pos=None):
|
||||
"""Toggle between live and scrub mode.
|
||||
|
||||
When leaving live, cursor stays at the live player's actual position
|
||||
if provided, otherwise stays at current cursor.
|
||||
"""
|
||||
if self.state.live:
|
||||
# Enter scrub mode at the live player's current position
|
||||
self.state.live = False
|
||||
self.state.paused = True
|
||||
if live_player_pos is not None and live_player_pos > 0:
|
||||
pos = max(0.0, min(live_player_pos, self.state.duration))
|
||||
self.state.cursor = pos
|
||||
else:
|
||||
self.state.live = True
|
||||
self.state.paused = False
|
||||
self.state.cursor = self.state.duration
|
||||
self._emit()
|
||||
|
||||
def play(self):
|
||||
if self.state.live:
|
||||
return
|
||||
self.state.paused = False
|
||||
self._emit()
|
||||
|
||||
def pause(self):
|
||||
if self.state.live:
|
||||
return
|
||||
self.state.paused = True
|
||||
self._emit()
|
||||
|
||||
def seek(self, position):
|
||||
"""Seek to position — enters scrub mode, pauses."""
|
||||
self.state.live = False
|
||||
self.state.paused = True
|
||||
position = max(0.0, min(position, self.state.duration))
|
||||
if abs(position - self.state.cursor) >= 0.05:
|
||||
self.state.cursor = position
|
||||
self._emit()
|
||||
|
||||
def reset(self):
|
||||
"""Reset all state (called on disconnect)."""
|
||||
self.state = ViewState()
|
||||
self._emit()
|
||||
|
||||
|
||||
class TimelineControls(Gtk.Box):
|
||||
"""Shared slider + play/pause/live controls.
|
||||
|
||||
Play/Pause and slider are insensitive in live mode.
|
||||
LIVE button is a toggle — active style when live=True.
|
||||
"""
|
||||
|
||||
def __init__(self, timeline, **kwargs):
|
||||
super().__init__(orientation=Gtk.Orientation.HORIZONTAL, spacing=4, **kwargs)
|
||||
self._timeline = timeline
|
||||
self._updating_slider = False
|
||||
self._dragging = False
|
||||
self._wall_clock_start = None
|
||||
|
||||
self.set_margin_start(4)
|
||||
self.set_margin_end(4)
|
||||
self.set_margin_top(2)
|
||||
self.set_margin_bottom(4)
|
||||
|
||||
# Play/Pause button
|
||||
self._play_btn = Gtk.Button(label="Play")
|
||||
self._play_btn.connect("clicked", self._on_play_clicked)
|
||||
self.append(self._play_btn)
|
||||
|
||||
# Current time label
|
||||
self._time_label = Gtk.Label(label="00:00")
|
||||
self._time_label.set_width_chars(6)
|
||||
self.append(self._time_label)
|
||||
|
||||
# Slider
|
||||
self._slider = Gtk.Scale(orientation=Gtk.Orientation.HORIZONTAL)
|
||||
self._slider.set_hexpand(True)
|
||||
self._slider.set_range(0, 1)
|
||||
self._slider.set_draw_value(False)
|
||||
self._slider.connect("value-changed", self._on_slider_value_changed)
|
||||
|
||||
press_ctrl = Gtk.GestureClick()
|
||||
press_ctrl.connect("pressed", self._on_slider_pressed)
|
||||
press_ctrl.connect("released", self._on_slider_released)
|
||||
press_ctrl.set_propagation_phase(Gtk.PropagationPhase.CAPTURE)
|
||||
self._slider.add_controller(press_ctrl)
|
||||
self.append(self._slider)
|
||||
|
||||
# Duration label
|
||||
self._duration_label = Gtk.Label(label="00:00 / 00:00")
|
||||
self._duration_label.set_width_chars(14)
|
||||
self.append(self._duration_label)
|
||||
|
||||
# LIVE toggle button
|
||||
self._live_btn = Gtk.Button(label="LIVE")
|
||||
self._live_btn.connect("clicked", self._on_live_clicked)
|
||||
self.append(self._live_btn)
|
||||
|
||||
timeline.connect("changed", self._on_changed)
|
||||
GLib.timeout_add(1000, self._tick_total)
|
||||
|
||||
def _on_play_clicked(self, btn):
|
||||
s = self._timeline.state
|
||||
if s.paused:
|
||||
self._timeline.play()
|
||||
else:
|
||||
self._timeline.pause()
|
||||
|
||||
def set_live_toggle_callback(self, cb):
|
||||
"""Override the LIVE button handler. cb() should return the live player
|
||||
position (float or None) and call timeline.toggle_live() itself."""
|
||||
self._live_toggle_cb = cb
|
||||
|
||||
def _on_live_clicked(self, btn):
|
||||
if hasattr(self, "_live_toggle_cb"):
|
||||
self._live_toggle_cb()
|
||||
else:
|
||||
self._timeline.toggle_live()
|
||||
|
||||
def _on_slider_value_changed(self, slider):
|
||||
if self._dragging:
|
||||
self._time_label.set_text(self._fmt_time(slider.get_value()))
|
||||
|
||||
def _on_slider_pressed(self, gesture, n_press, x, y):
|
||||
self._dragging = True
|
||||
|
||||
def _on_slider_released(self, gesture, n_press, x, y):
|
||||
if self._dragging:
|
||||
self._dragging = False
|
||||
self._timeline.seek(self._slider.get_value())
|
||||
|
||||
def _on_changed(self, timeline):
|
||||
s = timeline.state
|
||||
|
||||
# Start wall clock when first going live (not on duration, which arrives ~30s later)
|
||||
if s.live and self._wall_clock_start is None:
|
||||
self._wall_clock_start = time.monotonic()
|
||||
|
||||
# Live mode: disable scrub controls
|
||||
self._play_btn.set_sensitive(not s.live)
|
||||
self._slider.set_sensitive(not s.live)
|
||||
|
||||
if s.live:
|
||||
self._live_btn.add_css_class("suggested-action")
|
||||
else:
|
||||
self._live_btn.remove_css_class("suggested-action")
|
||||
|
||||
# Play button label (only relevant in scrub mode)
|
||||
self._play_btn.set_label("Pause" if not s.paused else "Play")
|
||||
|
||||
# Slider position
|
||||
if not self._dragging:
|
||||
self._updating_slider = True
|
||||
self._slider.set_range(0, max(s.duration, 0.1))
|
||||
self._slider.set_value(s.cursor)
|
||||
self._updating_slider = False
|
||||
|
||||
self._time_label.set_text(self._fmt_time(s.cursor))
|
||||
self._update_duration_label()
|
||||
|
||||
def _tick_total(self):
|
||||
self._update_duration_label()
|
||||
return True
|
||||
|
||||
def _update_duration_label(self):
|
||||
s = self._timeline.state
|
||||
loaded = s.duration
|
||||
total = (time.monotonic() - self._wall_clock_start) if self._wall_clock_start else loaded
|
||||
self._duration_label.set_text(
|
||||
f"{self._fmt_time(loaded)} / {self._fmt_time(total)}"
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _fmt_time(seconds):
|
||||
m, s = divmod(int(seconds), 60)
|
||||
h, m = divmod(m, 60)
|
||||
if h:
|
||||
return f"{h}:{m:02d}:{s:02d}"
|
||||
return f"{m:02d}:{s:02d}"
|
||||
Reference in New Issue
Block a user