IQ.Pilot Prebuilt Release @ 27f668a

iqpilot/selfdrive/ui/widgets/setup.py

@@ -0,0 +1,85 @@
+import math
+import pyray as rl
+from iqpilot.common.time_helpers import system_time_valid
+from iqpilot.selfdrive.ui.ui_state import ui_state
+from iqpilot.selfdrive.ui.widgets.pairing_dialog import PairingDialog
+from iqpilot.system.ui.lib.application import gui_app, FontWeight, FONT_SCALE
+from iqpilot.system.ui.lib.multilang import tr
+from iqpilot.system.ui.lib.text_measure import measure_text_cached
+from iqpilot.system.ui.lib.wrap_text import wrap_text
+from iqpilot.system.ui.widgets import Widget
+from iqpilot.system.ui.widgets.confirm_dialog import alert_dialog
+from iqpilot.system.ui.widgets.button import Button, ButtonStyle
+
+SETUP_CARD_BG = rl.Color(34, 36, 42, 255)
+SETUP_CARD_BORDER = rl.Color(255, 255, 255, 26)
+SETUP_ACCENT = rl.Color(16, 185, 169, 255)
+
+
+class SetupWidget(Widget):
+  def __init__(self):
+    super().__init__()
+    self._pairing_dialog: PairingDialog | None = None
+    self._pair_device_btn = Button(lambda: tr("Pair device"), self._show_pairing, button_style=ButtonStyle.PRIMARY)
+
+  def _render(self, rect: rl.Rectangle):
+    if not ui_state.prime_state.is_paired():
+      self._render_registration(rect)
+
+  def _render_registration(self, rect: rl.Rectangle):
+    """Render registration prompt."""
+
+    t = rl.get_time()
+    pulse = 0.5 + 0.5 * math.sin(t * 2.3)
+    glow_alpha = int(24 + pulse * 34)
+    border_alpha = int(28 + pulse * 38)
+
+    glow_rect = rl.Rectangle(rect.x - 4, rect.y - 4, rect.width + 8, rect.height + 8)
+    rl.draw_rectangle_rounded_lines_ex(glow_rect, 0.06, 24, 5, rl.Color(SETUP_ACCENT.r, SETUP_ACCENT.g, SETUP_ACCENT.b, glow_alpha))
+    rl.draw_rectangle_rounded(rl.Rectangle(rect.x, rect.y, rect.width, rect.height), 0.06, 24, SETUP_CARD_BG)
+    rl.draw_rectangle_rounded_lines_ex(rl.Rectangle(rect.x, rect.y, rect.width, rect.height), 0.06, 24, 2, SETUP_CARD_BORDER)
+    rl.draw_rectangle_rounded_lines_ex(rl.Rectangle(rect.x, rect.y, rect.width, rect.height), 0.06, 24, 2,
+                                       rl.Color(SETUP_ACCENT.r, SETUP_ACCENT.g, SETUP_ACCENT.b, border_alpha))
+
+    x = rect.x + 64
+    w = rect.width - 128
+
+    font = gui_app.font(FontWeight.BOLD)
+    title = tr("Finish Setup")
+    title_size = measure_text_cached(font, title, 75)
+    desc = tr("Pair your device in the Konn3kt app.")
+    light_font = gui_app.font(FontWeight.NORMAL)
+    wrapped = wrap_text(light_font, desc, 50, int(w))
+
+    desc_line_h = 50 * FONT_SCALE
+    content_h = title_size.y + 38 + len(wrapped) * desc_line_h + 30 + 200
+    y = rect.y + (rect.height - content_h) / 2 - 44
+
+    title_x = x + (w - title_size.x) / 2
+    rl.draw_text_ex(font, title, rl.Vector2(int(title_x), int(y)), 75, 0, rl.WHITE)
+    y += title_size.y + 38
+
+    for line in wrapped:
+      line_size = measure_text_cached(light_font, line, 50)
+      line_x = x + (w - line_size.x) / 2
+      rl.draw_text_ex(light_font, line, rl.Vector2(int(line_x), int(y)), 50, 0, rl.WHITE)
+      y += desc_line_h
+
+    button_rect = rl.Rectangle(x, y + 30, w, 200)
+    cta_glow = rl.Rectangle(button_rect.x - 8, button_rect.y - 8, button_rect.width + 16, button_rect.height + 16)
+    rl.draw_rectangle_rounded(cta_glow, 0.5, 24, rl.Color(SETUP_ACCENT.r, SETUP_ACCENT.g, SETUP_ACCENT.b, int(16 + pulse * 20)))
+    self._pair_device_btn.render(button_rect)
+
+  def _show_pairing(self):
+    if not system_time_valid():
+      dlg = alert_dialog(tr("Please connect to Wi-Fi to complete initial pairing"))
+      gui_app.set_modal_overlay(dlg)
+      return
+
+    if not self._pairing_dialog:
+      self._pairing_dialog = PairingDialog()
+    gui_app.set_modal_overlay(self._pairing_dialog, lambda result: setattr(self, '_pairing_dialog', None))
+
+  def __del__(self):
+    if self._pairing_dialog:
+      del self._pairing_dialog