You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.4 KiB
84 lines
2.4 KiB
8 years ago
|
#!/usr/bin/env python
|
||
|
|
import numpy as np
|
||
|
|
from selfdrive.car.honda.can_parser import CANParser
|
||
|
|
from selfdrive.boardd.boardd import can_capnp_to_can_list_old
|
||
|
|
|
||
|
|
from cereal import car
|
||
|
|
|
||
|
|
import zmq
|
||
|
|
from common.services import service_list
|
||
|
|
import selfdrive.messaging as messaging
|
||
|
|
|
||
|
|
def _create_radard_can_parser():
|
||
|
|
dbc_f = 'acura_ilx_2016_nidec.dbc'
|
||
|
|
radar_messages = range(0x430, 0x43A) + range(0x440, 0x446)
|
||
|
|
signals = zip(['LONG_DIST'] * 16 + ['NEW_TRACK'] * 16 + ['LAT_DIST'] * 16 +
|
||
|
|
['REL_SPEED'] * 16, radar_messages * 4,
|
||
|
|
[255] * 16 + [1] * 16 + [0] * 16 + [0] * 16)
|
||
|
|
checks = zip(radar_messages, [20]*16)
|
||
|
|
|
||
|
|
return CANParser(dbc_f, signals, checks)
|
||
|
|
|
||
|
|
class RadarInterface(object):
|
||
|
|
def __init__(self):
|
||
|
|
# radar
|
||
|
|
self.pts = {}
|
||
|
|
self.track_id = 0
|
||
|
|
|
||
|
|
# Nidec
|
||
|
|
self.rcp = _create_radard_can_parser()
|
||
|
|
|
||
|
|
context = zmq.Context()
|
||
|
|
self.logcan = messaging.sub_sock(context, service_list['can'].port)
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
canMonoTimes = []
|
||
|
|
can_pub_radar = []
|
||
|
|
|
||
|
|
# TODO: can hang if no packets show up
|
||
|
|
while 1:
|
||
|
|
for a in messaging.drain_sock(self.logcan, wait_for_one=True):
|
||
|
|
canMonoTimes.append(a.logMonoTime)
|
||
|
|
can_pub_radar.extend(can_capnp_to_can_list_old(a.can, [1, 3]))
|
||
|
|
|
||
|
|
# only run on the 0x445 packets, used for timing
|
||
|
|
if any(x[0] == 0x445 for x in can_pub_radar):
|
||
|
|
break
|
||
|
|
|
||
|
|
self.rcp.update_can(can_pub_radar)
|
||
|
|
|
||
|
|
ret = car.RadarState.new_message()
|
||
|
|
errors = []
|
||
|
|
if not self.rcp.can_valid:
|
||
|
|
errors.append("notValid")
|
||
|
|
ret.errors = errors
|
||
|
|
ret.canMonoTimes = canMonoTimes
|
||
|
|
|
||
|
|
for ii in self.rcp.msgs_upd:
|
||
|
|
cpt = self.rcp.vl[ii]
|
||
|
|
if cpt['LONG_DIST'] < 255:
|
||
|
|
if ii not in self.pts or cpt['NEW_TRACK']:
|
||
|
|
self.pts[ii] = car.RadarState.RadarPoint.new_message()
|
||
|
|
self.pts[ii].trackId = self.track_id
|
||
|
|
self.track_id += 1
|
||
|
|
self.pts[ii].dRel = cpt['LONG_DIST'] # from front of car
|
||
|
|
self.pts[ii].yRel = -cpt['LAT_DIST'] # in car frame's y axis, left is positive
|
||
|
|
self.pts[ii].vRel = cpt['REL_SPEED']
|
||
|
|
self.pts[ii].aRel = float('nan')
|
||
|
|
self.pts[ii].yvRel = float('nan')
|
||
|
|
else:
|
||
|
|
if ii in self.pts:
|
||
|
|
del self.pts[ii]
|
||
|
|
|
||
|
|
ret.points = self.pts.values()
|
||
|
|
return ret
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
RI = RadarInterface()
|
||
|
|
while 1:
|
||
|
|
ret = RI.update()
|
||
|
|
print(chr(27) + "[2J")
|
||
|
|
print ret
|
||
|
|
|
||
|
|
|