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

Add vclass=density cmdline arguments

Browse Source
This commit is contained in:
Thibaud Gasser 2018-12-14 19:18:53 +01:00
parent d92971cb61
commit 2c38eb051e
1 changed files with 32 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
configurator/configurator.py
View File

@ -6,8 +6,8 @@ import subprocess
import tempfile
from xml.etree import ElementTree
import sumolib
import randomTrips
import sumolib
# Absolute path of the directory the script is in
SCRIPTDIR = os.path.dirname(__file__)
@ -72,11 +72,21 @@ class RandomTripsGenerator:
if edge.allows(vclass):
length += edge.getLaneNumber() * edge.getLength()
print(f'density = {density}')
period = 3600 / (length / 1000) / density
print(f'Period computed for network : {period}, vclass={self.vclass}')
self.flags.extend(['-p', period])
class StoreDictKeyPair(argparse.Action):
def __call__(self, parser, namespace, values, option_string=None):
pairs = {}
for kv in values:
k, v = kv.split("=")
pairs[k] = v
setattr(namespace, self.dest, pairs)
def load_netconvert_template(osm_input, out_name):
netconfig = ElementTree.parse(os.path.join(TEMPLATEDIR, 'simul.netcfg'))
root = netconfig.getroot()
@ -130,19 +140,18 @@ def generate_scenario(osm_file, out_path, scenario_name):
shutil.copytree(tmpdirname, out_path, ignore=ignore_patterns)
def generate_mobility(out_path, name):
def generate_mobility(out_path, name, vclasses):
netfile = f'{name}.net.xml'
netpath = os.path.join(out_path, netfile)
output = os.path.join(out_path, f'{name}.trips.xml')
routefiles = []
end_time = 200
classes = ('passenger', 'bus')
for vclass in classes:
for vclass in vclasses:
# simname.bus.rou.xml, simname.passenger.rou.xml, ...
routefile = f'{name}.{vclass}.rou.xml'
routepath = os.path.join(out_path, routefile)
routefiles.append(routefile)
generator = RandomTripsGenerator(netpath, routepath, output, vclass, 10, '-l', **{'--end': end_time})
generator = RandomTripsGenerator(netpath, routepath, output, vclass, 10.0, '-l', **{'--end': end_time})
generator.generate()
return routefiles
@ -152,11 +161,11 @@ def generate_sumo_configuration(routefiles, path, scenario_name):
sumo_template.write(os.path.join(path, f'{scenario_name}.sumocfg'))
def generate_all(osm_file, path, simulation_name):
def generate_all(osm_file, path, simulation_name, vclasses):
simulation_dir = os.path.join(path, simulation_name)
logs_dir = os.path.join(simulation_dir, 'log')
generate_scenario(osm_file, simulation_dir, simulation_name)
routefiles = generate_mobility(simulation_dir, simulation_name)
routefiles = generate_mobility(simulation_dir, simulation_name, vclasses)
generate_sumo_configuration(routefiles, simulation_dir, simulation_name)
# Move all logs to logdir
move_logs(simulation_dir, logs_dir)
@ -174,17 +183,23 @@ def dict_to_list(d):
def main():
parser = argparse.ArgumentParser()
parser.add_argument('osmfile', type=str)
parser.add_argument('--path', type=str, help='Where to generate the files')
parser.add_argument('--name', type=str, required=True, help='Name of the scenario to generate')
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('--name', required=True, help='Name of the scenario to generate')
parser.add_argument('--vclass', dest='vclasses', action=StoreDictKeyPair,
nargs="+", metavar="VCLASS=DENSITY",
help='Generate this vclass with given density, in pair form vclass=density. The density is '
'given in vehicles per hour per kilometer.')
args = parser.parse_args()
generate_all(args.osmfile, args.path, args.name)
# If no vehicle classes are specified, use 'passenger' as a default
vclasses = args.vclasses or ('passenger',)
# FIXME Delete simul_dir if it already exists
simul_dir = os.path.join(args.path, args.name)
if os.path.isdir(simul_dir):
input(f'{simul_dir} already exists ! Press Enter to overwrite...')
shutil.rmtree(simul_dir)
generate_all(args.osmfile, args.path, args.name, vclasses)
if __name__ == '__main__':
if os.path.isdir('/tmp/scenario/foo'):
shutil.rmtree('/tmp/scenario/foo')
path = '/tmp/scenario/'
osm = '/tmp/scenario/map.osm'
generate_all(osm, path, 'foo')
# main()
main()