From 9b7aa02e78a85d275ba5f44e429c33b44d64ac2b Mon Sep 17 00:00:00 2001
From: Eric Teunis de Boone <ericteunis@deboone.nl>
Date: Fri, 2 Dec 2022 19:07:46 +0100
Subject: [PATCH] ZH: new script to check phase residuals

---
 .../bc_phase_sigmas.py                        | 120 ++++++++++++++++++
 1 file changed, 120 insertions(+)

diff --git a/simulations/airshower_beacon_simulation/bc_phase_sigmas.py b/simulations/airshower_beacon_simulation/bc_phase_sigmas.py
index e69de29..b69bc9c 100755
--- a/simulations/airshower_beacon_simulation/bc_phase_sigmas.py
+++ b/simulations/airshower_beacon_simulation/bc_phase_sigmas.py
@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+# vim: fdm=indent ts=4
+
+import h5py
+from itertools import combinations, zip_longest
+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"
+    c_light = 3e8*1e-9
+    show_plots = not True
+    ref_ant_id = None # leave None for all baselines
+
+    ####
+    fname_dir = path.dirname(fname)
+    antennas_fname = path.join(fname_dir, beacon.antennas_fname)
+    time_diffs_fname = 'time_diffs.hdf5' if not True else 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)
+
+    # run over all baselines
+    if ref_ant_id is None:
+        print("Doing all baselines")
+        baselines = list(combinations(antennas,2))
+    # use ref_ant
+    else:
+        ref_ant = antennas[ref_ant_id]
+        print(f"Doing all baselines with {ref_ant.name}")
+        baselines = list(zip_longest([], antennas, fillvalue=ref_ant))
+
+    # For now, only one beacon_frequency is supported
+    freq_names = antennas[0].beacon_info.keys()
+    if len(freq_names) > 1:
+        raise NotImplementedError
+
+    freq_name = next(iter(freq_names))
+
+    # Get phase difference per baseline
+    phase_diffs = np.empty( (len(baselines), 2) )
+    for i, base in enumerate(baselines):
+        if i%50==0:
+            print(i, "out of", len(baselines))
+
+        # read f_beacon from the first antenna
+        f_beacon = base[0].beacon_info[freq_name]['freq']
+
+        # Get true phase diffs
+        try:
+
+            true_phases = np.array([ant.beacon_info[freq_name]['phase'] for ant in base])
+            true_phases_diff = lib.phase_mod(lib.phase_mod(true_phases[1]) - lib.phase_mod(true_phases[0]))
+        except IndexError:
+            # true_phase not determined yet
+            print(f"Missing true_phases for {freq_name} in baseline {base[0].name},{base[1].name}")
+            true_phases_diff = np.nan
+
+        # save phase difference with antenna names
+        phase_diffs[i] = [f_beacon, true_phases_diff]
+
+    # Read actual phases from antenna hdf5
+    actual_antenna_phases = { a.name: 2*np.pi*a.attrs['clock_offset']*f_beacon for a in antennas }
+
+
+    # Compare actual time shifts
+    my_phase_diffs = []
+    for i,b in enumerate(baselines):
+        actual_phase_diff = lib.phase_mod( lib.phase_mod(actual_antenna_phases[b[1].name]) - lib.phase_mod(actual_antenna_phases[b[0].name]))
+
+        if True: # remove time axis phase difference
+            t0s = np.array([ant.t_AxB[0] for ant in base])
+            delta_t0 = t0s[1] - t0s[1]
+            actual_phase_diff = lib.phase_mod( actual_phase_diff - lib.phase_mod(delta_t0*2*np.pi*f_beacon))
+        my_phase_diffs.append(actual_phase_diff)
+
+    # Make a plot
+    if True:
+        N_base = len(baselines)
+        N_ant = len(antennas)
+
+        phase_residuals = lib.phase_mod(phase_diffs[:,1] - my_phase_diffs)
+
+        fig, axs = plt.subplots(2, 1, sharex=True)
+        axs[0].set_title("Measured phase difference - Actual phase difference")
+        axs[0].set_xlabel("Phase residuals $\\varphi$")
+
+        if True:
+            forward = lambda x: x/(2*np.pi*f_beacon)
+            inverse = lambda x: 2*np.pi*x*f_beacon
+            secax = axs[0].secondary_xaxis('top', functions=(forward, inverse))
+            secax.set_xlabel('$\\varphi/(2\\pi f_{beacon})$ [ns]')
+
+        i=0
+        axs[i].set_ylabel("#")
+        axs[i].hist(phase_residuals, bins='sqrt', density=False)
+
+        i=1
+        axs[i].set_ylabel("Baseline combination #")
+        if not True:
+            axs[i].plot(my_phase_diffs, np.arange(N_base), ls='none', marker='+', label='actual time shifts')
+            l = axs[i].plot(phase_diffs[:,1], np.arange(N_base), ls='none', marker='x', label='calculated')
+
+            axs[i].legend()
+
+        else:
+            axs[i].plot(phase_residuals, np.arange(N_base), ls='none', marker='x')
+
+        plt.show()