75 lines
2.4 KiB
Python
Executable File
75 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from cereal import car
|
|
from opendbc.can.parser import CANParser
|
|
from selfdrive.car.ford.values import DBC
|
|
from selfdrive.config import Conversions as CV
|
|
from selfdrive.car.interfaces import RadarInterfaceBase
|
|
|
|
RADAR_MSGS = list(range(0x500, 0x540))
|
|
|
|
def _create_radar_can_parser(car_fingerprint):
|
|
dbc_f = DBC[car_fingerprint]['radar']
|
|
msg_n = len(RADAR_MSGS)
|
|
signals = list(zip(['X_Rel'] * msg_n + ['Angle'] * msg_n + ['V_Rel'] * msg_n,
|
|
RADAR_MSGS * 3,
|
|
[0] * msg_n + [0] * msg_n + [0] * msg_n))
|
|
checks = list(zip(RADAR_MSGS, [20]*msg_n))
|
|
|
|
return CANParser(dbc_f, signals, checks, 1)
|
|
|
|
class RadarInterface(RadarInterfaceBase):
|
|
def __init__(self, CP):
|
|
super().__init__(CP)
|
|
self.validCnt = {key: 0 for key in RADAR_MSGS}
|
|
self.track_id = 0
|
|
|
|
self.rcp = _create_radar_can_parser(CP.carFingerprint)
|
|
self.trigger_msg = 0x53f
|
|
self.updated_messages = set()
|
|
|
|
def update(self, can_strings):
|
|
vls = self.rcp.update_strings(can_strings)
|
|
self.updated_messages.update(vls)
|
|
|
|
if self.trigger_msg not in self.updated_messages:
|
|
return None
|
|
|
|
|
|
ret = car.RadarData.new_message()
|
|
errors = []
|
|
if not self.rcp.can_valid:
|
|
errors.append("canError")
|
|
ret.errors = errors
|
|
|
|
for ii in sorted(self.updated_messages):
|
|
cpt = self.rcp.vl[ii]
|
|
|
|
if cpt['X_Rel'] > 0.00001:
|
|
self.validCnt[ii] = 0 # reset counter
|
|
|
|
if cpt['X_Rel'] > 0.00001:
|
|
self.validCnt[ii] += 1
|
|
else:
|
|
self.validCnt[ii] = max(self.validCnt[ii] - 1, 0)
|
|
#print ii, self.validCnt[ii], cpt['VALID'], cpt['X_Rel'], cpt['Angle']
|
|
|
|
# radar point only valid if there have been enough valid measurements
|
|
if self.validCnt[ii] > 0:
|
|
if ii not in self.pts:
|
|
self.pts[ii] = car.RadarData.RadarPoint.new_message()
|
|
self.pts[ii].trackId = self.track_id
|
|
self.track_id += 1
|
|
self.pts[ii].dRel = cpt['X_Rel'] # from front of car
|
|
self.pts[ii].yRel = cpt['X_Rel'] * cpt['Angle'] * CV.DEG_TO_RAD # in car frame's y axis, left is positive
|
|
self.pts[ii].vRel = cpt['V_Rel']
|
|
self.pts[ii].aRel = float('nan')
|
|
self.pts[ii].yvRel = float('nan')
|
|
self.pts[ii].measured = True
|
|
else:
|
|
if ii in self.pts:
|
|
del self.pts[ii]
|
|
|
|
ret.points = list(self.pts.values())
|
|
self.updated_messages.clear()
|
|
return ret
|