matsim-proto/main.py
2024-03-05 15:14:36 -05:00

41 lines
1.4 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 import Matsim
from matsim_visualizer import MatsimVisualizer
try:
file_path = (Path(__file__).parent / 'input_files' / 'mtlbld_v4.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='ETAGE_HORS',
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()
Matsim(city, 'output_files').export()
MatSimEngine('output_files/Montreal_config.xml').run()
visualizer = MatsimVisualizer(
'output_files/Montreal',
'output_files',
(3854635.608067, 6040255.047194, 3855550.392442, 6039416.667303)
)
visualizer.visualize()
except Exception as ex:
print('error: ', ex)
print('[simulation abort]')