ZH: View some antennas and their beacons

simulations/airshower_beacon_simulation/view_beaconed_antenna.py

@@ -5,26 +5,51 @@ import matplotlib.pyplot as plt
 import numpy.fft as ft
 
 from earsim import Antenna
+
 import aa_generate_beacon as beacon
+from view_orig_ant0 import plot_antenna_geometry
+import lib
 
 if __name__ == "__main__":
     import os.path as path
 
     fname = "ZH_airshower/mysim.sry"
 
+    plot_ft_amplitude = True
+    plot_geometry = True
+    
     ####
     fname_dir = path.dirname(fname)
     antennas_fname = path.join(fname_dir, beacon.antennas_fname)
 
     f_beacon, tx, antennas = beacon.read_beacon_hdf5(antennas_fname)
 
-    idx = [0, len(antennas)//2, -1]
+    if not True:
-    fig1, axs = plt.subplots(2)
+        idx = [0, 1, len(antennas)//2, len(antennas)//2+1, -2, -1]
+    elif not True:
+        idx = np.arange(1, 20, 2, dtype=int)
+    elif True:
+        # center 6 antennas
+        names = [55, 56, 57, 65, 66, 45, 46]
+
+        idx = [ i for i, ant in enumerate(antennas) if int(ant.name) in names ]
+
+        [ print(antennas[i].name) for i in names ]
+
+    fig1, axs = plt.subplots(1+plot_ft_amplitude*2)
+    if not plot_ft_amplitude:
+        axs = [axs]
     axs[0].set_xlabel('t [ns]')
     axs[0].set_ylabel('[$\mu$V/m]')
+    if plot_ft_amplitude:
         axs[1].set_xlabel('f [GHz]')
         axs[1].set_ylabel('Power')
 
+        axs[2].set_ylabel("Phase")
+        axs[2].set_xlabel('f [GHz]')
+        axs[2].set_ylim(-np.pi,+np.pi)
+
+    colorlist = []
     for i in idx:
         ant = antennas[i]
 
@@ -33,14 +58,48 @@ if __name__ == "__main__":
 
         axs[0].axvline(ant.t[0], color='k', alpha=0.5)
 
-        for direction, trace in dict(x=ant.Ex, y=ant.Ex, z=ant.Ez).items():
+        if not True:
+            mydict = dict(x=ant.Ex, y=ant.Ex, z=ant.Ez)
+        else:
+            mydict = dict(b=ant.beacon)
+
+        for j, (direction, trace) in enumerate(mydict.items()):
+            l = axs[0].plot(ant.t, trace, label=f"E{direction} {ant.name}")
+
+            if j == 0 and 't0' in ant.attrs:
+                axs[0].axvline(ant.attrs['t0'], color=l[0].get_color(), alpha=0.5)
+
+            colorlist.append(l[0].get_color())
+
+            if not plot_ft_amplitude:
+                continue
+
             freqs = ft.fftfreq(n_samples, 1/samplerate)[:n_samples//2]
             fft = 2*ft.fft(trace)[:n_samples//2]/n_samples
         
-            axs[0].plot(ant.t, trace, label=f"E{direction} {i}")
+            #axs[1].plot(freqs, np.abs(fft)**2, color=l[0].get_color())
-            axs[1].plot(freqs, np.abs(fft))
 
+            if True:
+                cft = lib.direct_fourier_transform(f_beacon, ant.t, trace)
+                amp = 2*len(ant.t) * (cft[0]**2 + cft[1]**2)
+
+                #axs[0].axhline(amp, color=l[0].get_color())
+
+                print(amp)
+                phase = np.arctan2(cft[0],cft[1])
+                axs[1].plot(f_beacon, amp, color=l[0].get_color(), marker='3')
+                axs[2].plot(f_beacon, phase, color=l[0].get_color(), marker='3')
+
+    if plot_ft_amplitude:
         fig1.legend(loc='center right', ncol=min(3, len(idx)))
+    else:
+        fig1.legend(loc='upper right', ncol=min(3, len(idx)))
+
+    if plot_geometry:
+        fig2, axs2 = plt.subplots(1)
+        plot_antenna_geometry(antennas, ax=axs2, plot_max_values=False, color='grey', plot_names=False)
+        plot_antenna_geometry([ antennas[i] for i in idx], ax=axs2, colors=colorlist, plot_max_values=False)
+        axs2.plot(tx.x, tx.y, marker='X', color='k')
 
     #fig1.savefig('./fig1.png')
     plt.show()