pim/road-gen
1
0
Fork 0
mirror of https://github.com/pim-n/road-gen.git synced 2026-02-03 09:23:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add CLI options for a segmented road

Browse Source
This commit is contained in:
Pim Nelissen
2026-01-30 19:45:03 +01:00
parent 8fbd25519c
commit 21be94c94f
1 changed files with 42 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/road_gen/main.py
View File

@ -4,9 +4,8 @@ from matplotlib import pyplot as plt
from .generators.random_road_generator import RandomRoadGenerator from .generators.random_road_generator import RandomRoadGenerator
from .generators.segmented_road_generator import SegmentedRoadGenerator from .generators.segmented_road_generator import SegmentedRoadGenerator
from .plotting.plot_road import plot_road from .plotting.plot_road import plot_road
from .prefabs import prefabs
from .utils import export from .utils import export
def add_common_args(parser): def add_common_args(parser):
@ -30,9 +29,13 @@ def main():
random_parser.add_argument("--straight_section_max_rel_size", type=float, required=False, help="The maximum size that straight section(s) can have relative to the total length of the path. Defaults to 0.1.") random_parser.add_argument("--straight_section_max_rel_size", type=float, required=False, help="The maximum size that straight section(s) can have relative to the total length of the path. Defaults to 0.1.")
add_common_args(random_parser) add_common_args(random_parser)
segment_parser = subparsers.add_parser("segments", help="Generate a road according to a list of segments.")
segment_parser.add_argument("--segments", nargs="+", type=str, required=True, help=f"List of segments. Choose from {str(prefabs.PREFABS.keys())}")
segment_parser.add_argument("--alpha", type=float, required=False, help="Dirichlet distribution concentration parameter. A high alpha distributes total length more evenly across segments.")
add_common_args(segment_parser)
args = parser.parse_args() args = parser.parse_args()
if args.method == "random":
try: try:
if not all(v > 0 for v in (args.length, args.ds, args.velocity)): if not all(v > 0 for v in (args.length, args.ds, args.velocity)):
raise ValueError("Length, step size, and velocity must be positive values.") raise ValueError("Length, step size, and velocity must be positive values.")
@ -50,15 +53,23 @@ def main():
if args.seed: if args.seed:
init_args["seed"] = args.seed init_args["seed"] = args.seed
generator = RandomRoadGenerator(**init_args)
generate_args = {} generate_args = {}
if args.method == "random":
generator = RandomRoadGenerator(**init_args)
if args.straight_section_prob: if args.straight_section_prob:
generate_args["straight_section_prob"] = float(args.straight_section_prob) generate_args["straight_section_prob"] = float(args.straight_section_prob)
if args.straight_section_max_rel_size: if args.straight_section_max_rel_size:
generate_args["straight_section_max_rel_size"] = float(args.straight_section_max_rel_size) generate_args["straight_section_max_rel_size"] = float(args.straight_section_max_rel_size)
elif args.method == "segments":
generator = SegmentedRoadGenerator(**init_args)
if args.alpha:
generate_args["alpha"] = args.alpha
generate_args["segments"] = list(args.segments)
x, y = generator.generate(**generate_args) x, y = generator.generate(**generate_args)
if args.save: if args.save: