diff --git a/harnessed_jobs/BOT_EO_analysis/v0/producer_BOT_EO_analysis.py b/harnessed_jobs/BOT_EO_analysis/v0/producer_BOT_EO_analysis.py index 897167e..ec8224a 100755 --- a/harnessed_jobs/BOT_EO_analysis/v0/producer_BOT_EO_analysis.py +++ b/harnessed_jobs/BOT_EO_analysis/v0/producer_BOT_EO_analysis.py @@ -330,6 +330,14 @@ def cte_task(det_name): output.write('{}\n'.format(item)) +def find_flat2_bot(file1): + basename_pattern = '*_' + file1[-len('R22_S11.fits'):] + pattern = os.path.join(file1.split('flat1')[0] + 'flat2', + basename_pattern) + flat2 = glob.glob(pattern)[0] + return flat2 + + def flat_pairs_task(det_name): """Single sensor execution of the flat pairs task.""" run = siteUtils.getRunNumber() @@ -350,7 +358,8 @@ def flat_pairs_task(det_name): task = sensorTest.FlatPairTask() task.run(file_prefix, flat_files, mask_files, gains, - linearity_spec_range=(1e4, 9e4), use_exptime=use_exptime) + linearity_spec_range=(1e4, 9e4), use_exptime=use_exptime, + flat2_finder=find_flat2_bot) results_file = '%s_eotest_results.fits' % file_prefix plots = sensorTest.EOTestPlots(file_prefix, results_file=results_file) @@ -383,7 +392,8 @@ def ptc_task(det_name): gains = get_amplifier_gains(eotest_results_file) task = sensorTest.PtcTask() - task.run(file_prefix, flat_files, mask_files, gains) + task.run(file_prefix, flat_files, mask_files, gains, + flat2_finder=find_flat2_bot) results_file = '%s_eotest_results.fits' % file_prefix plots = sensorTest.EOTestPlots(file_prefix, results_file=results_file)