IQ.Pilot Release Commit @ d2ce8a8

iqpilot/ui/mici/onroad/emac_source.py

@@ -0,0 +1,43 @@
+"""
+Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
+"""
+
+import time
+
+import pyray as rl
+
+from iqpilot.common.params import Params
+from iqpilot.ui.onroad.big_model_status import SourceState, draw_source_label, resolve_source
+from iqpilot.selfdrive.ui.ui_state import ui_state
+from iqpilot.system.ui.lib.application import FontWeight, gui_app
+from iqpilot.system.ui.lib.text_measure import measure_text_cached
+
+_POLL_S = 1.0
+_FONT_SIZE = 44
+_WHEEL_H = 50
+_MARGIN_R = 12
+_MARGIN_B = 14
+
+
+class EmacSourceIndicator:
+  def __init__(self):
+    self._params = Params()
+    self._font = gui_app.font(FontWeight.SEMI_BOLD)
+    self._last_poll = 0.0
+    self._label = ""
+    self._state = SourceState.HIDDEN
+
+  def update(self) -> None:
+    now = time.monotonic()
+    if now - self._last_poll < _POLL_S:
+      return
+    self._last_poll = now
+    self._label, self._state = resolve_source(self._params, ui_state.engaged)
+
+  def render(self, rect: rl.Rectangle) -> None:
+    if self._state == SourceState.HIDDEN:
+      return
+    size = measure_text_cached(self._font, self._label, _FONT_SIZE)
+    x = rect.x + rect.width - _MARGIN_R - size.x
+    y = rect.y + rect.height - _MARGIN_B - (_WHEEL_H + size.y) / 2
+    draw_source_label(self._font, self._label, self._state, rl.Vector2(x, y), _FONT_SIZE)

iqpilot/ui/mici/onroad/hud_renderer.py

@@ -5,11 +5,12 @@ import pyray as rl
 
 from iqpilot.selfdrive.ui.mici.onroad.hud_renderer import HudRenderer
 from iqpilot.ui.onroad.hud_overlays import IQBlindSpotOverlay
+from iqpilot.ui.mici.onroad.emac_source import EmacSourceIndicator
 
 class IQMiciHudRenderer(HudRenderer):
   def __init__(self):
     super().__init__()
-    self._overlays = [IQBlindSpotOverlay()]
+    self._overlays = [IQBlindSpotOverlay(), EmacSourceIndicator()]
 
   def _update_state(self) -> None:
     super()._update_state()