124 lines
5.1 KiB
Python
124 lines
5.1 KiB
Python
from iqdbc.can import CANDefine, CANParser
|
|
from iqdbc.car import Bus, create_button_events, structs
|
|
from iqdbc.car.chrysler.values import DBC, STEER_THRESHOLD, RAM_CARS
|
|
from iqdbc.car.common.conversions import Conversions as CV
|
|
from iqdbc.car.interfaces import CarStateBase
|
|
|
|
from iqdbc.lvbs.car.chrysler.aol import AolCarState
|
|
from iqdbc.lvbs.car.chrysler.iq_carstate import IQCarState
|
|
|
|
ButtonType = structs.CarState.ButtonEvent.Type
|
|
|
|
|
|
class CarState(CarStateBase, AolCarState, IQCarState):
|
|
def __init__(self, CP, CP_IQ):
|
|
CarStateBase.__init__(self, CP, CP_IQ)
|
|
AolCarState.__init__(self, CP, CP_IQ)
|
|
IQCarState.__init__(self, CP, CP_IQ)
|
|
self.CP = CP
|
|
can_define = CANDefine(DBC[CP.carFingerprint][Bus.pt])
|
|
|
|
self.auto_high_beam = 0
|
|
self.button_counter = 0
|
|
self.lkas_car_model = -1
|
|
|
|
if CP.carFingerprint in RAM_CARS:
|
|
self.shifter_values = can_define.dv["Transmission_Status"]["Gear_State"]
|
|
else:
|
|
self.shifter_values = can_define.dv["GEAR"]["PRNDL"]
|
|
|
|
self.distance_button = 0
|
|
|
|
def update(self, can_parsers) -> tuple[structs.CarState, structs.IQCarState]:
|
|
cp = can_parsers[Bus.pt]
|
|
cp_cam = can_parsers[Bus.cam]
|
|
|
|
ret = structs.CarState()
|
|
ret_iq = structs.IQCarState()
|
|
|
|
prev_distance_button = self.distance_button
|
|
self.distance_button = cp.vl["CRUISE_BUTTONS"]["ACC_Distance_Dec"]
|
|
|
|
# lock info
|
|
ret.doorOpen = any([cp.vl["BCM_1"]["DOOR_OPEN_FL"],
|
|
cp.vl["BCM_1"]["DOOR_OPEN_FR"],
|
|
cp.vl["BCM_1"]["DOOR_OPEN_RL"],
|
|
cp.vl["BCM_1"]["DOOR_OPEN_RR"]])
|
|
ret.seatbeltUnlatched = cp.vl["ORC_1"]["SEATBELT_DRIVER_UNLATCHED"] == 1
|
|
|
|
# brake pedal
|
|
ret.brake = 0
|
|
ret.brakePressed = cp.vl["ESP_1"]['Brake_Pedal_State'] == 1 # Physical brake pedal switch
|
|
|
|
# gas pedal
|
|
ret.gasPressed = cp.vl["ECM_5"]["Accelerator_Position"] > 1e-5
|
|
|
|
# car speed
|
|
if self.CP.carFingerprint in RAM_CARS:
|
|
ret.vEgoRaw = cp.vl["ESP_8"]["Vehicle_Speed"] * CV.KPH_TO_MS
|
|
ret.gearShifter = self.parse_gear_shifter(self.shifter_values.get(cp.vl["Transmission_Status"]["Gear_State"], None))
|
|
else:
|
|
ret.vEgoRaw = (cp.vl["SPEED_1"]["SPEED_LEFT"] + cp.vl["SPEED_1"]["SPEED_RIGHT"]) / 2.
|
|
ret.gearShifter = self.parse_gear_shifter(self.shifter_values.get(cp.vl["GEAR"]["PRNDL"], None))
|
|
ret.vEgo, ret.aEgo = self.update_speed_kf(ret.vEgoRaw)
|
|
ret.standstill = not ret.vEgoRaw > 0.001
|
|
|
|
# button presses
|
|
ret.leftBlinker, ret.rightBlinker = self.update_blinker_from_stalk(200, cp.vl["STEERING_LEVERS"]["TURN_SIGNALS"] == 1,
|
|
cp.vl["STEERING_LEVERS"]["TURN_SIGNALS"] == 2)
|
|
ret.genericToggle = cp.vl["STEERING_LEVERS"]["HIGH_BEAM_PRESSED"] == 1
|
|
|
|
# steering wheel
|
|
ret.steeringAngleDeg = cp.vl["STEERING"]["STEERING_ANGLE"] + cp.vl["STEERING"]["STEERING_ANGLE_HP"]
|
|
ret.steeringRateDeg = cp.vl["STEERING"]["STEERING_RATE"]
|
|
ret.steeringTorque = cp.vl["EPS_2"]["COLUMN_TORQUE"]
|
|
ret.steeringTorqueEps = cp.vl["EPS_2"]["EPS_TORQUE_MOTOR"]
|
|
ret.steeringPressed = abs(ret.steeringTorque) > STEER_THRESHOLD
|
|
|
|
# cruise state
|
|
cp_cruise = cp_cam if self.CP.carFingerprint in RAM_CARS else cp
|
|
|
|
ret.cruiseState.available = cp_cruise.vl["DAS_3"]["ACC_AVAILABLE"] == 1
|
|
ret.cruiseState.enabled = cp_cruise.vl["DAS_3"]["ACC_ACTIVE"] == 1
|
|
ret.cruiseState.speed = cp_cruise.vl["DAS_4"]["ACC_SET_SPEED_KPH"] * CV.KPH_TO_MS
|
|
ret.cruiseState.nonAdaptive = cp_cruise.vl["DAS_4"]["ACC_STATE"] in (1, 2) # 1 NormalCCOn and 2 NormalCCSet
|
|
ret.cruiseState.standstill = cp_cruise.vl["DAS_3"]["ACC_STANDSTILL"] == 1
|
|
ret.accFaulted = cp_cruise.vl["DAS_3"]["ACC_FAULTED"] != 0
|
|
|
|
if self.CP.carFingerprint in RAM_CARS:
|
|
# Auto High Beam isn't Located in this message on chrysler or jeep currently located in 729 message
|
|
self.auto_high_beam = cp_cam.vl["DAS_6"]['AUTO_HIGH_BEAM_ON']
|
|
ret.steerFaultTemporary = cp.vl["EPS_3"]["DASM_FAULT"] == 1
|
|
else:
|
|
ret.steerFaultTemporary = cp.vl["EPS_2"]["LKAS_TEMPORARY_FAULT"] == 1
|
|
ret.steerFaultPermanent = cp.vl["EPS_2"]["LKAS_STATE"] == 4
|
|
|
|
# blindspot sensors
|
|
if self.CP.enableBsm:
|
|
ret.leftBlindspot = cp.vl["BSM_1"]["LEFT_STATUS"] == 1
|
|
ret.rightBlindspot = cp.vl["BSM_1"]["RIGHT_STATUS"] == 1
|
|
|
|
self.lkas_car_model = cp_cam.vl["DAS_6"]["CAR_MODEL"]
|
|
self.button_counter = cp.vl["CRUISE_BUTTONS"]["COUNTER"]
|
|
|
|
AolCarState.update_aol(self, ret, can_parsers)
|
|
IQCarState.update(self, ret, ret_iq, can_parsers)
|
|
|
|
ret.buttonEvents = [
|
|
*create_button_events(self.distance_button, prev_distance_button, {1: ButtonType.gapAdjustCruise}),
|
|
*create_button_events(self.lkas_button, self.prev_lkas_button, {1: ButtonType.lkas}),
|
|
*self.button_events,
|
|
]
|
|
|
|
return ret, ret_iq
|
|
|
|
@staticmethod
|
|
def get_can_parsers(CP, CP_IQ):
|
|
pt_messages: list = []
|
|
cam_messages: list = []
|
|
AolCarState.get_parser(CP, pt_messages, cam_messages)
|
|
return {
|
|
Bus.pt: CANParser(DBC[CP.carFingerprint][Bus.pt], pt_messages, 0),
|
|
Bus.cam: CANParser(DBC[CP.carFingerprint][Bus.pt], cam_messages, 2),
|
|
}
|