From 95b0cc4033a36c2925cd701bae2bdc4c02ffc973 Mon Sep 17 00:00:00 2001 From: Thibault Barnouin Date: Wed, 19 Mar 2025 16:38:36 +0100 Subject: [PATCH] save raw flux in fits file and display forward fixes to display with WCS --- package/FOC_reduction.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/package/FOC_reduction.py b/package/FOC_reduction.py index a700b49..b98886f 100755 --- a/package/FOC_reduction.py +++ b/package/FOC_reduction.py @@ -20,6 +20,7 @@ import lib.reduction as proj_red # Functions used in reduction pipeline import numpy as np from lib.utils import princ_angle, sci_not from matplotlib.colors import LogNorm +from astropy.wcs import WCS def main(target=None, proposal_id=None, infiles=None, output_dir="./data", crop=False, interactive=False): @@ -273,7 +274,7 @@ def main(target=None, proposal_id=None, infiles=None, output_dir="./data", crop= flux_data=np.array([flux_data, flux_error, flux_mask]), flux_head=flux_head, ) - outfiles.append("/".join([data_folder, Stokes_hdul["I_STOKES"].header["FILENAME"] + ".fits"])) + outfiles.append("/".join([data_folder, Stokes_hdul[0].header["FILENAME"] + ".fits"])) # Step 5: # crop to desired region of interest (roi)