from stdatamodels.jwst import datamodels
from jwst.coron import stack_refs
from jwst.stpipe import Step
__all__ = ["StackRefsStep"]
[docs]
class StackRefsStep(Step):
"""Stack multiple PSF reference exposures into a single CubeModel."""
class_alias = "stack_refs"
spec = """
""" # noqa: E501
[docs]
def process(self, input_files):
"""
Execute the StackRefs calibration step.
Parameters
----------
input_files : ModelContainer
ModelContainer containing input science exposures
Returns
-------
output_model : CubeModel
PSF reference exposures stacked into a CubeModel
"""
# Open the inputs
with datamodels.open(input_files) as input_models:
# Call the stacking routine
output_model = stack_refs.make_cube(input_models)
output_model.meta.cal_step.stack_psfs = "COMPLETE"
return output_model