82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
|
|
"""
|
|
|
|
import math
|
|
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 import UI_BORDER_SIZE
|
|
from iqpilot.selfdrive.ui.onroad.driver_state import BTN_SIZE
|
|
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
|
|
from iqpilot.system.ui.widgets import Widget
|
|
|
|
_POLL_S = 1.0
|
|
_FONT_SIZE = 70
|
|
_ICON_H = 76
|
|
_ICONS = {"MAC": ("mac", 62 / 46), "GPU": ("egpu", 1.0)}
|
|
_GREY = rl.Color(165, 165, 170, 235)
|
|
|
|
|
|
class EmacStatusRenderer(Widget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._params = Params()
|
|
self._font = gui_app.font(FontWeight.SEMI_BOLD)
|
|
self._last_poll = 0.0
|
|
self._label = ""
|
|
self._state = SourceState.HIDDEN
|
|
self._icons: dict[str, dict[str, rl.Texture]] = {}
|
|
|
|
def _icon_set(self, label: str) -> dict[str, rl.Texture] | None:
|
|
spec = _ICONS.get(label)
|
|
if spec is None:
|
|
return None
|
|
if label not in self._icons:
|
|
base, aspect = spec
|
|
w = int(_ICON_H * aspect)
|
|
self._icons[label] = {
|
|
"base": gui_app.texture(f"icons_mici/{base}.png", w, _ICON_H),
|
|
"green": gui_app.texture(f"icons_mici/{base}_green.png", w, _ICON_H),
|
|
"orange": gui_app.texture(f"icons_mici/{base}_orange.png", int(w * 1.26), _ICON_H),
|
|
}
|
|
return self._icons[label]
|
|
|
|
def update(self):
|
|
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):
|
|
if self._state == SourceState.HIDDEN:
|
|
return
|
|
icons = self._icon_set(self._label)
|
|
if icons is None:
|
|
size = measure_text_cached(self._font, self._label, _FONT_SIZE)
|
|
x = rect.x + UI_BORDER_SIZE + BTN_SIZE // 2 - size.x / 2
|
|
y = rect.y + rect.height / 2 - size.y / 2
|
|
draw_source_label(self._font, self._label, self._state, rl.Vector2(x, y), _FONT_SIZE)
|
|
return
|
|
if self._state == SourceState.ACTIVE:
|
|
tex, tint = icons["green"], rl.Color(255, 255, 255, 255)
|
|
elif self._state == SourceState.FAILED:
|
|
tex, tint = icons["orange"], rl.Color(255, 255, 255, 255)
|
|
elif self._state == SourceState.CROSSED:
|
|
tex, tint = icons["base"], rl.Color(255, 255, 255, 165)
|
|
else:
|
|
pulse = 0.35 + 0.65 * (0.5 - 0.5 * math.cos(rl.get_time() * 6.0))
|
|
tex, tint = icons["base"], rl.Color(_GREY.r, _GREY.g, _GREY.b, int(_GREY.a * pulse))
|
|
x = int(rect.x + UI_BORDER_SIZE + BTN_SIZE // 2 - tex.width / 2)
|
|
y = int(rect.y + rect.height / 2 - tex.height / 2)
|
|
rl.draw_texture(tex, x, y, tint)
|
|
if self._state == SourceState.CROSSED:
|
|
cy = y + tex.height // 2
|
|
rl.draw_line_ex(rl.Vector2(x - 4, cy), rl.Vector2(x + tex.width + 4, cy), 4, rl.Color(255, 255, 255, 165))
|