edited mask and sorted, there is a bug on line 574 of bbVVSkimmer due…

… to it accessing vbf_jets.pt.layout.content.offsets
rkansal47 · Sep 11, 2023 · d20bf37 · d20bf37
1 parent a65fb52
commit d20bf37
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/src/HHbbVV/processors/bbVVSkimmer.py b/src/HHbbVV/processors/bbVVSkimmer.py
@@ -102,6 +102,8 @@ class bbVVSkimmer(processor.ProcessorABC):
         "eta": 2.7,
         "jetId": "tight",
         "puId": "medium",
+        "dR_fatjetbb": 1.2,
+        "dR_fatjetVV": 0.8,
     }
 
     jecs = common.jecs
@@ -322,6 +324,7 @@ def process(self, events: ak.Array):
         ak4_jet_vars = {}
 
         jets, _ = get_jec_jets(events, year, isData, self.jecs, fatjets=False)
+        print('fatjet',fatjets)
 
         vbf_jet_mask = (
             jets.isTight
@@ -331,14 +334,22 @@ def process(self, events: ak.Array):
             & ((jets.pt > 50) | ((jets.puId & 2) == 2))
             & (
                 ak.all(
-                    jets.metric_table(ak.pad_none(fatjets, num_jets, axis=1, clip=True))
-                    > self.ak4_jet_selection["dR_fatjet"],
+                    jets.metric_table(ak.pad_none(fatjets, num_jets, axis=1, clip=True)[bb_mask])
+                    > self.ak4_jet_selection["dR_fatjetbb"],
+                    axis=-1,
+                )
+            )
+            & (
+                ak.all(
+                    jets.metric_table(ak.pad_none(fatjets, num_jets, axis=1, clip=True)[~bb_mask])
+                    > self.ak4_jet_selection["dR_fatjetVV"], 
                     axis=-1,
                 )
             )
         )
 
-        vbf_jets = jets[vbf_jet_mask]
+        vbf_jets = jets[vbf_jet_mask] # also needs pt sorting
+        vbf_jets = vbf_jets[ak.argsort(vbf_jets.pt, ascending=False)]
 
         VBFJetVars = {
             f"VBFJet{key}": pad_val(vbf_jets[var], num_ak4_jets, axis=1)
@@ -555,7 +566,12 @@ def process(self, events: ak.Array):
             weights.add("genweight", gen_weights)
 
             add_pileup_weight(weights, year, events.Pileup.nPU.to_numpy())
-            add_pileupid_weights(weights, year, vbf_jets, events.GenJet, wp="M")
+            #print(len(vbf_jets),type(vbf_jets.pt.layout.content),vbf_jets.pt.layout.content)
+            #print(dir(vbf_jets.pt.layout.content))
+            #print(dir(JetArray(vbf_jets).pt.layout.content))
+            #print(JetArray(vbf_jets).pt.layout.content.offsets)
+            #print(vbf_jets.pt.layout.content.offsets)
+            add_pileupid_weights(weights, year, vbf_jets, events.GenJet, wp="M") # this gives error
             add_VJets_kFactors(weights, events.GenPart, dataset)
 
             # if dataset.startswith("TTTo"):