sumo-emissions/configurator/configurator.py

# -*- coding: utf-8 -*-
import os
import shutil
import subprocess
import tempfile
from xml.etree import ElementTree

import randomTrips

# Absolute path of the directory the script is in
SCRIPTDIR = os.path.dirname(__file__)
STATICDIR = os.path.join(SCRIPTDIR, 'static')
SIMNAME = 'simul'
NETCONVERTCMD = ['netconvert', '-c', os.path.join(STATICDIR, f'{SIMNAME}.netcfg')]
POLYCONVERTCMD = ['polyconvert', '-c', os.path.join(STATICDIR, f'{SIMNAME}.polycfg')]


def clean():
    # [os.remove(os.path.join('simul', f)) for f in os.listdir('simul')]
    [os.remove(os.path.join('static', f))
     for f in os.listdir('static') if f.endswith('.log')]


def load_netconvert_template(osm_input, out_name):
    tree = ElementTree.parse(os.path.join(STATICDIR, 'simul.netcfg'))
    root = tree.getroot()
    root.find('input/osm-files').set('value', osm_input)
    root.find('output/output-file').set('value', f'{out_name}.net.xml')
    root.find('report/log').set('value', f'{out_name}.netconvert.log')
    return tree


def generate_network(osm_file, out_path, net_name):
    template = load_netconvert_template(osm_file, net_name)
    with tempfile.TemporaryDirectory() as tmpdirname:
        netconfig = os.path.join(tmpdirname, f'{net_name}.netcfg')
        template.write(netconfig)
        # copy typemaps to tempdir
        shutil.copytree(os.path.join(STATICDIR, 'typemap'), os.path.join(tmpdirname, 'typemap'))
        print('Creating the network…')
        netconvertcmd = ['netconvert', '-c', netconfig]
        subprocess.run(netconvertcmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        print('Copying files')
        shutil.copy(os.path.join(tmpdirname, f'{net_name}.net.xml'), os.path.join(out_path, f'{net_name}.net.xml'))
        print('created')


def generate_scenario(out_path, name):
    print('Creating the network…')
    subprocess.run(NETCONVERTCMD, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    print('Extracting polygons…')
    subprocess.run(POLYCONVERTCMD, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    print('Moving files')
    shutil.move(os.path.join(STATICDIR, 'simul.net.xml'), os.path.join(out_path, f'{name}.net.xml'))
    shutil.move(os.path.join(STATICDIR, 'simul.poly.xml'), os.path.join(out_path, f'{name}.poly.xml'))
    shutil.copyfile(os.path.join(STATICDIR, 'simul.sumocfg'), os.path.join(out_path, f'{name}.sumocfg'))
    # Move log files
    logdir = os.path.join(out_path, 'log')
    if not os.path.exists(logdir):
        os.mkdir(logdir)
    for f in os.listdir(STATICDIR):
        if f.endswith('.log'):
            shutil.move(os.path.join(STATICDIR, f), os.path.join(logdir, f))


def generate_mobility(path, name):
    routefile = os.path.join(path, f'{name}.rou.xml')
    netfile = os.path.join(path, f'{name}.net.xml')
    output = os.path.join(path, f'{name}.trips.xml')
    end_time = 200
    veh_class = 'passenger'
    options = [
        f'--net-file={netfile}',
        f'--route-file={routefile}',
        f'--end={end_time}',
        f'--vehicle-class={veh_class}',
        f'--output-trip-file={output}',
        '--validate',
        '--length',
    ]
    print('Generating mobility…')
    randomTrips.main(randomTrips.get_options(options))


def generate_all(output_path, simulation_name):
    output_path = os.path.join(output_path, simulation_name)
    if not os.path.exists(output_path):
        os.mkdir(output_path)
    generate_scenario(output_path, simulation_name)
    generate_mobility(output_path, simulation_name)


if __name__ == '__main__':
    # clean()
    # generate_all('/tmp/', 'simul')
    output_file = '/tmp/simul'
    if not os.path.exists(output_file):
        os.mkdir(output_file)
    osm_file = os.path.join(STATICDIR, 'simul.raw.osm')
    generate_network(osm_file, output_file, 'test-sc')