From 80cc240f23f4b63873bb5947d755e0222e69fd89 Mon Sep 17 00:00:00 2001 From: Eric Teunis de Boone Date: Fri, 28 Apr 2023 16:11:08 +0200 Subject: [PATCH] Pulse: move thresholding plot code around --- simulations/11_pulsed_timing.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/simulations/11_pulsed_timing.py b/simulations/11_pulsed_timing.py index 7c31b82..c55bfcf 100755 --- a/simulations/11_pulsed_timing.py +++ b/simulations/11_pulsed_timing.py @@ -265,7 +265,7 @@ def get_time_residuals_for_template( plt.close(fig) axs2 = None - if True: # upsampled trace + if True: # simple and dumb trace upsampling upsampled_trace, upsampled_t = trace_upsampler(antenna.signal, template.t, antenna.t) if do_plots: # Show upsampled traces fig2, axs2 = plt.subplots(1, sharex=True) @@ -423,6 +423,7 @@ if __name__ == "__main__": normalise_noise = False h5_cache_fname = f'11_pulsed_timing.hdf5' + use_cache = True # # Interpolation Template @@ -474,7 +475,7 @@ if __name__ == "__main__": N_residuals, template, interpolation_template=interp_template, antenna_dt=antenna_dt, antenna_timelength=antenna_timelength, snr_sigma_factor=snr_sigma_factor, bp_freq=bp_freq, normalise_noise=normalise_noise, - h5_cache_fname=h5_cache_fname, rng=rng, tqdm=tqdm) + h5_cache_fname=h5_cache_fname, rng=rng, tqdm=tqdm, read_cache=use_cache) print()# separating tqdm print()# separating tqdm @@ -565,6 +566,9 @@ if __name__ == "__main__": # SNR time accuracy plot if True: + threshold_markers = ['^', 'v', '8', 'o'] + mask_thresholds = [np.inf, N_residuals*0.5, N_residuals*0.1, 1, 0] + fig, ax = plt.subplots() ax.set_title(f"Template matching SNR vs time accuracy") ax.set_xlabel("Signal to Noise Factor") @@ -586,10 +590,10 @@ if __name__ == "__main__": for k, template_dt in enumerate(template_dts): # indicate masking values - for j, mask_threshold in enumerate(pairwise([np.inf, 250, 50, 1, 0])): + for j, mask_threshold in enumerate(pairwise(mask_thresholds)): kwargs = dict( ls='none', - marker=['^', 'v','8', 'o',][j], + marker=threshold_markers[j], color= None if template_dt_colors[k] is None else template_dt_colors[k] ) mask = mask_counts[k] >= mask_threshold[1] @@ -613,6 +617,9 @@ if __name__ == "__main__": ax.set_ylim([None, 1e1]) fig.tight_layout() - fig.savefig(f"figures/11_time_res_vs_snr_tdt{template_dt:0.1e}.pdf") + if len(template_dts) == 1: + fig.savefig(f"figures/11_time_res_vs_snr_tdt{template_dt:0.1e}.pdf") + else: + fig.savefig(f"figures/11_time_res_vs_snr.pdf") plt.show()