matsim-proto/main.py
2024-02-06 05:11:26 +01:00

35 lines
1.3 KiB
Python

from pathlib import Path
from hub.imports.geometry_factory import GeometryFactory
from hub.imports.construction_factory import ConstructionFactory
from hub.imports.usage_factory import UsageFactory
from hub.helpers.dictionaries import Dictionaries
from matsim_engine import MatSimEngine
from matsim_visualizer import MatSimVisualizer
try:
file_path = (Path(__file__).parent / 'input_files' / 'summerschool_all_buildings.geojson')
construction_format = 'nrcan'
usage_format = 'nrcan'
energy_systems_format = 'montreal_custom'
out_path = (Path(__file__).parent / 'output_files')
city = GeometryFactory('geojson',
path=file_path,
height_field='citygml_me',
year_of_construction_field='ANNEE_CONS',
function_field='CODE_UTILI',
function_to_hub=Dictionaries().montreal_function_to_hub_function).city
ConstructionFactory(construction_format, city).enrich()
UsageFactory(usage_format, city).enrich()
MatSimEngine(city, 'output_files')
# visualizer = MatSimVisualizer('output_files/network.xml.gz', 'output_files/output_events.xml.gz')
# visualizer.visualize()
except Exception as ex:
print('error: ', ex)
print('[simulation abort]')