Python Registry Exemples

Exemple #1

"""Built-in models in Gammapy."""
from gammapy.utils.registry import Registry
from .core import *
from .cube import *
from .spatial import *
from .spectral import *
from .spectral_cosmic_ray import *
from .spectral_crab import *
from .temporal import *

SPATIAL_MODEL_REGISTRY = Registry(
    [
        ConstantSpatialModel,
        TemplateSpatialModel,
        DiskSpatialModel,
        GaussianSpatialModel,
        GeneralizedGaussianSpatialModel,
        PointSpatialModel,
        ShellSpatialModel,
        Shell2SpatialModel,
    ]
)
"""Registry of spatial model classes."""

SPECTRAL_MODEL_REGISTRY = Registry(
    [
        ConstantSpectralModel,
        CompoundSpectralModel,
        PowerLawSpectralModel,
        PowerLaw2SpectralModel,
        BrokenPowerLawSpectralModel,
        SmoothBrokenPowerLawSpectralModel,

Exemple #2

# Licensed under a 3-clause BSD style license - see LICENSE.rst
"""Source catalogs."""
from gammapy.utils.registry import Registry
from .core import *
from .fermi import *
from .gammacat import *
from .hawc import *
from .hess import *


SOURCE_CATALOGS = Registry([
    SourceCatalogGammaCat,
    SourceCatalogHGPS,
    SourceCatalog2HWC,
    SourceCatalog3FGL,
    SourceCatalog4FGL,
    SourceCatalog2FHL,
    SourceCatalog3FHL
])
"""Registry of source catalogs in Gammapy."""

__all__ = [
    "SOURCE_CATALOGS",
    "SourceCatalog",
    "SourceCatalogObject",
    "SourceCatalogObjectHGPS",
    "SourceCatalogObject2FHL",
    "SourceCatalogObject3FHL",
    "SourceCatalogObject3FGL",
    "SourceCatalogObject4FGL",
    "SourceCatalogObject2HWC",

Exemple #3

# Licensed under a 3-clause BSD style license - see LICENSE.rst
"""
Instrument response functions (IRFs).
"""
from gammapy.utils.registry import Registry
from .background import *
from .edisp_kernel import *
from .edisp_map import *
from .effective_area import *
from .energy_dispersion import *
from .io import *
from .psf import *


IRF_REGISTRY = Registry(
    [
        EffectiveAreaTable2D,
        EnergyDispersion2D,
        PSF3D,
        EnergyDependentMultiGaussPSF,
        PSFKing,
        Background3D,
        Background2D,
        PSFMap,
        EDispKernelMap,
    ]
)

Exemple #4

from .core import *
from .excess_map import *
from .excess_profile import *
from .flux_map import *
from .flux_point import *
from .lightcurve import *
from .profile import *
from .sensitivity import *
from .ts_map import *

ESTIMATOR_REGISTRY = Registry(
    [
        ExcessMapEstimator,
        TSMapEstimator,
        ASmoothMapEstimator,
        FluxPointsEstimator,
        LightCurveEstimator,
        SensitivityEstimator,
        ImageProfileEstimator,
        ExcessProfileEstimator,
    ]
)
"""Registry of estimator classes in Gammapy."""

__all__ = [
    "ESTIMATOR_REGISTRY",
    "FluxPoints",
    "ImageProfile",
    "Estimator",
]
__all__.extend(cls.__name__ for cls in ESTIMATOR_REGISTRY)

Exemple #5

Fichier : __init__.py Projet : vikasj78/gammapy

# Licensed under a 3-clause BSD style license - see LICENSE.rst
"""Source catalogs."""
from gammapy.utils.registry import Registry
from .core import *
from .fermi import *
from .gammacat import *
from .hawc import *
from .hess import *

CATALOG_REGISTRY = Registry([
    SourceCatalogGammaCat,
    SourceCatalogHGPS,
    SourceCatalog2HWC,
    SourceCatalog3FGL,
    SourceCatalog4FGL,
    SourceCatalog2FHL,
    SourceCatalog3FHL,
])
"""Registry of source catalogs in Gammapy."""

__all__ = [
    "CATALOG_REGISTRY",
    "SourceCatalog",
    "SourceCatalogObject",
    "SourceCatalogObjectHGPS",
    "SourceCatalogObject2FHL",
    "SourceCatalogObject3FHL",
    "SourceCatalogObject3FGL",
    "SourceCatalogObject4FGL",
    "SourceCatalogObject2HWC",
    "SourceCatalogObjectGammaCat",

Exemple #6

Fichier : __init__.py Projet : mireianievas/gammapy

from gammapy.utils.registry import Registry
from .core import Dataset, Datasets
from .flux_points import FluxPointsDataset
from .map import MapDataset, MapDatasetOnOff, create_map_dataset_geoms
from .simulate import MapDatasetEventSampler
from .spectrum import SpectrumDataset, SpectrumDatasetOnOff

DATASET_REGISTRY = Registry([MapDataset, MapDatasetOnOff, SpectrumDatasetOnOff, FluxPointsDataset])
"""Registry of dataset classes in Gammapy."""

__all__ = [
    "create_map_dataset_geoms",
    "Dataset",
    "DATASET_REGISTRY",
    "Datasets",
    "FluxPointsDataset",
    "MapDataset",
    "MapDatasetEventSampler",
    "MapDatasetOnOff",
    "SpectrumDataset",
    "SpectrumDatasetOnOff",
]

Exemple #7

# Licensed under a 3-clause BSD style license - see LICENSE.rst
"""Built-in models in Gammapy."""
from gammapy.utils.registry import Registry
from .core import *
from .cube import *
from .spatial import *
from .spectral import *
from .spectral_cosmic_ray import *
from .spectral_crab import *
from .temporal import *

SPATIAL_MODEL_REGISTRY = Registry([
    ConstantSpatialModel,
    TemplateSpatialModel,
    DiskSpatialModel,
    GaussianSpatialModel,
    PointSpatialModel,
    ShellSpatialModel,
])
"""Registry of spatial model classes."""

SPECTRAL_MODEL_REGISTRY = Registry([
    ConstantSpectralModel,
    CompoundSpectralModel,
    PowerLawSpectralModel,
    PowerLaw2SpectralModel,
    BrokenPowerLawSpectralModel,
    SmoothBrokenPowerLawSpectralModel,
    ExpCutoffPowerLawSpectralModel,
    ExpCutoffPowerLaw3FGLSpectralModel,
    SuperExpCutoffPowerLaw3FGLSpectralModel,

Exemple #8

Fichier : __init__.py Projet : renlliang3/gammapy

from gammapy.utils.registry import Registry
from .core import *
from .flux_points import *
from .map import *
from .simulate import *
from .spectrum import *

DATASETS = Registry([MapDataset, SpectrumDatasetOnOff, FluxPointsDataset])

__all__ = ["Dataset", "Datasets", "MapDatasetOnOff", "SpectrumDataset"]
__all__.extend(cls.__name__ for cls in DATASETS)

Exemple #9

from gammapy.utils.registry import Registry
from .background import *
from .core import *
from .map import *
from .reduce import *
from .safe import *
from .spectrum import *

MAKER_REGISTRY = Registry([
    ReflectedRegionsBackgroundMaker,
    AdaptiveRingBackgroundMaker,
    FoVBackgroundMaker,
    PhaseBackgroundMaker,
    RingBackgroundMaker,
    SpectrumDatasetMaker,
    MapDatasetMaker,
    SafeMaskMaker,
    DatasetsMaker,
])
"""Registry of maker classes in Gammapy."""

__all__ = ["MAKER_REGISTRY", "Maker"]
__all__.extend(cls.__name__ for cls in MAKER_REGISTRY)