mirror of
https://github.com/MoreTore/openpilot.git
synced 2026-08-04 15:56:08 +08:00
FrogAi
240 lines
9.4 KiB
Python
240 lines
9.4 KiB
Python
#!/usr/bin/env python3
|
|
import random
|
|
|
|
from openpilot.common.conversions import Conversions as CV
|
|
from openpilot.common.realtime import DT_MDL
|
|
from openpilot.selfdrive.controls.controlsd import Desire
|
|
from openpilot.selfdrive.controls.lib.events import ET, EVENTS, EventName, Events
|
|
|
|
from openpilot.frogpilot.assets.theme_manager import update_wheel_image
|
|
from openpilot.frogpilot.common.frogpilot_variables import CRUISING_SPEED, NON_DRIVING_GEARS, params, params_memory
|
|
|
|
RANDOM_EVENTS_CHANCE = 0.01 * DT_MDL
|
|
|
|
class FrogPilotEvents:
|
|
def __init__(self, FrogPilotPlanner):
|
|
self.frogpilot_planner = FrogPilotPlanner
|
|
|
|
self.events = Events()
|
|
|
|
self.accel30_played = False
|
|
self.accel35_played = False
|
|
self.accel40_played = False
|
|
self.always_on_lateral_enabled_previously = False
|
|
self.dejaVu_played = False
|
|
self.fcw_played = False
|
|
self.firefox_played = False
|
|
self.goat_played = False
|
|
self.holiday_theme_played = False
|
|
self.nnff_played = False
|
|
self.no_entry_alert_played = False
|
|
self.previous_traffic_mode = False
|
|
self.random_event_playing = False
|
|
self.startup_seen = False
|
|
self.stopped_for_light = False
|
|
self.this_is_fine_played = False
|
|
self.vCruise69_played = False
|
|
self.youveGotMail_played = False
|
|
|
|
self.max_acceleration = 0
|
|
self.random_event_timer = 0
|
|
self.tracking_lead_distance = 0
|
|
|
|
def update(self, v_cruise, sm, frogpilot_toggles):
|
|
self.events.clear()
|
|
|
|
if self.random_event_playing:
|
|
self.random_event_timer += DT_MDL
|
|
|
|
if self.random_event_timer >= 5:
|
|
update_wheel_image(frogpilot_toggles.wheel_image, frogpilot_toggles.current_holiday_theme, False)
|
|
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.random_event_playing = False
|
|
self.random_event_timer = 0
|
|
|
|
if self.frogpilot_planner.frogpilot_vcruise.forcing_stop:
|
|
self.events.add(EventName.forcingStop)
|
|
|
|
if not self.frogpilot_planner.tracking_lead and sm["carState"].standstill and sm["carState"].gearShifter not in NON_DRIVING_GEARS and frogpilot_toggles.green_light_alert:
|
|
if not self.frogpilot_planner.model_stopped and self.stopped_for_light:
|
|
self.events.add(EventName.greenLight)
|
|
|
|
self.stopped_for_light = self.frogpilot_planner.cem.stop_light_detected
|
|
else:
|
|
self.stopped_for_light = False
|
|
|
|
if not self.holiday_theme_played and self.startup_seen and sm["controlsState"].alertText1 == "" and frogpilot_toggles.current_holiday_theme != "stock" and len(self.events) == 0:
|
|
self.events.add(EventName.holidayActive)
|
|
|
|
self.holiday_theme_played = True
|
|
|
|
if self.frogpilot_planner.tracking_lead and sm["carState"].standstill and sm["carState"].gearShifter not in NON_DRIVING_GEARS and frogpilot_toggles.lead_departing_alert:
|
|
if self.tracking_lead_distance == 0:
|
|
self.tracking_lead_distance = self.frogpilot_planner.lead_one.dRel
|
|
|
|
lead_departing = self.frogpilot_planner.lead_one.dRel - self.tracking_lead_distance > 1
|
|
lead_departing &= self.frogpilot_planner.lead_one.vLead > 1
|
|
|
|
if lead_departing:
|
|
self.events.add(EventName.leadDeparting)
|
|
else:
|
|
self.tracking_lead_distance = 0
|
|
|
|
if not self.nnff_played and self.startup_seen and sm["controlsState"].alertText1 == "" and len(self.events) == 0 and params.get("NNFFModelName", encoding="utf-8") is not None:
|
|
self.events.add(EventName.torqueNNLoad)
|
|
|
|
self.nnff_played = True
|
|
|
|
if not self.random_event_playing and frogpilot_toggles.random_events:
|
|
acceleration = sm["carState"].aEgo
|
|
|
|
if not sm["carState"].gasPressed:
|
|
self.max_acceleration = max(acceleration, self.max_acceleration)
|
|
else:
|
|
self.max_acceleration = 0
|
|
|
|
if not self.accel30_played and 3.5 > self.max_acceleration >= 3.0 and acceleration < 1.5:
|
|
self.events.add(EventName.accel30)
|
|
|
|
update_wheel_image("weeb_wheel")
|
|
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.accel30_played = True
|
|
self.random_event_playing = True
|
|
|
|
self.max_acceleration = 0
|
|
|
|
elif not self.accel35_played and 4.0 > self.max_acceleration >= 3.5 and acceleration < 1.5:
|
|
self.events.add(EventName.accel35)
|
|
|
|
update_wheel_image("tree_fiddy")
|
|
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.accel35_played = True
|
|
self.random_event_playing = True
|
|
|
|
self.max_acceleration = 0
|
|
|
|
elif not self.accel40_played and self.max_acceleration >= 4.0 and acceleration < 1.5:
|
|
self.events.add(EventName.accel40)
|
|
|
|
update_wheel_image("great_scott")
|
|
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.accel40_played = True
|
|
self.random_event_playing = True
|
|
|
|
self.max_acceleration = 0
|
|
|
|
|
|
if not self.dejaVu_played and sm["carState"].vEgo > CRUISING_SPEED * 2 and self.frogpilot_planner.road_curvature_detected:
|
|
if sm["carState"].vEgo > (1 / abs(self.frogpilot_planner.road_curvature))**0.75 * 2 > CRUISING_SPEED * 2 and abs(sm["carState"].steeringAngleDeg) > 30:
|
|
self.events.add(EventName.dejaVuCurve)
|
|
|
|
self.dejaVu_played = True
|
|
self.random_event_playing = True
|
|
|
|
if not self.no_entry_alert_played and sm["controlsState"].alertType == ET.NO_ENTRY:
|
|
self.events.add(EventName.hal9000)
|
|
|
|
self.no_entry_alert_played = True
|
|
self.random_event_playing = True
|
|
|
|
saturated_event_names = [EventName.steerSaturated, EventName.goatSteerSaturated]
|
|
saturated_alert_match = any((sm["controlsState"].alertText1, sm["controlsState"].alertText2) == (EVENTS[e][ET.WARNING].alert_text_1, EVENTS[e][ET.WARNING].alert_text_2) for e in saturated_event_names)
|
|
if saturated_alert_match:
|
|
event_choices = []
|
|
|
|
if not self.firefox_played:
|
|
event_choices.append("firefoxSteerSaturated")
|
|
if not self.goat_played:
|
|
event_choices.append("goatSteerSaturated")
|
|
if not self.this_is_fine_played:
|
|
event_choices.append("thisIsFineSteerSaturated")
|
|
|
|
if event_choices and random.random() < RANDOM_EVENTS_CHANCE:
|
|
event_choice = random.choice(event_choices)
|
|
|
|
if event_choice == "firefoxSteerSaturated":
|
|
self.events.add(EventName.firefoxSteerSaturated)
|
|
|
|
update_wheel_image("firefox")
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.firefox_played = True
|
|
|
|
elif event_choice == "goatSteerSaturated":
|
|
self.events.add(EventName.goatSteerSaturated)
|
|
|
|
update_wheel_image("goat")
|
|
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.goat_played = True
|
|
elif event_choice == "thisIsFineSteerSaturated":
|
|
self.events.add(EventName.thisIsFineSteerSaturated)
|
|
|
|
update_wheel_image("this_is_fine")
|
|
|
|
params_memory.put_bool("UpdateWheelImage", True)
|
|
|
|
self.this_is_fine_played = True
|
|
|
|
self.random_event_playing = True
|
|
|
|
if not self.vCruise69_played and 70 > max(sm["controlsState"].vCruise, sm["controlsState"].vCruiseCluster) * (1 if frogpilot_toggles.is_metric else CV.KPH_TO_MPH) >= 69:
|
|
self.events.add(EventName.vCruise69)
|
|
|
|
self.vCruise69_played = True
|
|
self.random_event_playing = True
|
|
|
|
if not self.fcw_played:
|
|
fcw_alert_match = sm["controlsState"].alertText1 == EVENTS[EventName.fcw][ET.PERMANENT].alert_text_1 and sm["controlsState"].alertText2 == EVENTS[EventName.fcw][ET.PERMANENT].alert_text_2
|
|
stock_aeb_alert_match = sm["controlsState"].alertText1 == EVENTS[EventName.stockAeb][ET.PERMANENT].alert_text_1 and sm["controlsState"].alertText2 == EVENTS[EventName.stockAeb][ET.PERMANENT].alert_text_2
|
|
if fcw_alert_match or stock_aeb_alert_match:
|
|
event_choices = []
|
|
|
|
event_choices.append("toBeContinued")
|
|
event_choices.append("yourFrogTriedToKillMe")
|
|
|
|
event_choice = random.choice(event_choices)
|
|
|
|
if event_choice == "toBeContinued":
|
|
self.events.add(EventName.toBeContinued)
|
|
elif event_choice == "yourFrogTriedToKillMe":
|
|
self.events.add(EventName.yourFrogTriedToKillMe)
|
|
|
|
self.fcw_played = True
|
|
self.random_event_playing = True
|
|
|
|
if not self.youveGotMail_played and sm["frogpilotCarState"].alwaysOnLateralEnabled and not self.always_on_lateral_enabled_previously:
|
|
if random.random() < RANDOM_EVENTS_CHANCE:
|
|
self.events.add(EventName.youveGotMail)
|
|
|
|
self.youveGotMail_played = True
|
|
self.random_event_playing = True
|
|
self.always_on_lateral_enabled_previously = sm["frogpilotCarState"].alwaysOnLateralEnabled
|
|
|
|
if frogpilot_toggles.speed_limit_changed_alert and self.frogpilot_planner.frogpilot_vcruise.slc.speed_limit_changed_timer == DT_MDL:
|
|
self.events.add(EventName.speedLimitChanged)
|
|
|
|
startup_alert_match = sm["controlsState"].alertText1 == frogpilot_toggles.startup_alert_top and sm["controlsState"].alertText2 == frogpilot_toggles.startup_alert_bottom
|
|
self.startup_seen |= startup_alert_match
|
|
|
|
if sm["frogpilotCarState"].trafficModeEnabled != self.previous_traffic_mode:
|
|
if self.previous_traffic_mode:
|
|
self.events.add(EventName.trafficModeInactive)
|
|
else:
|
|
self.events.add(EventName.trafficModeActive)
|
|
self.previous_traffic_mode = sm["frogpilotCarState"].trafficModeEnabled
|
|
|
|
if sm["modelV2"].meta.turnDirection == Desire.turnLeft:
|
|
self.events.add(EventName.turningLeft)
|
|
elif sm["modelV2"].meta.turnDirection == Desire.turnRight:
|
|
self.events.add(EventName.turningRight)
|