Source code for jwst.linearity.linearity_step

from ..stpipe import Step
from .. import datamodels
from . import linearity

__all__ = ["LinearityStep"]

[docs]class LinearityStep(Step): """ LinearityStep: This step performs a correction for non-linear detector response, using the "classic" polynomial method. """ reference_file_types = ['linearity']
[docs] def process(self, input): # Open the input data model with datamodels.RampModel(input) as input_model: # Get the name of the linearity reference file to use self.lin_name = self.get_reference_file(input_model, 'linearity')'Using Linearity reference file %s', self.lin_name) # Check for a valid reference file if self.lin_name == 'N/A': self.log.warning('No Linearity reference file found') self.log.warning('Linearity step will be skipped') result = input_model.copy() result.meta.cal_step.linearity = 'SKIPPED' return result # Open the linearity reference file data model lin_model = datamodels.LinearityModel(self.lin_name) # Do the linearity correction result = linearity.do_correction(input_model, lin_model) # Close the reference file and update the step status lin_model.close() result.meta.cal_step.linearity = 'COMPLETE' return result