def test_manual_geometry(data, expdata):
planview = pyodrx.PlanView()
planview.add_fixed_geometry(pyodrx.Line(data[0]), data[1], data[2],
data[3])
x, y, h = planview.get_end_point()
assert pytest.approx(x, 0.000001) == expdata[0]
assert pytest.approx(y, 0.000001) == expdata[1]
assert pytest.approx(h, 0.000001) == expdata[2]
x, y, h = planview.get_start_point()
assert pytest.approx(x, 0.000001) == data[1]
assert pytest.approx(y, 0.000001) == data[2]
assert pytest.approx(h, 0.000001) == data[3]
def test_simple_road():
line1 = pyodrx.Line(100)
planview = pyodrx.PlanView()
planview.add_geometry(line1)
rm = pyodrx.RoadMark(pyodrx.RoadMarkType.solid,
0.2,
rule=pyodrx.MarkRule.no_passing)
lane1 = pyodrx.Lane(a=2)
lane1.add_roadmark(rm)
lanesec = pyodrx.LaneSection(0, lane1)
lanes = pyodrx.Lanes()
lanes.add_lanesection(lanesec)
road = pyodrx.Road(1, planview, lanes)
prettyprint(road.get_element())
def test_link_road():
line1 = pyodrx.Line(100)
planview = pyodrx.PlanView()
planview.add_geometry(line1)
rm = pyodrx.RoadMark(pyodrx.RoadMarkType.solid,
0.2,
rule=pyodrx.MarkRule.no_passing)
lane1 = pyodrx.Lane(a=2)
lane1.add_roadmark(rm)
lanesec = pyodrx.LaneSection(0, lane1)
lanes = pyodrx.Lanes()
lanes.add_lanesection(lanesec)
road = pyodrx.Road(1, planview, lanes)
road.add_predecessor(pyodrx.ElementType.road, '1',
pyodrx.ContactPoint.start)
prettyprint(road.get_element())
from scenariogeneration import xodr, prettyprint
import numpy as np
import os
odr = xodr.OpenDrive('myroad')
#---------------- Road 1
planview = xodr.PlanView(0,0,0)
# create some geometries and add to the planview
planview.add_geometry(xodr.Line(100))
# create a solid roadmark
rm = xodr.RoadMark(xodr.RoadMarkType.solid, 0.2)
# create centerlane
centerlane_1 = xodr.Lane(a=2)
centerlane_1.add_roadmark(rm)
lanesec_1 = xodr.LaneSection(0,centerlane_1)
# add a driving lane
lane2_1 = xodr.Lane(a=3.1)
lane2_1.add_roadmark(rm)
lanesec_1.add_left_lane(lane2_1)
lane3_1 = xodr.Lane(a=3.1)
lane3_1.add_roadmark(rm)
lanesec_1.add_right_lane(lane3_1)
## finalize the road
def test_create_lane_links_normalroad1():
planview = []
lanec = []
lanel = []
laner = []
lanesec = []
lanes = []
rm = pyodrx.RoadMark(pyodrx.RoadMarkType.solid,
0.2,
rule=pyodrx.MarkRule.no_passing)
geom = []
geom.append(pyodrx.Line(50))
geom.append(pyodrx.Arc(0.01, angle=np.pi / 2))
geom.append(pyodrx.Line(50))
# create planviews
for i in range(len(geom)):
planview.append(pyodrx.PlanView())
planview[i].add_geometry(geom[i])
# create centerlanes
for i in range(len(geom)):
lanec.append(pyodrx.Lane(a=3))
lanel.append(pyodrx.Lane(a=3))
laner.append(pyodrx.Lane(a=3))
#add roadmarks
for i in range(len(geom)):
lanec[i].add_roadmark(rm)
lanel[i].add_roadmark(rm)
laner[i].add_roadmark(rm)
# create lanesections
for i in range(len(geom)):
lanesec.append(pyodrx.LaneSection(0, lanec[i]))
lanesec[i].add_right_lane(lanel[i])
lanesec[i].add_left_lane(laner[i])
#create lanes
for i in range(len(geom)):
lanes.append(pyodrx.Lanes())
lanes[i].add_lanesection(lanesec[i])
#create roads
road1 = pyodrx.Road(1, planview[0], lanes[0])
road1.add_successor(pyodrx.ElementType.road, 2, pyodrx.ContactPoint.start)
road2 = pyodrx.Road(2, planview[1], lanes[1])
road2.add_predecessor(pyodrx.ElementType.road, 1, pyodrx.ContactPoint.end)
road2.add_successor(pyodrx.ElementType.road, 3, pyodrx.ContactPoint.start)
road3 = pyodrx.Road(3, planview[2], lanes[2])
road3.add_predecessor(pyodrx.ElementType.road, 2, pyodrx.ContactPoint.end)
# create the opendrive and add roads
odr = pyodrx.OpenDrive('myroad')
odr.add_road(road1)
odr.add_road(road2)
odr.add_road(road3)
odr.adjust_roads_and_lanes()
assert road1.lanes.lanesections[0].rightlanes[0].links.get_predecessor_id(
) == None
assert int(road1.lanes.lanesections[0].rightlanes[0].links.
get_successor_id()) == -1
assert road1.lanes.lanesections[0].leftlanes[0].links.get_predecessor_id(
) == None
assert int(
road1.lanes.lanesections[0].leftlanes[0].links.get_successor_id()) == 1
assert int(road2.lanes.lanesections[0].rightlanes[0].links.
get_predecessor_id()) == -1
assert int(road2.lanes.lanesections[0].rightlanes[0].links.
get_successor_id()) == -1
assert int(road2.lanes.lanesections[0].leftlanes[0].links.
get_predecessor_id()) == 1
assert int(
road2.lanes.lanesections[0].leftlanes[0].links.get_successor_id()) == 1
assert int(road3.lanes.lanesections[0].rightlanes[0].links.
get_predecessor_id()) == -1
assert road3.lanes.lanesections[0].rightlanes[0].links.get_successor_id(
) == None
assert int(road3.lanes.lanesections[0].leftlanes[0].links.
get_predecessor_id()) == 1
assert road3.lanes.lanesections[0].leftlanes[0].links.get_successor_id(
) == None
from scenariogeneration import xodr, prettyprint import numpy as np import os ## EXAMPLE 1 ## Multiple geometries in one only road. ##1. Create the planview planview = xodr.PlanView() ##2. Create some geometries and add them to the planview line1 = xodr.Line(100) arc1 = xodr.Arc(0.05, angle=np.pi / 2) line2 = xodr.Line(100) cloth1 = xodr.Spiral(0.05, -0.1, 30) line3 = xodr.Line(100) planview.add_geometry(line1) planview.add_geometry(arc1) planview.add_geometry(line2) planview.add_geometry(cloth1) planview.add_geometry(line3) ##3. Create a solid roadmark rm = xodr.RoadMark(xodr.RoadMarkType.solid, 0.2) ##4. Create centerlane centerlane = xodr.Lane(a=2) centerlane.add_roadmark(rm) ##5. Create lane section form the centerlane
# create geometries
geoms = []
geoms.append(xodr.Line(100))
geoms.append(xodr.Spiral(0.001, 0.019, 30))
geoms.append(xodr.Line(100))
geoms.append(xodr.Spiral(-0.001, -0.1, 30))
geoms.append(xodr.Line(100))
geoms.append(xodr.Line(20))
geoms.append(xodr.Line(100))
numberofroads = len(geoms)
# create planviews
planviews = []
for g in geoms:
pv = xodr.PlanView()
pv.add_geometry(g)
planviews.append(pv)
# create centerlanes
lanecenters = []
for i in range(numberofroads):
lc = xodr.Lane(a=3)
lc.add_roadmark(rm)
lanecenters.append(lc)
# create lanes
rightlanes = []
leftlanes = []
for i in range(numberofroads):
right = xodr.Lane(a=3)