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

ZH: improve cc reconstruction script from ffb1fa8

Browse source
This commit is contained in:
Eric Teunis de Boone 2023-01-11 02:59:04 +01:00
parent 32f7445fc4
commit 056ead83e3
1 changed files with 22 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
simulations/airshower_beacon_simulation/cc_power_along_axis_reconstruction.py
View file

@ -7,8 +7,10 @@ clock offsets.
""" """
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D # required for projection='3d' on old matplotliblib versions
import numpy as np import numpy as np
from os import path from os import path
import pickle
from earsim import REvent from earsim import REvent
from atmocal import AtmoCal from atmocal import AtmoCal
@ -31,12 +33,19 @@ if __name__ == "__main__":
fig_dir = "./figures/" fig_dir = "./figures/"
fig_subdir = path.join(fig_dir, 'reconstruction') fig_subdir = path.join(fig_dir, 'reconstruction')
show_plots = False show_plots = False
pickle_fname = path.join(fig_subdir, 'res.pkl')
#### ####
fname_dir = path.dirname(fname) fname_dir = path.dirname(fname)
antennas_fname = path.join(fname_dir, beacon.antennas_fname) antennas_fname = path.join(fname_dir, beacon.antennas_fname)
time_diffs_fname = 'time_diffs.hdf5' if not True else antennas_fname time_diffs_fname = 'time_diffs.hdf5' if not True else antennas_fname
# create fig_dir
if fig_dir:
os.makedirs(fig_dir, exist_ok=True)
if fig_subdir:
os.makedirs(fig_subdir, exist_ok=True)
# Read in antennas from file # Read in antennas from file
_, tx, antennas = beacon.read_beacon_hdf5(antennas_fname) _, tx, antennas = beacon.read_beacon_hdf5(antennas_fname)
# Read original REvent # Read original REvent
@ -44,6 +53,14 @@ if __name__ == "__main__":
# .. patch in our antennas # .. patch in our antennas
ev.antennas = antennas ev.antennas = antennas
# For now only implement using one freq_name
freq_names = antennas[0].beacon_info.keys()
if len(freq_names) > 1:
raise NotImplementedError
freq_name = next(iter(freq_names))
f_beacon = ev.antennas[0].beacon_info[freq_name]['freq']
# Repair clock offsets with the measured offsets # Repair clock offsets with the measured offsets
for i, ant in enumerate(ev.antennas): for i, ant in enumerate(ev.antennas):
clock_phase_time = ant.beacon_info[freq_name]['sigma_phase_mean']/(2*np.pi*f_beacon) clock_phase_time = ant.beacon_info[freq_name]['sigma_phase_mean']/(2*np.pi*f_beacon)
@ -52,12 +69,12 @@ if __name__ == "__main__":
total_clock_time = best_k_time + clock_phase_time total_clock_time = best_k_time + clock_phase_time
ev.antennas[i].t += total_clock_time ev.antennas[i].t += total_clock_time
## Start a pickle N_X, Xlow, Xhigh = 23, 100, 1200
with open(pickle_fname, 'wb') as fp: res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir+'/', Xlow=Xlow, N_X=N_X, Xhigh=Xhigh)
N_X, Xlow, Xhigh = 23, 100, 1200
res = rit.reconstruction(ev, outfile=fig_subdir+'/fig.pdf', slice_outdir=fig_subdir, Xlow=Xlow, N_X=N_X, Xhigh=Xhigh)
## Save a pickle
with open(pickle_fname, 'wb') as fp:
pickle.dump(res,fp) pickle.dump(res,fp)
if show_plots(): if show_plots:
plt.show() plt.show()