from ..IRadiant import IRadiant
from ..SpectralQty import SpectralQty
from ...lib.logger import logger
from abc import abstractmethod
import astropy.units as u
from astropy.modeling.models import BlackBody
from typing import Union, Callable, Tuple
from ..Entry import Entry


class AOpticalComponent(IRadiant):
    """
    Abstract super class for an optical component
    """

    @abstractmethod
    @u.quantity_input(obstructor_temp=[u.K, u.Celsius])
    def __init__(self, parent: IRadiant, transreflectivity: Union[SpectralQty, int, float, u.Quantity] = None,
                 noise: Union[SpectralQty, int, float, u.Quantity, Callable] = None, obstruction: float = 0,
                 obstructor_temp: u.Quantity = 0 * u.K, obstructor_emissivity: float = 1):
        """
        Initialize a new optical component

        Parameters
        ----------
        parent : IRadiant
            The parent element of the optical component from which the electromagnetic radiation is received
        transreflectivity : Union[SpectralQty, int, float, u.Quantity]
            The spectral transmission / reflectivity coefficient of the component. This coefficient is multiplied with
            the incoming radiation from the parent element in order to propagate the incoming radiation through the
            optical component.
        noise : Union[SpectralQty, int, float, u.Quantity, Callable]
            The noise created by the optical component as spectral radiance. This noise will be added to the propagated
            incoming noise in order to calculate the overall noise.
        obstruction : float
            The additional obstruction factor of the optical component. 0 means the component is not obstructed, 1
            denotes a completely obstructed component with therefore no incoming flux. It is important to note, that
            the obstruction factor reflects the obstruction of the optical component additionally to the obstruction
            factors of the prior elements in the beam.
        obstructor_temp : Quantity in Kelvin / Celsius
            Temperature of the obstructing component.
        obstructor_emissivity : float
            Emissivity of the obstructing component.
        """
        self.__parent = parent
        if transreflectivity is not None:
            self.__transreflectivity = transreflectivity
        if noise is not None:
            self.__noise = noise
        self.__obstruction = obstruction
        self.__obstructor_temp = obstructor_temp
        self.__obstructor_emissivity = obstructor_emissivity

    def calcSignal(self) -> Tuple[SpectralQty, float]:
        """
        Calculate the spectral flux density of the target's signal

        Returns
        -------
        signal : SpectralQty
            The spectral flux density of the target's signal
        obstruction : float
            The obstruction factor as A_ob / A_ap.
        """
        signal, obstruction = self.__parent.calcSignal()
        logger.info("Calculating signal for class '" + self.__class__.__name__ + "'.")
        signal = self._propagate(signal) * (1 - self.__obstruction)
        obstruction = obstruction + self.__obstruction
        logger.debug(signal)
        return signal, obstruction

    def calcBackground(self) -> SpectralQty:
        """
        Calculate the spectral radiance of the background

        Returns
        -------
        background : SpectralQty
            The spectral radiance of the background
        """
        parent = self.__parent.calcBackground()
        logger.info("Calculating background for class '" + self.__class__.__name__ + "'.")
        parent = self._propagate(parent)
        if self.__obstructor_temp > 0 * u.K:
            bb = BlackBody(temperature=self.__obstructor_temp, scale=1. * u.W / (u.m ** 2 * u.nm * u.sr))
            obstructor = bb(parent.wl) * self.__obstructor_emissivity
            background = parent * (1. - self.__obstruction) + obstructor * self.__obstruction
        else:
            background = parent * (1. - self.__obstruction)
        background = background + self._ownNoise()
        logger.debug(background)
        return background

    def _propagate(self, rad: SpectralQty) -> SpectralQty:
        """
        Propagate incoming radiation through the optical component

        Parameters
        ----------
        rad : SpectralQty
            The incoming radiation

        Returns
        -------
        rad : SpectralQty
            Manipulated incoming radiation
        """
        try:
            return rad * self.__transreflectivity
        except AttributeError:
            logger.error("Transreflectivity not given. Method propagate() needs to be implemented.")

    def _ownNoise(self) -> Union[SpectralQty, Callable[[u.Quantity], u.Quantity], int, float]:
        """
        Calculate the noise created by the optical component

        Returns
        -------
        noise : Union[SpectralQty, Callable[[u.Quantity], u.Quantity], int, float]
            The noise created by the optical component
        """
        try:
            return self.__noise
        except AttributeError:
            logger.error("noise not given. Method ownNoise() needs to be implemented.")

    @staticmethod
    @abstractmethod
    def check_config(conf: Entry) -> Union[None, str]:
        """
        Check the configuration for this class

        Parameters
        ----------
        conf : Entry
            The configuration entry to be checked.

        Returns
        -------
        mes : Union[None, str]
            The error message of the check. This will be None if the check was successful.
        """
        pass