from geojson_creator import process_geojson
from pathlib import Path
from scripts.ep_workflow import energy_plus_workflow
from hub.imports.geometry_factory import GeometryFactory
from hub.helpers.dictionaries import Dictionaries
from hub.imports.construction_factory import ConstructionFactory
from hub.imports.usage_factory import UsageFactory
from hub.imports.weather_factory import WeatherFactory
from hub.imports.energy_systems_factory import EnergySystemsFactory
from scripts.random_assignation import call_random, residential_systems_percentage, residential_new_systems_percentage
import hub.helpers.constants as cte
from hub.imports.energy_systems.energy_system_sizing import SystemSizing
geojson_file = process_geojson(x=-73.567908, y=45.492783, diff=0.00045)
file_path = (Path(__file__).parent.parent / 'input_files' / f'{geojson_file}')
print('[simulation start]')
city = GeometryFactory('geojson',
                       path=file_path,
                       height_field='height',
                       year_of_construction_field='year_of_construction',
                       function_field='function',
                       function_to_hub=Dictionaries().montreal_function_to_hub_function).city
print(f'city created from {file_path}')
ConstructionFactory('nrcan', city).enrich()
print('enrich constructions... done')
UsageFactory('nrcan', city).enrich()
print('enrich usage... done')
WeatherFactory('epw', city).enrich()
print('enrich weather... done')
energy_plus_workflow(city)
city_buildings_peak_heating_demands = {}
for building in city.buildings:
  city_buildings_peak_heating_demands[f'{building.name}'] = building.heating_peak_load[cte.YEAR]
# call_random(city.buildings, residential_systems_percentage)
# EnergySystemsFactory('montreal_custom', city).enrich()
call_random(city.buildings, residential_new_systems_percentage)
EnergySystemsFactory('north_america', city).enrich()
SystemSizing(city.buildings).hvac_sizing()
print('test')