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

71
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,48 +29,60 @@ 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.")
init_args = { init_args = {
"length": args.length, "length": args.length,
"ds": args.ds, "ds": args.ds,
"velocity": args.velocity, "velocity": args.velocity,
} }
if args.mu: if args.mu:
init_args["mu"] = args.mu init_args["mu"] = args.mu
if args.g: if args.g:
init_args["g"] = args.g init_args["g"] = args.g
if args.seed: if args.seed:
init_args["seed"] = args.seed init_args["seed"] = args.seed
generate_args = {}
if args.method == "random":
generator = RandomRoadGenerator(**init_args) generator = RandomRoadGenerator(**init_args)
generate_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)
x, y = generator.generate(**generate_args) elif args.method == "segments":
generator = SegmentedRoadGenerator(**init_args)
if args.save: if args.alpha:
basename = str(generator.seed) generate_args["alpha"] = args.alpha
plot_road(x, y, generator, save = True, filename = basename+".jpg")
export.coords_to_json(x, y, filename = basename+".json")
export.params_to_json(generator, filename = basename+".params.json")
else:
plot_road(x, y, generator)
except ValueError as e: generate_args["segments"] = list(args.segments)
print(f"Error: {e}")
exit(1) x, y = generator.generate(**generate_args)
if args.save:
basename = str(generator.seed)
plot_road(x, y, generator, save = True, filename = basename+".jpg")
export.coords_to_json(x, y, filename = basename+".json")
export.params_to_json(generator, filename = basename+".params.json")
else:
plot_road(x, y, generator)
except ValueError as e:
print(f"Error: {e}")
exit(1)
if __name__ == "__main__": if __name__ == "__main__":
main() main()