"""
CaUsageParameters model the usage properties
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.geometry.helpers.geometry_helper import GeometryHelper as gh
from imports.usage.hft_usage_interface import HftUsageInterface
from city_model_structure.building_demand.usage_zone import UsageZone
from city_model_structure.building_demand.internal_gains import InternalGains


class CaUsageParameters(HftUsageInterface):
  """
  CaUsageParameters class
  """
  def __init__(self, city, base_path):
    super().__init__(base_path, 'ca_archetypes_reduced.xml')
    self._city = city
    # todo: this is a wrong location for self._min_air_change -> re-think where to place this info
    #  and where it comes from
    self._min_air_change = 0

  def enrich_buildings(self):
    """
    Returns the city with the usage parameters assigned to the buildings
    :return:
    """
    city = self._city
    for building in city.buildings:
      archetype = self._search_archetype(building.function)
      if archetype is None:
        sys.stderr.write(f'Building {building.name} has unknown archetype for building function:'
                         f' {building.function}, that assigns building usage as '
                         f'{gh.usage_from_function(building.function)}\n')
        continue
      # todo: what to do with mix-usage usage from gml?
      mix_usage = False
      if not mix_usage:
        # just one usage_zone
        for thermal_zone in building.thermal_zones:
          usage_zone = UsageZone()
          usage_zone.volume = thermal_zone.volume
          self._assign_values(usage_zone, archetype)
          thermal_zone.usage_zones = [usage_zone]

  def _search_archetype(self, building_usage):
    for building_archetype in self._usage_archetypes:
      if building_archetype.usage == building_usage:
        return building_archetype
    return None

  @staticmethod
  def _assign_values(usage_zone, archetype):
    usage_zone.usage = archetype.usage
    # Due to the fact that python is not a typed language, the wrong object type is assigned to
    # usage_zone.internal_gains when writing usage_zone.internal_gains = archetype.internal_gains.
    # Therefore, this walk around has been done.
    internal_gains = []
    for ig in archetype.internal_gains:
      internal_gain = InternalGains()
      internal_gain.average_internal_gain = ig.average_internal_gain
      internal_gain.convective_fraction = ig.convective_fraction
      internal_gain.radiative_fraction = ig.radiative_fraction
      internal_gain.latent_fraction = ig.latent_fraction
      internal_gains.append(internal_gain)
    usage_zone.internal_gains = internal_gains
    usage_zone.heating_setpoint = archetype.heating_setpoint
    usage_zone.heating_setback = archetype.heating_setback
    usage_zone.cooling_setpoint = archetype.cooling_setpoint
    usage_zone.occupancy_density = archetype.occupancy_density
    usage_zone.hours_day = archetype.hours_day
    usage_zone.days_year = archetype.days_year
    usage_zone.dhw_average_volume_pers_day = archetype.dhw_average_volume_pers_day
    usage_zone.dhw_preparation_temperature = archetype.dhw_preparation_temperature
    usage_zone.electrical_app_average_consumption_sqm_year = archetype.electrical_app_average_consumption_sqm_year
    usage_zone.mechanical_air_change = archetype.mechanical_air_change