def cut_in_scenarios():
with temp_scenario(name="straight",
map="maps/6lane.net.xml") as scenario_root:
ego_missions = [
t.Mission(
route=t.Route(begin=("edge-west-WE", 0, 30),
end=("edge-west-EW", 0, "max")),
task=t.CutIn(),
),
]
gen_scenario(
t.Scenario(ego_missions=ego_missions),
output_dir=scenario_root,
)
yield Scenario.variations_for_all_scenario_roots([str(scenario_root)],
[AGENT_ID])
]
)
social_agent_missions = {
"all": (
[
t.SocialAgentActor(
name="open-agent",
agent_locator="open_agent:open_agent-v0",
initial_speed=20,
),
],
[
t.Mission(
t.Route(begin=("west", 1, 10), end=("east", 0, "max")),
task=t.CutIn(),
)
],
),
}
ego_missions = [
t.Mission(
t.Route(begin=("west", 1, 5), end=("east", 0, "max")),
task=t.CutIn(),
)
]
scenario = t.Scenario(
traffic={"all": traffic},
ego_missions=ego_missions,