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

more accessible plotsé

Browse Source
This commit is contained in:
Thibault Barnouin
2025-09-25 16:16:32 +02:00
parent 5a48ea1b24
commit 561f226473
2 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
package/lib/background.py
View File

@@ -45,14 +45,14 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
date_time = np.array([datetime.strptime(d, "%Y-%m-%d %H:%M:%S") for d in date_time]) date_time = np.array([datetime.strptime(d, "%Y-%m-%d %H:%M:%S") for d in date_time])
date_err = np.array([timedelta(seconds=headers[i]["exptime"] / 2.0) for i in range(len(headers))]) date_err = np.array([timedelta(seconds=headers[i]["exptime"] / 2.0) for i in range(len(headers))])
filt = np.array([headers[i]["filtnam1"] for i in range(len(headers))]) filt = np.array([headers[i]["filtnam1"] for i in range(len(headers))])
dict_filt = {"POL0": "r", "POL60": "g", "POL120": "b"} dict_filt = {"POL0": "yo", "POL60": "bv", "POL120": "rs"}
c_filt = np.array([dict_filt[f] for f in filt]) c_filt = np.array([dict_filt[f][0] for f in filt])
fig, ax = plt.subplots(figsize=(10, 6), constrained_layout=True) fig, ax = plt.subplots(figsize=(10, 4), constrained_layout=True)
ax.errorbar(date_time, background * convert_flux, xerr=date_err, yerr=std_bkg * convert_flux, fmt="+k", markersize=0, ecolor=c_filt)
for f in np.unique(filt): for f in np.unique(filt):
mask = [fil == f for fil in filt] mask = [fil == f for fil in filt]
ax.scatter(date_time[mask], background[mask] * convert_flux[mask], color=dict_filt[f], label="{0:s}".format(f)) ax.scatter(date_time[mask], background[mask] * convert_flux[mask], color=dict_filt[f][0], marker=dict_filt[f][1], label="{0:s}".format(f))
ax.errorbar(date_time, background * convert_flux, xerr=date_err, yerr=std_bkg * convert_flux, fmt="+k", markersize=0, ecolor=c_filt)
# Date handling # Date handling
locator = mdates.AutoDateLocator() locator = mdates.AutoDateLocator()
formatter = mdates.ConciseDateFormatter(locator) formatter = mdates.ConciseDateFormatter(locator)
@@ -69,7 +69,7 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
this_savename += "_background_flux.pdf" this_savename += "_background_flux.pdf"
else: else:
this_savename = savename[:-4] + "_background_flux" + savename[-4:] this_savename = savename[:-4] + "_background_flux" + savename[-4:]
fig.savefig(path_join(plots_folder, this_savename), bbox_inches="tight") fig.savefig(path_join(plots_folder, this_savename), bbox_inches="tight", facecolor="None", edgecolor="None")
if histograms is not None: if histograms is not None:
filt_obs = {"POL0": 0, "POL60": 0, "POL120": 0} filt_obs = {"POL0": 0, "POL60": 0, "POL120": 0}
@@ -111,7 +111,7 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
this_savename += "_histograms.pdf" this_savename += "_histograms.pdf"
else: else:
this_savename = savename[:-4] + "_histograms" + savename[-4:] this_savename = savename[:-4] + "_histograms" + savename[-4:]
fig_h.savefig(path_join(plots_folder, this_savename), bbox_inches="tight") fig_h.savefig(path_join(plots_folder, this_savename), bbox_inches="tight", facecolor="None", edgecolor="None")
fig2, ax2 = plt.subplots(figsize=(10, 10)) fig2, ax2 = plt.subplots(figsize=(10, 10))
data0 = data[0] * convert_flux[0] data0 = data[0] * convert_flux[0]
@@ -144,7 +144,7 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
this_savename += "_" + filt + "_background_location.pdf" this_savename += "_" + filt + "_background_location.pdf"
else: else:
this_savename = savename[:-4] + "_" + filt + "_background_location" + savename[-4:] this_savename = savename[:-4] + "_" + filt + "_background_location" + savename[-4:]
fig2.savefig(path_join(plots_folder, this_savename), bbox_inches="tight") fig2.savefig(path_join(plots_folder, this_savename), bbox_inches="tight", facecolor="None", edgecolor="None")
if rectangle is not None: if rectangle is not None:
plot_obs( plot_obs(
data, data,

6
package/lib/plots.py
View File

@@ -2913,7 +2913,7 @@ class pol_map(object):
b_tf = Button(ax_tf, r"$F_{\lambda}$") b_tf = Button(ax_tf, r"$F_{\lambda}$")
b_pf = Button(ax_pf, r"$F_{\lambda} \cdot P$") b_pf = Button(ax_pf, r"$F_{\lambda} \cdot P$")
b_p = Button(ax_p, r"$P$") b_p = Button(ax_p, r"$P$")
b_pa = Button(ax_pa, r"$\theta_{P}$") b_pa = Button(ax_pa, r"$\Psi$")
b_snri = Button(ax_snri, r"$I / \sigma_{I}$") b_snri = Button(ax_snri, r"$I / \sigma_{I}$")
b_snrp = Button(ax_snrp, r"$P / \sigma_{P}$") b_snrp = Button(ax_snrp, r"$P / \sigma_{P}$")
@@ -3220,8 +3220,8 @@ class pol_map(object):
ax.set_ylim(0, self.data.shape[0]) ax.set_ylim(0, self.data.shape[0])
plt.rcParams.update({"font.size": 14}) plt.rcParams.update({"font.size": 14})
fig.colorbar(im, ax=ax, aspect=50, shrink=0.75, pad=0.025, label=label) fig.colorbar(im, ax=ax, aspect=50, shrink=0.75, pad=0.025, label=label)
if format_cursor_data is not None: # if format_cursor_data is not None:
self.im.format_cursor_data = format_cursor_data # self.im.format_cursor_data = format_cursor_data
plt.rcParams.update({"font.size": 10}) plt.rcParams.update({"font.size": 10})
fig.canvas.draw_idle() fig.canvas.draw_idle()
return im return im