thib8956/sumo-emissions
1
0
Fork 0
mirror of https://github.com/Ahp06/SUMO_Emissions.git synced 2024-11-22 03:26:30 +00:00
Code Issues Releases Wiki Activity

Add JSON config file support

Browse Source
This commit is contained in:
Thibaud Gasser 2019-01-30 15:40:53 +01:00
parent 128f8adebd
commit 49672fdcb9
1 changed files with 29 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
configurator.py
View File

@ -1,9 +1,12 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import argparse import argparse
import json
import os import os
import shutil import shutil
import subprocess import subprocess
import tempfile import tempfile
from sys import argv
from types import SimpleNamespace
from xml.etree import ElementTree from xml.etree import ElementTree
import randomTrips import randomTrips
@ -176,11 +179,15 @@ def generate_sumo_configuration(routefiles, path, scenario_name, generate_polygo
def generate_all(args): def generate_all(args):
simulation_name = args.name simulation_name = args.name
simulation_dir = os.path.join(args.path, simulation_name) simulation_dir = os.path.join(args.path, simulation_name)
try:
generate_polygons = args.generate_polygons
except AttributeError:
generate_polygons = False
osm_file = args.osmfile osm_file = args.osmfile
logs_dir = os.path.join(simulation_dir, 'log') logs_dir = os.path.join(simulation_dir, 'log')
generate_scenario(osm_file, simulation_dir, simulation_name, args.generate_polygons) generate_scenario(osm_file, simulation_dir, simulation_name, generate_polygons)
routefiles = generate_mobility(simulation_dir, simulation_name, args.vclasses) routefiles = generate_mobility(simulation_dir, simulation_name, args.vclasses)
generate_sumo_configuration(routefiles, simulation_dir, simulation_name, args.generate_polygons) generate_sumo_configuration(routefiles, simulation_dir, simulation_name, )
# Move all logs to logdir # Move all logs to logdir
move_logs(simulation_dir, logs_dir) move_logs(simulation_dir, logs_dir)
@ -208,6 +215,10 @@ def parse_command_line():
'given in vehicles per hour per kilometer. For now, the following vehicle classes are ' 'given in vehicles per hour per kilometer. For now, the following vehicle classes are '
'available: passenger, truck, bus.') 'available: passenger, truck, bus.')
options = parser.parse_args() options = parser.parse_args()
handle_args(options)
def handle_args(options):
# If no vehicle classes are specified, use 'passenger' as a default with a density of 10 cars/km/h. # If no vehicle classes are specified, use 'passenger' as a default with a density of 10 cars/km/h.
options.vclasses = options.vclasses or {'passenger': 10} options.vclasses = options.vclasses or {'passenger': 10}
# Delete simul_dir if it already exists # Delete simul_dir if it already exists
@ -218,5 +229,20 @@ def parse_command_line():
generate_all(options) generate_all(options)
def parse_json(json_file):
config = SimpleNamespace(**json.load(json_file))
handle_args(config)
if __name__ == '__main__': if __name__ == '__main__':
if len(argv) > 2:
# Try to load the config file
if argv[1] == '-c' or '--config' or '-config':
try:
with open(argv[2]) as jsonfile:
parse_json(jsonfile)
except FileNotFoundError:
raise FileNotFoundError(f'The config file {argv[2]} does not exist!')
else:
# Run with command line arguments
parse_command_line() parse_command_line()