mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-12 15:33:45 +08:00
firestarsdog
185 lines
10 KiB
Python
185 lines
10 KiB
Python
from collections.abc import Callable
|
|
|
|
import pyray as rl
|
|
|
|
from openpilot.selfdrive.ui.layouts.arabesque.theme import DEFAULT_THEME
|
|
from openpilot.selfdrive.ui.layouts.arabesque.tools.browser import Browser
|
|
from openpilot.selfdrive.ui.layouts.arabesque.tools.files import Files
|
|
from openpilot.selfdrive.ui.layouts.arabesque.tools.rebecca import Rebecca
|
|
from openpilot.selfdrive.ui.layouts.arabesque.tools.terminal import Terminal
|
|
from openpilot.system.ui.lib.application import gui_app, MouseEvent, MousePos, TextAlignment
|
|
from openpilot.system.ui.lib.multilang import tr
|
|
from openpilot.system.ui.widgets import Widget
|
|
from openpilot.system.ui.widgets.label import gui_label
|
|
|
|
|
|
_DOCK_SHADOW = rl.Color(0, 0, 0, 52)
|
|
_DOCK_FACE = rl.Color(72, 54, 98, 255)
|
|
_DOCK_RIM = rl.Color(113, 88, 147, 255)
|
|
|
|
|
|
class ArabesqueLayout(Widget):
|
|
def __init__(self, on_exit: Callable[[], None], permitted: Callable[[], bool] = lambda: True):
|
|
super().__init__()
|
|
self._on_exit = on_exit
|
|
self._permitted = permitted
|
|
self._menu_open = False
|
|
self._pressed_rect: rl.Rectangle | None = None
|
|
self._active_tool: Widget | None = None
|
|
self._active_name = ""
|
|
|
|
def _update_layout_rects(self):
|
|
rect = self._rect
|
|
# GuiApplication scales both drawing and input from these logical coordinates.
|
|
self._scale = s = min(rect.width / 2160, rect.height / 1080)
|
|
self._bar = rl.Rectangle(rect.x, rect.y, rect.width, 104 * s)
|
|
self._menu_button = rl.Rectangle(rect.x + 24 * s, rect.y + 8 * s, 88 * s, 88 * s)
|
|
self._exit_button = rl.Rectangle(rect.x + 24 * s, rect.y + 116 * s, 456 * s, 116 * s)
|
|
self._back_button = rl.Rectangle(rect.x + rect.width - 296 * s, rect.y + 8 * s, 272 * s, 88 * s)
|
|
self._dock = dock = rl.Rectangle(rect.x + (rect.width - 1080 * s) / 2, rect.y + rect.height - 272 * s, 1080 * s, 240 * s)
|
|
self._dock_shelf = rl.Rectangle(dock.x, dock.y + 178 * s, dock.width, 10 * s)
|
|
self._dock_top = (rl.Vector2(dock.x + 36 * s, dock.y + 118 * s), rl.Vector2(dock.x, self._dock_shelf.y),
|
|
rl.Vector2(dock.x + dock.width, self._dock_shelf.y), rl.Vector2(dock.x + dock.width - 36 * s, dock.y + 118 * s))
|
|
self._dock_shadows = (rl.Rectangle(dock.x - 10 * s, dock.y + 174 * s, dock.width + 20 * s, 32 * s),
|
|
rl.Rectangle(dock.x - 4 * s, dock.y + 178 * s, dock.width + 8 * s, 22 * s))
|
|
self._launchers = (
|
|
(rl.Rectangle(dock.x + 60 * s, dock.y, 240 * s, 200 * s), "Terminal", lambda: Terminal()),
|
|
(rl.Rectangle(dock.x + 300 * s, dock.y, 240 * s, 200 * s), "Files", lambda: Files(self._permitted)),
|
|
(rl.Rectangle(dock.x + 540 * s, dock.y, 240 * s, 200 * s), "Rebecca", lambda: Rebecca()),
|
|
(rl.Rectangle(dock.x + 780 * s, dock.y, 240 * s, 200 * s), "Firefox", lambda: Browser(self._permitted)),
|
|
)
|
|
self._tool_rect = rl.Rectangle(rect.x, self._bar.y + self._bar.height, rect.width, rect.height - self._bar.height)
|
|
|
|
def _open_tool(self, name: str, tool: Widget):
|
|
self._close_tool()
|
|
self._active_tool, self._active_name = tool, name
|
|
tool.set_rect(self._tool_rect)
|
|
tool.show_event()
|
|
|
|
def _close_tool(self):
|
|
tool, self._active_tool = self._active_tool, None
|
|
self._active_name = ""
|
|
self._menu_open = False
|
|
self._pressed_rect = None
|
|
gui_app.mouse_events.clear()
|
|
# Transient tools have one explicit lifecycle owner, rather than persistent _child registration.
|
|
if tool is not None:
|
|
tool.hide_event()
|
|
|
|
def hide_event(self):
|
|
super().hide_event()
|
|
self._close_tool()
|
|
|
|
def poll_runtime(self, awake: bool):
|
|
if isinstance(self._active_tool, Browser):
|
|
self._active_tool.poll_runtime(awake)
|
|
|
|
def _handle_mouse_press(self, pos: MousePos):
|
|
self._pressed_rect = None
|
|
if rl.check_collision_point_rec(pos, self._menu_button):
|
|
self._pressed_rect = self._menu_button
|
|
elif self._menu_open:
|
|
if rl.check_collision_point_rec(pos, self._exit_button):
|
|
self._pressed_rect = self._exit_button
|
|
elif self._active_tool is not None and rl.check_collision_point_rec(pos, self._back_button):
|
|
self._pressed_rect = self._back_button
|
|
elif self._active_tool is None:
|
|
self._pressed_rect = next((target for target, _, _ in self._launchers if rl.check_collision_point_rec(pos, target)), None)
|
|
|
|
def _handle_mouse_release(self, pos: MousePos):
|
|
pressed = self._pressed_rect
|
|
self._pressed_rect = None
|
|
if pressed is self._menu_button and rl.check_collision_point_rec(pos, self._menu_button):
|
|
self._menu_open = not self._menu_open
|
|
elif self._menu_open:
|
|
self._menu_open = False
|
|
if pressed is self._exit_button and rl.check_collision_point_rec(pos, self._exit_button):
|
|
self._on_exit()
|
|
elif pressed is self._back_button and rl.check_collision_point_rec(pos, self._back_button):
|
|
self._close_tool()
|
|
elif self._active_tool is None:
|
|
for target, name, create in self._launchers:
|
|
if pressed is target and rl.check_collision_point_rec(pos, target):
|
|
self._open_tool(tr(name), create())
|
|
break
|
|
|
|
def _handle_mouse_event(self, mouse_event: MouseEvent):
|
|
if mouse_event.left_released and not rl.check_collision_point_rec(mouse_event.pos, self._rect):
|
|
self._pressed_rect = None
|
|
self._menu_open = False
|
|
|
|
def _render(self, rect: rl.Rectangle):
|
|
s, theme = self._scale, DEFAULT_THEME
|
|
rl.draw_rectangle_rec(rect, theme.background)
|
|
if self._active_tool is not None:
|
|
# Widget handles input after drawing: gate the whole frame before the tool sees chrome gestures.
|
|
tool_enabled = self.enabled and not self._menu_open and self._pressed_rect is None and not any(
|
|
event.slot == 0 and not rl.check_collision_point_rec(event.pos, self._tool_rect) for event in gui_app.mouse_events)
|
|
self._active_tool.set_enabled(tool_enabled)
|
|
rl.begin_scissor_mode(int(self._tool_rect.x), int(self._tool_rect.y), int(self._tool_rect.width), int(self._tool_rect.height))
|
|
self._active_tool.render(self._tool_rect)
|
|
rl.end_scissor_mode()
|
|
|
|
rl.draw_rectangle_rec(self._bar, theme.surface)
|
|
rl.draw_line_ex(rl.Vector2(rect.x, self._bar.y + self._bar.height),
|
|
rl.Vector2(rect.x + rect.width, self._bar.y + self._bar.height), 2 * s, theme.control)
|
|
menu_pressed = self._menu_open or self._pressed_rect is self._menu_button and self.is_pressed
|
|
rl.draw_rectangle_rounded(self._menu_button, 0.28, 12, theme.pressed if menu_pressed else theme.control)
|
|
|
|
center = rl.Vector2(self._menu_button.x + 42 * s, self._menu_button.y + 44 * s)
|
|
rl.draw_ring(center, 15 * s, 20 * s, 0, 360, 24, theme.background)
|
|
rl.draw_line_ex(rl.Vector2(center.x + 18 * s, center.y - 20 * s), rl.Vector2(center.x + 18 * s, center.y + 20 * s),
|
|
5 * s, theme.background)
|
|
gui_label(rl.Rectangle(rect.x + 136 * s, rect.y, self._back_button.x - rect.x - 160 * s, self._bar.height),
|
|
self._active_name or "Arabesque", round(40 * s), theme.text, theme.font_weight)
|
|
|
|
if self._active_tool is not None:
|
|
back_pressed = self._pressed_rect is self._back_button and self.is_pressed
|
|
rl.draw_rectangle_rounded(self._back_button, 0.28, 12, theme.pressed if back_pressed else theme.control)
|
|
gui_label(self._back_button, "< " + tr("Desktop"), round(32 * s), theme.background, theme.font_weight,
|
|
alignment=TextAlignment.CENTER)
|
|
else:
|
|
# Two fixed translucent shapes soften the shelf's shadow without a blur pass or texture.
|
|
for shadow in self._dock_shadows:
|
|
rl.draw_rectangle_rounded(shadow, 0.9, 12, _DOCK_SHADOW)
|
|
rl.draw_rectangle_rec(self._dock_shelf, theme.surface)
|
|
top = self._dock_top
|
|
rl.draw_triangle(top[0], top[1], top[2], _DOCK_FACE)
|
|
rl.draw_triangle(top[0], top[2], top[3], _DOCK_FACE)
|
|
rl.draw_line_ex(top[1], top[2], 2 * s, _DOCK_RIM)
|
|
for target, name, _ in self._launchers:
|
|
icon = rl.Vector2(target.x + target.width / 2, self._dock.y + 72 * s)
|
|
rl.draw_ellipse(int(icon.x), int(icon.y + 68 * s), 52 * s, 9 * s, _DOCK_SHADOW)
|
|
tint = theme.pressed if self._pressed_rect is target and self.is_pressed else theme.control
|
|
rl.draw_circle_v(rl.Vector2(icon.x, icon.y + 5 * s), 64 * s, theme.surface)
|
|
rl.draw_circle_v(icon, 64 * s, tint)
|
|
rl.draw_circle_v(icon, 54 * s, theme.background)
|
|
if name == "Terminal":
|
|
rl.draw_line_ex(rl.Vector2(icon.x - 32 * s, icon.y - 24 * s), rl.Vector2(icon.x - 8 * s, icon.y), 6 * s, tint)
|
|
rl.draw_line_ex(rl.Vector2(icon.x - 8 * s, icon.y), rl.Vector2(icon.x - 32 * s, icon.y + 24 * s), 6 * s, tint)
|
|
rl.draw_line_ex(rl.Vector2(icon.x + 8 * s, icon.y + 24 * s), rl.Vector2(icon.x + 36 * s, icon.y + 24 * s), 6 * s, tint)
|
|
elif name == "Files":
|
|
rl.draw_rectangle_rounded(rl.Rectangle(icon.x - 32 * s, icon.y - 26 * s, 30 * s, 18 * s), 0.3, 8, tint)
|
|
rl.draw_rectangle_rounded(rl.Rectangle(icon.x - 36 * s, icon.y - 16 * s, 72 * s, 44 * s), 0.2, 8, tint)
|
|
rl.draw_rectangle_rounded(rl.Rectangle(icon.x - 29 * s, icon.y - 9 * s, 58 * s, 30 * s), 0.2, 8, theme.background)
|
|
elif name == "Firefox":
|
|
rl.draw_ring(icon, 35 * s, 40 * s, 0, 360, 40, tint)
|
|
rl.draw_ellipse_lines(int(icon.x), int(icon.y), 16 * s, 39 * s, tint)
|
|
rl.draw_line_ex(rl.Vector2(icon.x - 38 * s, icon.y), rl.Vector2(icon.x + 38 * s, icon.y), 4 * s, tint)
|
|
else:
|
|
rl.draw_circle_v(rl.Vector2(icon.x - 19 * s, icon.y - 6 * s), 20 * s, tint)
|
|
rl.draw_circle_v(rl.Vector2(icon.x + 19 * s, icon.y - 6 * s), 20 * s, tint)
|
|
for offset in (-19, 19):
|
|
rl.draw_circle_v(rl.Vector2(icon.x + offset * s, icon.y - 8 * s), 11 * s, theme.background)
|
|
rl.draw_line_ex(rl.Vector2(icon.x - 8 * s, icon.y + 25 * s), rl.Vector2(icon.x + 8 * s, icon.y + 25 * s), 3 * s, tint)
|
|
gui_label(rl.Rectangle(icon.x - 96 * s, icon.y + 68 * s, 192 * s, 36 * s),
|
|
tr(name), round(28 * s), theme.text, theme.font_weight, alignment=TextAlignment.CENTER)
|
|
|
|
if self._menu_open:
|
|
exit_pressed = self._pressed_rect is self._exit_button and self.is_pressed
|
|
color = theme.pressed if exit_pressed else theme.surface
|
|
rl.draw_rectangle_rounded(self._exit_button, 0.22, 12, color)
|
|
rl.draw_rectangle_rounded_lines_ex(self._exit_button, 0.22, 12, 2 * s, theme.control)
|
|
gui_label(rl.Rectangle(self._exit_button.x + 28 * s, self._exit_button.y, self._exit_button.width - 56 * s, self._exit_button.height),
|
|
tr("Exit Arabesque"), round(40 * s), theme.background if exit_pressed else theme.text, theme.font_weight)
|