""" 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