thib8956/sumo-emissions
1
0
Fork 0
mirror of https://github.com/Ahp06/SUMO_Emissions.git synced 2024-11-24 20:46:29 +00:00
Code Issues Releases Wiki Activity

Fix running configuration, only SUMO_HOME required

Browse Source
This commit is contained in:
Thibaud Gasser 2019-02-01 17:58:55 +01:00
parent 112c90ff08
commit d5d6463c72
1 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
sumo_project/configurator.py
View File

@ -1,12 +1,20 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import os
import sys
if 'SUMO_HOME' in os.environ:
TOOLSDIR = os.path.join(os.environ['SUMO_HOME'], 'tools')
sys.path.append(TOOLSDIR)
else:
sys.exit("Please declare environment variable 'SUMO_HOME'")
import argparse import argparse
import datetime import datetime
import json import json
import logging import logging
import os
import shutil import shutil
import subprocess import subprocess
import sys
import tempfile import tempfile
import time import time
from sys import argv from sys import argv
@ -16,15 +24,10 @@ from xml.etree import ElementTree
import randomTrips import randomTrips
import sumolib import sumolib
if 'SUMO_HOME' in os.environ:
TOOLSDIR = os.path.join(os.environ['SUMO_HOME'], 'tools')
sys.path.append(TOOLSDIR)
else:
sys.exit("Please declare environment variable 'SUMO_HOME'")
# Absolute path of the directory the script is in # Absolute path of the directory the script is in
SCRIPTDIR = os.path.dirname(__file__) SCRIPTDIR = os.path.dirname(__file__)
TEMPLATEDIR = os.path.join(SCRIPTDIR, 'templates') TEMPLATEDIR = os.path.join(SCRIPTDIR, 'templates')
SUMOBIN = os.path.join(os.environ['SUMO_HOME'], 'bin')
# Init logger # Init logger
logfile = os.path.join(SCRIPTDIR, f'files/logs/configurator_{datetime.datetime.utcnow().isoformat()}.log') logfile = os.path.join(SCRIPTDIR, f'files/logs/configurator_{datetime.datetime.utcnow().isoformat()}.log')
@ -155,7 +158,7 @@ def generate_scenario(osm_file, out_path, scenario_name, generate_polygons=False
shutil.copytree(os.path.join(TEMPLATEDIR, 'typemap'), os.path.join(tmpdirname, 'typemap')) shutil.copytree(os.path.join(TEMPLATEDIR, 'typemap'), os.path.join(tmpdirname, 'typemap'))
# Call NETCONVERT # Call NETCONVERT
logging.info("Generating network…") logging.info("Generating network…")
netconvertcmd = ['netconvert', '-c', netconfig] netconvertcmd = [os.path.join(SUMOBIN, 'netconvert'), '-c', netconfig]
logging.debug(f'Calling {" ".join(netconvertcmd)}') logging.debug(f'Calling {" ".join(netconvertcmd)}')
subprocess.run(netconvertcmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) subprocess.run(netconvertcmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
# Optionaly generate polygons # Optionaly generate polygons
@ -172,7 +175,7 @@ def generate_polygons_(osm_file, scenario_name, dest):
poly_template.write(polyconfig) poly_template.write(polyconfig)
# Call POLYCONVERT # Call POLYCONVERT
logging.info('Generating polygons…') logging.info('Generating polygons…')
polyconvert_cmd = ['polyconvert', '-c', polyconfig] polyconvert_cmd = [os.path.join(SUMOBIN, 'polyconvert'), '-c', polyconfig]
logging.debug(f'Calling {" ".join(polyconvert_cmd)}') logging.debug(f'Calling {" ".join(polyconvert_cmd)}')
subprocess.run(polyconvert_cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) subprocess.run(polyconvert_cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
@ -229,7 +232,7 @@ def dict_to_list(d):
return [item for k in d for item in (k, d[k])] return [item for k in d for item in (k, d[k])]
def parse_command_line(): def parse_command_line(args=None):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('osmfile', help='Path to the .osm file to convert to a SUMO simulation') parser.add_argument('osmfile', help='Path to the .osm file to convert to a SUMO simulation')
parser.add_argument('--path', help='Where to generate the files') parser.add_argument('--path', help='Where to generate the files')
@ -242,8 +245,7 @@ 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.')
parser.add_argument('--seed', help='Initializes the random number generator.') parser.add_argument('--seed', help='Initializes the random number generator.')
options = parser.parse_args() return parser.parse_args(args=args)
handle_args(options)
def handle_args(options): def handle_args(options):
@ -262,20 +264,21 @@ def parse_json(json_file):
logging.info(f'Loading config from {json_file}') logging.info(f'Loading config from {json_file}')
config = SimpleNamespace(**json.load(json_file)) config = SimpleNamespace(**json.load(json_file))
logging.debug(f'Config {config}') logging.debug(f'Config {config}')
handle_args(config) return config
if __name__ == '__main__': if __name__ == '__main__':
if len(argv) > 2:
# Try to load the config file # Try to load the config file
if argv[1] == '-c' or '--config' or '-config': if len(argv) > 2 and argv[1] == '-c' or argv[1] == '--config' or argv[1] == '-config':
try: try:
with open(argv[2]) as jsonfile: with open(argv[2]) as jsonfile:
parse_json(jsonfile) config = parse_json(jsonfile)
handle_args(config)
except FileNotFoundError: except FileNotFoundError:
msg = f'The config file {argv[2]} does not exist!' msg = f'The config file {argv[2]} does not exist!'
logging.fatal(msg) logging.fatal(msg)
raise FileNotFoundError(msg) raise FileNotFoundError(msg)
else: else:
# Run with command line arguments # Run with command line arguments
parse_command_line() config = parse_command_line()
handle_args(config)