forked from IQ.Lvbs/IQ.Pilot
91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos
|
|
|
|
IQ.Pilot road-view HUD: extends the stock renderer and layers on the IQ overlays
|
|
(developer bar, nav map, road name, speed + speed-limit, turn signals, rocket-fuel
|
|
accel bar, soft warnings, steering arc).
|
|
"""
|
|
import pyray as rl
|
|
|
|
from openpilot.selfdrive.ui.mici.onroad.torque_bar import TorqueBar
|
|
from openpilot.selfdrive.ui.ui_state import ui_state
|
|
from openpilot.selfdrive.ui.onroad.hud_renderer import HudRenderer
|
|
from openpilot.iqpilot.ui.onroad.hud_overlays import (
|
|
IQDevMetricsOverlay,
|
|
RoadNameRenderer,
|
|
IQAccelBar,
|
|
IQSpeedLimitOverlay,
|
|
IQTurnSignalOverlay,
|
|
IQSpeedOverlay,
|
|
)
|
|
from openpilot.iqpilot.ui.onroad.nav_map_panel import NavMapPanel
|
|
from openpilot.iqpilot.ui.onroad.soft_warning import SoftWarningRenderer
|
|
|
|
ENABLE_FLOATING_NAV_MAP_PANEL = False
|
|
ENABLE_SPLIT_NAV_MAP_PANEL = True
|
|
|
|
|
|
class IQHudRenderer(HudRenderer):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.developer_ui = IQDevMetricsOverlay()
|
|
self.nav_map_panel = NavMapPanel()
|
|
self.road_name_renderer = RoadNameRenderer()
|
|
self.rocket_fuel = IQAccelBar()
|
|
self.speed_limit_renderer = IQSpeedLimitOverlay()
|
|
self.turn_signal_controller = IQTurnSignalOverlay()
|
|
self.speed_renderer = IQSpeedOverlay()
|
|
self.soft_warning_renderer = SoftWarningRenderer()
|
|
self._torque_bar = TorqueBar(scale=3.0, always=True)
|
|
|
|
def _update_state(self) -> None:
|
|
super()._update_state()
|
|
if ENABLE_FLOATING_NAV_MAP_PANEL or ENABLE_SPLIT_NAV_MAP_PANEL:
|
|
self.nav_map_panel.update()
|
|
self.road_name_renderer.update()
|
|
self.speed_limit_renderer.update()
|
|
has_limit = self.speed_limit_renderer.speed_limit_valid or self.speed_limit_renderer.speed_limit_last_valid
|
|
self.limit_available = has_limit
|
|
self.limit_speed_text = str(round(self.speed_limit_renderer.speed_limit_last)) if has_limit else "---"
|
|
offset = round(self.speed_limit_renderer.speed_limit_offset)
|
|
self.limit_offset_text = f"{offset:+d}" if has_limit and offset != 0 else ""
|
|
self.turn_signal_controller.update()
|
|
self.speed_renderer.update()
|
|
self.soft_warning_renderer.update()
|
|
|
|
def _draw_current_speed(self, rect: rl.Rectangle) -> None:
|
|
self.speed_renderer.render(rect)
|
|
|
|
def _render(self, rect: rl.Rectangle) -> None:
|
|
super()._render(rect)
|
|
|
|
if ui_state.torque_bar:
|
|
torque_rect = rect
|
|
if ui_state.developer_ui in (IQDevMetricsOverlay.DEV_UI_BOTTOM, IQDevMetricsOverlay.DEV_UI_BOTH):
|
|
torque_rect = rl.Rectangle(rect.x, rect.y, rect.width, rect.height - IQDevMetricsOverlay.BOTTOM_BAR_HEIGHT)
|
|
self._torque_bar.render(torque_rect)
|
|
|
|
if not self.split_nav_enabled():
|
|
self.developer_ui.render(rect)
|
|
if ENABLE_FLOATING_NAV_MAP_PANEL:
|
|
self.nav_map_panel.render(rect)
|
|
self.road_name_renderer.render(rect)
|
|
self.turn_signal_controller.render(rect)
|
|
self.soft_warning_renderer.render(rect)
|
|
self.rocket_fuel.render(rect, ui_state.sm)
|
|
|
|
def split_nav_enabled(self) -> bool:
|
|
if not ENABLE_SPLIT_NAV_MAP_PANEL:
|
|
return False
|
|
if hasattr(self.nav_map_panel, "maps_enabled"):
|
|
return bool(self.nav_map_panel.maps_enabled())
|
|
return bool(getattr(self.nav_map_panel, "_maps_enabled", False))
|
|
|
|
def render_split_nav(self, rect: rl.Rectangle) -> None:
|
|
if self.split_nav_enabled():
|
|
self.nav_map_panel.render_split(rect)
|
|
|
|
def render_full_width_overlays(self, rect: rl.Rectangle) -> None:
|
|
if self.split_nav_enabled():
|
|
self.developer_ui.render(rect)
|