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: precalculate E_AxB (bandpassed) for hdf5 file

Browse source
This commit is contained in:
Eric Teunis de Boone 2022-11-22 09:40:58 +01:00
parent 22a54da1c8
commit b91971368d
1 changed files with 32 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
simulations/airshower_beacon_simulation/aa_generate_beacon.py
View file

@ -12,7 +12,7 @@ import h5py
import os.path as path import os.path as path
from copy import deepcopy as copy from copy import deepcopy as copy
from earsim import REvent, Antenna from earsim import REvent, Antenna, block_filter
import lib import lib
tx_fname = 'tx.json' tx_fname = 'tx.json'
@ -42,7 +42,7 @@ def read_tx_file(fname):
return tx, f_beacon return tx, f_beacon
def read_beacon_hdf5(fname, traces_key='traces', raise_exception=True): def read_beacon_hdf5(fname, traces_key='traces', raise_exception=True, read_AxB=True):
with h5py.File(fname, 'r') as h5: with h5py.File(fname, 'r') as h5:
tx_attrs = h5['tx'].attrs tx_attrs = h5['tx'].attrs
f_beacon = tx_attrs.get('f_beacon') f_beacon = tx_attrs.get('f_beacon')
@ -51,25 +51,29 @@ def read_beacon_hdf5(fname, traces_key='traces', raise_exception=True):
tx = Antenna(**mydict) tx = Antenna(**mydict)
antennas = [] antennas = []
for k, ant in h5['antennas'].items(): for k, h5ant in h5['antennas'].items():
mydict = { k:ant.attrs.get(k) for k in ['x', 'y', 'z', 'name'] } mydict = { k:h5ant.attrs.get(k) for k in ['x', 'y', 'z', 'name'] }
antenna = Antenna(**mydict) ant = Antenna(**mydict)
if traces_key not in ant: if traces_key not in h5ant:
if raise_exception: if raise_exception:
raise ValueError("Traces_key not in file") raise ValueError("Traces_key not in file")
else: else:
antenna.t = ant[traces_key][0] ant.t = h5ant[traces_key][0]
antenna.Ex = ant[traces_key][1] ant.Ex = h5ant[traces_key][1]
antenna.Ey = ant[traces_key][2] ant.Ey = h5ant[traces_key][2]
antenna.Ez = ant[traces_key][3] ant.Ez = h5ant[traces_key][3]
if len(ant[traces_key]) > 4: if len(h5ant[traces_key]) > 4:
antenna.beacon = ant[traces_key][4] ant.beacon = h5ant[traces_key][4]
if ant.attrs: if read_AxB and 'E_AxB' in h5ant:
antenna.attrs = {**ant.attrs} ant.t_AxB = h5ant['E_AxB'][0]
ant.E_AxB = h5ant['E_AxB'][1]
antennas.append(antenna) if h5ant.attrs:
ant.attrs = {**h5ant.attrs}
antennas.append(ant)
return f_beacon, tx, antennas return f_beacon, tx, antennas
@ -141,6 +145,10 @@ if __name__ == "__main__":
tx = Antenna(x=-500,y=0,z=0,name='tx') # m tx = Antenna(x=-500,y=0,z=0,name='tx') # m
f_beacon = 51.53e-3 # GHz f_beacon = 51.53e-3 # GHz
# Bandpass for E field blockfilter
low_bp = 0.03 # GHz
high_bp = 0.08 # GHz
beacon_amplitudes = 1e-6*np.array([1e2, 0, 0]) # mu V/m beacon_amplitudes = 1e-6*np.array([1e2, 0, 0]) # mu V/m
beacon_radiate_rsq = True beacon_radiate_rsq = True
@ -177,12 +185,19 @@ if __name__ == "__main__":
beacon = lib.beacon_from(tx, antenna, f_beacon, antenna.t, t0=t0, c_light=np.inf, radiate_rsq=beacon_radiate_rsq) beacon = lib.beacon_from(tx, antenna, f_beacon, antenna.t, t0=t0, c_light=np.inf, radiate_rsq=beacon_radiate_rsq)
E = np.array([antenna.Ex, antenna.Ey, antenna.Ez, beacon]) E = np.array([antenna.Ex, antenna.Ey, antenna.Ez, beacon])
append_antenna_hdf5( antennas_fname, antenna, E, name='orig_traces', prepend_time=True)
# add to relevant polarisation # add to relevant polarisation
# and apply block filter
dt = antenna.t[1] - antenna.t[0]
for j, _ in enumerate(beacon_amplitudes): for j, _ in enumerate(beacon_amplitudes):
E[j] += beacon_amplitudes[j]*beacon E[j] += block_filter(beacon_amplitudes[j]*beacon, dt, low_bp, high_bp)
append_antenna_hdf5( antennas_fname, antenna, E, name='traces', prepend_time=True, attrs_dict=dict(t0=t0)) append_antenna_hdf5( antennas_fname, antenna, E, name='traces', prepend_time=True, attrs_dict=dict(t0=t0))
# Save E field in E_AxB
E = [np.dot(ev.uAxB,[ex,ey,ez]) for ex,ey,ez in zip(E[0], E[1], E[2])]
append_antenna_hdf5( antennas_fname, antenna, [E], name='E_AxB', prepend_time=True)
print("Antenna HDF5 file written as " + str(antennas_fname)) print("Antenna HDF5 file written as " + str(antennas_fname))