Merge branch 'auxpowersave' into ccnc-port-auxpowersave

This reverts commit 01420fc08488374ec8fe3d17b757ccd1564d1328.

openpilot/cereal/custom.capnp

@@ -353,6 +353,7 @@ struct OnroadEventSP @0xda96579883444c35 {
     speedLimitPending @22;
     e2eChime @23;
     laneChangeRoadEdge @24;
+    bigModelReady @25;
   }
 }
 

openpilot/common/params_keys.h

@@ -132,6 +132,7 @@ inline static std::unordered_map<std::string, ParamKeyAttributes> keys = {
     {"UptimeOnroad", {PERSISTENT, FLOAT, "0.0"}},
     {"ChestnutActive", {CLEAR_ON_MANAGER_START | CLEAR_ON_OFFROAD_TRANSITION | CLEAR_ON_IGNITION_ON, BOOL}},
     {"ChestnutLoading", {CLEAR_ON_MANAGER_START | CLEAR_ON_OFFROAD_TRANSITION | CLEAR_ON_IGNITION_ON, BOOL}},
+    {"AuxPowerSave", {PERSISTENT | BACKUP, BOOL}},
     {"Version", {PERSISTENT, STRING}},
 
     // --- sunnypilot params --- //

openpilot/selfdrive/selfdrived/selfdrived.py

@@ -198,6 +198,7 @@ class SelfdriveD(CruiseHelper):
     loading = self.params.get_bool("ChestnutLoading")
     if self.big_model_loading and not loading:
       self.big_model_ready_t = time.monotonic()
+      self.events_sp.add(custom.OnroadEventSP.EventName.bigModelReady)
     self.big_model_loading = loading
     if self.big_model_loading:
       self.events.add(EventName.bigModelLoading)

openpilot/selfdrive/ui/sunnypilot/layouts/settings/developer.py

@@ -20,7 +20,6 @@ from openpilot.system.ui.widgets.list_view import button_item
 
 from openpilot.system.ui.sunnypilot.widgets.html_render import HtmlModalSP
 from openpilot.system.ui.sunnypilot.widgets.list_view import toggle_item_sp
-from openpilot.selfdrive.ui.sunnypilot.layouts.settings.external_storage import external_storage_item
 
 PREBUILT_PATH = os.path.join(Paths.comma_home(), "prebuilt") if PC else "/data/openpilot/prebuilt"
 
@@ -53,11 +52,7 @@ class DeveloperLayoutSP(DeveloperLayout):
 
     self.error_log_btn = button_item(tr("Error Log"), tr("VIEW"), tr("View the error log for sunnypilot crashes."), callback=self._on_error_log_clicked)
 
-    self.external_storage = external_storage_item(tr("External Storage"), description=tr("Extend your comma device's storage by inserting a USB drive " +
-                                                                                         "into the aux port."))
-
-    self.items: list = [self.show_advanced_controls, self.enable_github_runner_toggle, self.enable_copyparty_toggle, self.prebuilt_toggle,
-                        self.external_storage, self.error_log_btn,]
+    self.items: list = [self.show_advanced_controls, self.enable_github_runner_toggle, self.enable_copyparty_toggle, self.prebuilt_toggle, self.error_log_btn,]
 
   @staticmethod
   def _on_prebuilt_toggled(state):

openpilot/selfdrive/ui/sunnypilot/layouts/settings/external_storage.py

@@ -1,261 +0,0 @@
-"""
-Copyright (c) 2021-, Haibin Wen, sunnypilot, and a number of other contributors.
-
-This file is part of sunnypilot and is licensed under the MIT License.
-See the LICENSE.md file in the root directory for more details.
-"""
-import pyray as rl
-import threading
-import subprocess
-import copy
-from enum import Enum
-from collections.abc import Callable
-
-from openpilot.common.params import Params
-from openpilot.common.hardware import PC
-from openpilot.system.ui.lib.application import gui_app, FontWeight
-from openpilot.system.ui.lib.text_measure import measure_text_cached
-from openpilot.system.ui.lib.multilang import tr, tr_noop
-from openpilot.system.ui.widgets import DialogResult
-from openpilot.system.ui.widgets.button import Button, ButtonStyle
-from openpilot.system.ui.widgets.confirm_dialog import alert_dialog, ConfirmDialog
-from openpilot.system.ui.widgets.list_view import (
-  ItemAction,
-  ListItem,
-  BUTTON_HEIGHT,
-  BUTTON_BORDER_RADIUS,
-  BUTTON_FONT_SIZE,
-  BUTTON_WIDTH,
-)
-
-VALUE_FONT_SIZE = 48
-
-
-class ExternalStorageState(Enum):
-  DISABLED = tr_noop("DISABLED")
-  LOADING = tr_noop("LOADING")
-  CHECK = tr_noop("CHECK")
-  MOUNT = tr_noop("MOUNT")
-  UNMOUNT = tr_noop("UNMOUNT")
-  FORMAT = tr_noop("FORMAT")
-
-
-class ExternalStorageAction(ItemAction):
-  MAX_WIDTH = 500
-
-  def __init__(self):
-    super().__init__(self.MAX_WIDTH, True)
-
-    self._params = Params()
-    self._error_message = ""
-    self._text_font = gui_app.font(FontWeight.NORMAL)
-
-    self._button = Button(
-      "",
-      click_callback=self._handle_button_click,
-      button_style=ButtonStyle.LIST_ACTION,
-      border_radius=BUTTON_BORDER_RADIUS,
-      font_size=BUTTON_FONT_SIZE,
-    )
-
-    self._value_text = ""
-    self._formatting = False
-    self._refresh_pending = False
-
-    self._state = ExternalStorageState.CHECK
-    self._refresh_state()
-    self.refresh()
-
-  def set_touch_valid_callback(self, callback):
-    def wrapped():
-      if self._state == ExternalStorageState.DISABLED:
-        return False
-      return callback()
-    super().set_touch_valid_callback(wrapped)
-    self._button.set_touch_valid_callback(wrapped)
-
-  def _run(self, cmd: str) -> bool:
-    return subprocess.call(["sh", "-c", cmd]) == 0
-
-  def _run_output(self, cmd: str) -> str:
-    try:
-      out = subprocess.check_output(["sh", "-c", cmd], universal_newlines=True)
-      return out.strip()
-    except Exception:
-      return ""
-
-  def _render(self, rect: rl.Rectangle) -> bool:
-    if self._error_message:
-      msg = copy.copy(self._error_message)
-      gui_app.set_modal_overlay(alert_dialog(msg))
-      self._error_message = ""
-
-    if self._value_text:
-      text_size = measure_text_cached(self._text_font, self._value_text, VALUE_FONT_SIZE)
-      rl.draw_text_ex(
-        self._text_font,
-        self._value_text,
-        (rect.x + rect.width - BUTTON_WIDTH - text_size.x - 30,
-         rect.y + (rect.height - text_size.y) / 2),
-        VALUE_FONT_SIZE,
-        1.0,
-        rl.Color(170, 170, 170, 255),
-      )
-
-    button_rect = rl.Rectangle(
-      rect.x + rect.width - BUTTON_WIDTH,
-      rect.y + (rect.height - BUTTON_HEIGHT) / 2,
-      BUTTON_WIDTH,
-      BUTTON_HEIGHT
-    )
-    self._button.set_rect(button_rect)
-    self._button.set_text(tr(self._state.value))
-    self._button.set_enabled(self._state not in (ExternalStorageState.LOADING,
-                                                 ExternalStorageState.DISABLED))
-    self._button.render(button_rect)
-    return False
-
-  def _refresh_state(self):
-    if PC:
-      self._state = ExternalStorageState.DISABLED
-      self._button.set_enabled(False)
-      self._value_text = ""
-
-  def debounced_refresh(self):
-    if self._refresh_pending:
-      return
-    self._refresh_pending = True
-
-    def _timer():
-      import time
-      time.sleep(0.25)
-      self._refresh_pending = False
-      self.refresh()
-
-    threading.Thread(target=_timer, daemon=True).start()
-
-  def refresh(self):
-    def _work():
-      is_mounted = self._run("findmnt -n /mnt/external_realdata")
-      has_drive = self._run("lsblk -f /dev/sdg")
-      has_fs = self._run("lsblk -f /dev/sdg1 | grep -q ext4")
-      has_label = self._run("blkid /dev/sdg1 | grep -q 'LABEL=\"openpilot\"'")
-
-      info = ""
-      if is_mounted and has_label:
-        info = self._run_output(
-          "df -h /mnt/external_realdata | awk 'NR==2 {print $3 \"/\" $2}'"
-        )
-
-      def apply():
-        if self._formatting:
-          self._value_text = tr("formatting")
-          self._state = ExternalStorageState.FORMAT
-          return
-
-        if not has_drive:
-          self._value_text = tr("insert drive")
-          self._state = ExternalStorageState.CHECK
-
-        elif not has_fs or not has_label:
-          self._value_text = tr("needs format")
-          self._state = ExternalStorageState.FORMAT
-
-        elif is_mounted:
-          self._value_text = info
-          self._state = ExternalStorageState.UNMOUNT
-
-        else:
-          self._value_text = tr("drive detected")
-          self._state = ExternalStorageState.MOUNT
-
-      apply()
-
-    threading.Thread(target=_work, daemon=True).start()
-
-  def _handle_button_click(self):
-    st = self._state
-
-    if st == ExternalStorageState.DISABLED:
-      return
-
-    if st in (ExternalStorageState.CHECK, ExternalStorageState.MOUNT):
-      self.mount_storage()
-
-    elif st == ExternalStorageState.UNMOUNT:
-      self.unmount_storage()
-
-    elif st == ExternalStorageState.FORMAT:
-      dialog = ConfirmDialog(
-          tr("Are you sure you want to format this drive? This will erase all data."),
-          confirm_text=tr("Format"),
-          cancel_text=tr("Cancel"),
-        )
-      gui_app.set_modal_overlay(dialog, callback=self._confirm_format)
-
-  def _confirm_format(self, result: DialogResult):
-    if result == DialogResult.CONFIRM:
-      self.format_storage()
-
-  def mount_storage(self):
-    self._value_text = tr("mounting")
-    self._state = ExternalStorageState.LOADING
-
-    def _work():
-      cmd = """
-        sudo mount -o remount,rw / &&
-        sudo mkdir -p /mnt/external_realdata &&
-        (grep -q '/dev/sdg1 /mnt/external_realdata' /etc/fstab ||
-         echo '/dev/sdg1 /mnt/external_realdata ext4 defaults,nofail 0 2' >> /etc/fstab) &&
-        sudo systemctl daemon-reexec &&
-        sudo mount /mnt/external_realdata &&
-        sudo chown -R comma:comma /mnt/external_realdata &&
-        sudo chmod -R 775 /mnt/external_realdata &&
-        sudo mount -o remount,ro /
-      """
-      subprocess.call(["sh", "-c", cmd])
-      self.debounced_refresh()
-
-    threading.Thread(target=_work, daemon=True).start()
-
-  def unmount_storage(self):
-    self._value_text = tr("unmounting")
-    self._state = ExternalStorageState.LOADING
-
-    def _work():
-      subprocess.call(["sh", "-c", "sudo umount /mnt/external_realdata"])
-      self.debounced_refresh()
-
-    threading.Thread(target=_work, daemon=True).start()
-
-  def format_storage(self):
-    self._formatting = True
-    self._value_text = tr("formatting")
-    self._state = ExternalStorageState.LOADING
-
-    def _work():
-      cmd = """
-        sudo wipefs -a /dev/sdg &&
-        sudo parted -s /dev/sdg mklabel gpt mkpart primary ext4 0% 100% &&
-        sudo mkfs.ext4 -F -L openpilot /dev/sdg1
-      """
-      exitcode = subprocess.call(["sh", "-c", cmd])
-
-      def apply():
-        self._formatting = False
-        if exitcode == 0:
-          self.mount_storage()
-        else:
-          self._value_text = tr("needs format")
-          self._state = ExternalStorageState.FORMAT
-
-      apply()
-
-    threading.Thread(target=_work, daemon=True).start()
-
-def external_storage_item(title: str | Callable[[], str], description: str | Callable[[], str]) -> ListItem:
-  return ListItem(
-    title=title,
-    description=description,
-    action_item=ExternalStorageAction()
-  )

openpilot/selfdrive/ui/ui_state.py

@@ -220,6 +220,7 @@ class UIState(UIStateSP):
                              ChestnutState.UNCOMPILED if detected else ChestnutState.DISCONNECTED)
       return
 
