save raw flux in fits file and display

rebase display on main
2025-03-19 16:38:36 +01:00
parent 4e9733b0df
commit 88be9529d4
2 changed files with 14 additions and 2 deletions
--- a/package/lib/plots.py
+++ b/package/lib/plots.py
@@ -2951,7 +2951,19 @@ class pol_map(object):

    @property
    def wcs(self):
-        return WCS(self.Stokes[0].header).celestial.deepcopy()
+        return WCS(self.Stokes["I_STOKES"].header).celestial.deepcopy()
+
+    @property
+    def Flux(self):
+        return self.Stokes[0].data[0] * self.Stokes[0].header["PHOTFLAM"]
+
+    @property
+    def Flux_err(self):
+        return self.Stokes[0].data[1] * self.Stokes[0].header["PHOTFLAM"]
+
+    @property
+    def Flux_mask(self):
+        return self.Stokes[0].data[2].astype(bool)

    @property
    def Flux(self):