#! /usr/bin/env python
from stdatamodels.jwst import datamodels
from ..stpipe import Step
from .white_light import white_light
__all__ = ["WhiteLightStep"]
[docs]
class WhiteLightStep(Step):
"""
WhiteLightStep: Computes integrated flux as a function of time for a
multi-integration spectroscopic observation.
"""
class_alias = "white_light"
spec = """
min_wavelength = float(default=None) # Default wavelength minimum for integration
max_wavelength = float(default=None) # Default wavelength maximum for integration
output_ext = string(default='.ecsv') # Output file type
suffix = string(default='whtlt') # Default suffix for output files
"""
[docs]
def process(self, input):
# Load the input
with datamodels.open(input) as input_model:
# Call the white light curve generation routine
result = white_light(input_model, self.min_wavelength, self.max_wavelength)
# Write the output catalog
if self.save_results:
output_path = self.make_output_path()
result.write(output_path, format='ascii.ecsv', overwrite=True)
return result