mirror of
https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git
synced 2024-11-13 18:13:31 +01:00
65 lines
2 KiB
Python
65 lines
2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# vim: fdm=indent ts=4
|
||
|
|
|
||
|
|
import h5py
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
import aa_generate_beacon as beacon
|
||
|
|
import lib
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
from os import path
|
||
|
|
import sys
|
||
|
|
|
||
|
|
fname = "ZH_airshower/mysim.sry"
|
||
|
|
|
||
|
|
####
|
||
|
|
fname_dir = path.dirname(fname)
|
||
|
|
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)
|
||
|
|
|
||
|
|
# Read in antennas from file
|
||
|
|
f_beacon, tx, antennas = beacon.read_beacon_hdf5(antennas_fname)
|
||
|
|
|
||
|
|
if True and 'beacon_phase_true' in antennas[0].attrs:
|
||
|
|
true_phases = np.array([a.attrs['beacon_phase_true'] for a in antennas])
|
||
|
|
else:
|
||
|
|
true_phases = np.empty( (len(antennas)) )
|
||
|
|
|
||
|
|
for i, ant in enumerate(antennas):
|
||
|
|
measured_phase = ant.attrs['beacon_phase_measured']
|
||
|
|
|
||
|
|
geom_time = lib.geometry_time(tx, ant, c_light=3e8*1e-9)
|
||
|
|
geom_phase = geom_time * 2*np.pi*f_beacon
|
||
|
|
|
||
|
|
true_phases[i] = lib.phase_mod(measured_phase) - lib.phase_mod(geom_phase)
|
||
|
|
|
||
|
|
ant.attrs['beacon_phase_true'] = true_phases[i]
|
||
|
|
|
||
|
|
# True phase (without geometry determined)
|
||
|
|
|
||
|
|
if True: # show antenna phases
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
spatial_unit=None
|
||
|
|
fig.suptitle('f= {:2.0f}MHz'.format(f_beacon*1e3))
|
||
|
|
|
||
|
|
antenna_locs = list(zip(*[(ant.x, ant.y) for ant in antennas]))
|
||
|
|
ax.set_xlabel('x' if spatial_unit is None else 'x [{}]'.format(spatial_unit))
|
||
|
|
ax.set_ylabel('y' if spatial_unit is None else 'y [{}]'.format(spatial_unit))
|
||
|
|
scatter_kwargs = {}
|
||
|
|
scatter_kwargs['cmap'] = 'Spectral_r'
|
||
|
|
scatter_kwargs['vmin'] = -np.pi
|
||
|
|
scatter_kwargs['vmax'] = +np.pi
|
||
|
|
color_label='$\\varphi$'
|
||
|
|
|
||
|
|
sc = ax.scatter(*antenna_locs, c=true_phases, **scatter_kwargs)
|
||
|
|
fig.colorbar(sc, ax=ax, label=color_label)
|
||
|
|
|
||
|
|
plt.show()
|
||
|
|
# Report back to CLI
|
||
|
|
#print("Period Multiples resolved in", antenna_fname)
|