mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-08-21 00:03:45 +08:00
thenhnn
77d6d1a8f1
* get platform_str from the enum name * fix tests * add migration table * remove impossible todo * Add link to PR in MIGRATION table Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> * Remove useless brand name comments and rename RAM_1500 to RAM_1500_5TH_GEN * rename RAM_HD to RAM_HD_5TH_GEN * rename references to RAM_HD and RAM_1500 * change "mock" to "MOCK" and rename torque data of Nissan Leaf 2018 IC * remove MOCK from fingerprints.py * change hard-coded car model in test_can_fingerprint.py/test_timing * migration * update ref * space * prius --------- Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> Co-authored-by: justin newberry <justin@comma.ai> Co-authored-by: Justin Newberry <jnewberry0502@gmail.com> old-commit-hash: 489528dcae3888311f92075ba9e0f68626725121
76 lines
2.5 KiB
Python
Executable File
76 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
from collections import defaultdict
|
|
from openpilot.selfdrive.debug.format_fingerprints import format_brand_fw_versions
|
|
|
|
from openpilot.selfdrive.car.fw_versions import match_fw_to_car
|
|
from openpilot.selfdrive.car.interfaces import get_interface_attr
|
|
from openpilot.tools.lib.logreader import LogReader, ReadMode
|
|
|
|
|
|
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()
|
|
|
|
lr = LogReader(args.route, ReadMode.QLOG)
|
|
|
|
carFw = None
|
|
carVin = None
|
|
carPlatform = None
|
|
|
|
platform: str | None = None
|
|
|
|
CP = lr.first("carParams")
|
|
|
|
if CP is None:
|
|
raise Exception("No fw versions in the provided route...")
|
|
|
|
carFw = CP.carFw
|
|
carVin = CP.carVin
|
|
carPlatform = CP.carFingerprint
|
|
|
|
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)
|