ZH: configurable max_k in ca_period_from_shower

This commit is contained in:
Eric Teunis de Boone 2023-01-24 15:21:47 +01:00
parent c660cb9c1a
commit f3531402c5

View file

@ -151,6 +151,7 @@ if __name__ == "__main__":
from scriptlib import MyArgumentParser from scriptlib import MyArgumentParser
parser = MyArgumentParser(default_fig_dir="./figures/periods_from_shower_figures/") parser = MyArgumentParser(default_fig_dir="./figures/periods_from_shower_figures/")
parser.add_argument('--max-k', type=float, default=2, help='Maximum abs(k) allowed to be shifted. (Default: %(default)d)')
args = parser.parse_args() args = parser.parse_args()
fname = "ZH_airshower/mysim.sry" fname = "ZH_airshower/mysim.sry"
@ -159,7 +160,8 @@ if __name__ == "__main__":
fig_subdir = path.join(fig_dir, 'shifts/') fig_subdir = path.join(fig_dir, 'shifts/')
show_plots = args.show_plots show_plots = args.show_plots
allowed_ks = [ -2, -1, 0, 1, 2] max_k = int(args.max_k)
allowed_ks = np.arange(-max_k, max_k+1, dtype=int)
Xref = 400 Xref = 400
N_runs = 3 N_runs = 3
@ -465,7 +467,7 @@ if __name__ == "__main__":
h5attrs = h5beacon_info[freq_name].attrs h5attrs = h5beacon_info[freq_name].attrs
h5attrs['best_k'] = old_ks_per_loc[i] h5attrs['best_k'] = old_ks_per_loc[i]
h5attrs['best_k_time'] = old_ks_per_loc[i]*dt/f_beacon h5attrs['best_k_time'] = old_ks_per_loc[i]/f_beacon
if show_plots: if show_plots:
plt.show() plt.show()