ZH: restructure reconstruction script

simulations/airshower_beacon_simulation/do_reconstruction.py

@@ -24,6 +24,7 @@ if __name__ == "__main__":
 
     ####
     fname_dir = path.dirname(fname)
+    max_clock_offset = None
     if True: # use the original files
         ev = REvent(fname)
 
@@ -33,6 +34,13 @@ if __name__ == "__main__":
         rng = np.random.default_rng(seed)
 
         dirname += f'rand{max_clock_offset}ns'
+    else:
+        raise NotImplementedError
+        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)
 
     # Full reconstruction?
     if False:
@@ -42,36 +50,43 @@ if __name__ == "__main__":
             print(f"Randomising timing: {max_clock_offset}")
             clock_offsets = max_clock_offset * (2*rng.uniform(size=len(ev.antennas)) - 1)
             for i, ant in enumerate(ev.antennas):
-                    ant.t -= clock_offsets[i]
+                ev.antennas[i].t += clock_offsets[i]
         else:
             print(f"Not randomising timing")
 
         with open(dirname + '/res.pkl', 'wb') as fp:
-                res = rit.reconstruction(ev, outfile=dirname+'/fig.pdf', slice_outdir=dirname+'/', Xlow=100, N_X=15, Xhigh=800)
+            res = rit.reconstruction(ev, outfile=dirname+'/fig.pdf', slice_outdir=dirname+'/', Xlow=100, N_X=23, Xhigh=1200)
 
             pickle.dump(res, fp)
         sys.exit(0)
 
-        # Else do single slice for original timing
-        # and randomised timing
-        print("Shower plane reconstruction + randomised timing")
+    # Only a single slice
+    else:
+        print("Shower plane reconstruction")
 
-        atm = AtmoCal()
-        X = 750
-        rit.set_pol_and_bp(ev)
-        dXref = atm.distance_to_slant_depth(np.deg2rad(ev.zenith),X,0)
-        scale2d = dXref*np.tan(np.deg2rad(2.))
+#        atm = AtmoCal()
+#        X = 750
+#        rit.set_pol_and_bp(ev)
+#        dXref = atm.distance_to_slant_depth(np.deg2rad(ev.zenith),750,0)
+#        scale2d = dXref*np.tan(np.deg2rad(2.))
+
+        if max_clock_offset is None:
+            print(" + randomised timing")
 
             ants_copy = deepcopy(ev.antennas)
             clock_offsets = max_clock_offset * (2*rng.uniform(size=len(ants_copy)) - 1)
             for i, ant in enumerate(ants_copy):
                 ant.t -= clock_offsets[i]
 
-        fig, axs = plt.subplots(1,2, sharex=True, sharey=True)
+        fig, axs = plt.subplots(1,1+(max_clock_offset is not None), sharex=True, sharey=True)
         fig.suptitle(f"Reconstructed Shower plane slice at X={X}")
+        if max_clock_offset is None:
+            axs = [ax]
         for ax in axs:
             ax.set_xlabel("-v x B (m)")
             ax.set_ylabel(" vxvxB (m)")
+
+        if max_clock_offset is not None:
             axs[0].set_title("Simulated timing")
             axs[1].set_title(f"Randomised timing [0, {max_clock_offset}]")
 
@@ -97,19 +112,11 @@ if __name__ == "__main__":
             fig.colorbar(sc, ax=axs[0])
         # Always make sure to show the plot
         except:
+            if True:
+                fig.savefig(dirname + f"/shower_plane_slice_X{X}_T{max_clock_offset}_exception.pdf")
             plt.show()
 
         if True:
             fig.savefig(dirname + f"/shower_plane_slice_X{X}_T{max_clock_offset}.pdf")
 
-
-    else:
-        raise NotImplementedError
-        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)
-
-
     plt.show()