354 lines
16 KiB
Python
354 lines
16 KiB
Python
"""
|
||
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
|
||
"""
|
||
|
||
import time
|
||
|
||
from iqpilot.cereal import log
|
||
import pyray as rl
|
||
from collections.abc import Callable
|
||
from iqpilot.system.ui.widgets.label import gui_label, MiciLabel, UnifiedLabel
|
||
from iqpilot.system.ui.widgets import Widget
|
||
from iqpilot.system.ui.lib.application import gui_app, FontWeight, DEFAULT_TEXT_COLOR, MousePos
|
||
from iqpilot.system.ui.lib.text_measure import measure_text_cached
|
||
from iqpilot.selfdrive.ui.ui_state import ui_state
|
||
from iqpilot.system.ui.text import wrap_text
|
||
from iqpilot.system.version import training_version, RELEASE_IQ_BRANCHES
|
||
|
||
HEAD_BUTTON_FONT_SIZE = 40
|
||
HOME_PADDING = 8
|
||
HOME_TITLE_MAX_FONT_SIZE = 72
|
||
HOME_TITLE_MIN_FONT_SIZE = 36
|
||
HOME_TITLE_TEXT = "IQ.Pilot"
|
||
|
||
NetworkType = log.DeviceState.NetworkType
|
||
|
||
NETWORK_TYPES = {
|
||
NetworkType.none: "Offline",
|
||
NetworkType.wifi: "WiFi",
|
||
NetworkType.cell2G: "2G",
|
||
NetworkType.cell3G: "3G",
|
||
NetworkType.cell4G: "LTE",
|
||
NetworkType.cell5G: "5G",
|
||
NetworkType.ethernet: "Ethernet",
|
||
}
|
||
|
||
|
||
class DeviceStatus(Widget):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.set_rect(rl.Rectangle(0, 0, 300, 175))
|
||
self._update_state()
|
||
self._version_text = self._get_version_text()
|
||
|
||
self._do_welcome()
|
||
|
||
def _do_welcome(self):
|
||
ui_state.params.put("CompletedTrainingVersion", training_version)
|
||
|
||
def refresh(self):
|
||
self._update_state()
|
||
self._version_text = self._get_version_text()
|
||
|
||
def _get_version_text(self) -> str:
|
||
brand = "IQ.Pilot"
|
||
description = ui_state.params.get("UpdaterCurrentDescription")
|
||
return f"{brand} {description}" if description else brand
|
||
|
||
def _update_state(self):
|
||
# TODO: refresh function that can be called periodically, not at 60 fps, so we can update version
|
||
# update system status
|
||
self._system_status = "SYSTEM READY ✓" if ui_state.panda_type != log.PandaState.PandaType.unknown else "BOOTING UP..."
|
||
|
||
# update network status
|
||
strength = ui_state.sm['deviceState'].networkStrength.raw
|
||
strength_text = "● " * strength + "○ " * (4 - strength) # ◌ also works
|
||
network_type = NETWORK_TYPES[ui_state.sm['deviceState'].networkType.raw]
|
||
self._network_status = f"{network_type} {strength_text}"
|
||
|
||
def _render(self, _):
|
||
# draw status
|
||
status_rect = rl.Rectangle(self._rect.x, self._rect.y, self._rect.width, 40)
|
||
gui_label(status_rect, self._system_status, font_size=HEAD_BUTTON_FONT_SIZE, color=DEFAULT_TEXT_COLOR,
|
||
font_weight=FontWeight.BOLD, alignment=rl.GuiTextAlignment.TEXT_ALIGN_CENTER)
|
||
|
||
# draw network status
|
||
network_rect = rl.Rectangle(self._rect.x, self._rect.y + 60, self._rect.width, 40)
|
||
gui_label(network_rect, self._network_status, font_size=40, color=DEFAULT_TEXT_COLOR,
|
||
font_weight=FontWeight.MEDIUM, alignment=rl.GuiTextAlignment.TEXT_ALIGN_CENTER)
|
||
|
||
# draw version
|
||
version_font_size = 30
|
||
version_rect = rl.Rectangle(self._rect.x, self._rect.y + 140, self._rect.width + 20, 40)
|
||
wrapped_text = '\n'.join(wrap_text(self._version_text, version_font_size, version_rect.width))
|
||
gui_label(version_rect, wrapped_text, font_size=version_font_size, color=DEFAULT_TEXT_COLOR,
|
||
font_weight=FontWeight.MEDIUM, alignment=rl.GuiTextAlignment.TEXT_ALIGN_LEFT)
|
||
|
||
|
||
class MiciHomeLayout(Widget):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self._on_settings_click: Callable | None = None
|
||
|
||
self._last_refresh = 0
|
||
self._mouse_down_t: None | float = None
|
||
self._did_long_press = False
|
||
self._is_pressed_prev = False
|
||
|
||
self._version_text = None
|
||
self._experimental_mode = False
|
||
|
||
self._settings_txt = gui_app.texture("icons_mici/settings.png", 48, 48)
|
||
self._experimental_txt = gui_app.texture("icons_mici/experimental_mode_mici.png", 48, 48)
|
||
self._iqdynamic_txt = gui_app.texture("icons_mici/iqdynamic_mode_mici.png", 48, 48)
|
||
self._iqstandard_txt = gui_app.texture("icons_mici/iqstandard_mode_mici.png", 48, 48)
|
||
self._mode_txt = None
|
||
self._mic_txt = gui_app.texture("icons_mici/microphone.png", 32, 46)
|
||
|
||
self._net_type = NETWORK_TYPES.get(NetworkType.none)
|
||
self._net_strength = 0
|
||
|
||
self._wifi_slash_txt = gui_app.texture("icons_mici/settings/network/wifi_strength_slash.png", 50, 44)
|
||
self._wifi_none_txt = gui_app.texture("icons_mici/settings/network/wifi_strength_none.png", 50, 37)
|
||
self._wifi_low_txt = gui_app.texture("icons_mici/settings/network/wifi_strength_low.png", 50, 37)
|
||
self._wifi_medium_txt = gui_app.texture("icons_mici/settings/network/wifi_strength_medium.png", 50, 37)
|
||
self._wifi_full_txt = gui_app.texture("icons_mici/settings/network/wifi_strength_full.png", 50, 37)
|
||
|
||
self._cell_none_txt = gui_app.texture("icons_mici/settings/network/cell_strength_none.png", 54, 36)
|
||
self._cell_low_txt = gui_app.texture("icons_mici/settings/network/cell_strength_low.png", 54, 36)
|
||
self._cell_medium_txt = gui_app.texture("icons_mici/settings/network/cell_strength_medium.png", 54, 36)
|
||
self._cell_high_txt = gui_app.texture("icons_mici/settings/network/cell_strength_high.png", 54, 36)
|
||
self._cell_full_txt = gui_app.texture("icons_mici/settings/network/cell_strength_full.png", 54, 36)
|
||
self._lte_label = UnifiedLabel("LTE", font_size=22, text_color=rl.Color(255, 255, 255, int(255 * 0.82)),
|
||
font_weight=FontWeight.BOLD)
|
||
|
||
self._openpilot_label = MiciLabel(HOME_TITLE_TEXT, font_size=HOME_TITLE_MAX_FONT_SIZE, color=rl.Color(255, 255, 255, int(255 * 0.9)),
|
||
font_weight=FontWeight.SYNCOPATE)
|
||
self._version_label = MiciLabel("", font_size=36, font_weight=FontWeight.ROMAN)
|
||
self._large_version_label = MiciLabel("", font_size=64, color=rl.GRAY, font_weight=FontWeight.ROMAN)
|
||
self._date_label = MiciLabel("", font_size=36, color=rl.GRAY, font_weight=FontWeight.ROMAN)
|
||
self._branch_label = UnifiedLabel("", font_size=36, text_color=rl.GRAY, font_weight=FontWeight.ROMAN, scroll=True)
|
||
self._version_commit_label = UnifiedLabel("", font_size=36, text_color=rl.GRAY,
|
||
font_weight=FontWeight.ROMAN, wrap_text=False, scroll=True)
|
||
|
||
def show_event(self):
|
||
self._version_text = self._get_version_text()
|
||
self._update_network_status(ui_state.sm['deviceState'])
|
||
self._update_params()
|
||
|
||
def _update_params(self):
|
||
p = ui_state.params
|
||
self._experimental_mode = p.get_bool("ExperimentalMode")
|
||
if not p.get_bool("AlphaLongitudinalEnabled"):
|
||
self._mode_txt = None
|
||
elif not self._experimental_mode:
|
||
self._mode_txt = self._iqstandard_txt
|
||
elif p.get_bool("IQDynamicMode"):
|
||
self._mode_txt = self._iqdynamic_txt
|
||
else:
|
||
self._mode_txt = self._experimental_txt
|
||
|
||
def _update_state(self):
|
||
if self.is_pressed and not self._is_pressed_prev:
|
||
self._mouse_down_t = time.monotonic()
|
||
elif not self.is_pressed and self._is_pressed_prev:
|
||
self._mouse_down_t = None
|
||
self._did_long_press = False
|
||
self._is_pressed_prev = self.is_pressed
|
||
|
||
if self._mouse_down_t is not None:
|
||
if time.monotonic() - self._mouse_down_t > 0.5:
|
||
# long gating for experimental mode - only allow toggle if longitudinal control is available
|
||
if ui_state.has_longitudinal_control:
|
||
self._experimental_mode = not self._experimental_mode
|
||
ui_state.params.put("ExperimentalMode", self._experimental_mode)
|
||
if not self._experimental_mode:
|
||
ui_state.params.put_bool("IQDynamicMode", False)
|
||
self._update_params()
|
||
self._mouse_down_t = None
|
||
self._did_long_press = True
|
||
|
||
if rl.get_time() - self._last_refresh > 5.0:
|
||
device_state = ui_state.sm['deviceState']
|
||
self._update_network_status(device_state)
|
||
|
||
# Update version text
|
||
self._version_text = self._get_version_text()
|
||
self._last_refresh = rl.get_time()
|
||
self._update_params()
|
||
|
||
def _update_network_status(self, device_state):
|
||
self._net_type = device_state.networkType
|
||
strength = device_state.networkStrength
|
||
self._net_strength = max(0, min(5, strength.raw + 1)) if strength.raw > 0 else 0
|
||
|
||
def set_callbacks(self, on_settings: Callable | None = None):
|
||
self._on_settings_click = on_settings
|
||
|
||
def _handle_mouse_release(self, mouse_pos: MousePos):
|
||
if not self._did_long_press:
|
||
if self._on_settings_click:
|
||
self._on_settings_click()
|
||
self._did_long_press = False
|
||
|
||
def _get_version_text(self) -> tuple[str, str, str, str] | None:
|
||
description = ui_state.params.get("UpdaterCurrentDescription")
|
||
|
||
if description is not None and len(description) > 0:
|
||
# Expect "version / branch / commit / date"; be tolerant of other formats
|
||
try:
|
||
version, branch, commit, date = description.split(" / ")
|
||
return version, branch, commit, date
|
||
except Exception:
|
||
return None
|
||
|
||
return None
|
||
|
||
def _fit_title_font(self, title_x: float) -> None:
|
||
font = gui_app.font(FontWeight.SYNCOPATE)
|
||
viewport_width = min(w for w in (self.rect.width, gui_app.width, rl.get_screen_width()) if w > 0)
|
||
max_width = max(100, int(viewport_width - title_x - HOME_PADDING - 16))
|
||
fit_width = max_width * 0.9
|
||
text_width = measure_text_cached(font, HOME_TITLE_TEXT, HOME_TITLE_MAX_FONT_SIZE).x
|
||
if text_width <= fit_width:
|
||
target_size = HOME_TITLE_MAX_FONT_SIZE
|
||
else:
|
||
target_size = int(HOME_TITLE_MAX_FONT_SIZE * (fit_width / text_width))
|
||
target_size = max(HOME_TITLE_MIN_FONT_SIZE, min(HOME_TITLE_MAX_FONT_SIZE, target_size))
|
||
|
||
if target_size != self._openpilot_label.font_size:
|
||
self._openpilot_label.set_font_size(target_size)
|
||
|
||
def _render_title_gradient(self, x: float, y: float) -> None:
|
||
"""Render HOME_TITLE_TEXT with a left-to-right gradient (#7400b8 → #80ffdb).
|
||
|
||
Technique: render white text to an offscreen RenderTexture, then draw it
|
||
tinted by sampling the gradient per-column using draw_texture_pro with a
|
||
tint. Since draw_texture_pro only supports a single tint color, we use the
|
||
BLEND_MULTIPLIED trick:
|
||
1. Draw white text normally onto the framebuffer.
|
||
2. Draw gradient rect with BLEND_MULTIPLIED on top — this multiplies each
|
||
existing pixel by the gradient color, turning white text into the gradient
|
||
while the dark background (near-zero RGB) stays dark.
|
||
"""
|
||
font = gui_app.font(FontWeight.SYNCOPATE)
|
||
font_size = self._openpilot_label.font_size
|
||
text = HOME_TITLE_TEXT
|
||
|
||
text_size = measure_text_cached(font, text, font_size)
|
||
tw = int(text_size.x) + 4
|
||
th = int(text_size.y) + 4
|
||
|
||
# Step 1: draw white text at full opacity
|
||
rl.draw_text_ex(font, text, rl.Vector2(x + 2, y + 2), font_size, 0,
|
||
rl.Color(255, 255, 255, 230))
|
||
|
||
# Step 2: multiply gradient over the text region — white → gradient color,
|
||
# black background → stays black (0 × anything = 0)
|
||
rl.begin_blend_mode(rl.BlendMode.BLEND_MULTIPLIED)
|
||
rl.draw_rectangle_gradient_h(
|
||
int(x), int(y), tw, th,
|
||
rl.Color(0x80, 0xff, 0xdb, 255), # #80ffdb — left
|
||
rl.Color(0x74, 0x00, 0xb8, 255), # #7400b8 — right
|
||
)
|
||
rl.end_blend_mode()
|
||
|
||
def _render(self, _):
|
||
text_pos = rl.Vector2(self.rect.x - 2 + HOME_PADDING, self.rect.y + HOME_PADDING)
|
||
self._fit_title_font(text_pos.x)
|
||
self._render_title_gradient(text_pos.x, text_pos.y)
|
||
|
||
if self._version_text is not None:
|
||
# release branch
|
||
release_branch = self._version_text[1] in RELEASE_IQ_BRANCHES
|
||
version_pos = rl.Rectangle(text_pos.x, text_pos.y + self._openpilot_label.font_size + 16, 100, 44)
|
||
self._version_label.set_text(self._version_text[0])
|
||
self._version_label.set_position(version_pos.x, version_pos.y)
|
||
self._version_label.render()
|
||
|
||
self._date_label.set_text(" " + self._version_text[3])
|
||
self._date_label.set_position(version_pos.x + self._version_label.rect.width + 10, version_pos.y)
|
||
self._date_label.render()
|
||
|
||
viewport_right = min(w for w in (self.rect.x + self.rect.width, gui_app.width, rl.get_screen_width()) if w > 0)
|
||
branch_x = version_pos.x + self._version_label.rect.width + self._date_label.rect.width + 20
|
||
self._branch_label.set_max_width(max(80, int(viewport_right - branch_x - HOME_PADDING)))
|
||
self._branch_label.set_text(" " + ("release" if release_branch else self._version_text[1]))
|
||
self._branch_label.set_position(branch_x, version_pos.y)
|
||
self._branch_label.render()
|
||
|
||
if not release_branch:
|
||
# 2nd line
|
||
self._version_commit_label.set_text(self._version_text[2])
|
||
commit_y = version_pos.y + self._date_label.font_size + 7
|
||
commit_rect = rl.Rectangle(version_pos.x, commit_y, max(100, viewport_right - version_pos.x - HOME_PADDING), 44)
|
||
self._version_commit_label.render(commit_rect)
|
||
|
||
self._render_bottom_status_bar()
|
||
|
||
def _render_bottom_status_bar(self):
|
||
# ***** Center-aligned bottom section icons *****
|
||
|
||
# TODO: refactor repeated icon drawing into a small loop
|
||
ITEM_SPACING = 18
|
||
Y_CENTER = 24
|
||
|
||
last_x = self.rect.x + HOME_PADDING
|
||
|
||
# Draw settings icon in bottom left corner
|
||
rl.draw_texture(self._settings_txt, int(last_x), int(self._rect.y + self.rect.height - self._settings_txt.height / 2 - Y_CENTER),
|
||
rl.Color(255, 255, 255, int(255 * 0.9)))
|
||
last_x = last_x + self._settings_txt.width + ITEM_SPACING
|
||
|
||
# draw network
|
||
if self._net_type == NetworkType.wifi:
|
||
# There is no 1
|
||
draw_net_txt = {0: self._wifi_none_txt,
|
||
2: self._wifi_low_txt,
|
||
3: self._wifi_medium_txt,
|
||
4: self._wifi_full_txt,
|
||
5: self._wifi_full_txt}.get(self._net_strength, self._wifi_low_txt)
|
||
rl.draw_texture(draw_net_txt, int(last_x),
|
||
int(self._rect.y + self.rect.height - draw_net_txt.height / 2 - Y_CENTER), rl.Color(255, 255, 255, int(255 * 0.9)))
|
||
last_x += draw_net_txt.width + ITEM_SPACING
|
||
|
||
elif self._net_type in (NetworkType.cell2G, NetworkType.cell3G, NetworkType.cell4G, NetworkType.cell5G):
|
||
last_x = self._draw_cellular_cluster(last_x, ITEM_SPACING, Y_CENTER, connected=True)
|
||
|
||
else:
|
||
last_x = self._draw_cellular_cluster(last_x, ITEM_SPACING, Y_CENTER, connected=False)
|
||
|
||
if self._mode_txt is not None:
|
||
rl.draw_texture(self._mode_txt, int(last_x),
|
||
int(self._rect.y + self.rect.height - self._mode_txt.height / 2 - Y_CENTER), rl.Color(255, 255, 255, 255))
|
||
last_x += self._mode_txt.width + ITEM_SPACING
|
||
|
||
# draw microphone icon when recording audio is enabled
|
||
if ui_state.recording_audio:
|
||
rl.draw_texture(self._mic_txt, int(last_x),
|
||
int(self._rect.y + self.rect.height - self._mic_txt.height / 2 - Y_CENTER), rl.Color(255, 255, 255, 255))
|
||
last_x += self._mic_txt.width + ITEM_SPACING
|
||
|
||
def _draw_cellular_cluster(self, start_x: float, spacing: int, y_center: int, connected: bool) -> float:
|
||
draw_net_txt = {0: self._cell_none_txt,
|
||
2: self._cell_low_txt,
|
||
3: self._cell_medium_txt,
|
||
4: self._cell_high_txt,
|
||
5: self._cell_full_txt}.get(self._net_strength, self._cell_none_txt)
|
||
|
||
icon_y = int(self._rect.y + self.rect.height - draw_net_txt.height / 2 - y_center)
|
||
rl.draw_texture(draw_net_txt, int(start_x), icon_y, rl.Color(255, 255, 255, int(255 * 0.9)))
|
||
|
||
label_x = start_x + draw_net_txt.width + 12
|
||
label_y = self._rect.y + self.rect.height - 37
|
||
label_rect = rl.Rectangle(label_x, label_y, 44, 24)
|
||
self._lte_label.render(label_rect)
|
||
|
||
if not connected:
|
||
slash_start = rl.Vector2(start_x + 8, icon_y + draw_net_txt.height - 3)
|
||
slash_end = rl.Vector2(start_x + draw_net_txt.width - 2, icon_y + 3)
|
||
rl.draw_line_ex(slash_start, slash_end, 8, rl.Color(255, 255, 255, int(255 * 0.22)))
|
||
rl.draw_line_ex(slash_start, slash_end, 5, rl.Color(255, 255, 255, int(255 * 0.82)))
|
||
|
||
return label_x + label_rect.width + spacing
|