mirror of
https://gitlvb.teallvbs.xyz/IQ.Lvbs/IQ.Pilot.git
synced 2026-07-24 04:52:13 +08:00
IQ.Lvbs CI [bot]
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos
|
|
"""
|
|
import pyray as rl
|
|
from cereal import log
|
|
|
|
from openpilot.selfdrive.ui import UI_BORDER_SIZE
|
|
from openpilot.selfdrive.ui.ui_state import ui_state
|
|
from openpilot.selfdrive.ui.onroad.driver_state import BTN_SIZE
|
|
from openpilot.system.ui.lib.application import gui_app
|
|
|
|
EventName = log.OnroadEvent.EventName
|
|
|
|
# Events that trigger the soft warning triangle instead of a disruptive alert
|
|
SOFT_WARNING_EVENTS = {
|
|
EventName.commIssue,
|
|
EventName.commIssueAvgFreq,
|
|
EventName.selfdrivedLagging,
|
|
}
|
|
|
|
ICON_SIZE = 96
|
|
|
|
# Speed box geometry (mirrors hud_renderer._draw_set_speed)
|
|
_SPEED_BOX_X_OFFSET = 60
|
|
_SPEED_BOX_Y_OFFSET = 45
|
|
_SPEED_BOX_WIDTH = 180 # midpoint between metric (186) and imperial (174)
|
|
_SPEED_BOX_HEIGHT = 228
|
|
|
|
_DM_OFFSET = UI_BORDER_SIZE + BTN_SIZE // 2 # = 126
|
|
|
|
|
|
class SoftWarningRenderer:
|
|
def __init__(self):
|
|
self._icon = gui_app.texture("icons_mici/offroad_alerts/orange_warning.png", ICON_SIZE, ICON_SIZE)
|
|
self._active = False
|
|
|
|
def update(self) -> None:
|
|
sm = ui_state.sm
|
|
self._active = any(e.name in SOFT_WARNING_EVENTS for e in sm['onroadEvents'])
|
|
|
|
def render(self, rect: rl.Rectangle) -> None:
|
|
if not self._active:
|
|
return
|
|
|
|
# Centre of speed box (top-left of screen)
|
|
speed_cx = rect.x + _SPEED_BOX_X_OFFSET + _SPEED_BOX_WIDTH / 2
|
|
speed_cy = rect.y + _SPEED_BOX_Y_OFFSET + _SPEED_BOX_HEIGHT / 2
|
|
|
|
# Centre of driver monitoring icon (bottom-left of screen, LHD)
|
|
dm_cx = rect.x + _DM_OFFSET
|
|
dm_cy = rect.y + rect.height - _DM_OFFSET
|
|
|
|
# Midpoint between the two
|
|
mid_x = (speed_cx + dm_cx) / 2
|
|
mid_y = (speed_cy + dm_cy) / 2
|
|
|
|
draw_x = int(mid_x - ICON_SIZE / 2)
|
|
draw_y = int(mid_y - ICON_SIZE / 2)
|
|
|
|
rl.draw_texture(self._icon, draw_x, draw_y, rl.WHITE)
|