def plot_hanle_fits_for_power_dependence(
data, envir, curve_loader, fit_plot_filename, db_conn, constrain_background=False
):
exp_dir = exp_dir_from_env(envir)
fit_plots_dir = os.path.join(exp_dir, "fit_plots")
if not os.path.exists(fit_plots_dir):
os.makedirs(fit_plots_dir)
for s, w, pup in zip(data.Timestamp.values, data.W.values, data.PuP):
curve = curve_loader(s)
result = hanle_fit_for_data(curve, db_conn)
if result is not None:
plot_hanle_curve(curve, result)
# title(hanle_curve_title_from_stored(curve_id))
plt.savefig(os.path.join(fit_plots_dir, fit_plot_filename(s, w)))
plt.close()
def plot_hanle_fits_for_energy_dependence(
data, envir, curve_loader, fit_plot_filename, db_conn, constrain_background=False, **kwargs
):
exp_dir = exp_dir_from_env(envir)
fit_plots_dir = os.path.join(exp_dir, "fit_plots")
if not os.path.exists(fit_plots_dir):
os.makedirs(fit_plots_dir)
for s, w, pp, z, bg in zip(data.Timestamp.values, data.W.values, data.PP, data.Z.values, data.BG.values):
curve = curve_loader(s)
result = hanle_fit_for_data(curve, db_conn)
if result is not None:
plot_hanle_curve(curve, result, **kwargs)
# title(hanle_curve_title_from_stored(curve_id))
if not constrain_background:
plt.savefig(os.path.join(fit_plots_dir, fit_plot_filename(s, w)))
else:
if not os.path.exists(os.path.join(fit_plots_dir, "constrained_bg")):
os.path.makedirs(os.path.join(fit_plots_dir, "constrained_bg"))
plt.savefig(os.path.exists(os.path.join(fit_plots_dir, "constrained_bg", "%04d.pdf" % s)))
plt.close()