Files
IQ.Pilot/selfdrive/ui/widgets/exp_mode_button.py
T
2026-06-25 08:41:49 -05:00

93 lines
3.9 KiB
Python

import pyray as rl
from openpilot.common.params import Params
from openpilot.system.ui.lib.application import gui_app, FontWeight, FONT_SCALE
from openpilot.system.ui.lib.multilang import tr
from openpilot.system.ui.widgets import Widget
class ExperimentalModeButton(Widget):
def __init__(self):
super().__init__()
self.img_width = 80
self.horizontal_padding = 25
self.button_height = 125
self.params = Params()
self.experimental_mode = self.params.get_bool("ExperimentalMode")
self.iq_dynamic_mode = self.params.get_bool("IQDynamicMode")
self.alpha_longitudinal_enabled = self.params.get_bool("AlphaLongitudinalEnabled")
self.chill_pixmap = gui_app.texture("icons/couch.png", self.img_width, self.img_width)
self.experimental_pixmap = gui_app.texture("icons_mici/experimental_mode_tizi.png", self.img_width, self.img_width)
self.iqstandard_pixmap = gui_app.texture("icons_mici/iqstandard_mode_tizi.png", self.img_width, self.img_width)
self.iqdynamic_pixmap = gui_app.texture("icons_mici/iqdynamic_mode_tizi.png", self.img_width, self.img_width)
def show_event(self):
self.experimental_mode = self.params.get_bool("ExperimentalMode")
self.iq_dynamic_mode = self.params.get_bool("IQDynamicMode")
self.alpha_longitudinal_enabled = self.params.get_bool("AlphaLongitudinalEnabled")
def _get_gradient_colors(self):
alpha = 0xCC if self.is_pressed else 0xFF
if not self.alpha_longitudinal_enabled:
return rl.Color(112, 112, 112, alpha), rl.Color(78, 78, 78, alpha)
if self.experimental_mode:
# IQ.Pilot / IQ.Dynamic active gradient
return rl.Color(0x13, 0xC3, 0xE2, alpha), rl.Color(0xE2, 0x13, 0xAD, alpha)
else:
return rl.Color(0xCC, 0x00, 0xCC, alpha), rl.Color(0xFF, 0xDD, 0x00, alpha)
def _draw_gradient_background(self, rect):
start_color, end_color = self._get_gradient_colors()
rl.draw_rectangle_gradient_h(int(rect.x), int(rect.y), int(rect.width), int(rect.height),
start_color, end_color)
def _render(self, rect):
rl.begin_scissor_mode(int(rect.x), int(rect.y), int(rect.width), int(rect.height))
self._draw_gradient_background(rect)
rl.draw_rectangle_rounded_lines_ex(self._rect, 0.19, 10, 5, rl.BLACK)
rl.end_scissor_mode()
# Draw vertical separator line
line_x = rect.x + rect.width - self.img_width - (2 * self.horizontal_padding)
separator_color = rl.Color(0, 0, 0, 77) # 0x4d = 77
rl.draw_line_ex(rl.Vector2(line_x, rect.y), rl.Vector2(line_x, rect.y + rect.height), 3, separator_color)
# Draw text label (left aligned)
if not self.alpha_longitudinal_enabled:
text = tr("STOCK ACC")
font_size = 45
else:
if self.experimental_mode and self.iq_dynamic_mode:
text = tr("IQ.DYNAMIC")
elif self.experimental_mode:
text = tr("IQ.PILOT")
else:
text = tr("IQ.STANDARD")
font_size = 45
text_x = rect.x + self.horizontal_padding
text_y = rect.y + rect.height / 2 - font_size * FONT_SCALE // 2 # Center vertically
rl.draw_text_ex(gui_app.font(FontWeight.NORMAL), text, rl.Vector2(int(text_x), int(text_y)), font_size, 0, rl.BLACK)
# Draw icon (right aligned)
icon_x = rect.x + rect.width - self.horizontal_padding - self.img_width
icon_y = rect.y + (rect.height - self.img_width) / 2
icon_rect = rl.Rectangle(icon_x, icon_y, self.img_width, self.img_width)
# Draw current mode icon
if not self.alpha_longitudinal_enabled:
current_icon = self.chill_pixmap
elif self.experimental_mode and self.iq_dynamic_mode:
current_icon = self.iqdynamic_pixmap
elif self.experimental_mode:
current_icon = self.experimental_pixmap
else:
current_icon = self.iqstandard_pixmap
source_rect = rl.Rectangle(0, 0, current_icon.width, current_icon.height)
rl.draw_texture_pro(current_icon, source_rect, icon_rect, rl.Vector2(0, 0), 0, rl.WHITE)