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
import hub.helpers.constants as cte
from hub.imports.energy_systems_factory import EnergySystemsFactory
from hub.exports.exports_factory import ExportsFactory
import pandas as pd
# Specify the GeoJSON file path
input_files_path = (Path(__file__).parent / 'input_files')
geojson_file_path = input_files_path / 'Lachine_New_Developments.geojson'
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='maximum_roof_height',
                       year_of_construction_field='year_built',
                       function_field='building_type',
                       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('obj', city, output_path).export()
energy_plus_workflow(city)
for building in city.buildings:
  print(building.heating_demand[cte.YEAR][0] / 3.6e6)
  print(building.name)
  total_floor_area = 0
  for thermal_zone in building.thermal_zones_from_internal_zones:
    total_floor_area += thermal_zone.total_floor_area
  print(building.heating_demand[cte.YEAR][0] / (3.6e6 * total_floor_area))
  building.energy_systems_archetype_name = 'system 1 gas'
print('test')
EnergySystemsFactory('montreal_custom', city).enrich()
print('test')
for building in city.buildings:
  energy_systems = building.energy_systems
  for energy_system in energy_systems:
    generation_units = energy_system.generation_systems
    if cte.HEATING in energy_system.demand_types:
      for generation_unit in generation_units:
        generation_unit.heat_efficiency = 0.96
for building in city.buildings:
  building.function = cte.COMMERCIAL

ConstructionFactory('nrcan', city).enrich()
UsageFactory('nrcan', city).enrich()
energy_plus_workflow(city)
for building in city.buildings:
  print(building.heating_demand[cte.YEAR][0] / 3.6e6)
  print(building.name)
  total_floor_area = 0
  for thermal_zone in building.thermal_zones_from_internal_zones:
    total_floor_area += thermal_zone.total_floor_area
  print(building.heating_demand[cte.YEAR][0] / (3.6e6 * total_floor_area))
building_names = []
for building in city.buildings:
  building_names.append(building.name)

df = pd.DataFrame(columns=building_names)
df.to_csv('selected_buildings.csv')