'''
Created May 2016

@author: TEASER Development Team
'''

"""annex.py

This module contains functions to sort and partly rewrite zone attributes
specific for annex simulation, including the export of boundary conditions
and AHU operation values
"""
import scipy.io
import teaser.logic.utilities as utilitis
import numpy as np


def compare_orientation(bldg, number_of_elements=3):
    """Fills the zone weightfactors according to orientation and tilt of
        building

        compares orientation and tilt of all outer building elements and then
        creates lists for zone weightfactors and building orientation and tilt

        Parameters
        ----------

        bldg: Building()
            TEASER instance of Building()

        number_of_elements: int()
            The number of elements calculated
    """

    orient_tilt_help1 = []
    orient_tilt_help2 = []
    bldg.orientation_bldg = []
    bldg.tilt_bldg = []

    for zone in bldg.thermal_zones:
        for wall in zone.outer_walls:
            if wall.orientation != -2:
                orient_tilt_help1.append([wall.orientation, wall.tilt])
            else:
                pass
        for win in zone.windows:
            if win.orientation != -2:
                orient_tilt_help1.append([win.orientation, win.tilt])
            else:
                pass

        for i in orient_tilt_help1:
            if i in orient_tilt_help2:
                pass
            else:
                orient_tilt_help2.append(i)

        orient_tilt_help2.sort(key=lambda x: x[0])

        if orient_tilt_help2[0][0] == -1:
            orient_tilt_help2.insert(len(orient_tilt_help2), orient_tilt_help2.pop(0))

        for i in orient_tilt_help2:
            bldg.orientation_bldg.append(i[0])
            bldg.tilt_bldg.append(i[1])

        groundfloors = zone.find_walls(-2, 0)
        if groundfloors == [] or number_of_elements in [3, 4]:
            zone.weightfactor_ground.append(0.0)
        else:
            zone.weightfactor_ground.append(
                sum([groundfl.wf_out for groundfl in groundfloors]))

        for i in orient_tilt_help2:
            walls = zone.find_walls(i[0], i[1])
            wins = zone.find_wins(i[0], i[1])

            rts = zone.find_rts(i[0], i[1])

            if walls == []:
                zone.weightfactor_ow.append(0.0)
                zone.outer_walls_areas.append(0.0)
            else:
                if number_of_elements != 4:
                    zone.weightfactor_ow.append(
                        sum([wall.wf_out for wall in walls]))
                    [zone.outer_walls_areas.append(i.area) for i in walls]
                    zone.tilt_wall.append(i[1])
                    zone.orientation_wall.append(i[0])
                elif i[1] >= 90:
                    zone.weightfactor_ow.append(
                        sum([wall.wf_out for wall in walls]))
                    [zone.outer_walls_areas.append(i.area) for i in walls]
                    zone.tilt_wall.append(i[1])
                    zone.orientation_wall.append(i[0])
                else:
                    pass
            if wins == []:
                if number_of_elements != 4:
                    zone.weightfactor_win.append(0.0)
                    zone.window_area_list.append(0.0)
                    zone.g_sunblind_list.append(0.0)
                    zone.window_areas.append(0.0)
                elif number_of_elements == 4 and i[1] >= 90:
                    zone.weightfactor_win.append(0.0)
                    zone.window_area_list.append(0.0)
                    zone.g_sunblind_list.append(0.0)
                    zone.window_areas.append(0.0)

            else:
                if number_of_elements != 4:
                    zone.weightfactor_win.append(
                        sum([win.wf_out for win in wins]))
                    zone.window_area_list.append(
                        sum([win.area for win in wins]))
                    zone.g_sunblind_list.append(
                        sum([win.shading_g_total for win in wins]))
                    [zone.window_areas.append(i.area) for i in wins]
                    zone.tilt_win.append(i[1])
                    zone.orientation_win.append(i[0])
                elif i[1] >= 90:
                    zone.weightfactor_win.append(
                        sum([win.wf_out for win in wins]))
                    zone.window_area_list.append(
                        sum([win.area for win in wins]))
                    zone.g_sunblind_list.append(
                        sum([win.shading_g_total for win in wins]))
                    [zone.window_areas.append(i.area) for i in wins]
                    zone.tilt_win.append(i[1])
                    zone.orientation_win.append(i[0])

            # if rts == []:
            #     zone.weightfactor_rt.append(0.0)
            if rts:
                zone.orientation_rt.append(i[0])
                zone.tilt_rt.append(i[1])
                [zone.weightfactor_rt.append(i.wf_out) for i in rts]
