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

ZH: beacon removal figure saved with zoomed parts

Browse source
This commit is contained in:
Eric Teunis de Boone 2023-01-16 18:40:59 +01:00
parent 1e7c469bb8
commit 7c570c462e
1 changed files with 18 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
simulations/airshower_beacon_simulation/ca_period_from_shower.py
View file

@ -234,13 +234,22 @@ if __name__ == "__main__":
# save # save
if fig_dir: if fig_dir:
fig.tight_layout() fig.tight_layout()
if True: # zoom if True: # zoom
old_xlim = ax.get_xlim() old_xlim = ax.get_xlim()
if True: # zoomed on part without peak of this trace
wx, x = 100, 0#ant.t_AxB[np.argmax(ant.E_AxB)] wx, x = 100, 0#ant.t_AxB[np.argmax(ant.E_AxB)]
ax.set_xlim(x-wx, x+wx) ax.set_xlim(x-wx, x+wx)
fig.savefig(path.join(fig_dir, __file__+f'.traces.zoomed.A{i}.pdf')) fig.savefig(path.join(fig_dir, __file__+f'.traces.A{i}.zoomed.beacon.pdf'))
if True: # zoomed on peak of this trace
idx = np.argmax(ev.antennas[i].E_AxB)
x = ev.antennas[i].t_AxB[idx]
wx = 100
ax.set_xlim(x-wx, x+wx)
fig.savefig(path.join(fig_dir, __file__+f".traces.A{i}.zoomed.peak.pdf"))
ax.set_xlim(*old_xlim) ax.set_xlim(*old_xlim)
@ -263,7 +272,7 @@ if __name__ == "__main__":
scale2d = dXref*np.tan(np.deg2rad(2.)) scale2d = dXref*np.tan(np.deg2rad(2.))
scale4d = dXref*np.tan(np.deg2rad(4.)) scale4d = dXref*np.tan(np.deg2rad(4.))
if not True: #quicky if False: #quicky
x_coarse = np.linspace(-scale2d, scale2d, 4) x_coarse = np.linspace(-scale2d, scale2d, 4)
y_coarse = np.linspace(-scale2d, scale2d, 4) y_coarse = np.linspace(-scale2d, scale2d, 4)
@ -289,15 +298,15 @@ if __name__ == "__main__":
for r in range(N_runs): for r in range(N_runs):
# Setup Plane grid to test # Setup Plane grid to test
xoff, yoff = 0,0
if r == 0: if r == 0:
xoff, yoff = 0,0
x = x_coarse x = x_coarse
y = y_coarse y = y_coarse
else: else:
# zooming in # zooming in
# best_idx is defined at the end of the loop
old_ks_per_loc = ks_per_loc[best_idx] old_ks_per_loc = ks_per_loc[best_idx]
xoff = xx[best_idx] xoff, yoff = locs[best_idx]
yoff = yy[best_idx]
if r == 1: if r == 1:
x = x_fine x = x_fine
y = y_fine y = y_fine
@ -305,7 +314,7 @@ if __name__ == "__main__":
x /= 4 x /= 4
y /= 4 y /= 4
print(f"Testing grid run {r} centered on {xoff}, {yoff}") print(f"Testing grid run {r} centered on ({xoff}, {yoff})")
ks_per_loc = np.zeros( (len(x)*len(y), len(ev.antennas)) , dtype=int) ks_per_loc = np.zeros( (len(x)*len(y), len(ev.antennas)) , dtype=int)
maxima_per_loc = np.zeros( (len(x)*len(y))) maxima_per_loc = np.zeros( (len(x)*len(y)))
@ -343,7 +352,7 @@ if __name__ == "__main__":
if True: #plot maximum at test locations if True: #plot maximum at test locations
fig, axs = plt.subplots() fig, axs = plt.subplots()
axs.set_title(f"Optimizing signal strength varying k per antenna,\n Grid Run {r}") axs.set_title(f"Optimizing signal strength by varying $k$ per antenna,\n Grid Run {r}")
axs.set_ylabel("vxvxB [km]") axs.set_ylabel("vxvxB [km]")
axs.set_xlabel(" vxB [km]") axs.set_xlabel(" vxB [km]")
axs.set_aspect('equal', 'datalim') axs.set_aspect('equal', 'datalim')