185 lines
8.7 KiB
Python
185 lines
8.7 KiB
Python
import os
|
|
import operator
|
|
import platform
|
|
|
|
from cereal import car, custom
|
|
from openpilot.common.params import Params
|
|
from openpilot.system.hardware import PC, TICI
|
|
from openpilot.system.manager.process import PythonProcess, NativeProcess, DaemonProcess
|
|
from openpilot.system.hardware.hw import Paths
|
|
|
|
from openpilot.iqpilot.mapd.mapd_manager import MAPD_PATH
|
|
|
|
from openpilot.iqpilot.models.helpers import get_active_model_runner
|
|
from iqpilot.konn3kt.utils import hephaestus_ready
|
|
|
|
WEBCAM = os.getenv("USE_WEBCAM") is not None
|
|
|
|
def driverview(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started or params.get_bool("IsDriverViewEnabled")
|
|
|
|
def notcar(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and CP.notCar
|
|
|
|
def iscar(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and not CP.notCar
|
|
|
|
def logging(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
run = (not CP.notCar) or not params.get_bool("DisableLogging")
|
|
return started and run
|
|
|
|
def ublox_available() -> bool:
|
|
return os.path.exists('/dev/ttyHS0') and not os.path.exists('/persist/comma/use-quectel-gps')
|
|
|
|
def ublox(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
use_ublox = ublox_available()
|
|
if use_ublox != params.get_bool("UbloxAvailable"):
|
|
params.put_bool("UbloxAvailable", use_ublox)
|
|
return started and use_ublox
|
|
|
|
def joystick(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and params.get_bool("JoystickDebugMode")
|
|
|
|
def not_joystick(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and not params.get_bool("JoystickDebugMode")
|
|
|
|
def long_maneuver(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and params.get_bool("LongitudinalManeuverMode")
|
|
|
|
def not_long_maneuver(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and not params.get_bool("LongitudinalManeuverMode")
|
|
|
|
def qcomgps(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started and not ublox_available()
|
|
|
|
def always_run(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return True
|
|
|
|
def only_onroad(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return started
|
|
|
|
def only_offroad(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return not started
|
|
|
|
def use_github_runner(started, params, CP: car.CarParams) -> bool:
|
|
return not PC and params.get_bool("EnableGithubRunner") and (
|
|
not params.get_bool("NetworkMetered") and not params.get_bool("GithubRunnerSufficientVoltage"))
|
|
|
|
def use_copyparty(started, params, CP: car.CarParams) -> bool:
|
|
return bool(params.get_bool("EnableCopyparty"))
|
|
|
|
def is_snpe_model(started, params, CP: car.CarParams) -> bool:
|
|
"""Check if the active model runner is SNPE."""
|
|
return bool(get_active_model_runner(params, not started) == custom.IQModelManager.Runner.snpe)
|
|
|
|
def is_tinygrad_model(started, params, CP: car.CarParams) -> bool:
|
|
"""Check if the active model runner is SNPE."""
|
|
return bool(get_active_model_runner(params, not started) == custom.IQModelManager.Runner.tinygrad)
|
|
|
|
def is_stock_model(started, params, CP: car.CarParams) -> bool:
|
|
"""Check if the active model runner is stock."""
|
|
return bool(get_active_model_runner(params, not started) == custom.IQModelManager.Runner.stock)
|
|
|
|
def hephaestus_ready_shim(started, params, CP: car.CarParams) -> bool:
|
|
return hephaestus_ready(params)
|
|
|
|
def mapd_ready(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
return bool(os.path.exists(Paths.mapd_root()))
|
|
|
|
def uploader_ready(started: bool, params: Params, CP: car.CarParams) -> bool:
|
|
if not params.get_bool("OnroadUploads"):
|
|
return only_offroad(started, params, CP)
|
|
|
|
return always_run(started, params, CP)
|
|
|
|
def or_(*fns):
|
|
return lambda *args: operator.or_(*(fn(*args) for fn in fns))
|
|
|
|
def and_(*fns):
|
|
return lambda *args: operator.and_(*(fn(*args) for fn in fns))
|
|
|
|
procs = [
|
|
#NativeProcess("loggerd", "system/loggerd", ["./loggerd"], logging),
|
|
NativeProcess("encoderd", "system/loggerd", ["./encoderd"], only_onroad),
|
|
NativeProcess("stream_encoderd", "system/loggerd", ["./encoderd", "--stream"], notcar),
|
|
PythonProcess("logmessaged", "system.logmessaged", always_run),
|
|
|
|
NativeProcess("camerad", "system/camerad", ["./camerad"], driverview, enabled=not WEBCAM),
|
|
PythonProcess("webcamerad", "tools.webcam.camerad", driverview, enabled=WEBCAM),
|
|
PythonProcess("proclogd", "system.proclogd", only_onroad, enabled=platform.system() != "Darwin"),
|
|
PythonProcess("journald", "system.journald", only_onroad, platform.system() != "Darwin"),
|
|
#PythonProcess("micd", "system.micd", iscar),
|
|
PythonProcess("timed", "system.timed", always_run, enabled=not PC),
|
|
|
|
PythonProcess("modeld", "selfdrive.modeld.modeld", and_(only_onroad, is_stock_model)),
|
|
PythonProcess("dmonitoringmodeld", "selfdrive.modeld.dmonitoringmodeld", driverview, enabled=(WEBCAM or not PC)),
|
|
|
|
PythonProcess("sensord", "system.sensord.sensord", only_onroad, enabled=not PC),
|
|
PythonProcess("ui", "selfdrive.ui.ui", always_run, restart_if_crash=True),
|
|
PythonProcess("soundd", "selfdrive.ui.soundd", driverview),
|
|
PythonProcess("locationd", "selfdrive.locationd.locationd", only_onroad),
|
|
NativeProcess("_pandad", "selfdrive/pandad", ["./pandad"], always_run, enabled=False),
|
|
PythonProcess("calibrationd", "selfdrive.locationd.calibrationd", only_onroad),
|
|
PythonProcess("torqued", "selfdrive.locationd.torqued", only_onroad),
|
|
PythonProcess("controlsd", "selfdrive.controls.controlsd", and_(not_joystick, iscar)),
|
|
PythonProcess("joystickd", "tools.joystick.joystickd", or_(joystick, notcar)),
|
|
PythonProcess("selfdrived", "selfdrive.selfdrived.selfdrived", only_onroad),
|
|
PythonProcess("card", "selfdrive.car.card", only_onroad),
|
|
PythonProcess("deleter", "system.loggerd.deleter", always_run),
|
|
PythonProcess("dmonitoringd", "selfdrive.monitoring.dmonitoringd", driverview, enabled=(WEBCAM or not PC)),
|
|
PythonProcess("qcomgpsd", "system.qcomgpsd.qcomgpsd", qcomgps, enabled=TICI),
|
|
PythonProcess("pandad", "selfdrive.pandad.pandad", always_run),
|
|
PythonProcess("paramsd", "selfdrive.locationd.paramsd", only_onroad),
|
|
PythonProcess("lagd", "selfdrive.locationd.lagd", only_onroad),
|
|
PythonProcess("ubloxd", "system.ubloxd.ubloxd", ublox, enabled=TICI),
|
|
PythonProcess("pigeond", "system.ubloxd.pigeond", ublox, enabled=TICI),
|
|
PythonProcess("plannerd", "selfdrive.controls.plannerd", not_long_maneuver),
|
|
PythonProcess("maneuversd", "tools.longitudinal_maneuvers.maneuversd", long_maneuver),
|
|
PythonProcess("radard", "selfdrive.controls.radard", only_onroad),
|
|
PythonProcess("hardwared", "system.hardware.hardwared", always_run),
|
|
PythonProcess("tombstoned", "system.tombstoned", always_run, enabled=not PC),
|
|
PythonProcess("updated", "system.updated.updated", only_offroad, enabled=not PC),
|
|
PythonProcess("uploader", "system.loggerd.uploader", uploader_ready),
|
|
PythonProcess("statsd", "system.statsd", always_run),
|
|
PythonProcess("beep", "selfdrive.selfdrived.beep", always_run),
|
|
PythonProcess("feedbackd", "selfdrive.ui.feedback.feedbackd", only_onroad),
|
|
|
|
# debug procs
|
|
NativeProcess("bridge", "cereal/messaging", ["./bridge"], notcar),
|
|
PythonProcess("webrtcd", "system.webrtc.webrtcd", notcar),
|
|
PythonProcess("webjoystick", "tools.bodyteleop.web", notcar),
|
|
]
|
|
|
|
# iqpilot
|
|
procs += [
|
|
# Models
|
|
PythonProcess("models_manager", "iqpilot.models.manager", only_offroad),
|
|
NativeProcess("modeld_snpe", "iqpilot/modeld", ["./modeld"], and_(only_onroad, is_snpe_model)),
|
|
NativeProcess("modeld_tinygrad", "iqpilot/modeld_v2", ["./modeld"], and_(only_onroad, is_tinygrad_model)),
|
|
|
|
PythonProcess("backup_manager_k3", "iqpilot.konn3kt.backups.manager", and_(only_offroad, hephaestus_ready_shim)),
|
|
|
|
# mapd
|
|
NativeProcess("mapd", Paths.mapd_root(), ["bash", "-c", f"{MAPD_PATH} > /dev/null 2>&1"], mapd_ready),
|
|
PythonProcess("mapd_manager", "iqpilot.mapd.mapd_manager", always_run),
|
|
|
|
# locationd
|
|
NativeProcess("locationd_llk", "iqpilot/selfdrive/locationd", ["./locationd"], only_onroad),
|
|
]
|
|
|
|
if os.path.exists("./github_runner.sh"):
|
|
procs += [NativeProcess("github_runner_start", "system/manager", ["./github_runner.sh", "start"], and_(only_offroad, use_github_runner), sigkill=False)]
|
|
|
|
if os.path.exists("../../third_party/copyparty/copyparty-sfx.py"):
|
|
iqpilot_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
|
copyparty_args = [f"-v{Paths.crash_log_root()}:/swaglogs:r"]
|
|
copyparty_args += [f"-v{Paths.log_root()}:/routes:r"]
|
|
copyparty_args += [f"-v{Paths.model_root()}:/models:rw"]
|
|
copyparty_args += [f"-v{iqpilot_root}:/iqpilot:rw"]
|
|
copyparty_args += ["-p8080"]
|
|
copyparty_args += ["-z"]
|
|
copyparty_args += ["-q"]
|
|
procs += [NativeProcess("copyparty-sfx", "third_party/copyparty", ["./copyparty-sfx.py", *copyparty_args], and_(only_offroad, use_copyparty))]
|
|
|
|
managed_processes = {p.name: p for p in procs}
|