mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
a82935f326
Integrate onroad views, configurable widgets, local settings, steering-profile contracts, Bluetooth and Android Auto with their device and runtime services. Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> Co-authored-by: Daniel Koepping <elkoled@gmail.com> Co-authored-by: Danny <dannyirwin31@yahoo.com> Co-authored-by: DimitriChrysafis <Dimitri.chrysafis@gmail.com> Co-authored-by: Dom <168790843+firestar5683@users.noreply.github.com> Co-authored-by: Harald Schäfer <harald.the.engineer@gmail.com> Co-authored-by: Prabhaav Pillai <prabhaav@msu.edu> Co-authored-by: Shane Smiskol <shane@smiskol.com> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com> Co-authored-by: firestarsdog <229254897+firestarsdog@users.noreply.github.com> Co-authored-by: stef <19478336+stefpi@users.noreply.github.com> Co-authored-by: whoisdomi <dcantu83@gmail.com> Co-authored-by: Intelli <contact@intelli.software> Co-authored-by: pharmacomaniac <blittle65@gmail.com> Co-authored-by: prabhaavp <prabhaav@msu.edu>
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
import time
|
|
import pyray as rl
|
|
from openpilot.common.params import Params
|
|
from openpilot.selfdrive.ui.ui_state import ui_state
|
|
from openpilot.system.ui.lib.application import gui_app
|
|
from openpilot.system.ui.widgets import Widget
|
|
|
|
|
|
class ExpButton(Widget):
|
|
def __init__(self, button_size: int, icon_size: int):
|
|
super().__init__()
|
|
self._params = Params()
|
|
self._experimental_mode: bool = False
|
|
self._engageable: bool = False
|
|
|
|
# State hold mechanism
|
|
self._hold_duration = 2.0 # seconds
|
|
self._held_mode: bool | None = None
|
|
self._hold_end_time: float | None = None
|
|
|
|
self._white_color: rl.Color = rl.Color(255, 255, 255, 255)
|
|
self._black_bg: rl.Color = rl.Color(0, 0, 0, 166)
|
|
self._txt_wheel: rl.Texture = gui_app.texture('icons/chffr_wheel.png', icon_size, icon_size)
|
|
self._txt_exp: rl.Texture = gui_app.texture('icons/experimental.png', icon_size, icon_size)
|
|
self._rect = rl.Rectangle(0, 0, button_size, button_size)
|
|
|
|
def set_rect(self, rect: rl.Rectangle) -> None:
|
|
self._rect.x, self._rect.y = rect.x, rect.y
|
|
|
|
def _update_state(self) -> None:
|
|
selfdrive_state = ui_state.sm["selfdriveState"]
|
|
self._experimental_mode = selfdrive_state.experimentalMode
|
|
self._engageable = selfdrive_state.engageable or selfdrive_state.enabled
|
|
|
|
def _handle_mouse_release(self, _):
|
|
super()._handle_mouse_release(_)
|
|
self.request_toggle()
|
|
|
|
def request_toggle(self) -> bool:
|
|
self._update_state()
|
|
if ui_state.has_longitudinal_control:
|
|
new_mode = not self._experimental_mode
|
|
self._params.put_bool("ExperimentalMode", new_mode)
|
|
|
|
# Hold new state temporarily
|
|
self._held_mode = new_mode
|
|
self._hold_end_time = time.monotonic() + self._hold_duration
|
|
return True
|
|
return False
|
|
|
|
def _render(self, rect: rl.Rectangle) -> None:
|
|
center_x = int(self._rect.x + self._rect.width // 2)
|
|
center_y = int(self._rect.y + self._rect.height // 2)
|
|
|
|
self._white_color.a = 180 if self.is_pressed or not self._engageable else 255
|
|
|
|
texture = self._txt_exp if self._held_or_actual_mode() else self._txt_wheel
|
|
rl.draw_circle(center_x, center_y, self._rect.width / 2, self._black_bg)
|
|
rl.draw_texture_ex(texture, rl.Vector2(center_x - texture.width / 2, center_y - texture.height / 2), 0.0, 1.0, self._white_color)
|
|
|
|
def _held_or_actual_mode(self):
|
|
now = time.monotonic()
|
|
if self._hold_end_time and now < self._hold_end_time:
|
|
return self._held_mode
|
|
|
|
if self._hold_end_time and now >= self._hold_end_time:
|
|
self._hold_end_time = self._held_mode = None
|
|
|
|
return self._experimental_mode
|