57 lines
2.7 KiB
Python
57 lines
2.7 KiB
Python
from cereal import car
|
|
from opendbc.can.parser import CANParser
|
|
from common.numpy_fast import mean
|
|
from selfdrive.config import Conversions as CV
|
|
from selfdrive.car.interfaces import CarStateBase
|
|
from selfdrive.car.ford.values import DBC
|
|
|
|
WHEEL_RADIUS = 0.33
|
|
|
|
class CarState(CarStateBase):
|
|
def update(self, cp):
|
|
ret = car.CarState.new_message()
|
|
ret.wheelSpeeds.rr = cp.vl["WheelSpeed_CG1"]['WhlRr_W_Meas'] * WHEEL_RADIUS
|
|
ret.wheelSpeeds.rl = cp.vl["WheelSpeed_CG1"]['WhlRl_W_Meas'] * WHEEL_RADIUS
|
|
ret.wheelSpeeds.fr = cp.vl["WheelSpeed_CG1"]['WhlFr_W_Meas'] * WHEEL_RADIUS
|
|
ret.wheelSpeeds.fl = cp.vl["WheelSpeed_CG1"]['WhlFl_W_Meas'] * WHEEL_RADIUS
|
|
ret.vEgoRaw = mean([ret.wheelSpeeds.rr, ret.wheelSpeeds.rl, ret.wheelSpeeds.fr, ret.wheelSpeeds.fl])
|
|
ret.vEgo, ret.aEgo = self.update_speed_kf(ret.vEgoRaw)
|
|
ret.standstill = not ret.vEgoRaw > 0.001
|
|
ret.steeringAngleDeg = cp.vl["Steering_Wheel_Data_CG1"]['SteWhlRelInit_An_Sns']
|
|
ret.steeringPressed = not cp.vl["Lane_Keep_Assist_Status"]['LaHandsOff_B_Actl']
|
|
ret.steerError = cp.vl["Lane_Keep_Assist_Status"]['LaActDeny_B_Actl'] == 1
|
|
ret.cruiseState.speed = cp.vl["Cruise_Status"]['Set_Speed'] * CV.MPH_TO_MS
|
|
ret.cruiseState.enabled = not (cp.vl["Cruise_Status"]['Cruise_State'] in [0, 3])
|
|
ret.cruiseState.available = cp.vl["Cruise_Status"]['Cruise_State'] != 0
|
|
ret.gas = cp.vl["EngineData_14"]['ApedPosScal_Pc_Actl'] / 100.
|
|
ret.gasPressed = ret.gas > 1e-6
|
|
ret.brakePressed = bool(cp.vl["Cruise_Status"]["Brake_Drv_Appl"])
|
|
ret.brakeLights = bool(cp.vl["BCM_to_HS_Body"]["Brake_Lights"])
|
|
ret.genericToggle = bool(cp.vl["Steering_Buttons"]["Dist_Incr"])
|
|
# TODO: we also need raw driver torque, needed for Assisted Lane Change
|
|
self.lkas_state = cp.vl["Lane_Keep_Assist_Status"]['LaActAvail_D_Actl']
|
|
|
|
return ret
|
|
|
|
@staticmethod
|
|
def get_can_parser(CP):
|
|
signals = [
|
|
# sig_name, sig_address, default
|
|
("WhlRr_W_Meas", "WheelSpeed_CG1", 0.),
|
|
("WhlRl_W_Meas", "WheelSpeed_CG1", 0.),
|
|
("WhlFr_W_Meas", "WheelSpeed_CG1", 0.),
|
|
("WhlFl_W_Meas", "WheelSpeed_CG1", 0.),
|
|
("SteWhlRelInit_An_Sns", "Steering_Wheel_Data_CG1", 0.),
|
|
("Cruise_State", "Cruise_Status", 0.),
|
|
("Set_Speed", "Cruise_Status", 0.),
|
|
("LaActAvail_D_Actl", "Lane_Keep_Assist_Status", 0),
|
|
("LaHandsOff_B_Actl", "Lane_Keep_Assist_Status", 0),
|
|
("LaActDeny_B_Actl", "Lane_Keep_Assist_Status", 0),
|
|
("ApedPosScal_Pc_Actl", "EngineData_14", 0.),
|
|
("Dist_Incr", "Steering_Buttons", 0.),
|
|
("Brake_Drv_Appl", "Cruise_Status", 0.),
|
|
("Brake_Lights", "BCM_to_HS_Body", 0.),
|
|
]
|
|
checks = []
|
|
return CANParser(DBC[CP.carFingerprint]['pt'], signals, checks, 0, enforce_checks=False)
|