mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-03-01 02:23:56 +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>
87 lines
2.5 KiB
Python
Executable File
87 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import jinja2
|
|
import os
|
|
|
|
from cereal import car
|
|
from openpilot.common.basedir import BASEDIR
|
|
from openpilot.selfdrive.car.interfaces import get_interface_attr
|
|
|
|
Ecu = car.CarParams.Ecu
|
|
|
|
CARS = get_interface_attr('CAR')
|
|
FW_VERSIONS = get_interface_attr('FW_VERSIONS')
|
|
FINGERPRINTS = get_interface_attr('FINGERPRINTS')
|
|
ECU_NAME = {v: k for k, v in Ecu.schema.enumerants.items()}
|
|
|
|
FINGERPRINTS_PY_TEMPLATE = jinja2.Template("""
|
|
{%- if FINGERPRINTS[brand] %}
|
|
# ruff: noqa: E501
|
|
{% endif %}
|
|
{% if FW_VERSIONS[brand] %}
|
|
from cereal import car
|
|
{% endif %}
|
|
from openpilot.selfdrive.car.{{brand}}.values import CAR
|
|
{% if FW_VERSIONS[brand] %}
|
|
|
|
Ecu = car.CarParams.Ecu
|
|
{% endif %}
|
|
{% if comments +%}
|
|
{{ comments | join() }}
|
|
{% endif %}
|
|
{% if FINGERPRINTS[brand] %}
|
|
|
|
FINGERPRINTS = {
|
|
{% for car, fingerprints in FINGERPRINTS[brand].items() %}
|
|
CAR.{{car.name}}: [{
|
|
{% for fingerprint in fingerprints %}
|
|
{% if not loop.first %}
|
|
{{ "{" }}
|
|
{% endif %}
|
|
{% for key, value in fingerprint.items() %}{{key}}: {{value}}{% if not loop.last %}, {% endif %}{% endfor %}
|
|
|
|
}{% if loop.last %}]{% endif %},
|
|
{% endfor %}
|
|
{% endfor %}
|
|
}
|
|
{% endif %}
|
|
{% if FW_VERSIONS[brand] %}
|
|
|
|
FW_VERSIONS = {
|
|
{% for car, _ in FW_VERSIONS[brand].items() %}
|
|
CAR.{{car.name}}: {
|
|
{% for key, fw_versions in FW_VERSIONS[brand][car].items() %}
|
|
(Ecu.{{ECU_NAME[key[0]]}}, 0x{{"%0x" | format(key[1] | int)}}, \
|
|
{% if key[2] %}0x{{"%0x" | format(key[2] | int)}}{% else %}{{key[2]}}{% endif %}): [
|
|
{% for fw_version in fw_versions %}
|
|
{{fw_version}},
|
|
{% endfor %}
|
|
],
|
|
{% endfor %}
|
|
},
|
|
{% endfor %}
|
|
}
|
|
{% endif %}
|
|
""", trim_blocks=True)
|
|
|
|
|
|
def format_brand_fw_versions(brand, extra_fw_versions: None | dict[str, dict[tuple, list[bytes]]] = None):
|
|
fw_versions = FW_VERSIONS.copy()
|
|
# TODO: this modifies FW_VERSIONS in place
|
|
if extra_fw_versions is not None:
|
|
for platform, ecus in extra_fw_versions.items():
|
|
for ecu, fws in ecus.items():
|
|
fw_versions[brand][platform][ecu] += set(fws) - set(fw_versions[brand][platform][ecu])
|
|
|
|
fingerprints_file = os.path.join(BASEDIR, f"selfdrive/car/{brand}/fingerprints.py")
|
|
with open(fingerprints_file, "r") as f:
|
|
comments = [line for line in f.readlines() if line.startswith("#") and "noqa" not in line]
|
|
|
|
with open(fingerprints_file, "w") as f:
|
|
f.write(FINGERPRINTS_PY_TEMPLATE.render(brand=brand, comments=comments, ECU_NAME=ECU_NAME,
|
|
FINGERPRINTS=FINGERPRINTS, FW_VERSIONS=fw_versions))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for brand in FW_VERSIONS.keys():
|
|
format_brand_fw_versions(brand)
|