Merge pull request cms-nanoAOD#110 from daiktas/HNL

add raw jet pt
bainbrid · Feb 9, 2021 · c6fb045 · c6fb045
2 parents b8acd96 + 9e3ab46
commit c6fb045
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/processors/HNL.py b/processors/HNL.py
@@ -261,7 +261,7 @@
             jerResolutionFileName=jerResolutionFile[year],
             jerSFUncertaintyFileName=jerSFUncertaintyFile[year],
             propagateJER = False,
-            jetKeys = ['pt', 'eta', 'phi' , 'jetId', 'nConstituents'],
+            jetKeys = ['pt', 'eta', 'phi' , 'jetId', 'nConstituents', 'rawFactor'],
         )
     )
 

diff --git a/python/modules/JetSelection.py b/python/modules/JetSelection.py
@@ -30,7 +30,7 @@ def __init__(
          dRCleaning=0.4,
          dRP4Subtraction=0.4,
          flagDA=False,
-         storeKinematics=['pt', 'eta', 'phi', 'minDeltaRSubtraction', 'ptLepton', 'ptSubtracted'],
+         storeKinematics=['pt', 'eta', 'phi', 'minDeltaRSubtraction', 'ptLepton', 'ptSubtracted', 'rawFactor', 'ptRaw'],
          globalOptions={"isData": False, "year": 2016},
          jetId=TIGHT
      ):
@@ -86,6 +86,8 @@ def analyze(self, event):
             flagsDA = [0.]*event.nJet
 
         for jet in jets:    
+            jet.ptRaw = jet.pt*(1. - jet.rawFactor)
+
             if jet.pt<self.jetMinPt:
                 unselectedJets.append(jet)
                 continue