from cereal import car , log
from common . realtime import DT_CTRL
from selfdrive . swaglog import cloudlog
from selfdrive . controls . lib . alerts import ALERTS
import copy
AlertSize = log . ControlsState . AlertSize
AlertStatus = log . ControlsState . AlertStatus
VisualAlert = car . CarControl . HUDControl . VisualAlert
AudibleAlert = car . CarControl . HUDControl . AudibleAlert
class AlertManager ( ) :
def __init__ ( self ) :
self . activealerts = [ ]
self . alerts = { alert . alert_type : alert for alert in ALERTS }
def alertPresent ( self ) :
return len ( self . activealerts ) > 0
def add ( self , frame , alert_type , enabled = True , extra_text_1 = ' ' , extra_text_2 = ' ' ) :
alert_type = str ( alert_type )
added_alert = copy . copy ( self . alerts [ alert_type ] )
added_alert . alert_text_1 + = extra_text_1
added_alert . alert_text_2 + = extra_text_2
added_alert . start_time = frame * DT_CTRL
# if new alert is higher priority, log it
if not self . alertPresent ( ) or added_alert . alert_priority > self . activealerts [ 0 ] . alert_priority :
cloudlog . event ( ' alert_add ' , alert_type = alert_type , enabled = enabled )
self . activealerts . append ( added_alert )
# sort by priority first and then by start_time
self . activealerts . sort ( key = lambda k : ( k . alert_priority , k . start_time ) , reverse = True )
def process_alerts ( self , frame ) :
cur_time = frame * DT_CTRL
# first get rid of all the expired alerts
self . activealerts = [ a for a in self . activealerts if a . start_time +
max ( a . duration_sound , a . duration_hud_alert , a . duration_text ) > cur_time ]
current_alert = self . activealerts [ 0 ] if self . alertPresent ( ) else None
# start with assuming no alerts
self . alert_type = " "
self . alert_text_1 = " "
self . alert_text_2 = " "
self . alert_status = AlertStatus . normal
self . alert_size = AlertSize . none
self . visual_alert = VisualAlert . none
self . audible_alert = AudibleAlert . none
self . alert_rate = 0.
if current_alert :
self . alert_type = current_alert . alert_type
if current_alert . start_time + current_alert . duration_sound > cur_time :
self . audible_alert = current_alert . audible_alert
if current_alert . start_time + current_alert . duration_hud_alert > cur_time :
self . visual_alert = current_alert . visual_alert
if current_alert . start_time + current_alert . duration_text > cur_time :
self . alert_text_1 = current_alert . alert_text_1
self . alert_text_2 = current_alert . alert_text_2
self . alert_status = current_alert . alert_status
self . alert_size = current_alert . alert_size
self . alert_rate = current_alert . alert_rate
