"""
CaPhysicsParameters import the construction and material information for Canada
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca
"""
import sys
from imports.construction.helpers.construction_helper import ConstructionHelper
from imports.construction.nrel_physics_interface import NrelPhysicsInterface
from imports.construction.helpers.storeys_generation import StoreysGeneration


class CaPhysicsParameters(NrelPhysicsInterface):
  """
  CaPhysicsParameters class
  """
  def __init__(self, city, base_path):
    super().__init__(base_path, 'ca_constructions_reduced.xml', 'ca_archetypes_reduced.xml')
    self._city = city

  def enrich_buildings(self):
    """
    Returns the city with the construction parameters assigned to the buildings
    :return:
    """
    city = self._city
    # it is assumed that all buildings have the same archetypes' keys
    for building in city.buildings:
      archetype = self._search_archetype(ConstructionHelper.nrcan_from_function(building.function),
                                         building.year_of_construction)
      if archetype is None:
        sys.stderr.write(f'Building {building.name} has unknown archetype for building function: '
                         f'{ConstructionHelper.nrcan_from_function(building.function)} '
                         f'and building year of construction: {building.year_of_construction}\n')
        continue

      self._create_storeys(building, archetype)
      self._assign_values(building, archetype)

  def _search_archetype(self, function, year_of_construction):
    for building_archetype in self._building_archetypes:
      a_ft = str(building_archetype.archetype_keys['@function'])
      a_pc = str(building_archetype.archetype_keys['@periodOfConstruction'])
      a_yc1 = int(a_pc.split(sep='-')[0])
      a_yc2 = int(a_pc.split(sep='-')[1])
      if a_ft == str(function):
        if a_yc1 <= int(year_of_construction) <= a_yc2:
          return building_archetype
    return None

  def _assign_values(self, building, archetype):
    for thermal_zone in building.thermal_zones:
      thermal_zone.additional_thermal_bridge_u_value = archetype.additional_thermal_bridge_u_value
      thermal_zone.effective_thermal_capacity = archetype.effective_thermal_capacity
      thermal_zone.indirectly_heated_area_ratio = archetype.indirectly_heated_area_ratio
      thermal_zone.infiltration_rate_system_on = archetype.infiltration_rate_system_on
      thermal_zone.infiltration_rate_system_off = archetype.infiltration_rate_system_off
      for thermal_boundary in thermal_zone.thermal_boundaries:
        construction_type = ConstructionHelper.nrcan_construction_types[thermal_boundary.type]
        thermal_boundary_archetype = self._search_construction_in_archetype(archetype, construction_type)
        thermal_boundary.u_value = thermal_boundary_archetype.overall_u_value
        thermal_boundary.outside_solar_absorptance = thermal_boundary_archetype.outside_solar_absorptance
        thermal_boundary.construction_name = thermal_boundary_archetype.construction_name
        thermal_boundary.window_ratio = thermal_boundary_archetype.window_ratio
        if thermal_boundary.thermal_openings is not None:
          for thermal_opening in thermal_boundary.thermal_openings:
            if thermal_boundary_archetype.thermal_opening is not None:
              thermal_opening_archetype = thermal_boundary_archetype.thermal_opening
              thermal_opening.frame_ratio = thermal_opening_archetype.frame_ratio
              thermal_opening.g_value = thermal_opening_archetype.g_value
              thermal_opening.overall_u_value = thermal_opening_archetype.overall_u_value

  @staticmethod
  def _create_storeys(building, archetype):
    building.average_storey_height = archetype.average_storey_height
    building.storeys_above_ground = archetype.storeys_above_ground
    storeys_generation = StoreysGeneration(building)
    storeys = storeys_generation.storeys
    building.storeys = storeys
    storeys_generation.assign_thermal_zones_delimited_by_thermal_boundaries()