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 11:33:32 +01:00
Code Issues Projects Releases Packages Wiki Activity

ZH: restructure reconstruction script

Browse source
This commit is contained in:
Eric Teunis de Boone 2022-11-24 17:37:54 +01:00
parent 8fa0f268db
commit 0405ddb1c7
1 changed files with 47 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
simulations/airshower_beacon_simulation/do_reconstruction.py
View file

@ -24,6 +24,7 @@ if __name__ == "__main__":
#### ####
fname_dir = path.dirname(fname) fname_dir = path.dirname(fname)
max_clock_offset = None
if True: # use the original files if True: # use the original files
ev = REvent(fname) ev = REvent(fname)
@ -33,6 +34,13 @@ if __name__ == "__main__":
rng = np.random.default_rng(seed) rng = np.random.default_rng(seed)
dirname += f'rand{max_clock_offset}ns' dirname += f'rand{max_clock_offset}ns'
else:
raise NotImplementedError
antennas_fname = path.join(fname_dir, beacon.antennas_fname)
if not path.isfile(antennas_fname):
print("Antenna file cannot be found, did you try generating a beacon?")
sys.exit(1)
# Full reconstruction? # Full reconstruction?
if False: if False:
@ -42,36 +50,43 @@ if __name__ == "__main__":
print(f"Randomising timing: {max_clock_offset}") print(f"Randomising timing: {max_clock_offset}")
clock_offsets = max_clock_offset * (2*rng.uniform(size=len(ev.antennas)) - 1) clock_offsets = max_clock_offset * (2*rng.uniform(size=len(ev.antennas)) - 1)
for i, ant in enumerate(ev.antennas): for i, ant in enumerate(ev.antennas):
ant.t -= clock_offsets[i] ev.antennas[i].t += clock_offsets[i]
else: else:
print(f"Not randomising timing") print(f"Not randomising timing")
with open(dirname + '/res.pkl', 'wb') as fp: with open(dirname + '/res.pkl', 'wb') as fp:
res = rit.reconstruction(ev, outfile=dirname+'/fig.pdf', slice_outdir=dirname+'/', Xlow=100, N_X=15, Xhigh=800) res = rit.reconstruction(ev, outfile=dirname+'/fig.pdf', slice_outdir=dirname+'/', Xlow=100, N_X=23, Xhigh=1200)
pickle.dump(res, fp) pickle.dump(res, fp)
sys.exit(0) sys.exit(0)
# Else do single slice for original timing # Only a single slice
# and randomised timing else:
print("Shower plane reconstruction + randomised timing") print("Shower plane reconstruction")
atm = AtmoCal() # atm = AtmoCal()
X = 750 # X = 750
rit.set_pol_and_bp(ev) # rit.set_pol_and_bp(ev)
dXref = atm.distance_to_slant_depth(np.deg2rad(ev.zenith),X,0) # dXref = atm.distance_to_slant_depth(np.deg2rad(ev.zenith),750,0)
scale2d = dXref*np.tan(np.deg2rad(2.)) # scale2d = dXref*np.tan(np.deg2rad(2.))
if max_clock_offset is None:
print(" + randomised timing")
ants_copy = deepcopy(ev.antennas) ants_copy = deepcopy(ev.antennas)
clock_offsets = max_clock_offset * (2*rng.uniform(size=len(ants_copy)) - 1) clock_offsets = max_clock_offset * (2*rng.uniform(size=len(ants_copy)) - 1)
for i, ant in enumerate(ants_copy): for i, ant in enumerate(ants_copy):
ant.t -= clock_offsets[i] ant.t -= clock_offsets[i]
fig, axs = plt.subplots(1,2, sharex=True, sharey=True) fig, axs = plt.subplots(1,1+(max_clock_offset is not None), sharex=True, sharey=True)
fig.suptitle(f"Reconstructed Shower plane slice at X={X}") fig.suptitle(f"Reconstructed Shower plane slice at X={X}")
if max_clock_offset is None:
axs = [ax]
for ax in axs: for ax in axs:
ax.set_xlabel("-v x B (m)") ax.set_xlabel("-v x B (m)")
ax.set_ylabel(" vxvxB (m)") ax.set_ylabel(" vxvxB (m)")
if max_clock_offset is not None:
axs[0].set_title("Simulated timing") axs[0].set_title("Simulated timing")
axs[1].set_title(f"Randomised timing [0, {max_clock_offset}]") axs[1].set_title(f"Randomised timing [0, {max_clock_offset}]")
@ -97,19 +112,11 @@ if __name__ == "__main__":
fig.colorbar(sc, ax=axs[0]) fig.colorbar(sc, ax=axs[0])
# Always make sure to show the plot # Always make sure to show the plot
except: except:
if True:
fig.savefig(dirname + f"/shower_plane_slice_X{X}_T{max_clock_offset}_exception.pdf")
plt.show() plt.show()
if True: if True:
fig.savefig(dirname + f"/shower_plane_slice_X{X}_T{max_clock_offset}.pdf") fig.savefig(dirname + f"/shower_plane_slice_X{X}_T{max_clock_offset}.pdf")
else:
raise NotImplementedError
antennas_fname = path.join(fname_dir, beacon.antennas_fname)
if not path.isfile(antennas_fname):
print("Antenna file cannot be found, did you try generating a beacon?")
sys.exit(1)
plt.show() plt.show()