IQ.Pilot Release Commit @ bec7652

iqpilot/selfdrive/ui/mici/layouts/main.py

@@ -0,0 +1,134 @@
+"""
+Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
+"""
+
+import pyray as rl
+import iqpilot.cereal.messaging as messaging
+from iqpilot.selfdrive.ui.mici.layouts.home import MiciHomeLayout
+from iqpilot.selfdrive.ui.mici.layouts.settings.settings import SettingsLayout
+from iqpilot.selfdrive.ui.mici.layouts.offroad_alerts import MiciOffroadAlerts
+from iqpilot.selfdrive.ui.mici.onroad.augmented_road_view import AugmentedRoadView
+from iqpilot.selfdrive.ui.ui_state import device, ui_state
+from iqpilot.selfdrive.ui.mici.layouts.onboarding import OnboardingWindow
+from iqpilot.system.ui.widgets import Widget
+from iqpilot.system.ui.widgets.scroller import Scroller
+from iqpilot.system.ui.lib.application import gui_app
+from iqpilot.system.ui.lib.multilang import multilang
+from iqpilot.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._rebuild_settings = 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)
+    multilang.add_change_callback(self._on_language_changed)
+
+  def _on_language_changed(self):
+    self._rebuild_settings = True
+
+  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
+
+    if self._rebuild_settings:
+      self._rebuild_settings = False
+      gui_app.pop_widgets_to(self, instant=True)
+      self._settings_layout = SettingsLayout()
+      self._settings_layout.set_rect(rl.Rectangle(0, 0, gui_app.width, gui_app.height))
+
+  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)