Tibeuleu/FOC_Reduction
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix pol_map.set_data_mask and some keyword saving

Browse Source
This commit is contained in:
Thibault Barnouin
2025-04-11 14:17:42 +02:00
parent 4be32a54ac
commit c8bbcc9d3b
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
package/lib/fits.py
View File

@@ -217,7 +217,7 @@ def save_Stokes(
# Add Flux density as PrimaryHDU # Add Flux density as PrimaryHDU
if flux_data is None: if flux_data is None:
header["datatype"] = ("I_stokes", "type of data stored in the HDU") header["DATATYPE"] = ("I_stokes", "type of data stored in the HDU")
I_stokes[(1 - data_mask).astype(bool)] = 0.0 I_stokes[(1 - data_mask).astype(bool)] = 0.0
primary_hdu = fits.PrimaryHDU(data=I_stokes, header=header) primary_hdu = fits.PrimaryHDU(data=I_stokes, header=header)
primary_hdu.name = "I_stokes" primary_hdu.name = "I_stokes"
@@ -230,11 +230,11 @@ def save_Stokes(
head[key] = flux_head[key] head[key] = flux_head[key]
except KeyError: except KeyError:
head[key] = header[key] head[key] = header[key]
header["datatype"] = ("Flux_density", "type of data stored in the HDU") header["DATATYPE"] = ("Flux_density", "type of data stored in the HDU")
primary_hdu = fits.PrimaryHDU(data=flux_data, header=head) primary_hdu = fits.PrimaryHDU(data=flux_data, header=head)
primary_hdu.name = "Flux_density" primary_hdu.name = "Flux_density"
hdul.append(primary_hdu) hdul.append(primary_hdu)
header["datatype"] = ("I_stokes", "type of data stored in the HDU") header["DATATYPE"] = ("I_stokes", "type of data stored in the HDU")
I_stokes[(1 - data_mask).astype(bool)] = 0.0 I_stokes[(1 - data_mask).astype(bool)] = 0.0
image_hdu = fits.ImageHDU(data=I_stokes, header=header) image_hdu = fits.ImageHDU(data=I_stokes, header=header)
image_hdu.name = "I_stokes" image_hdu.name = "I_stokes"
@@ -255,7 +255,7 @@ def save_Stokes(
[data_mask, "Data_mask"], [data_mask, "Data_mask"],
]: ]:
hdu_header = header.copy() hdu_header = header.copy()
hdu_header["datatype"] = name hdu_header["DATATYPE"] = name
if not name == "IQU_cov_matrix": if not name == "IQU_cov_matrix":
data[(1 - data_mask).astype(bool)] = 0.0 data[(1 - data_mask).astype(bool)] = 0.0
hdu = fits.ImageHDU(data=data, header=hdu_header) hdu = fits.ImageHDU(data=data, header=hdu_header)

4
package/lib/plots.py
View File

@@ -2951,7 +2951,7 @@ class pol_map(object):
@property @property
def wcs(self): def wcs(self):
return WCS(self.Stokes[0].header).celestial.deepcopy() return WCS(self.Stokes["I_STOKES"].header).celestial.deepcopy()
@property @property
def Flux(self): def Flux(self):
@@ -3042,7 +3042,7 @@ class pol_map(object):
return self.Stokes["DATA_MASK"].data.astype(bool) return self.Stokes["DATA_MASK"].data.astype(bool)
def set_data_mask(self, mask): def set_data_mask(self, mask):
self.Stokes[np.argmax([self.Stokes[i].header["datatype"] == "Data_mask" for i in range(len(self.Stokes))])].data = mask.astype(float) self.Stokes["DATA_MASK"].data = mask.astype(float)
@property @property
def cut(self): def cut(self):