From 05f800b2820c55d3fe352e87eea6837a1743303b 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 fix rebase display on main fix rebase display on main --- package/FOC_reduction.py | 2 +- package/lib/plots.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/package/FOC_reduction.py b/package/FOC_reduction.py index 5abef77..d3aee2c 100755 --- a/package/FOC_reduction.py +++ b/package/FOC_reduction.py @@ -274,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) diff --git a/package/lib/plots.py b/package/lib/plots.py index 3797cbb..994b6b6 100755 --- a/package/lib/plots.py +++ b/package/lib/plots.py @@ -3575,7 +3575,7 @@ class pol_map(object): @property def wcs(self): - return WCS(self.Stokes["I_STOKES"].header).celestial.deepcopy() + return WCS(self.Stokes[0].header).celestial.deepcopy() @property def Flux(self): @@ -3735,7 +3735,7 @@ class pol_map(object): back_length=0.0, head_length=10.0, head_width=10.0, - angle=-self.Stokes["I_STOKES"].header["orientat"], + angle=-self.Stokes[0].header["orientat"], color="white", text_props={"ec": None, "fc": "w", "alpha": 1, "lw": 0.4}, arrow_props={"ec": None, "fc": "w", "alpha": 1, "lw": 1},