ui(raylib): setup.py (#35140)
* setup.py * better font * use gui_button * btn * fix button and triangle * low voltage text color * fix network page * HARDWARE.get_os_version() * typing * white title * update default text color * use default font color * fix software screen * fix software screen * radio font size * line length * fix regex * draw svgs * comment is out of date * add cairosvg * use cairosvg * remove unused import * support other image types * revert origin * fix setup warning icon * fix * remove cairosvg * use pngs * wrap * fix disabled style * TODO * revert uv.lock * use new file paths (not rasterized yet) * oops * fixes * params not used * network check thread * oops * fix custom URL and download failed screens * clear keyboard * rm * fixes * show full error message * check network typepull/35247/head
Cameron Clough
6 months ago
committed by
GitHub
parent
83679bd856
commit
512d83cc36
2 changed files with 345 additions and 1 deletions
@ -0,0 +1,344 @@ |
|||||||
|
#!/usr/bin/env python3 |
||||||
|
import os |
||||||
|
import re |
||||||
|
import threading |
||||||
|
import time |
||||||
|
import urllib.request |
||||||
|
from enum import IntEnum |
||||||
|
import pyray as rl |
||||||
|
|
||||||
|
from cereal import log |
||||||
|
from openpilot.system.hardware import HARDWARE |
||||||
|
from openpilot.system.ui.lib.application import gui_app, FontWeight |
||||||
|
from openpilot.system.ui.lib.button import gui_button, ButtonStyle |
||||||
|
from openpilot.system.ui.lib.label import gui_label, gui_text_box |
||||||
|
from openpilot.system.ui.widgets.network import WifiManagerUI, WifiManagerWrapper |
||||||
|
from openpilot.system.ui.widgets.keyboard import Keyboard |
||||||
|
|
||||||
|
NetworkType = log.DeviceState.NetworkType |
||||||
|
|
||||||
|
MARGIN = 50 |
||||||
|
TITLE_FONT_SIZE = 116 |
||||||
|
TITLE_FONT_WEIGHT = FontWeight.MEDIUM |
||||||
|
NEXT_BUTTON_WIDTH = 310 |
||||||
|
BODY_FONT_SIZE = 96 |
||||||
|
BUTTON_HEIGHT = 160 |
||||||
|
BUTTON_SPACING = 50 |
||||||
|
|
||||||
|
OPENPILOT_URL = "https://openpilot.comma.ai" |
||||||
|
USER_AGENT = f"AGNOSSetup-{HARDWARE.get_os_version()}" |
||||||
|
|
||||||
|
|
||||||
|
class SetupState(IntEnum): |
||||||
|
LOW_VOLTAGE = 0 |
||||||
|
GETTING_STARTED = 1 |
||||||
|
NETWORK_SETUP = 2 |
||||||
|
SOFTWARE_SELECTION = 3 |
||||||
|
CUSTOM_URL = 4 |
||||||
|
DOWNLOADING = 5 |
||||||
|
DOWNLOAD_FAILED = 6 |
||||||
|
|
||||||
|
|
||||||
|
class Setup: |
||||||
|
def __init__(self): |
||||||
|
self.state = SetupState.GETTING_STARTED |
||||||
|
self.network_check_thread = None |
||||||
|
self.network_connected = threading.Event() |
||||||
|
self.wifi_connected = threading.Event() |
||||||
|
self.stop_network_check_thread = threading.Event() |
||||||
|
self.failed_url = "" |
||||||
|
self.failed_reason = "" |
||||||
|
self.download_url = "" |
||||||
|
self.download_progress = 0 |
||||||
|
self.download_thread = None |
||||||
|
self.wifi_manager = WifiManagerWrapper() |
||||||
|
self.wifi_ui = WifiManagerUI(self.wifi_manager) |
||||||
|
self.keyboard = Keyboard() |
||||||
|
self.selected_radio = None |
||||||
|
|
||||||
|
self.warning = gui_app.texture("icons/warning.png", 150, 150) |
||||||
|
self.checkmark = gui_app.texture("icons/circled_check.png", 100, 100) |
||||||
|
|
||||||
|
try: |
||||||
|
with open("/sys/class/hwmon/hwmon1/in1_input") as f: |
||||||
|
voltage = float(f.read().strip()) / 1000.0 |
||||||
|
if voltage < 7: |
||||||
|
self.state = SetupState.LOW_VOLTAGE |
||||||
|
except (FileNotFoundError, ValueError): |
||||||
|
self.state = SetupState.LOW_VOLTAGE |
||||||
|
|
||||||
|
def render(self, rect: rl.Rectangle): |
||||||
|
if self.state == SetupState.LOW_VOLTAGE: |
||||||
|
self.render_low_voltage(rect) |
||||||
|
elif self.state == SetupState.GETTING_STARTED: |
||||||
|
self.render_getting_started(rect) |
||||||
|
elif self.state == SetupState.NETWORK_SETUP: |
||||||
|
self.render_network_setup(rect) |
||||||
|
elif self.state == SetupState.SOFTWARE_SELECTION: |
||||||
|
self.render_software_selection(rect) |
||||||
|
elif self.state == SetupState.CUSTOM_URL: |
||||||
|
self.render_custom_url() |
||||||
|
elif self.state == SetupState.DOWNLOADING: |
||||||
|
self.render_downloading(rect) |
||||||
|
elif self.state == SetupState.DOWNLOAD_FAILED: |
||||||
|
self.render_download_failed(rect) |
||||||
|
|
||||||
|
def render_low_voltage(self, rect: rl.Rectangle): |
||||||
|
rl.draw_texture(self.warning, int(rect.x + 150), int(rect.y + 110), rl.WHITE) |
||||||
|
|
||||||
|
title_rect = rl.Rectangle(rect.x + 150, rect.y + 110 + 150 + 100, rect.width - 500 - 150, TITLE_FONT_SIZE) |
||||||
|
gui_label(title_rect, "WARNING: Low Voltage", TITLE_FONT_SIZE, rl.Color(255, 89, 79, 255), FontWeight.MEDIUM) |
||||||
|
|
||||||
|
body_rect = rl.Rectangle(rect.x + 150, rect.y + 110 + 150 + 100 + TITLE_FONT_SIZE + 25, rect.width - 500 - 150, BODY_FONT_SIZE * 3) |
||||||
|
gui_text_box(body_rect, "Power your device in a car with a harness or proceed at your own risk.", BODY_FONT_SIZE) |
||||||
|
|
||||||
|
button_width = (rect.width - MARGIN * 3) / 2 |
||||||
|
button_y = rect.height - MARGIN - BUTTON_HEIGHT |
||||||
|
|
||||||
|
if gui_button(rl.Rectangle(rect.x + MARGIN, button_y, button_width, BUTTON_HEIGHT), "Power off"): |
||||||
|
HARDWARE.shutdown() |
||||||
|
|
||||||
|
if gui_button(rl.Rectangle(rect.x + MARGIN * 2 + button_width, button_y, button_width, BUTTON_HEIGHT), "Continue"): |
||||||
|
self.state = SetupState.GETTING_STARTED |
||||||
|
|
||||||
|
def render_getting_started(self, rect: rl.Rectangle): |
||||||
|
title_rect = rl.Rectangle(rect.x + 165, rect.y + 280, rect.width - 265, TITLE_FONT_SIZE) |
||||||
|
gui_label(title_rect, "Getting Started", TITLE_FONT_SIZE, font_weight=FontWeight.MEDIUM) |
||||||
|
|
||||||
|
desc_rect = rl.Rectangle(rect.x + 165, rect.y + 280 + TITLE_FONT_SIZE + 90, rect.width - 500, BODY_FONT_SIZE * 3) |
||||||
|
gui_text_box(desc_rect, "Before we get on the road, let's finish installation and cover some details.", BODY_FONT_SIZE) |
||||||
|
|
||||||
|
btn_rect = rl.Rectangle(rect.width - NEXT_BUTTON_WIDTH, 0, NEXT_BUTTON_WIDTH, rect.height) |
||||||
|
|
||||||
|
ret = gui_button(btn_rect, "", button_style=ButtonStyle.PRIMARY, border_radius=0) |
||||||
|
triangle = gui_app.texture("images/button_continue_triangle.png", 54, int(btn_rect.height)) |
||||||
|
rl.draw_texture_v(triangle, rl.Vector2(btn_rect.x + btn_rect.width / 2 - triangle.width / 2, btn_rect.height / 2 - triangle.height / 2), rl.WHITE) |
||||||
|
|
||||||
|
if ret: |
||||||
|
self.state = SetupState.NETWORK_SETUP |
||||||
|
self.wifi_manager.request_scan() |
||||||
|
self.start_network_check() |
||||||
|
|
||||||
|
def check_network_connectivity(self): |
||||||
|
while not self.stop_network_check_thread.is_set(): |
||||||
|
if self.state == SetupState.NETWORK_SETUP: |
||||||
|
try: |
||||||
|
urllib.request.urlopen("https://google.com", timeout=2) |
||||||
|
self.network_connected.set() |
||||||
|
if HARDWARE.get_network_type() == NetworkType.wifi: |
||||||
|
self.wifi_connected.set() |
||||||
|
else: |
||||||
|
self.wifi_connected.clear() |
||||||
|
except Exception: |
||||||
|
self.network_connected.clear() |
||||||
|
time.sleep(1) |
||||||
|
|
||||||
|
def start_network_check(self): |
||||||
|
if self.network_check_thread is None or not self.network_check_thread.is_alive(): |
||||||
|
self.network_check_thread = threading.Thread(target=self.check_network_connectivity, daemon=True) |
||||||
|
self.network_check_thread.start() |
||||||
|
|
||||||
|
def close(self): |
||||||
|
if self.network_check_thread is not None: |
||||||
|
self.stop_network_check_thread.set() |
||||||
|
self.network_check_thread.join() |
||||||
|
|
||||||
|
def render_network_setup(self, rect: rl.Rectangle): |
||||||
|
title_rect = rl.Rectangle(rect.x + MARGIN, rect.y + MARGIN, rect.width - MARGIN * 2, TITLE_FONT_SIZE) |
||||||
|
gui_label(title_rect, "Connect to Wi-Fi", TITLE_FONT_SIZE, font_weight=FontWeight.MEDIUM) |
||||||
|
|
||||||
|
wifi_rect = rl.Rectangle(rect.x + MARGIN, rect.y + TITLE_FONT_SIZE + MARGIN + 25, rect.width - MARGIN * 2, |
||||||
|
rect.height - TITLE_FONT_SIZE - 25 - BUTTON_HEIGHT - MARGIN * 3) |
||||||
|
rl.draw_rectangle_rounded(wifi_rect, 0.05, 10, rl.Color(51, 51, 51, 255)) |
||||||
|
wifi_content_rect = rl.Rectangle(wifi_rect.x + MARGIN, wifi_rect.y, wifi_rect.width - MARGIN * 2, wifi_rect.height) |
||||||
|
self.wifi_ui.render(wifi_content_rect) |
||||||
|
|
||||||
|
button_width = (rect.width - BUTTON_SPACING - MARGIN * 2) / 2 |
||||||
|
button_y = rect.height - BUTTON_HEIGHT - MARGIN |
||||||
|
|
||||||
|
if gui_button(rl.Rectangle(rect.x + MARGIN, button_y, button_width, BUTTON_HEIGHT), "Back"): |
||||||
|
self.state = SetupState.GETTING_STARTED |
||||||
|
|
||||||
|
# Check network connectivity status |
||||||
|
continue_enabled = self.network_connected.is_set() |
||||||
|
continue_text = ("Continue" if self.wifi_connected.is_set() else "Continue without Wi-Fi") if continue_enabled else "Waiting for internet" |
||||||
|
|
||||||
|
if gui_button( |
||||||
|
rl.Rectangle(rect.x + MARGIN + button_width + BUTTON_SPACING, button_y, button_width, BUTTON_HEIGHT), |
||||||
|
continue_text, |
||||||
|
button_style=ButtonStyle.PRIMARY if continue_enabled else ButtonStyle.NORMAL, |
||||||
|
): |
||||||
|
self.state = SetupState.SOFTWARE_SELECTION |
||||||
|
self.stop_network_check_thread.set() |
||||||
|
|
||||||
|
def render_software_selection(self, rect: rl.Rectangle): |
||||||
|
title_rect = rl.Rectangle(rect.x + MARGIN, rect.y + MARGIN, rect.width - MARGIN * 2, TITLE_FONT_SIZE) |
||||||
|
gui_label(title_rect, "Choose Software to Install", TITLE_FONT_SIZE, font_weight=FontWeight.MEDIUM) |
||||||
|
|
||||||
|
radio_height = 230 |
||||||
|
radio_spacing = 30 |
||||||
|
|
||||||
|
openpilot_rect = rl.Rectangle(rect.x + MARGIN, rect.y + TITLE_FONT_SIZE + MARGIN * 2, rect.width - MARGIN * 2, radio_height) |
||||||
|
openpilot_selected = self.selected_radio == "openpilot" |
||||||
|
|
||||||
|
rl.draw_rectangle_rounded(openpilot_rect, 0.1, 10, rl.Color(70, 91, 234, 255) if openpilot_selected else rl.Color(79, 79, 79, 255)) |
||||||
|
gui_label(rl.Rectangle(openpilot_rect.x + 100, openpilot_rect.y, openpilot_rect.width - 200, radio_height), "openpilot", BODY_FONT_SIZE) |
||||||
|
|
||||||
|
if openpilot_selected: |
||||||
|
checkmark_pos = rl.Vector2(openpilot_rect.x + openpilot_rect.width - 100 - self.checkmark.width, |
||||||
|
openpilot_rect.y + radio_height / 2 - self.checkmark.height / 2) |
||||||
|
rl.draw_texture_v(self.checkmark, checkmark_pos, rl.WHITE) |
||||||
|
|
||||||
|
custom_rect = rl.Rectangle(rect.x + MARGIN, rect.y + TITLE_FONT_SIZE + MARGIN * 2 + radio_height + radio_spacing, rect.width - MARGIN * 2, radio_height) |
||||||
|
custom_selected = self.selected_radio == "custom" |
||||||
|
|
||||||
|
rl.draw_rectangle_rounded(custom_rect, 0.1, 10, rl.Color(70, 91, 234, 255) if custom_selected else rl.Color(79, 79, 79, 255)) |
||||||
|
gui_label(rl.Rectangle(custom_rect.x + 100, custom_rect.y, custom_rect.width - 200, radio_height), "Custom Software", BODY_FONT_SIZE) |
||||||
|
|
||||||
|
if custom_selected: |
||||||
|
checkmark_pos = rl.Vector2(custom_rect.x + custom_rect.width - 100 - self.checkmark.width, custom_rect.y + radio_height / 2 - self.checkmark.height / 2) |
||||||
|
rl.draw_texture_v(self.checkmark, checkmark_pos, rl.WHITE) |
||||||
|
|
||||||
|
mouse_pos = rl.get_mouse_position() |
||||||
|
if rl.is_mouse_button_released(rl.MouseButton.MOUSE_BUTTON_LEFT): |
||||||
|
if rl.check_collision_point_rec(mouse_pos, openpilot_rect): |
||||||
|
self.selected_radio = "openpilot" |
||||||
|
elif rl.check_collision_point_rec(mouse_pos, custom_rect): |
||||||
|
self.selected_radio = "custom" |
||||||
|
|
||||||
|
button_width = (rect.width - BUTTON_SPACING - MARGIN * 2) / 2 |
||||||
|
button_y = rect.height - BUTTON_HEIGHT - MARGIN |
||||||
|
|
||||||
|
if gui_button(rl.Rectangle(rect.x + MARGIN, button_y, button_width, BUTTON_HEIGHT), "Back"): |
||||||
|
self.state = SetupState.NETWORK_SETUP |
||||||
|
|
||||||
|
continue_enabled = self.selected_radio is not None |
||||||
|
if gui_button( |
||||||
|
rl.Rectangle(rect.x + MARGIN + button_width + BUTTON_SPACING, button_y, button_width, BUTTON_HEIGHT), |
||||||
|
"Continue", |
||||||
|
button_style=ButtonStyle.PRIMARY, |
||||||
|
is_enabled=continue_enabled, |
||||||
|
): |
||||||
|
if continue_enabled: |
||||||
|
if self.selected_radio == "openpilot": |
||||||
|
self.download(OPENPILOT_URL) |
||||||
|
else: |
||||||
|
self.state = SetupState.CUSTOM_URL |
||||||
|
|
||||||
|
def render_downloading(self, rect: rl.Rectangle): |
||||||
|
title_rect = rl.Rectangle(rect.x, rect.y + rect.height / 2 - TITLE_FONT_SIZE / 2, rect.width, TITLE_FONT_SIZE) |
||||||
|
gui_label(title_rect, "Downloading...", TITLE_FONT_SIZE, font_weight=FontWeight.MEDIUM, alignment=rl.GuiTextAlignment.TEXT_ALIGN_CENTER) |
||||||
|
|
||||||
|
def render_download_failed(self, rect: rl.Rectangle): |
||||||
|
title_rect = rl.Rectangle(rect.x + 117, rect.y + 185, rect.width - 117, TITLE_FONT_SIZE) |
||||||
|
gui_label(title_rect, "Download Failed", TITLE_FONT_SIZE, font_weight=FontWeight.MEDIUM) |
||||||
|
|
||||||
|
url_rect = rl.Rectangle(rect.x + 117, rect.y + 185 + TITLE_FONT_SIZE + 67, rect.width - 117 - 100, 64) |
||||||
|
gui_label(url_rect, self.failed_url, 64, font_weight=FontWeight.NORMAL) |
||||||
|
|
||||||
|
error_rect = rl.Rectangle(rect.x + 117, rect.y + 185 + TITLE_FONT_SIZE + 67 + 64 + 48, |
||||||
|
rect.width - 117 - 100, rect.height - 185 + TITLE_FONT_SIZE + 67 + 64 + 48 - BUTTON_HEIGHT - MARGIN * 2) |
||||||
|
gui_text_box(error_rect, self.failed_reason, BODY_FONT_SIZE) |
||||||
|
|
||||||
|
button_width = (rect.width - BUTTON_SPACING - MARGIN * 2) / 2 |
||||||
|
button_y = rect.height - BUTTON_HEIGHT - MARGIN |
||||||
|
|
||||||
|
if gui_button(rl.Rectangle(rect.x + MARGIN, button_y, button_width, BUTTON_HEIGHT), "Reboot device"): |
||||||
|
HARDWARE.reboot() |
||||||
|
|
||||||
|
if gui_button(rl.Rectangle(rect.x + MARGIN + button_width + BUTTON_SPACING, button_y, button_width, BUTTON_HEIGHT), "Start over", |
||||||
|
button_style=ButtonStyle.PRIMARY): |
||||||
|
self.state = SetupState.GETTING_STARTED |
||||||
|
|
||||||
|
def render_custom_url(self): |
||||||
|
result = self.keyboard.render("Enter URL", "for Custom Software") |
||||||
|
|
||||||
|
# Enter pressed |
||||||
|
if result == 1: |
||||||
|
url = self.keyboard.text |
||||||
|
self.keyboard.clear() |
||||||
|
if url: |
||||||
|
self.download(url) |
||||||
|
|
||||||
|
# Cancel pressed |
||||||
|
elif result == 0: |
||||||
|
self.state = SetupState.SOFTWARE_SELECTION |
||||||
|
|
||||||
|
def download(self, url: str): |
||||||
|
# autocomplete incomplete URLs |
||||||
|
if re.match("^([^/.]+)/([^/]+)$", url): |
||||||
|
url = f"https://installer.comma.ai/{url}" |
||||||
|
|
||||||
|
self.download_url = url |
||||||
|
self.state = SetupState.DOWNLOADING |
||||||
|
|
||||||
|
self.download_thread = threading.Thread(target=self._download_thread, daemon=True) |
||||||
|
self.download_thread.start() |
||||||
|
|
||||||
|
def _download_thread(self): |
||||||
|
try: |
||||||
|
import tempfile |
||||||
|
|
||||||
|
_, tmpfile = tempfile.mkstemp(prefix="installer_") |
||||||
|
|
||||||
|
headers = {"User-Agent": USER_AGENT, "X-openpilot-serial": HARDWARE.get_serial()} |
||||||
|
req = urllib.request.Request(self.download_url, headers=headers) |
||||||
|
|
||||||
|
with open(tmpfile, 'wb') as f, urllib.request.urlopen(req, timeout=30) as response: |
||||||
|
total_size = int(response.headers.get('content-length', 0)) |
||||||
|
downloaded = 0 |
||||||
|
block_size = 8192 |
||||||
|
|
||||||
|
while True: |
||||||
|
buffer = response.read(block_size) |
||||||
|
if not buffer: |
||||||
|
break |
||||||
|
|
||||||
|
downloaded += len(buffer) |
||||||
|
f.write(buffer) |
||||||
|
|
||||||
|
if total_size: |
||||||
|
self.download_progress = int(downloaded * 100 / total_size) |
||||||
|
|
||||||
|
is_elf = False |
||||||
|
with open(tmpfile, 'rb') as f: |
||||||
|
header = f.read(4) |
||||||
|
is_elf = header == b'\x7fELF' |
||||||
|
|
||||||
|
if not is_elf: |
||||||
|
self.download_failed(self.download_url, "No custom software found at this URL.") |
||||||
|
return |
||||||
|
|
||||||
|
os.rename(tmpfile, "/tmp/installer") |
||||||
|
os.chmod("/tmp/installer", 0o755) |
||||||
|
|
||||||
|
with open("/tmp/installer_url", "w") as f: |
||||||
|
f.write(self.download_url) |
||||||
|
|
||||||
|
gui_app.request_close() |
||||||
|
|
||||||
|
except Exception: |
||||||
|
error_msg = "Ensure the entered URL is valid, and the device's internet connection is good." |
||||||
|
self.download_failed(self.download_url, error_msg) |
||||||
|
|
||||||
|
def download_failed(self, url: str, reason: str): |
||||||
|
self.failed_url = url |
||||||
|
self.failed_reason = reason |
||||||
|
self.state = SetupState.DOWNLOAD_FAILED |
||||||
|
|
||||||
|
|
||||||
|
def main(): |
||||||
|
try: |
||||||
|
gui_app.init_window("Setup") |
||||||
|
setup = Setup() |
||||||
|
for _ in gui_app.render(): |
||||||
|
setup.render(rl.Rectangle(0, 0, gui_app.width, gui_app.height)) |
||||||
|
setup.close() |
||||||
|
except Exception as e: |
||||||
|
print(f"Setup error: {e}") |
||||||
|
finally: |
||||||
|
gui_app.close() |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__": |
||||||
|
main() |
||||||
Loading…
Reference in new issue