mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-18 20:03:53 +08:00
b6dbb0fd8d
* pt 2 * fix line height * fixup html renderer * fix sidebar * fix label line height * firehose fixups * fix ssh value font styling * fixup inputbot * do experimental mode * pairing dialog numbers * fix radius for prime user * add emoji to firehose mode * full screen registration * fix registration btn size * fix update and alerts * debugging * Revert "debugging" This reverts commit 0095372e9479d8c727bcc8a78061f582d852133d. * firehose styling * fix offroad alerts missing bottom spacing expansion * huge oof * huge oof
95 lines
3.4 KiB
Python
95 lines
3.4 KiB
Python
import pyray as rl
|
|
from openpilot.selfdrive.ui.ui_state import ui_state
|
|
from openpilot.selfdrive.ui.widgets.pairing_dialog import PairingDialog
|
|
from openpilot.system.ui.lib.application import gui_app, FontWeight, FONT_SCALE
|
|
from openpilot.system.ui.lib.wrap_text import wrap_text
|
|
from openpilot.system.ui.widgets import Widget
|
|
from openpilot.system.ui.widgets.button import Button, ButtonStyle
|
|
from openpilot.system.ui.widgets.label import Label
|
|
|
|
|
|
class SetupWidget(Widget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._open_settings_callback = None
|
|
self._pairing_dialog: PairingDialog | None = None
|
|
self._pair_device_btn = Button("Pair device", self._show_pairing, button_style=ButtonStyle.PRIMARY)
|
|
self._open_settings_btn = Button("Open", lambda: self._open_settings_callback() if self._open_settings_callback else None,
|
|
button_style=ButtonStyle.PRIMARY)
|
|
self._firehose_label = Label("🔥Firehose Mode 🔥", font_weight=FontWeight.MEDIUM, font_size=64)
|
|
|
|
def set_open_settings_callback(self, callback):
|
|
self._open_settings_callback = callback
|
|
|
|
def _render(self, rect: rl.Rectangle):
|
|
if not ui_state.prime_state.is_paired():
|
|
self._render_registration(rect)
|
|
else:
|
|
self._render_firehose_prompt(rect)
|
|
|
|
def _render_registration(self, rect: rl.Rectangle):
|
|
"""Render registration prompt."""
|
|
|
|
rl.draw_rectangle_rounded(rl.Rectangle(rect.x, rect.y, rect.width, rect.height), 0.02, 20, rl.Color(51, 51, 51, 255))
|
|
|
|
x = rect.x + 64
|
|
y = rect.y + 48
|
|
w = rect.width - 128
|
|
|
|
# Title
|
|
font = gui_app.font(FontWeight.BOLD)
|
|
rl.draw_text_ex(font, "Finish Setup", rl.Vector2(x, y), 75, 0, rl.WHITE)
|
|
y += 113 # 75 + 38 spacing
|
|
|
|
# Description
|
|
desc = "Pair your device with comma connect (connect.comma.ai) and claim your comma prime offer."
|
|
light_font = gui_app.font(FontWeight.LIGHT)
|
|
wrapped = wrap_text(light_font, desc, 50, int(w))
|
|
for line in wrapped:
|
|
rl.draw_text_ex(light_font, line, rl.Vector2(x, y), 50, 0, rl.WHITE)
|
|
y += 50 * FONT_SCALE
|
|
|
|
button_rect = rl.Rectangle(x, y + 30, w, 200)
|
|
self._pair_device_btn.render(button_rect)
|
|
|
|
def _render_firehose_prompt(self, rect: rl.Rectangle):
|
|
"""Render firehose prompt widget."""
|
|
|
|
rl.draw_rectangle_rounded(rl.Rectangle(rect.x, rect.y, rect.width, 500), 0.02, 20, rl.Color(51, 51, 51, 255))
|
|
|
|
# Content margins (56, 40, 56, 40)
|
|
x = rect.x + 56
|
|
y = rect.y + 40
|
|
w = rect.width - 112
|
|
spacing = 42
|
|
|
|
# Title with fire emojis
|
|
# TODO: fix Label centering with emojis
|
|
self._firehose_label.render(rl.Rectangle(x - 40, y, w, 64))
|
|
y += 64 + spacing
|
|
|
|
# Description
|
|
desc_font = gui_app.font(FontWeight.NORMAL)
|
|
desc_text = "Maximize your training data uploads to improve openpilot's driving models."
|
|
wrapped_desc = wrap_text(desc_font, desc_text, 40, int(w))
|
|
|
|
for line in wrapped_desc:
|
|
rl.draw_text_ex(desc_font, line, rl.Vector2(x, y), 40, 0, rl.WHITE)
|
|
y += 40 * FONT_SCALE
|
|
|
|
y += spacing
|
|
|
|
# Open button
|
|
button_height = 48 + 64 # font size + padding
|
|
button_rect = rl.Rectangle(x, y, w, button_height)
|
|
self._open_settings_btn.render(button_rect)
|
|
|
|
def _show_pairing(self):
|
|
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
|