mirror of
https://github.com/Ahp06/SUMO_Emissions.git
synced 2024-11-22 03:26:30 +00:00
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
|
|
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 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')
|