from scripts.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.retrofit_factory_virtual import RetrofitFactory
import json
# Specify the GeoJSON file path
input_files_path = (Path(__file__).parent / 'input_files')
input_files_path.mkdir(parents=True, exist_ok=True)
geojson_file = process_geojson(x=-73.5681295982132, y=45.49218262677643, diff=0.0001)
geojson_file_path = input_files_path / 'input_one_buildings.geojson'

retrofit_json_file_path = input_files_path / 'retrofit_scenarios.json'
with open(retrofit_json_file_path, 'r') as f:
    retrofit_data = json.load(f)
output_path = (Path(__file__).parent / 'out_files').resolve()
output_path.mkdir(parents=True, exist_ok=True)
# Create city object from GeoJSON file
city = GeometryFactory('geojson',
                       path=geojson_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
# Enrich city data
ConstructionFactory('cerc', city).enrich()
UsageFactory('cerc', city).enrich()
# Apply retrofits based on JSON data using building IDs
RetrofitFactory(retrofit_data, city).enrich()

WeatherFactory('epw', city).enrich()
energy_plus_workflow(city)