mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-01 03:22:07 +08:00
Dean Lee
f824e6c0ec
* implement reset calibration feature in device settings * check confirm dialog result * fix null check * use enum * use enum
104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
import os
|
|
import json
|
|
|
|
from openpilot.common.basedir import BASEDIR
|
|
from openpilot.common.params import Params
|
|
from openpilot.selfdrive.ui.onroad.driver_camera_dialog import DriverCameraDialog
|
|
from openpilot.selfdrive.ui.ui_state import ui_state
|
|
from openpilot.system.hardware import TICI
|
|
from openpilot.system.ui.lib.application import gui_app, Widget, DialogResult
|
|
from openpilot.system.ui.lib.list_view import ListView, text_item, button_item
|
|
from openpilot.system.ui.widgets.option_dialog import MultiOptionDialog
|
|
from openpilot.system.ui.widgets.confirm_dialog import confirm_dialog, alert_dialog
|
|
|
|
|
|
# Description constants
|
|
DESCRIPTIONS = {
|
|
'pair_device': "Pair your device with comma connect (connect.comma.ai) and claim your comma prime offer.",
|
|
'driver_camera': "Preview the driver facing camera to ensure that driver monitoring has good visibility. (vehicle must be off)",
|
|
'reset_calibration': (
|
|
"openpilot requires the device to be mounted within 4° left or right and within 5° " +
|
|
"up or 9° down. openpilot is continuously calibrating, resetting is rarely required."
|
|
),
|
|
'review_guide': "Review the rules, features, and limitations of openpilot",
|
|
}
|
|
|
|
|
|
class DeviceLayout(Widget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
self._params = Params()
|
|
dongle_id = self._params.get("DongleId", encoding="utf-8") or "N/A"
|
|
serial = self._params.get("HardwareSerial") or "N/A"
|
|
|
|
items = [
|
|
text_item("Dongle ID", dongle_id),
|
|
text_item("Serial", serial),
|
|
button_item("Pair Device", "PAIR", DESCRIPTIONS['pair_device'], self._on_pair_device),
|
|
button_item("Driver Camera", "PREVIEW", DESCRIPTIONS['driver_camera'], callback=self._on_driver_camera),
|
|
button_item("Reset Calibration", "RESET", DESCRIPTIONS['reset_calibration'], callback=self._on_reset_calibration),
|
|
button_item("Review Training Guide", "REVIEW", DESCRIPTIONS['review_guide'], self._on_review_training_guide),
|
|
]
|
|
|
|
if TICI:
|
|
items.append(button_item("Regulatory", "VIEW", callback=self._on_regulatory))
|
|
|
|
items.append(button_item("Change Language", "CHANGE", callback=self._on_change_language))
|
|
|
|
self._list_widget = ListView(items)
|
|
self._select_language_dialog: MultiOptionDialog | None = None
|
|
self._driver_camera: DriverCameraDialog | None = None
|
|
|
|
def _render(self, rect):
|
|
self._list_widget.render(rect)
|
|
|
|
def _on_change_language(self):
|
|
try:
|
|
languages_file = os.path.join(BASEDIR, "selfdrive/ui/translations/languages.json")
|
|
with open(languages_file, encoding='utf-8') as f:
|
|
languages = json.load(f)
|
|
|
|
self._select_language_dialog = MultiOptionDialog("Select a language", languages)
|
|
gui_app.set_modal_overlay(self._select_language_dialog, callback=self._on_select_lang_dialog_closed)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
def _on_select_lang_dialog_closed(self, result: int):
|
|
if result == 1 and self._select_language_dialog:
|
|
selected_language = self._select_language_dialog.selection
|
|
self._params.put("LanguageSetting", selected_language)
|
|
|
|
self._select_language_dialog = None
|
|
|
|
def _on_driver_camera(self):
|
|
if not self._driver_camera:
|
|
self._driver_camera = DriverCameraDialog()
|
|
|
|
gui_app.set_modal_overlay(self._driver_camera, callback=lambda result: setattr(self, '_driver_camera', None))
|
|
|
|
def _on_reset_calibration(self):
|
|
if ui_state.engaged:
|
|
gui_app.set_modal_overlay(lambda: alert_dialog("Disengage to Reset Calibration"))
|
|
return
|
|
|
|
gui_app.set_modal_overlay(
|
|
lambda: confirm_dialog("Are you sure you want to reset calibration?", "Reset"),
|
|
callback=self._reset_calibration,
|
|
)
|
|
|
|
def _reset_calibration(self, result: int):
|
|
if ui_state.engaged or result != DialogResult.CONFIRM:
|
|
return
|
|
|
|
self._params.remove("CalibrationParams")
|
|
self._params.remove("LiveTorqueParameters")
|
|
self._params.remove("LiveParameters")
|
|
self._params.remove("LiveParametersV2")
|
|
self._params.remove("LiveDelay")
|
|
self._params.put_bool("OnroadCycleRequested", True)
|
|
|
|
def _on_pair_device(self): pass
|
|
def _on_review_training_guide(self): pass
|
|
def _on_regulatory(self): pass
|