Merging full workflow into Tanay's HiggsDNA

Don't merge this yet! Let's do some testing first before we merge to master

This should work out of the box (fingers crossed). This will have the full h->aa->4gamma workflow included in the base class. This should also not affect any corrections that you are using currently. That still needs to be tested. There may be bugs that I haven't caught yet, so I need to do some testing... I also included some nice helper scripts for you in the scripts folder.

higgs_dna/selections/photon_selections.py

@@ -188,12 +188,22 @@ def photon_preselection_h4g(
     self, photons: awkward.Array, events: awkward.Array, year="2022"
 ) -> awkward.Array:
 
+    """
+    print(awkward.num(events.Photon[awkward.num(events.Photon, axis=1) >= 2], axis=1), awkward.num(photons[awkward.num(photons, axis=1) >= 2], axis=1), set(photons.fields).symmetric_difference(set(events.Photon.fields)), photons[awkward.num(photons, axis=1) >= 2], events.Photon[awkward.num(events.Photon, axis=1) >= 2])
+    print(awkward.num(events.Photon[awkward.num(events.Photon, axis=1) >= 2], axis=1), awkward.num(photons[awkward.num(photons, axis=1) >= 2], axis=1), set(photons.fields).symmetric_difference(set(events.Photon.fields)), photons[awkward.num(photons, axis=1) >= 2], events.Photon[awkward.num(events.Photon, axis=1) >= 2])
+    print(awkward.num(events.Photon[awkward.num(events.Photon, axis=1) >= 2], axis=1), awkward.num(photons[awkward.num(photons, axis=1) >= 2], axis=1), set(photons.fields).symmetric_difference(set(events.Photon.fields)), photons[awkward.num(photons, axis=1) >= 2], events.Photon[awkward.num(events.Photon, axis=1) >= 2])
+    assert len(events.Photon[awkward.num(events.Photon, axis=1) >= 2]) == len(photons[awkward.num(photons, axis=1) >= 2])
+    """
+
+    photons = photons[awkward.num(photons, axis=1) >= 2]
+
     # run3 = True if "2022" in year or "2023" in year else False
     run3 = True
     if not run3:
         rho = events.fixedGridRhoAll * awkward.ones_like(photons.pt)
     else:
-        rho = events.Rho.fixedGridRhoAll * awkward.ones_like(photons.pt)
+        #rho = events.Rho.fixedGridRhoAll * awkward.ones_like(photons.pt)
+        rho = photons.fixedGridRhoAll
 
     photons = photons[awkward.argsort(photons.pt, ascending=False, axis=1)]
 
@@ -319,5 +329,8 @@ def photon_preselection_h4g(
     iso = iso_r9 | iso_charged | iso_charged_rel
 
     all_cuts = pt_cuts & eta & e_veto & hlt_all & iso
+    #events.Photon = events.Photon[all_cuts]
+    #events = events[~awkward.is_none(events.Photon)]
+    #events = events[awkward.num(events.Photon, axis=1) >= 2]
 
     return events, photons[all_cuts], all_cuts