Source code for jwst.datamodels.source_container

from . import ModelContainer, MultiExposureModel, SlitModel

__all__ = ['SourceModelContainer']


# Models that can initiliaze into a SourceModelContainer
VALID_INITS = (
    MultiExposureModel,
)


[docs]class SourceModelContainer(ModelContainer): """ A container to make MultiExposureModel look like ModelContainer The `MultiExposureModel.exposures` list contains the data for each exposure from a common slit id. Though the information is the same, the structures are not true `SlitModel` instances. This container creates a `SlitModel` wrapper around each exposure, such that pipeline code can treat each as a `DataModel`. """ def __init__(self, init=None, **kwargs): if not isinstance(init, (self.__class__, ) + VALID_INITS): raise TypeError( 'Input {0!r} cannot initialize a SourceModelContainer'.format(init) ) if isinstance(init, SourceModelContainer): super(SourceModelContainer, self).__init__(init, **kwargs) self._multiexposure = init._multiexposure elif isinstance(init, MultiExposureModel): super(SourceModelContainer, self).__init__(init=None, **kwargs) # Convert each exposure to an actual SlitModel. # Note that the model is not instantiated # since a copy is not desired. models = [] for exposure in init.exposures: model = SlitModel() model._instance.update(exposure._instance) models.append(model) self._models = models self._multiexposure = init @property def multiexposure(self): """Return the MultiExposureModel that is being wrapped The result will be updated with any new data in the container. """ # Reapply models back to the exposures for exposure, model in zip(self._multiexposure.exposures, self._models): exposure._instance.update(model._instance) return self._multiexposure
[docs] def save(self, path=None, dir_path=None, save_model_func=None, *args, **kwargs): """Save out the container as a MultiExposureModel""" if save_model_func is None: self.multiexposure.save( path=path, dir_path=dir_path, *args, **kwargs ) else: save_model_func(self.multiexposure, output_file=path)