Source code for slsim.Sources.point_plus_extended_sources

from slsim.Sources.source_pop_base import SourcePopBase
from slsim.Sources.galaxies import Galaxies




class PointPlusExtendedSources(Galaxies, SourcePopBase):
    """Class to describe point and extended sources."""

    def __init__(
        self,
        point_plus_extended_sources_list,
        cosmo,
        sky_area,
        kwargs_cut,
        variability_model=None,
        kwargs_variability_model=None,
        list_type="astropy_table",
    ):
        """

        :param pes_list: list of dictionary with point and extended source parameters
        :param cosmo: cosmology
        :type cosmo: ~astropy.cosmology class
        :param sky_area: Sky area over which galaxies are sampled. Must be in units of
            solid angle.
        :type sky_area: `~astropy.units.Quantity`
        :param variability_model: keyword for the variability model to be used. This is
         a population argument, not the light curve parameter for the individual
         source.
        :param kwargs_variability_model: keyword arguments for the variability of
         a source. This is a population argument, not the light curve parameter for
         the individual source.
        :param list_type: format of the source catalog file. Currently, it supports
         a single astropy table or a list of astropy tables.
        """
        Galaxies.__init__(
            self,
            point_plus_extended_sources_list,
            kwargs_cut,
            cosmo,
            sky_area,
            list_type=list_type,
        )
        SourcePopBase.__init__(
            self, cosmo, sky_area, variability_model, kwargs_variability_model
        )