mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-03-03 11:53:58 +08:00
fe983a7b8c
* rebased * cleaner, seems to drive better? * more stable * wrong import * new way of thinking * reports look nice * start move back * works at leas * good timestamps * step by step * somewhat work * tests pass * ALL CARS STOPPED * should work * fake a cruise obstacle * cleaner costs * pretty good except cruise braking * works pretty well now! * cleanup * add source * add source * that is needed for unit tests * nan recovery * little cleaner * stop wasting arrays * unreasonable without unfair init * this isnt needed without the exponential * that works too * unused * uses less * new ref * long enough * e2e long api * DONT PUT IN A VIEW INTO ACADOS * new ref for outside weights * remove debug prints
145 lines
3.6 KiB
Python
Executable File
145 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import unittest
|
|
|
|
from common.params import Params
|
|
from selfdrive.test.longitudinal_maneuvers.maneuver import Maneuver
|
|
|
|
|
|
# TODO: make new FCW tests
|
|
maneuvers = [
|
|
Maneuver(
|
|
'approach stopped car at 20m/s',
|
|
duration=20.,
|
|
initial_speed=25.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=120.,
|
|
speed_lead_values=[30., 0.],
|
|
breakpoints=[0., 1.],
|
|
),
|
|
Maneuver(
|
|
'approach stopped car at 20m/s',
|
|
duration=20.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=90.,
|
|
speed_lead_values=[20., 0.],
|
|
breakpoints=[0., 1.],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 1m/s^2',
|
|
duration=50.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
breakpoints=[0., 15., 35.0],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 2m/s^2',
|
|
duration=50.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
breakpoints=[0., 15., 25.0],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 3m/s^2',
|
|
duration=50.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
breakpoints=[0., 15., 21.66],
|
|
),
|
|
Maneuver(
|
|
'steady state following a car at 20m/s, then lead decel to 0mph at 4m/s^2',
|
|
duration=40.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=35.,
|
|
speed_lead_values=[20., 20., 0.],
|
|
prob_lead_values=[0., 1., 1.],
|
|
cruise_values=[20., 20., 20.],
|
|
breakpoints=[2., 2.01, 7.01],
|
|
),
|
|
Maneuver(
|
|
"approach stopped car at 20m/s",
|
|
duration=30.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=120.,
|
|
speed_lead_values=[0.0, 0., 0.],
|
|
prob_lead_values=[0.0, 0., 1.],
|
|
cruise_values=[20., 20., 20.],
|
|
breakpoints=[0.0, 2., 2.01],
|
|
),
|
|
Maneuver(
|
|
"approach slower cut-in car at 20m/s",
|
|
duration=20.,
|
|
initial_speed=20.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=50.,
|
|
speed_lead_values=[15., 15.],
|
|
breakpoints=[1., 11.],
|
|
only_lead2=True,
|
|
),
|
|
Maneuver(
|
|
"stay stopped behind radar override lead",
|
|
duration=20.,
|
|
initial_speed=0.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=10.,
|
|
speed_lead_values=[0., 0.],
|
|
prob_lead_values=[0., 0.],
|
|
breakpoints=[1., 11.],
|
|
only_radar=True,
|
|
),
|
|
Maneuver(
|
|
"NaN recovery",
|
|
duration=30.,
|
|
initial_speed=15.,
|
|
lead_relevancy=True,
|
|
initial_distance_lead=60.,
|
|
speed_lead_values=[0., 0., 0.0],
|
|
breakpoints=[1., 1.01, 11.],
|
|
cruise_values=[float("nan"), 15., 15.],
|
|
),
|
|
]
|
|
|
|
|
|
class LongitudinalControl(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
os.environ['SIMULATION'] = "1"
|
|
os.environ['SKIP_FW_QUERY'] = "1"
|
|
os.environ['NO_CAN_TIMEOUT'] = "1"
|
|
|
|
params = Params()
|
|
params.clear_all()
|
|
params.put_bool("Passive", bool(os.getenv("PASSIVE")))
|
|
params.put_bool("OpenpilotEnabledToggle", True)
|
|
params.put_bool("CommunityFeaturesToggle", True)
|
|
|
|
# hack
|
|
def test_longitudinal_setup(self):
|
|
pass
|
|
|
|
|
|
def run_maneuver_worker(k):
|
|
def run(self):
|
|
man = maneuvers[k]
|
|
print(man.title)
|
|
valid, _ = man.evaluate()
|
|
self.assertTrue(valid, msg=man.title)
|
|
return run
|
|
|
|
|
|
for k in range(len(maneuvers)):
|
|
setattr(LongitudinalControl, f"test_longitudinal_maneuvers_{k+1}",
|
|
run_maneuver_worker(k))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(failfast=True)
|