forked from IQ.Lvbs/IQ.Pilot
123 lines
4.4 KiB
Python
123 lines
4.4 KiB
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
|
|
"""
|
|
|
|
import pyray as rl
|
|
import cereal.messaging as messaging
|
|
from openpilot.selfdrive.ui.mici.layouts.home import MiciHomeLayout
|
|
from openpilot.selfdrive.ui.mici.layouts.settings.settings import SettingsLayout
|
|
from openpilot.selfdrive.ui.mici.layouts.offroad_alerts import MiciOffroadAlerts
|
|
from openpilot.selfdrive.ui.mici.onroad.augmented_road_view import AugmentedRoadView
|
|
from openpilot.selfdrive.ui.ui_state import device, ui_state
|
|
from openpilot.selfdrive.ui.mici.layouts.onboarding import OnboardingWindow
|
|
from openpilot.system.ui.widgets import Widget
|
|
from openpilot.system.ui.widgets.scroller import Scroller
|
|
from openpilot.system.ui.lib.application import gui_app
|
|
from openpilot.system.version import training_version
|
|
|
|
|
|
ONROAD_DELAY = 2.5 # seconds
|
|
|
|
|
|
class MiciMainLayout(Widget):
|
|
"""Root mici layout. Lives at the bottom of the nav stack; settings push on top.
|
|
|
|
Keeps the IQ.Pilot custom home + onroad as horizontally-scrolled pages, while
|
|
the (stock) settings open as a swipe-to-dismiss NavWidget on the nav stack.
|
|
"""
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
self._pm = messaging.PubMaster(['bookmarkButton'])
|
|
|
|
self._prev_onroad = False
|
|
self._prev_standstill = False
|
|
self._onroad_time_delay: float | None = None
|
|
self._setup = False
|
|
|
|
self._home_layout = MiciHomeLayout()
|
|
self._alerts_layout = MiciOffroadAlerts()
|
|
self._settings_layout = SettingsLayout()
|
|
self._onroad_layout = AugmentedRoadView(bookmark_callback=self._on_bookmark_clicked)
|
|
|
|
for widget in (self._home_layout, self._settings_layout, self._alerts_layout, self._onroad_layout):
|
|
widget.set_rect(rl.Rectangle(0, 0, gui_app.width, gui_app.height))
|
|
|
|
self._scroller = Scroller([
|
|
self._alerts_layout,
|
|
self._home_layout,
|
|
self._onroad_layout,
|
|
], spacing=0, pad_start=0, pad_end=0)
|
|
self._scroller.set_reset_scroll_at_show(False)
|
|
|
|
# Disable scrolling when onroad is interacting with bookmark
|
|
self._scroller.set_scrolling_enabled(lambda: not self._onroad_layout.is_swiping_left())
|
|
|
|
self._setup_callbacks()
|
|
|
|
if ui_state.params.get("CompletedTrainingVersion") != training_version:
|
|
ui_state.params.put("CompletedTrainingVersion", training_version)
|
|
|
|
gui_app.add_nav_stack_tick(self._handle_transitions)
|
|
gui_app.push_widget(self)
|
|
|
|
self._onboarding_window = OnboardingWindow()
|
|
if not self._onboarding_window.completed:
|
|
gui_app.set_modal_overlay(self._onboarding_window)
|
|
|
|
def _setup_callbacks(self):
|
|
self._home_layout.set_callbacks(on_settings=self._on_settings_clicked)
|
|
self._onroad_layout.set_click_callback(lambda: self._scroll_to(self._home_layout))
|
|
device.add_interactive_timeout_callback(self._on_interactive_timeout)
|
|
|
|
def _scroll_to(self, layout: Widget):
|
|
self._scroller.scroll_to(int(layout.rect.x), smooth=True)
|
|
|
|
def show_event(self):
|
|
super().show_event()
|
|
self._scroller.show_event()
|
|
|
|
def _render(self, _):
|
|
if not self._setup:
|
|
if self._alerts_layout.active_alerts() > 0:
|
|
self._scroller.scroll_to(self._alerts_layout.rect.x)
|
|
else:
|
|
self._scroller.scroll_to(self._rect.width)
|
|
self._setup = True
|
|
|
|
self._scroller.render(self._rect)
|
|
|
|
def _handle_transitions(self):
|
|
if ui_state.started != self._prev_onroad:
|
|
self._prev_onroad = ui_state.started
|
|
|
|
if ui_state.started:
|
|
self._onroad_time_delay = rl.get_time()
|
|
else:
|
|
self._scroll_to(self._home_layout)
|
|
|
|
if self._onroad_time_delay is not None and rl.get_time() - self._onroad_time_delay >= ONROAD_DELAY:
|
|
gui_app.pop_widgets_to(self, lambda: self._scroll_to(self._onroad_layout))
|
|
self._onroad_time_delay = None
|
|
|
|
CS = ui_state.sm["carState"]
|
|
if not CS.standstill and self._prev_standstill:
|
|
gui_app.pop_widgets_to(self, lambda: self._scroll_to(self._onroad_layout))
|
|
self._prev_standstill = CS.standstill
|
|
|
|
def _on_interactive_timeout(self):
|
|
if ui_state.started:
|
|
if not ui_state.sm["carState"].standstill:
|
|
gui_app.pop_widgets_to(self, lambda: self._scroll_to(self._onroad_layout))
|
|
else:
|
|
gui_app.pop_widgets_to(self, instant=True)
|
|
self._scroll_to(self._home_layout)
|
|
|
|
def _on_settings_clicked(self):
|
|
gui_app.push_widget(self._settings_layout)
|
|
|
|
def _on_bookmark_clicked(self):
|
|
user_bookmark = messaging.new_message('bookmarkButton')
|
|
user_bookmark.valid = True
|
|
self._pm.send('bookmarkButton', user_bookmark)
|