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: repair cases: on axis overlapping traces

Browse source
This commit is contained in:
Eric Teunis de Boone 2023-05-23 14:03:06 +02:00
parent 5243f09edc
commit c08e2ae73c
1 changed files with 26 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
airshower_beacon_simulation/dc_grid_power_time_fixes.py
View file

@ -18,17 +18,21 @@ import aa_generate_beacon as beacon
import lib
from lib import rit
def save_overlapping_traces_figure(test_location, ev, N_plot = 30, wx=200, title_extra=None, fname_distinguish='', fig_dir=None, **fig_kwargs):
def save_overlapping_traces_figure(test_location, ev, N_plot = 30, wx=200, title_extra=None, fname_distinguish='', fig_dir=None, location_in_plot_text_loc=(0.02, 0.95), location_text=None, **fig_kwargs):
P, t_, a_, a_sum, t_sum = rit.pow_and_time(test_location, ev, dt=1)
fig, axs = plt.subplots(**fig_kwargs)
axs.set_title("Antenna traces" + (("\n" + title_extra) if title_extra is not None else '') )
axs.set_xlabel("Time [ns]")
axs.set_ylabel("Amplitude [$\\mu V/m$]")
if True:
text_loc = (0.02, 0.95)
shower_plane_location = rit.location_to_shower_plane(test_location, ev=ev)
axs.text(*text_loc, '[' + ', '.join(['{:.1g}'.format(x) for x in shower_plane_location]) + ']', ha='left', transform=axs.transAxes)
if location_in_plot_text_loc:
text_loc = location_in_plot_text_loc
if not location_text:
shower_plane_location = rit.location_to_shower_plane(test_location, ev=ev)
location_text = '(' + ', '.join(['{:.1g}'.format(x) for x in shower_plane_location]) + ')'
axs.text(*text_loc, location_text, ha='left', transform=axs.transAxes)
a_max = [ np.amax(ant.E_AxB) for ant in ev.antennas ]
power_sort_idx = np.argsort(a_max)
@ -161,10 +165,10 @@ if __name__ == "__main__":
trace_zoom_wx = 100
plot_titling = {
'no_offset': "no clock offset",
'repair_none': "unrepaired clock offset",
'repair_phases': "phase resolved clock offsets repaired",
'repair_full': "final measured clock offsets repaired"
'no_offset': "no clock offsets",
'repair_none': "randomised clock offsets",
'repair_phases': "phase clock offsets",
'repair_full': "phase + period clock offsets"
}
# For now only implement using one freq_name
@ -253,6 +257,7 @@ if __name__ == "__main__":
transl_mode = transl_modes[case]
measured_offsets = beacon.read_antenna_clock_repair_offsets(antennas, mode=transl_mode, freq_name=freq_name)
else:
print(f"Warning: unknown repair case requested '{case}', defaulting to none.")
measured_offsets = [0]*len(ev.antennas)
for i, ant in enumerate(ev.antennas):
@ -269,7 +274,18 @@ if __name__ == "__main__":
#
# Plot overlapping traces at 0,0,0
#
fig = save_overlapping_traces_figure([0,0,0], ev, N_plot=N_plot, wx=trace_zoom_wx, title_extra = plot_titling[case], fname_distinguish=f'{case}.0', fig_dir=fig_dir, figsize=figsize)
loc = (0,0,0)
fig = save_overlapping_traces_figure(loc, ev, N_plot=N_plot, wx=trace_zoom_wx, title_extra = plot_titling[case], fname_distinguish=f'{case}.0', fig_dir=fig_dir, figsize=figsize)
plt.close(fig)
#
# Plot overlapping traces at simulation shower axis
#
dX = atm.distance_to_slant_depth(np.deg2rad(ev.zenith), X, zgr)
loc = (0)*ev.uAxB + (0)*ev.uAxAxB + dX * ev.uA
loc[0:1] = 0
fig = save_overlapping_traces_figure(loc, ev, N_plot=N_plot, wx=trace_zoom_wx, title_extra = plot_titling[case], fname_distinguish=f'{case}.axis', location_text=f"on simulation axis, $X={X}$", fig_dir=fig_dir, figsize=figsize)
plt.close(fig)
# Measure power on grid