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

fix WCS comparison and 2D display

Browse Source
This commit is contained in:
Thibault Barnouin
2026-02-13 14:31:56 +01:00
parent 561c5c2fec
commit 03f2c550e3
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
package/lib/fits.py
View File

@@ -89,7 +89,7 @@ def get_obs_data(infiles, data_folder="", compute_flux=False):
# force WCS for POL60 to have same pixel size as POL0 and POL120 # force WCS for POL60 to have same pixel size as POL0 and POL120
is_pol60 = np.array([head["filtnam1"].lower() == "pol60" for head in headers], dtype=bool) is_pol60 = np.array([head["filtnam1"].lower() == "pol60" for head in headers], dtype=bool)
cdelt = np.round(np.array([WCS(head).wcs.cdelt[:2] for head in headers]), 10) cdelt = np.round(np.abs(np.array([WCS(head).wcs.cdelt[:2] for head in headers])), 10)
if np.unique(cdelt[np.logical_not(is_pol60)], axis=0).size != 2: if np.unique(cdelt[np.logical_not(is_pol60)], axis=0).size != 2:
print(np.unique(cdelt[np.logical_not(is_pol60)], axis=0)) print(np.unique(cdelt[np.logical_not(is_pol60)], axis=0))
raise ValueError("Not all images have same pixel size") raise ValueError("Not all images have same pixel size")

6
package/lib/plots.py
View File

@@ -189,7 +189,7 @@ def plot_Stokes(Stokes, savename=None, plots_folder=""):
for dataset in [stkI, stkQ, stkU]: for dataset in [stkI, stkQ, stkU]:
dataset[np.logical_not(data_mask)] = np.nan dataset[np.logical_not(data_mask)] = np.nan
wcs = WCS(Stokes[0]).deepcopy() wcs = WCS(Stokes[0]).celestial.deepcopy()
# Plot figure # Plot figure
plt.rcParams.update({"font.size": 14}) plt.rcParams.update({"font.size": 14})
@@ -370,7 +370,9 @@ def polarization_map(
ax.coords[0].set_axislabel("Right Ascension (J2000)") ax.coords[0].set_axislabel("Right Ascension (J2000)")
ax.coords[0].set_axislabel_position("t") ax.coords[0].set_axislabel_position("t")
ax.coords[0].set_ticklabel_position("t") ax.coords[0].set_ticklabel_position("t")
ax.set_ylabel("Declination (J2000)", labelpad=-1) ax.coords[1].set_axislabel("Declination (J2000)")
ax.coords[1].set_axislabel_position("l")
ax.coords[1].set_ticklabel_position("l")
vmin, vmax = 0.0, stkI.max() * convert_flux vmin, vmax = 0.0, stkI.max() * convert_flux
for key, value in [["cmap", [["cmap", "inferno"]]], ["width", [["width", 0.4]]], ["linewidth", [["linewidth", 0.6]]]]: for key, value in [["cmap", [["cmap", "inferno"]]], ["width", [["width", 0.4]]], ["linewidth", [["linewidth", 0.6]]]]: