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.
57 lines
1.8 KiB
57 lines
1.8 KiB
5 years ago
|
#!/usr/bin/env python3
|
||
|
|
import argparse
|
||
|
|
import time
|
||
|
|
import zmq
|
||
|
|
|
||
|
|
import cereal.messaging as messaging
|
||
|
|
from cereal.services import service_list
|
||
|
|
from selfdrive.controls.lib.events import EVENTS, Alert
|
||
|
|
|
||
|
|
def now_millis(): return time.time() * 1000
|
||
|
|
|
||
|
|
ALERTS = [a for _, et in EVENTS.items() for _, a in et.items() if isinstance(a, Alert)]
|
||
|
|
|
||
|
|
#from cereal import car
|
||
|
|
#ALERTS = [a for a in ALERTS if a.audible_alert == car.CarControl.HUDControl.AudibleAlert.chimeWarningRepeat]
|
||
|
|
|
||
|
|
default_alerts = sorted(ALERTS, key=lambda alert: (alert.alert_size, len(alert.alert_text_2)))
|
||
|
|
|
||
|
|
def cycle_alerts(duration_millis, alerts=None):
|
||
|
|
if alerts is None:
|
||
|
|
alerts = default_alerts
|
||
|
|
|
||
|
|
controls_state = messaging.pub_sock('controlsState')
|
||
|
|
|
||
|
|
idx, last_alert_millis = 0, 0
|
||
|
|
alert = alerts[0]
|
||
|
|
while 1:
|
||
|
|
if (now_millis() - last_alert_millis) > duration_millis:
|
||
|
|
alert = alerts[idx]
|
||
|
|
idx = (idx + 1) % len(alerts)
|
||
|
|
last_alert_millis = now_millis()
|
||
|
|
print('sending {}'.format(str(alert)))
|
||
|
|
|
||
|
|
dat = messaging.new_message()
|
||
|
|
dat.init('controlsState')
|
||
|
|
|
||
|
|
dat.controlsState.alertType = alert.alert_type
|
||
|
|
dat.controlsState.alertText1 = alert.alert_text_1
|
||
|
|
dat.controlsState.alertText2 = alert.alert_text_2
|
||
|
|
dat.controlsState.alertSize = alert.alert_size
|
||
|
|
#dat.controlsState.alertStatus = alert.alert_status
|
||
|
|
dat.controlsState.alertSound = alert.audible_alert
|
||
|
|
controls_state.send(dat.to_bytes())
|
||
|
|
|
||
|
|
time.sleep(0.01)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('--duration', type=int, default=1000)
|
||
|
|
parser.add_argument('--alert-types', nargs='+')
|
||
|
|
args = parser.parse_args()
|
||
|
|
alerts = None
|
||
|
|
if args.alert_types:
|
||
|
|
alerts = [next(a for a in ALERTS if a.alert_type==alert_type) for alert_type in args.alert_types]
|
||
|
|
|
||
|
|
cycle_alerts(args.duration, alerts=alerts)
|