from cereal import log
from selfdrive . swaglog import cloudlog
from selfdrive . controls . lib . alerts import ALERTS
from common . realtime import sec_since_boot
import copy
AlertSize = log . ControlsState . AlertSize
AlertStatus = log . ControlsState . AlertStatus
class AlertManager ( object ) :
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 , 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 = sec_since_boot ( )
# 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 , cur_time ) :
# 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 = " none "
self . audible_alert = " 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
