add example of using the comma car segments database (#31117)
* add vin fingerprinting example
* perhaps!
* remove segment
* easier
old-commit-hash: 389b8ca30d
vw-mqb-aeb
Justin Newberry
2 years ago
committed by
GitHub
parent
f447062b66
commit
c03fe6bb13
3 changed files with 169 additions and 1 deletions
@ -0,0 +1,140 @@ |
|||||||
|
{ |
||||||
|
"cells": [ |
||||||
|
{ |
||||||
|
"cell_type": "code", |
||||||
|
"execution_count": 2, |
||||||
|
"metadata": {}, |
||||||
|
"outputs": [], |
||||||
|
"source": [ |
||||||
|
"\"\"\"In this example, we use the public comma car segments database to check if vin fingerprinting is feasible for ford.\"\"\"\n", |
||||||
|
"\n", |
||||||
|
"from openpilot.tools.lib.logreader import LogReader\n", |
||||||
|
"from openpilot.tools.lib.comma_car_segments import get_comma_car_segments_database\n", |
||||||
|
"from openpilot.selfdrive.car.ford.values import CAR\n", |
||||||
|
"\n", |
||||||
|
"database = get_comma_car_segments_database()\n", |
||||||
|
"\n", |
||||||
|
"platforms = [c.value for c in CAR]" |
||||||
|
] |
||||||
|
}, |
||||||
|
{ |
||||||
|
"cell_type": "code", |
||||||
|
"execution_count": 3, |
||||||
|
"metadata": {}, |
||||||
|
"outputs": [], |
||||||
|
"source": [ |
||||||
|
"# Adapted from https://github.com/commaai/openpilot/issues/31052#issuecomment-1902690083\n", |
||||||
|
"\n", |
||||||
|
"MODEL_YEAR_CODES = {'M': 2021, 'N': 2022, 'P': 2023, 'R': 2024, 'S': 2025}\n", |
||||||
|
"\n", |
||||||
|
"\n", |
||||||
|
"F150_CODES = ['F1C', 'F1E', 'W1C', 'W1E', 'X1C', 'X1E', 'W1R', 'W1P', 'W1S', 'W1T']\n", |
||||||
|
"LIGHTNING_CODES = ['L', 'V']\n", |
||||||
|
"MACHE_CODES = ['K1R', 'K1S', 'K2S', 'K3R', 'K3S', 'K4S']\n", |
||||||
|
"\n", |
||||||
|
"FORD_VIN_START = ['1FT', '3FM', '5LM']\n", |
||||||
|
"\n", |
||||||
|
"def ford_vin_fingerprint(vin): # Check if it's a Ford vehicle and determine the model\n", |
||||||
|
" vin_positions_567 = vin[4:7]\n", |
||||||
|
"\n", |
||||||
|
" if vin.startswith('1FT'):\n", |
||||||
|
" if vin_positions_567 in F150_CODES:\n", |
||||||
|
" if vin[7] in LIGHTNING_CODES:\n", |
||||||
|
" return f\"FORD F-150 LIGHTNING 1ST GEN\"\n", |
||||||
|
" else:\n", |
||||||
|
" return f\"FORD F-150 14TH GEN\"\n", |
||||||
|
" elif vin.startswith('3FM'):\n", |
||||||
|
" if vin_positions_567 in MACHE_CODES:\n", |
||||||
|
" return f\"FORD MUSTANG MACH-E 1ST GEN\"\n", |
||||||
|
" elif vin.startswith('5LM'):\n", |
||||||
|
" pass\n", |
||||||
|
"\n", |
||||||
|
" return \"mock\"" |
||||||
|
] |
||||||
|
}, |
||||||
|
{ |
||||||
|
"cell_type": "code", |
||||||
|
"execution_count": 4, |
||||||
|
"metadata": {}, |
||||||
|
"outputs": [ |
||||||
|
{ |
||||||
|
"name": "stdout", |
||||||
|
"output_type": "stream", |
||||||
|
"text": [ |
||||||
|
"Skipping platform: FORD F-150 14TH GEN, no data available\n" |
||||||
|
] |
||||||
|
} |
||||||
|
], |
||||||
|
"source": [ |
||||||
|
"from openpilot.tools.lib.logreader import comma_car_segments_source, get_first_message\n", |
||||||
|
"\n", |
||||||
|
"\n", |
||||||
|
"VINS_TO_CHECK = set()\n", |
||||||
|
"\n", |
||||||
|
"for platform in platforms:\n", |
||||||
|
" if platform not in database:\n", |
||||||
|
" print(f\"Skipping platform: {platform}, no data available\")\n", |
||||||
|
" continue\n", |
||||||
|
"\n", |
||||||
|
" for segment in database[platform]:\n", |
||||||
|
" lr = LogReader(segment, default_source=comma_car_segments_source)\n", |
||||||
|
" CP = get_first_message(lr, \"carParams\").carParams\n", |
||||||
|
" VINS_TO_CHECK.add((CP.carVin, CP.carFingerprint))" |
||||||
|
] |
||||||
|
}, |
||||||
|
{ |
||||||
|
"cell_type": "code", |
||||||
|
"execution_count": 6, |
||||||
|
"metadata": {}, |
||||||
|
"outputs": [ |
||||||
|
{ |
||||||
|
"name": "stdout", |
||||||
|
"output_type": "stream", |
||||||
|
"text": [ |
||||||
|
"vin: 1FM5K8GC7LGXXXXXX real platform: FORD EXPLORER 6TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 00000000000XXXXXX real platform: FORD ESCAPE 4TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 3FTTW8F98NRXXXXXX real platform: FORD MAVERICK 1ST GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 1FTVW1EL4NWXXXXXX real platform: FORD F-150 LIGHTNING 1ST GEN determined platform: FORD F-150 LIGHTNING 1ST GEN correct: True\n", |
||||||
|
"vin: 1FM5K7LC0MGXXXXXX real platform: FORD EXPLORER 6TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: WF0NXXGCHNJXXXXXX real platform: FORD FOCUS 4TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 1FMCU9J94MUXXXXXX real platform: FORD ESCAPE 4TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 5LM5J7XC9LGXXXXXX real platform: FORD EXPLORER 6TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 3FMCR9B69NRXXXXXX real platform: FORD BRONCO SPORT 1ST GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 3FMTK3SU0MMXXXXXX real platform: FORD MUSTANG MACH-E 1ST GEN determined platform: FORD MUSTANG MACH-E 1ST GEN correct: True\n", |
||||||
|
"vin: 1FM5K8HC7MGXXXXXX real platform: FORD EXPLORER 6TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 1FM5K8GC7NGXXXXXX real platform: FORD EXPLORER 6TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 5LM5J7XC8MGXXXXXX real platform: FORD EXPLORER 6TH GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 3FTTW8E31PRXXXXXX real platform: FORD MAVERICK 1ST GEN determined platform: mock correct: False\n", |
||||||
|
"vin: 3FTTW8E99NRXXXXXX real platform: FORD MAVERICK 1ST GEN determined platform: mock correct: False\n" |
||||||
|
] |
||||||
|
} |
||||||
|
], |
||||||
|
"source": [ |
||||||
|
"for vin, real_fingerprint in VINS_TO_CHECK:\n", |
||||||
|
" determined_fingerprint = ford_vin_fingerprint(vin)\n", |
||||||
|
" print(f\"vin: {vin} real platform: {real_fingerprint: <30} determined platform: {determined_fingerprint: <30} correct: {real_fingerprint == determined_fingerprint}\")" |
||||||
|
] |
||||||
|
} |
||||||
|
], |
||||||
|
"metadata": { |
||||||
|
"kernelspec": { |
||||||
|
"display_name": ".venv", |
||||||
|
"language": "python", |
||||||
|
"name": "python3" |
||||||
|
}, |
||||||
|
"language_info": { |
||||||
|
"codemirror_mode": { |
||||||
|
"name": "ipython", |
||||||
|
"version": 3 |
||||||
|
}, |
||||||
|
"file_extension": ".py", |
||||||
|
"mimetype": "text/x-python", |
||||||
|
"name": "python", |
||||||
|
"nbconvert_exporter": "python", |
||||||
|
"pygments_lexer": "ipython3", |
||||||
|
"version": "3.11.4" |
||||||
|
} |
||||||
|
}, |
||||||
|
"nbformat": 4, |
||||||
|
"nbformat_minor": 2 |
||||||
|
} |
||||||
Loading…
Reference in new issue