mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-19 22:53:55 +08:00
3713e4d5ea
* Add script * fix script * fix script * fix for subaddr * run the script * old style can fingerprints * cleanup * sort imports, make executable, fix path * newline w/o newline * match og can formatting * match og formatting * generate template once * standard name * less nested * can fingerprints comments * fix spacing * no need for PLATFORM_TO_ENUM_NAME! * prep for PRs * comments for all, add honda comments * Auto-generated fingerprint PR from fuzzy fingerprinting cars * Revert "Auto-generated fingerprint PR from fuzzy fingerprinting cars" This reverts commit 97bc9e3bdb9d819dcbe684ceba92ea702d40eaf0. * even closer to original * readd this comment * and run script * add to precommit * add comments * add comments * add to release * use for auto fingerprint * disable precommit for now --------- Co-authored-by: Shane Smiskol <shane@smiskol.com>
80 lines
2.7 KiB
Python
Executable File
80 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
from collections import defaultdict
|
|
from typing import Optional
|
|
from openpilot.selfdrive.debug.format_fingerprints import format_brand_fw_versions
|
|
|
|
from openpilot.tools.lib.logreader import MultiLogIterator
|
|
from openpilot.tools.lib.route import Route
|
|
from openpilot.selfdrive.car.fw_versions import match_fw_to_car
|
|
from openpilot.selfdrive.car.interfaces import get_interface_attr
|
|
|
|
|
|
ALL_FW_VERSIONS = get_interface_attr("FW_VERSIONS")
|
|
ALL_CARS = get_interface_attr("CAR")
|
|
|
|
PLATFORM_TO_PYTHON_CAR_NAME = {brand: {car.value: car.name for car in ALL_CARS[brand]} for brand in ALL_CARS}
|
|
BRAND_TO_PLATFORMS = {brand: [car.value for car in ALL_CARS[brand]] for brand in ALL_CARS}
|
|
PLATFORM_TO_BRAND = dict(sum([[(platform, brand) for platform in BRAND_TO_PLATFORMS[brand]] for brand in BRAND_TO_PLATFORMS], []))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Auto fingerprint from a route")
|
|
parser.add_argument("route", help="The route name to use")
|
|
parser.add_argument("platform", help="The platform, or leave empty to auto-determine using fuzzy", default=None, nargs='?')
|
|
args = parser.parse_args()
|
|
|
|
route = Route(args.route)
|
|
lr = MultiLogIterator(route.qlog_paths())
|
|
|
|
carFw = None
|
|
carVin = None
|
|
carPlatform = None
|
|
|
|
platform: Optional[str] = None
|
|
|
|
for msg in lr:
|
|
if msg.which() == "carParams":
|
|
carFw = msg.carParams.carFw
|
|
carVin = msg.carParams.carVin
|
|
carPlatform = msg.carParams.carFingerprint
|
|
break
|
|
|
|
if carFw is None:
|
|
raise Exception("No fw versions in the provided route...")
|
|
|
|
if args.platform is None: # attempt to auto-determine platform with other fuzzy fingerprints
|
|
_, possible_platforms = match_fw_to_car(carFw, log=False)
|
|
|
|
if len(possible_platforms) != 1:
|
|
print(f"Unable to auto-determine platform, possible platforms: {possible_platforms}")
|
|
|
|
if carPlatform != "mock":
|
|
print("Using platform from route")
|
|
platform = carPlatform
|
|
else:
|
|
platform = None
|
|
else:
|
|
platform = list(possible_platforms)[0]
|
|
else:
|
|
platform = args.platform
|
|
|
|
if platform is None:
|
|
raise Exception("unable to determine platform, try manually specifying the fingerprint.")
|
|
|
|
print("Attempting to add fw version for: ", platform)
|
|
|
|
fw_versions: dict[str, dict[tuple, list[bytes]]] = defaultdict(lambda: defaultdict(list))
|
|
brand = PLATFORM_TO_BRAND[platform]
|
|
|
|
for fw in carFw:
|
|
if fw.brand == brand and not fw.logging:
|
|
addr = fw.address
|
|
subAddr = None if fw.subAddress == 0 else fw.subAddress
|
|
key = (fw.ecu.raw, addr, subAddr)
|
|
|
|
fw_versions[platform][key].append(fw.fwVersion)
|
|
|
|
format_brand_fw_versions(brand, fw_versions)
|