Source code for jwst.assign_mtwcs.assign_mtwcs_step

#! /usr/bin/env python
from ..stpipe import Step
from .. import datamodels
import logging
from .moving_target_wcs import assign_moving_target_wcs

log = logging.getLogger(__name__)

__all__ = ["AssignMTWcsStep"]

[docs]class AssignMTWcsStep(Step): """ AssignMTWcsStep: Create a gWCS object for a moving target. Parameters ---------- input : `~jwst.associations.Association. A JWST association file. """ spec = """ suffix = string(default='assign_mtwcs') # Default suffix for output files output_use_model = boolean(default=True) # When saving use `DataModel.meta.filename` """
[docs] def process(self, input): if isinstance(input, str): input = if not isinstance(input, datamodels.ModelContainer): log.warning("Input data type is not supported.") # raise ValueError("Expected input to be an association file name or a ModelContainer.") input.meta.cal_step.assign_mtwcs = 'SKIPPED' return input result = assign_moving_target_wcs(input) return result