ericteunis/m-thesis-introduction
1
0
Fork 0
mirror of https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git synced 2024-11-14 02:23:32 +01:00
Code Issues Projects Releases Packages Wiki Activity

ZH: improve temporary correlation figures

Browse source
This commit is contained in:
Eric Teunis de Boone 2022-12-23 12:27:25 +01:00
parent 6eba992a3a
commit 0d8bce9953
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
simulations/airshower_beacon_simulation/ca_period_from_shower.py
View file

@ -93,18 +93,23 @@ def find_best_sample_shifts_summing_at_location(test_loc, antennas, allowed_samp
shift_maxima[j] = np.max(augmented_a + a_sum) shift_maxima[j] = np.max(augmented_a + a_sum)
if i in plot_iteration_with_shifted_trace: if i in plot_iteration_with_shifted_trace:
ax.plot(t_sum, augmented_a, alpha=0.8, label=f'{shift}') ax.plot(t_sum, augmented_a, alpha=0.7, ls='dashed', label=f'{shift}')
# transform maximum into best_sample_shift # transform maximum into best_sample_shift
best_idx = np.argmax(shift_maxima) best_idx = np.argmax(shift_maxima)
best_sample_shifts[i] = allowed_sample_shifts[best_idx] best_sample_shifts[i] = allowed_sample_shifts[best_idx]
a_sum += np.roll(a_int, best_sample_shifts[i]) best_augmented_a = np.roll(a_int, best_sample_shifts[i])
a_sum += best_augmented_a
# cleanup figure # cleanup figure
if i in plot_iteration_with_shifted_trace: if i in plot_iteration_with_shifted_trace:
if True: # plot best k again
ax.plot(t_sum, augmented_a, alpha=0.8, label=f'best k={best_sample_shifts[i]}', lw=2)
ax.legend( ncol=5 ) ax.legend( ncol=5 )
if fig_dir: if fig_dir:
fig.tight_layout()
fname = path.join(fig_dir, __file__ + f'.{fig_distinguish}i{i}' + '.loc{:.1f}-{:.1f}-{:.1f}'.format(*test_loc)) fname = path.join(fig_dir, __file__ + f'.{fig_distinguish}i{i}' + '.loc{:.1f}-{:.1f}-{:.1f}'.format(*test_loc))
if True: if True:
old_xlim = ax.get_xlim() old_xlim = ax.get_xlim()
@ -117,7 +122,7 @@ def find_best_sample_shifts_summing_at_location(test_loc, antennas, allowed_samp
if True: # zoomed on peak of this trace if True: # zoomed on peak of this trace
x = t_r[np.argmax(E_)] x = t_r[np.argmax(E_)]
wx = 10 + max(best_sample_shifts) - min(best_sample_shifts) wx = 50 + (max(best_sample_shifts) - min(best_sample_shifts))*dt
ax.set_xlim(x-wx, x+wx) ax.set_xlim(x-wx, x+wx)
fig.savefig(fname + ".zoomed.peak.pdf") fig.savefig(fname + ".zoomed.peak.pdf")