ESBO-ETC/esbo_etc/classes/target/FileTarget.py

from ..target.ATarget import ATarget
from ..SpectralQty import SpectralQty
import astropy.units as u
from ..Entry import Entry
from typing import Union


class FileTarget(ATarget):
    """
    A class to create a target from a file containing the spectral flux densities
    """

    @u.quantity_input(wl_bins="length")
    def __init__(self, file: str, wl_bins: u.Quantity, size: str = "Point"):
        """
        Initialize a new target from a file containing the spectral flux density values

        Parameters
        ----------
        file : str
            The file to read the spectral flux density values from. The file needs to provide two columns: wavelength
            and the corresponding spectral flux density. The format of the file will be guessed by
            `astropy.io.ascii.read(). If the file doesn't provide units via astropy's enhanced CSV format, the units
            will be read from the column headers or otherwise assumed to be *nm* and *W / m^2 / nm*.
        wl_bins : length-Quantity
            Wavelengths used for binning
        size : str
            The size of the target. Can be either point or extended.
        """
        # Create spectral quantity from file
        sfd = SpectralQty.fromFile(file, u.nm, u.W / (u.m ** 2 * u.nm))
        # Initialize the super class
        super().__init__(sfd, wl_bins, size)

    @staticmethod
    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.
        """
        mes = conf.check_file("file")
        if mes is not None:
            return mes
        mes = conf.check_selection("size", ["point", "extended"])
        if mes is not None:
            return mes