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: ba measure phase: mask strong peak for every trace

Browse source
This commit is contained in:
Eric Teunis de Boone 2023-02-03 11:56:11 +01:00
parent 8eb2431ec6
commit 583577b807
1 changed files with 55 additions and 43 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

98
simulations/airshower_beacon_simulation/ba_measure_beacon_phase.py
View file

@ -23,21 +23,28 @@ if __name__ == "__main__":
from scriptlib import MyArgumentParser from scriptlib import MyArgumentParser
parser = MyArgumentParser() parser = MyArgumentParser()
parser.add_argument('--no-use-AxB', dest='use_AxB_trace', action='store_false')
group1 = parser.add_mutually_exclusive_group()
group1.add_argument('--AxB', dest='use_AxB_trace', action='store_true', help='Only use AxB trace, if both AxB and beacon are not used, we use the antenna polarisations.')
group1.add_argument('--beacon', dest='use_beacon_trace', action='store_true', help='Only use the beacon trace')
parser.add_argument('--N-mask', type=float, default=500, help='Mask N_MASK samples around the absolute maximum of the trace. (Default: %(default)d)')
args = parser.parse_args() args = parser.parse_args()
f_beacon_band = (49e-3,55e-3) #GHz f_beacon_band = (49e-3,55e-3) #GHz
allow_frequency_fitting = False allow_frequency_fitting = False
read_frequency_from_file = True read_frequency_from_file = True
N_mask = int(args.N_mask)
use_AxB_trace = args.use_AxB_trace use_only_AxB_trace = args.use_AxB_trace
use_beacon_trace = True # only applicable if AxB = False use_only_beacon_trace = args.use_beacon_trace # only applicable if AxB = False
show_plots = args.show_plots show_plots = args.show_plots
figsize = (12,8) figsize = (12,8)
print("use_AxB_trace:", use_AxB_trace, "use_beacon_trace:",use_beacon_trace) print("use_only_AxB_trace:", use_only_AxB_trace, "use_only_beacon_trace:", use_only_beacon_trace)
#### ####
fname_dir = args.data_dir fname_dir = args.data_dir
@ -83,59 +90,64 @@ if __name__ == "__main__":
h5ant = group[name] h5ant = group[name]
# use E_AxB only instead of polarisations # use E_AxB only instead of polarisations
if use_AxB_trace: if use_only_AxB_trace:
if 'E_AxB' not in h5ant.keys(): traces_key = 'E_AxB'
print(f"Antenna does not have 'E_AxB' in {name}") if traces_key not in h5ant.keys():
print(f"Antenna does not have '{traces_key}' in {name}")
sys.exit(1) sys.exit(1)
traces = h5ant['E_AxB'] traces = h5ant[traces_key]
t_trace = traces[0] t_trace = traces[0]
test_traces = [ traces[1] ] test_traces = [ traces[1] ]
orients = ['E_AxB'] orients = ['E_AxB']
# TODO: refine masking # Only beacon
# use beacon but remove where E_AxB-Beacon != 0 elif use_only_beacon_trace:
if True: traces_key = 'filtered_traces'
N_pre, N_post = 250, 250 # TODO: make this configurable if traces_key not in h5ant.keys():
print(f"Antenna file corrupted? no '{traces_key}' in {name}")
sys.exit(1)
max_idx = np.argmax(test_traces[0]) traces = h5ant[traces_key]
t_trace = traces[0]
low_idx = max(0, max_idx-N_pre) test_traces = [traces[4]]
high_idx = min(len(t_trace), max_idx+N_post) orients = ['B']
t_mask = np.ones(len(t_trace), dtype=bool)
t_mask[low_idx:high_idx] = False
t_trace = t_trace[t_mask]
for j, t in enumerate(test_traces):
test_traces[j] = t[t_mask]
orients[j] = orients[j] + ' masked'
# use separate polarisations # use separate polarisations
else: else:
if 'traces' not in h5ant.keys(): traces_key = 'filtered_traces'
print(f"Antenna file corrupted? no 'traces' in {name}") if traces_key not in h5ant.keys():
print(f"Antenna file corrupted? no '{traces_key}' in {name}")
sys.exit(1) sys.exit(1)
traces = h5ant['traces'] traces = h5ant[traces_key]
t_trace = traces[0] t_trace = traces[0]
test_traces = [traces[i] for i in range(1,4)]
orients = ['Ex', 'Ey', 'Ez']
if use_beacon_trace: # Really only select the first component
# only take the Beacon trace
test_traces = [traces[4]]
orients = ['B']
else:
test_traces = traces[1:]
orients = ['Ex', 'Ey', 'Ez', 'B']
# modify the length of the traces
if False: if False:
t_trace = t_trace[:len(t_trace)//2] test_traces = [test_traces[0]]
half_traces = [] orients = [orients[0]]
for trace in test_traces:
half_traces.append( trace[:len(trace)//2]) # TODO: refine masking
test_traces = half_traces # use beacon but remove where E_AxB-Beacon != 0
# Uses the first traces as reference
if N_mask and orients[0] != 'B':
N_pre, N_post = N_mask//2, N_mask//2
max_idx = np.argmax(test_traces[0])
low_idx = max(0, max_idx-N_pre)
high_idx = min(len(t_trace), max_idx+N_post)
t_mask = np.ones(len(t_trace), dtype=bool)
t_mask[low_idx:high_idx] = False
t_trace = t_trace[t_mask]
for j, t in enumerate(test_traces):
test_traces[j] = t[t_mask]
orients[j] = orients[j] + ' masked'
# Do Fourier Transforms # Do Fourier Transforms
# to find phases and amplitudes # to find phases and amplitudes
@ -155,7 +167,7 @@ if __name__ == "__main__":
amps = [ 3e-7 ] amps = [ 3e-7 ]
# choose highest amp # choose highest amp
idx = 0 idx = np.argmax(amps)
if False and len(beacon_phases) > 1: if False and len(beacon_phases) > 1:
#idx = np.argmax(amplitudes, axis=-1) #idx = np.argmax(amplitudes, axis=-1)
raise NotImplementedError raise NotImplementedError