def run(stages_to_do,
pipeline_context,
image_types=[],
calibration_maker=False,
log_message=''):
"""
Main driver script for banzai.
"""
if len(log_message) > 0:
logger.info(log_message,
extra={'tags': {
'raw_path': pipeline_context.raw_path
}})
image_list = image_utils.make_image_list(pipeline_context)
image_list = image_utils.select_images(image_list, image_types)
images = banzai.images.read_images(image_list, pipeline_context)
for stage in stages_to_do:
stage_to_run = stage(pipeline_context)
images = stage_to_run.run(images)
output_files = image_utils.save_images(
pipeline_context, images, master_calibration=calibration_maker)
return output_files
def reduce_science_frames(pipeline_context=None):
stages_to_do = [
munge.DataMunger, qc.SaturationTest, crosstalk.CrosstalkCorrector,
bias.OverscanSubtractor, gain.GainNormalizer, mosaic.MosaicCreator,
trim.Trimmer, bias.BiasSubtractor, dark.DarkSubtractor,
flats.FlatDivider, photometry.SourceDetector, astrometry.WCSSolver,
headers.HeaderUpdater, pointing.PointingTest
]
image_list = image_utils.make_image_list(pipeline_context)
original_filename = pipeline_context.filename
for image in image_list:
pipeline_context.filename = os.path.basename(image)
try:
run(stages_to_do,
pipeline_context,
image_types=['EXPOSE', 'STANDARD'])
except Exception as e:
logger.error('{0}'.format(e),
extra={
'tags': {
'filename': pipeline_context.filename,
'filepath': pipeline_context.raw_path
}
})
pipeline_context.filename = original_filename
def reduce_frames_one_by_one(stages_to_do, pipeline_context, image_types=['EXPOSE', 'STANDARD']):
image_list = image_utils.make_image_list(pipeline_context)
original_filename = pipeline_context.filename
for image in image_list:
pipeline_context.filename = os.path.basename(image)
try:
run(stages_to_do, pipeline_context, image_types=image_types)
except Exception as e:
logger.error('{0}'.format(e), extra={'tags': {'filename': pipeline_context.filename,
'filepath': pipeline_context.raw_path}})
pipeline_context.filename = original_filename
def reduce_science_frames(pipeline_context=None):
stages_to_do = [munge.DataMunger, qc.SaturationTest, crosstalk.CrosstalkCorrector,
bias.OverscanSubtractor, gain.GainNormalizer, mosaic.MosaicCreator,
trim.Trimmer, bias.BiasSubtractor, dark.DarkSubtractor,
flats.FlatDivider, photometry.SourceDetector, astrometry.WCSSolver,
headers.HeaderUpdater, pointing.PointingTest]
image_list = image_utils.make_image_list(pipeline_context)
original_filename = pipeline_context.filename
for image in image_list:
pipeline_context.filename = os.path.basename(image)
run(stages_to_do, pipeline_context, image_types=['EXPOSE', 'STANDARD'])
pipeline_context.filename = original_filename
def run(stages_to_do, pipeline_context, image_types=[], calibration_maker=False, log_message=''):
"""
Main driver script for banzai.
"""
if len(log_message) > 0:
logger.info(log_message, extra={'tags': {'raw_path': pipeline_context.raw_path}})
image_list = image_utils.make_image_list(pipeline_context)
image_list = image_utils.select_images(image_list, image_types)
images = banzai.images.read_images(image_list, pipeline_context)
for stage in stages_to_do:
stage_to_run = stage(pipeline_context)
images = stage_to_run.run(images)
image_utils.save_images(pipeline_context, images, master_calibration=calibration_maker)
def reduce_frames_one_by_one(stages_to_do,
pipeline_context,
image_types=['EXPOSE', 'STANDARD']):
image_list = image_utils.make_image_list(pipeline_context)
original_filename = pipeline_context.filename
for image in image_list:
pipeline_context.filename = os.path.basename(image)
try:
run(stages_to_do, pipeline_context, image_types=image_types)
except Exception as e:
logger.error('{0}'.format(e),
extra={
'tags': {
'filename': pipeline_context.filename,
'filepath': pipeline_context.raw_path
}
})
pipeline_context.filename = original_filename