up

Author: farakiko

boostedhiggs/hwwprocessor.py

@@ -347,6 +347,9 @@ def _build_objects(self, events):
         jet1 = ak4_outside_ak8[:, 0:1]
         jet2 = ak4_outside_ak8[:, 1:2]
 
+        FirstFatjet = ak.firsts(good_fatjets[:, 0:1])
+        SecondFatjet = ak.firsts(good_fatjets[:, 1:2])
+
         objects = {
             "muons": muons,
             "good_muons": good_muons,
@@ -376,6 +379,8 @@ def _build_objects(self, events):
             "ak4_outside_ak8": ak4_outside_ak8,
             "jet1": jet1,
             "jet2": jet2,
+            "FirstFatjet": FirstFatjet,
+            "SecondFatjet": SecondFatjet,
         }
 
         return objects
@@ -406,6 +411,9 @@ def _derive_variables(self, events, objects):
         jet1 = objects["jet1"]
         jet2 = objects["jet2"]
 
+        FirstFatjet = objects["FirstFatjet"]
+        SecondFatjet = objects["SecondFatjet"]
+
         # VH jet
         minDeltaR = ak.argmin(candidatelep_p4.delta_r(good_fatjets), axis=1)  # similar to fj_idx_lep but without keepdims
         fatJetIndices = ak.local_index(good_fatjets, axis=1)
@@ -448,8 +456,6 @@ def _derive_variables(self, events, objects):
 
         # njets
         NumFatjets = ak.num(good_fatjets)
-        FirstFatjet = ak.firsts(good_fatjets[:, 0:1])
-        SecondFatjet = ak.firsts(good_fatjets[:, 1:2])
         NumOtherJets = ak.num(ak4_outside_ak8)
 
         # n-bjets
@@ -690,7 +696,7 @@ def _apply_JEC(self, events, objects, variables):
     def _store_genVars(self, dataset, events, objects, variables):
 
         candidatefj = objects["candidatefj"]
-        FirstFatjet = variables["FirstFatjet"]
+        FirstFatjet = objects["FirstFatjet"]
 
         # store gen-level matching variables
         signal_mask = None