+    self.chestnut_present = self.chestnut_present or detected
     model_seen = self.sm.recv_frame["modelV2"] > self.started_frame
     if not self.chestnut_present:
       self.chestnut_state = ChestnutState.DISCONNECTED

openpilot/sunnypilot/selfdrive/selfdrived/events.py

@@ -252,4 +252,12 @@ EVENTS_SP: dict[int, dict[str, Alert | AlertCallbackType]] = {
       AlertStatus.userPrompt, AlertSize.small,
       Priority.LOW, VisualAlert.none, AudibleAlert.prompt, 0.1),
   },
+
+  EventNameSP.bigModelReady: {
+    ET.PERMANENT: Alert(
+      "Big Model Ready",
+      "",
+      AlertStatus.normal, AlertSize.small,
+      Priority.LOW, VisualAlert.none, AudibleAlert.prompt, 2.),
+  },
 }

openpilot/sunnypilot/sunnylink/settings_ui.json

@@ -1675,6 +1675,12 @@
               "widget": "toggle",
               "title": "Onroad Uploads"
             },
+            {
+              "key": "AuxPowerSave",
+              "widget": "toggle",
+              "title": "Disable Aux Port When Offroad",
+              "description": "Power off the aux USB-C port while offroad to save power. It powers back on automatically when you go onroad."
+            },
             {
               "key": "MaxTimeOffroad",
               "widget": "option",

openpilot/sunnypilot/sunnylink/settings_ui_src/pages/device.yaml

@@ -30,6 +30,10 @@ sections:
   - key: OnroadUploads
     widget: toggle
     title: Onroad Uploads
+  - key: AuxPowerSave
+    widget: toggle
+    title: Disable Aux Port When Offroad
+    description: Power off the aux USB-C port while offroad to save power. It powers back on automatically when you go onroad.
   - key: MaxTimeOffroad
     widget: option
     title: Max Time Offroad

openpilot/system/hardware/hardwared.py

@@ -21,6 +21,7 @@ from openpilot.selfdrive.selfdrived.alertmanager import set_offroad_alert
 from openpilot.common.hardware import HARDWARE, COMMA_HARDWARE
 from openpilot.common.basedir import BASEDIR
 from openpilot.common.hardware.usb import CHESTNUT_FW_VERSION, CHESTNUT_ROM_USB_IDS, CHESTNUT_USB_IDS, get_usb_state, get_usb_topology, set_usb_state
+from openpilot.system.hardware.chestnut.flash import VBUS_PATH
 from openpilot.common.linux import LinuxSystemStats
 from openpilot.system.loggerd.config import get_available_percent
 from openpilot.common.swaglog import cloudlog
@@ -49,6 +50,10 @@ class Chestnut:
     self.attempts = 0
     self.last_attempt = 0.
     self.flashed = False
+    self.vbus_on = None
+    self.params = Params()
+    self.powersave = False
+    self.last_offroad = None
 
   def flash(self) -> None:
     ret = subprocess.run(["sudo", sys.executable, os.path.join(BASEDIR, "openpilot/system/hardware/chestnut/flash.py"), CHESTNUT_FW_VERSION],
@@ -56,9 +61,19 @@ class Chestnut:
     cloudlog.event("chestnut flash done", returncode=ret.returncode, output=ret.stdout[-1000:], error=ret.returncode != 0)
     self.flashed = ret.returncode == 0
 
+  def set_vbus(self, on: bool) -> None:
+    if on == self.vbus_on:
+      return
+    subprocess.run(["sudo", "tee", VBUS_PATH], input=b"1" if on else b"0", stdout=subprocess.DEVNULL, check=False)
+    self.vbus_on = on
+
   def update(self, offroad: bool, usb_state: list[dict]) -> None:
     mismatch = any((d["vendorId"], d["productId"]) in CHESTNUT_USB_IDS + CHESTNUT_ROM_USB_IDS and
                    d["product"] != f"custom {CHESTNUT_FW_VERSION}-CLEAN" for d in usb_state)
+    if offroad != self.last_offroad:
+      self.powersave = self.params.get_bool("AuxPowerSave")
+      self.last_offroad = offroad
+    self.set_vbus((not offroad or mismatch) or not self.powersave)
     if not mismatch:
       self.flashed = False
       return