import csv

from scripts.geojson_creator import process_geojson
from pathlib import Path
import subprocess
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.results_factory import ResultFactory
from hub.exports.exports_factory import ExportsFactory
import hub.helpers.constants as cte
# Specify the GeoJSON file path
geojson_file = process_geojson(x=-73.5681295982132, y=45.49218262677643, diff=0.0001)
file_path = (Path(__file__).parent.parent / 'input_files' / f'{geojson_file}')
# Specify the output path for the PDF file
output_path = (Path(__file__).parent / 'out_files').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('sra', city, output_path).export()
sra_path = (output_path / f'{city.name}_sra.xml').resolve()
subprocess.run(['sra', str(sra_path)])
ResultFactory('sra', city, output_path).enrich()
for building in city.buildings:
  beam = [x / 3600 for x in building.beam[cte.HOUR]]
  diffuse = [x / 3600 for x in building.diffuse[cte.HOUR]]
  roof = building.roofs[0].global_irradiance[cte.HOUR]
  data = list(zip(beam, diffuse, roof))
  file_name = f'solar_radiation_{building.name}.csv'
  with open(output_path / file_name, 'w', newline='') as csvfile:
    output_file = csv.writer(csvfile)
    # Write header
    output_file.writerow(['beam_component', 'diffuse_component', 'roof_global_irradiance'])
    # Write data
    output_file.writerows(data)
print('test')