from numpy import nan
from datatypes.epc.floor import EpcFloorDescriptions

floor_map = {
    # Solid floor
    ('Solid', 'AsBuilt'): None,  # Mapped
    ('Solid', 'Unknown'): None,  # Mapped
    ('Solid', nan): None,  # Mapped
    ('Solid', 'RetroFitted'): EpcFloorDescriptions.solid_insulated,

    # Suspended floor
    ('SuspendedTimber', nan): None,  # Mapped suspended_floor_as_built
    ('SuspendedTimber', 'AsBuilt'): None,  # Mapped suspended_floor_as_built
    ('SuspendedTimber', 'RetroFitted'): EpcFloorDescriptions.suspended_insulated,
    ('SuspendedTimber', 'Unknown'): None,  # Mapped suspended_floor_as_built
    ('SuspendedNotTimber', 'RetroFitted'): EpcFloorDescriptions.suspended_insulated,
    ('SuspendedNotTimber', nan): None,  # Mapped suspended_floor_as_built
    ('SuspendedNotTimber', 'Unknown'): None,  # Mapped suspended_floor_as_built
    ('SuspendedNotTimber', 'AsBuilt'): None,  # Mapped suspended_floor_as_built

    # Unknown type - mapped on age
    ('Unknown', 'Unknown'): None,  # Mapped unknown_floor_as_built
    ('Unknown', 'RetroFitted'): None,  # Mapped unknown_floor_retrofitted
    (nan, nan): None,  # No actual information!
    ('Unknown', 'AsBuilt'): None,  # Mapped unknown_floor_as_built
}