do driver state

pull/35543/head
Shane Smiskol 3 months ago
parent 98bfd45712
commit 9aa31fd3dc
  1. 11
      selfdrive/ui/onroad/driver_state.py

@ -76,10 +76,10 @@ class DriverStateRenderer(Widget):
self.engaged_color = rl.Color(26, 242, 66, 255) self.engaged_color = rl.Color(26, 242, 66, 255)
self.disengaged_color = rl.Color(139, 139, 139, 255) self.disengaged_color = rl.Color(139, 139, 139, 255)
def _render(self, rect): self.set_visible(lambda: (ui_state.sm.recv_frame['driverStateV2'] > ui_state.started_frame and
if not self._is_visible(ui_state.sm): ui_state.sm.seen['driverMonitoringState']))
return
def _render(self, rect):
self._update_state(ui_state.sm, rect) self._update_state(ui_state.sm, rect)
if not self.state_updated: if not self.state_updated:
return return
@ -108,11 +108,6 @@ class DriverStateRenderer(Widget):
if self.v_arc_data: if self.v_arc_data:
rl.draw_spline_linear(self.v_arc_lines, len(self.v_arc_lines), self.v_arc_data.thickness, self.arc_color) rl.draw_spline_linear(self.v_arc_lines, len(self.v_arc_lines), self.v_arc_data.thickness, self.arc_color)
def _is_visible(self, sm):
"""Check if the visualization should be rendered."""
return (sm.recv_frame['driverStateV2'] > ui_state.started_frame and
sm.seen['driverMonitoringState'])
def _update_state(self, sm, rect): def _update_state(self, sm, rect):
"""Update the driver monitoring state based on model data""" """Update the driver monitoring state based on model data"""
if not sm.updated["driverMonitoringState"]: if not sm.updated["driverMonitoringState"]:

Loading…
Cancel
Save