Source code for jwst.extract_2d.extract_2d_step

#! /usr/bin/env python

from ..stpipe import Step
from .. import datamodels
from . import extract_2d


__all__ = ["Extract2dStep"]


[docs]class Extract2dStep(Step): """ This Step performs a 2D extraction of spectra. """ spec = """ slit_name = string(default=None) extract_orders = int_list(default=None) # list of orders to extract tsgrism_extract_height = integer(default=None) # extraction height in pixels, TSGRISM mode wfss_extract_half_height = integer(default=5) # extraction half height in pixels, WFSS mode grism_objects = list(default=None) # list of grism objects to use mmag_extract = float(default=99.) # minimum abmag to extract """ reference_file_types = ['wavelengthrange']
[docs] def process(self, input_model, *args, **kwargs): reference_file_names = {} for reftype in self.reference_file_types: reffile = self.get_reference_file(input_model, reftype) reference_file_names[reftype] = reffile if reffile else "" with datamodels.open(input_model) as dm: output_model = extract_2d.extract2d(dm, self.slit_name, reference_files=reference_file_names, grism_objects=self.grism_objects, tsgrism_extract_height=self.tsgrism_extract_height, wfss_extract_half_height=self.wfss_extract_half_height, extract_orders=self.extract_orders, mmag_extract=self.mmag_extract) return output_model