from bpy import data, types
from .. import utilities, logger


def _lamp(func):
    """

    :param func:

    """

    def inner(name, *args, **kwargs):
        """

        :param name:
        :param *args:
        :param **kwargs:

        """

        if isinstance(name, types.Lamp):
            lamp = name
        else:
            lamp = data.lamps[name]

        return func(lamp, *args, **kwargs)

    return inner


@_lamp
def angle(lamp):
    """

    :param lamp:
    :rtype: float

    """
    logger.debug("light.angle(%s)", lamp)
    return lamp.spot_size


@_lamp
def color(lamp):
    """

    :param lamp:
    :rtype: int

    """
    logger.debug("light.color(%s)", lamp)
    colour = (lamp.color.r, lamp.color.g, lamp.color.b)
    return utilities.rgb2int(colour)


@_lamp
def distance(lamp):
    """

    :param lamp:
    :rtype: float

    """
    logger.debug("light.distance(%s)", lamp)
    return lamp.distance


@_lamp
def intensity(lamp):
    """

    :param lamp:
    :rtype: float

    """
    logger.debug("light.intensity(%s)", lamp)
    return round(lamp.energy, 2)

# mapping enum values to decay exponent
__FALLOFF_TO_EXP = {
    'CONSTANT': 0,
    'INVERSE_LINEAR': 1,
    'INVERSE_SQUARE': 2,
    'CUSTOM_CURVE': 0,
    'LINEAR_QUADRATIC_WEIGHTED': 2
}

@_lamp
def falloff(lamp):
    """

    :param lamp:
    :rtype: float

    """
    logger.debug("light.falloff(%s)", lamp)
    return __FALLOFF_TO_EXP[lamp.falloff_type]