You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.4 KiB
48 lines
1.4 KiB
#!/usr/bin/env python3
|
|
import gc
|
|
|
|
import cereal.messaging as messaging
|
|
from openpilot.common.params import Params
|
|
from openpilot.common.realtime import set_realtime_priority
|
|
from openpilot.selfdrive.monitoring.helpers import DriverMonitoring
|
|
|
|
|
|
def dmonitoringd_thread():
|
|
gc.disable()
|
|
set_realtime_priority(2)
|
|
|
|
params = Params()
|
|
pm = messaging.PubMaster(['driverMonitoringState'])
|
|
sm = messaging.SubMaster(['driverStateV2', 'liveCalibration', 'carState', 'controlsState', 'modelV2'], poll='driverStateV2')
|
|
|
|
DM = DriverMonitoring(rhd_saved=params.get_bool("IsRhdDetected"), always_on=params.get_bool("AlwaysOnDM"))
|
|
|
|
# 20Hz <- dmonitoringmodeld
|
|
while True:
|
|
sm.update()
|
|
if (not sm.updated['driverStateV2']) or (not sm.all_checks()):
|
|
# iterate when model has new output
|
|
continue
|
|
|
|
DM.run_step(sm)
|
|
|
|
# publish
|
|
dat = DM.get_state_packet()
|
|
pm.send('driverMonitoringState', dat)
|
|
|
|
# load live always-on toggle
|
|
if sm['driverStateV2'].frameId % 40 == 1:
|
|
DM.always_on = params.get_bool("AlwaysOnDM")
|
|
|
|
# save rhd virtual toggle every 5 mins
|
|
if (sm['driverStateV2'].frameId % 6000 == 0 and
|
|
DM.wheelpos_learner.filtered_stat.n > DM.settings._WHEELPOS_FILTER_MIN_COUNT and
|
|
DM.wheel_on_right == (DM.wheelpos_learner.filtered_stat.M > DM.settings._WHEELPOS_THRESHOLD)):
|
|
params.put_bool_nonblocking("IsRhdDetected", DM.wheel_on_right)
|
|
|
|
def main():
|
|
dmonitoringd_thread()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|