#! /usr/bin/env python
from ..stpipe import Step
from .. import datamodels
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.
"""
spec = """
output_ext = string(default='.ecsv') # Default type of output
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)
# 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