up

Author: farakiko

boostedhiggs/hwwprocessor.py

@@ -644,7 +644,7 @@ def _apply_JEC(self, objects, variables):
             if shift != "" and not self._systematics:
                 continue
             jecvariables = getJECVariables(
-                fatjetvars, objects["candidatelep_p4"], objects[" met"], pt_shift=shift, met_shift=None
+                fatjetvars, objects["candidatelep_p4"], objects["met"], pt_shift=shift, met_shift=None
             )
             variables = {**variables, **jecvariables}
 
@@ -747,14 +747,11 @@ def _apply_selections(self, events, trigger, metfilters, objects, variables):
 
     def _store_genVars(self, dataset, events, objects, variables):
 
-        candidatefj = objects["candidatefj"]
-        FirstFatjet = objects["FirstFatjet"]
-
         # store gen-level matching variables
         signal_mask = None
         if self.isMC:
             if self.isSignal:
-                genVars, signal_mask = match_H(events.GenPart, candidatefj, fatjet_pt=FirstFatjet)
+                genVars, signal_mask = match_H(events.GenPart, objects["candidatefj"], fatjet_pt=objects["FirstFatjet"])
                 # genVars = {**genVars, **match_H_alljets(events.GenPart, fatjets)}
                 # add signal mask and modify sum of genweights to only consider those events that pass the mask
                 self.add_selection(name="Signal", sel=signal_mask)
@@ -766,21 +763,21 @@ def _store_genVars(self, dataset, events, objects, variables):
                     genVars["STXS_cat"] = events.HTXS.stage1_2_cat_pTjet30GeV
                     genVars["STXS_finecat"] = events.HTXS.stage1_2_fine_cat_pTjet30GeV
             elif "HToTauTau" in dataset:
-                genVars, signal_mask = match_H(events.GenPart, candidatefj, dau_pdgid=15)
+                genVars, signal_mask = match_H(events.GenPart, objects["candidatefj"], dau_pdgid=15)
                 self.add_selection(name="Signal", sel=signal_mask)
             elif ("WJets" in dataset) or ("ZJets" in dataset) or ("DYJets" in dataset):
-                genVars, _ = match_V(events.GenPart, candidatefj)
+                genVars, _ = match_V(events.GenPart, objects["candidatefj"])
                 if "LHE_HT" in events.fields:
                     genVars["LHE_HT"] = events.LHE.HT
                 if "LHE_Vpt" in events.fields:
                     genVars["LHE_Vpt"] = events.LHE.Vpt
             elif "TT" in dataset:
-                genVars, _ = match_Top(events.GenPart, candidatefj)
+                genVars, _ = match_Top(events.GenPart, objects["candidatefj"])
             else:
                 genVars = {}
             # save gen jet mass (not msd)
-            genVars["fj_genjetmass"] = candidatefj.matched_gen.mass
-            genVars["fj_genjetpt"] = candidatefj.matched_gen.pt
+            genVars["fj_genjetmass"] = objects["candidatefj"].matched_gen.mass
+            genVars["fj_genjetpt"] = objects["candidatefj"].matched_gen.pt
             variables = {**variables, **genVars}
 
         return variables