from scripts.geojson_creator import process_geojson
from pathlib import Path
from scripts.ep_run_enrich import energy_plus_workflow
from scripts.CityBEM_run import CityBEM_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.exports.formats.simplified_radiosity_algorithm import SimplifiedRadiosityAlgorithm
from hub.imports.results_factory import ResultFactory
import hub.helpers.constants as cte
from hub.exports.exports_factory import ExportsFactory
from hub.exports.exports_factory import ExportsFactory
import subprocess

# Specify the GeoJSON file path
# geojson_file = process_geojson(a=-73.5681295982132, b=45.49218262677643, c=-73.5681295982132, d=45.51218262677643,
#                                e=-73.5881295982132, f=45.49218262677643,g=-73.5881295982132, h=45.51218262677643)
geojson_file = process_geojson(x=-73.5681295982132, y=45.49218262677643, diff=0.0001)

file_path = (Path(__file__).parent / 'input_files' / 'output_buildings.geojson')
# Specify the output path for the PDF file
output_path = (Path(__file__).parent / 'out_files').resolve()
# Create city object from GeoJSON file
tmp_folder = (Path(__file__).parent / 'tmp').resolve()
# Create city object from GeoJSON file
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
# Enrich city data
ConstructionFactory('nrcan', city).enrich()
UsageFactory('nrcan', city).enrich()
WeatherFactory('epw', city).enrich()

ExportsFactory('stl', city, output_path/'CityBEM_input_output').export()

ExportsFactory('sra', city, tmp_folder).export()
sra_path = (tmp_folder / f'{city.name}_sra.xml').resolve()
subprocess.run(['sra', str(sra_path)])
ResultFactory('sra', city, tmp_folder).enrich()
CityBEM_workflow(city) #run the city using a fast City-Building Energy Model, CityBEM
#energy_plus_workflow(city)
print ("test done")