Source code for jwst.saturation.saturation_step

#! /usr/bin/env python

from ..stpipe import Step
from .. import datamodels
from . import saturation

__all__ = ["SaturationStep"]

[docs]class SaturationStep(Step): """ This Step sets saturation flags. """ reference_file_types = ['saturation']
[docs] def process(self, input): # Open the input data model with datamodels.RampModel(input) as input_model: # Get the name of the saturation reference file self.ref_name = self.get_reference_file(input_model, 'saturation')'Using SATURATION reference file %s', self.ref_name) # Check for a valid reference file if self.ref_name == 'N/A': self.log.warning('No SATURATION reference file found') self.log.warning('Saturation step will be skipped') result = input_model.copy() result.meta.cal_step.saturation = 'SKIPPED' return result # Open the reference file data model ref_model = datamodels.SaturationModel(self.ref_name) # Do the saturation check sat = saturation.do_correction(input_model, ref_model) # Close the reference file and update the step status ref_model.close() sat.meta.cal_step.saturation = 'COMPLETE' return sat