79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
import os
|
|
from common.basedir import BASEDIR
|
|
|
|
def get_attr_from_cars(attr):
|
|
# read all the folders in selfdrive/car and return a dict where:
|
|
# - keys are all the car models
|
|
# - values are attr values from all car folders
|
|
result = {}
|
|
|
|
for car_folder in [x[0] for x in os.walk(BASEDIR + '/selfdrive/car')]:
|
|
try:
|
|
car_name = car_folder.split('/')[-1]
|
|
values = __import__('selfdrive.car.%s.values' % car_name, fromlist=[attr])
|
|
if hasattr(values, attr):
|
|
attr_values = getattr(values, attr)
|
|
else:
|
|
continue
|
|
|
|
for f, v in attr_values.items():
|
|
result[f] = v
|
|
|
|
except (ImportError, IOError):
|
|
pass
|
|
|
|
return result
|
|
|
|
|
|
def get_fw_versions_list():
|
|
return get_attr_from_cars('FW_VERSIONS')
|
|
|
|
|
|
def get_fingerprint_list():
|
|
# read all the folders in selfdrive/car and return a dict where:
|
|
# - keys are all the car models for which we have a fingerprint
|
|
# - values are lists dicts of messages that constitute the unique
|
|
# CAN fingerprint of each car model and all its variants
|
|
return get_attr_from_cars('FINGERPRINTS')
|
|
|
|
|
|
FW_VERSIONS = get_fw_versions_list()
|
|
_FINGERPRINTS = get_fingerprint_list()
|
|
|
|
_DEBUG_ADDRESS = {1880: 8} # reserved for debug purposes
|
|
|
|
def is_valid_for_fingerprint(msg, car_fingerprint):
|
|
adr = msg.address
|
|
# ignore addresses that are more than 11 bits
|
|
return (adr in car_fingerprint and car_fingerprint[adr] == len(msg.dat)) or adr >= 0x800
|
|
|
|
|
|
def eliminate_incompatible_cars(msg, candidate_cars):
|
|
"""Removes cars that could not have sent msg.
|
|
|
|
Inputs:
|
|
msg: A cereal/log CanData message from the car.
|
|
candidate_cars: A list of cars to consider.
|
|
|
|
Returns:
|
|
A list containing the subset of candidate_cars that could have sent msg.
|
|
"""
|
|
compatible_cars = []
|
|
|
|
for car_name in candidate_cars:
|
|
car_fingerprints = _FINGERPRINTS[car_name]
|
|
|
|
for fingerprint in car_fingerprints:
|
|
fingerprint.update(_DEBUG_ADDRESS) # add alien debug address
|
|
|
|
if is_valid_for_fingerprint(msg, fingerprint):
|
|
compatible_cars.append(car_name)
|
|
break
|
|
|
|
return compatible_cars
|
|
|
|
|
|
def all_known_cars():
|
|
"""Returns a list of all known car strings."""
|
|
return list(_FINGERPRINTS.keys())
|