CMS 3D CMS Logo

List of all members | Public Member Functions | Public Attributes
JetAnalyzer.JetAnalyzer Class Reference
Inheritance diagram for JetAnalyzer.JetAnalyzer:

Public Member Functions

def __init__ (self, cfg_ana, cfg_comp, looperName)
 
def beginLoop (self, setup)
 
def declareHandles (self)
 
def jerCorrection (self, jet)
 
def jesCorrection (self, jet, scale=0.)
 
def process (self, event)
 
def testBJet (self, jet)
 
def testJet (self, jet)
 
def testJetID (self, jet)
 

Public Attributes

 btagSF
 
 is2012
 

Detailed Description

Analyze jets ;-)

This analyzer filters the jets that do not correspond to the leptons
stored in event.selectedLeptons, and puts in the event:
- jets: all jets passing the pt and eta cuts
- cleanJets: the collection of jets away from the leptons
- cleanBJets: the jets passing testBJet, and away from the leptons

Example configuration:

jetAna = cfg.Analyzer(
  'JetAnalyzer',
  jetCol = 'slimmedJets'
  # cmg jet input collection
  # pt threshold
  jetPt = 30,
  # eta range definition
  jetEta = 5.0,
  # seed for the btag scale factor
  btagSFseed = 0xdeadbeef,
  # if True, the PF and PU jet ID are not applied, and the jets get flagged
  relaxJetId = False,
)

Definition at line 16 of file JetAnalyzer.py.

Constructor & Destructor Documentation

def JetAnalyzer.JetAnalyzer.__init__ (   self,
  cfg_ana,
  cfg_comp,
  looperName 
)

Definition at line 42 of file JetAnalyzer.py.

42  def __init__(self, cfg_ana, cfg_comp, looperName):
43  super(JetAnalyzer,self).__init__(cfg_ana, cfg_comp, looperName)
44  self.btagSF = BTagSF (cfg_ana.btagSFseed)
45  self.is2012 = isNewerThan('CMSSW_5_2_0')
46 
def __init__(self, cfg_ana, cfg_comp, looperName)
Definition: JetAnalyzer.py:42
def isNewerThan(release1, release2=None)
Definition: cmsswRelease.py:53

Member Function Documentation

def JetAnalyzer.JetAnalyzer.beginLoop (   self,
  setup 
)

Definition at line 59 of file JetAnalyzer.py.

59  def beginLoop(self, setup):
60  super(JetAnalyzer,self).beginLoop(setup)
61  self.counters.addCounter('jets')
62  count = self.counters.counter('jets')
63  count.register('all events')
64  count.register('at least 2 good jets')
65  count.register('at least 2 clean jets')
66  count.register('at least 1 b jet')
67  count.register('at least 2 b jets')
68 
def beginLoop(self, setup)
Definition: JetAnalyzer.py:59
def JetAnalyzer.JetAnalyzer.declareHandles (   self)

Definition at line 47 of file JetAnalyzer.py.

References core.Analyzer.Analyzer.handles, HTTP.RequestManager.handles, and core.Analyzer.Analyzer.mchandles.

47  def declareHandles(self):
48  super(JetAnalyzer, self).declareHandles()
49 
50  self.handles['jets'] = AutoHandle( self.cfg_ana.jetCol,
51  'std::vector<pat::Jet>' )
52  if self.cfg_comp.isMC:
53  # and ("BB" in self.cfg_comp.name):
54  self.mchandles['genParticles'] = AutoHandle( 'packedGenParticles',
55  'std::vector<pat::PackedGenParticle>' )
56  self.mchandles['genJets'] = AutoHandle('slimmedGenJets',
57  'std::vector<reco::GenJet>')
58 
def JetAnalyzer.JetAnalyzer.jerCorrection (   self,
  jet 
)
Adds JER correction according to first method at
https://twiki.cern.ch/twiki/bin/view/CMS/JetResolution

Requires some attention when genJet matching fails.

Definition at line 162 of file JetAnalyzer.py.

References funct.abs().

Referenced by JetAnalyzer.JetAnalyzer.process().

162  def jerCorrection(self, jet):
163  ''' Adds JER correction according to first method at
164  https://twiki.cern.ch/twiki/bin/view/CMS/JetResolution
165 
166  Requires some attention when genJet matching fails.
167  '''
168  if not hasattr(jet, 'matchedGenJet'):
169  return
170  #import pdb; pdb.set_trace()
171  corrections = [0.052, 0.057, 0.096, 0.134, 0.288]
172  maxEtas = [0.5, 1.1, 1.7, 2.3, 5.0]
173  eta = abs(jet.eta())
174  for i, maxEta in enumerate(maxEtas):
175  if eta < maxEta:
176  pt = jet.pt()
177  deltaPt = (pt - jet.matchedGenJet.pt()) * corrections[i]
178  totalScale = (pt + deltaPt) / pt
179 
180  if totalScale < 0.:
181  totalScale = 0.
182  jet.scaleEnergy(totalScale)
183  break
184 
def jerCorrection(self, jet)
Definition: JetAnalyzer.py:162
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
def JetAnalyzer.JetAnalyzer.jesCorrection (   self,
  jet,
  scale = 0. 
)
Adds JES correction in number of sigmas (scale)

Definition at line 185 of file JetAnalyzer.py.

Referenced by JetAnalyzer.JetAnalyzer.process().

185  def jesCorrection(self, jet, scale=0.):
186  ''' Adds JES correction in number of sigmas (scale)
187  '''
188  # Do nothing if nothing to change
189  if scale == 0.:
190  return
191  unc = jet.uncOnFourVectorScale()
192  totalScale = 1. + scale * unc
193  if totalScale < 0.:
194  totalScale = 0.
195  jet.scaleEnergy(totalScale)
196 
def jesCorrection(self, jet, scale=0.)
Definition: JetAnalyzer.py:185
def JetAnalyzer.JetAnalyzer.process (   self,
  event 
)

Definition at line 69 of file JetAnalyzer.py.

References funct.abs(), analyzer.Analyzer.cfg_ana, analyzer.Analyzer.cfg_comp, looper.Looper.cfg_comp, deltar.cleanObjectCollection(), reco.deltaR2(), core.Analyzer.Analyzer.handles, HTTP.RequestManager.handles, JetAnalyzer.JetAnalyzer.jerCorrection(), JetAnalyzer.JetAnalyzer.jesCorrection(), genParticles_cff.map, deltar.matchObjectCollection(), core.Analyzer.Analyzer.mchandles, cosmictrackingParticleSelector_cfi.pdgId, core.Analyzer.Analyzer.readCollections(), JetAnalyzer.JetAnalyzer.testBJet(), and JetAnalyzer.JetAnalyzer.testJet().

Referenced by ConfigBuilder.ConfigBuilder.addExtraStream(), ConfigBuilder.ConfigBuilder.completeInputCommand(), ConfigBuilder.ConfigBuilder.doNotInlineEventContent(), ConfigBuilder.ConfigBuilder.PrintAllModules.leave(), ConfigBuilder.ConfigBuilder.prepare(), ConfigBuilder.ConfigBuilder.prepare_ALCA(), ConfigBuilder.ConfigBuilder.prepare_DQM(), ConfigBuilder.ConfigBuilder.prepare_HLT(), ConfigBuilder.ConfigBuilder.prepare_LHE(), ConfigBuilder.ConfigBuilder.prepare_PATFILTER(), ConfigBuilder.ConfigBuilder.prepare_VALIDATION(), ConfigBuilder.ConfigBuilder.renameHLTprocessInSequence(), ConfigBuilder.ConfigBuilder.renameInputTagsInSequence(), and ConfigBuilder.ConfigBuilder.scheduleSequence().

69  def process(self, event):
70 
71  self.readCollections( event.input )
72  miniaodjets = self.handles['jets'].product()
73 
74  allJets = []
75  event.jets = []
76  event.bJets = []
77  event.cleanJets = []
78  event.cleanBJets = []
79 
80  leptons = []
81  if hasattr(event, 'selectedLeptons'):
82  leptons = event.selectedLeptons
83 
84  genJets = None
85  if self.cfg_comp.isMC:
86  genJets = map( GenJet, self.mchandles['genJets'].product() )
87 
88  for maodjet in miniaodjets:
89  jet = Jet( maodjet )
90  allJets.append( jet )
91  if self.cfg_comp.isMC and hasattr( self.cfg_comp, 'jetScale'):
92  scale = random.gauss( self.cfg_comp.jetScale,
93  self.cfg_comp.jetSmear )
94  jet.scaleEnergy( scale )
95  if genJets:
96  # Use DeltaR = 0.25 matching like JetMET
97  pairs = matchObjectCollection( [jet], genJets, 0.25*0.25)
98  if pairs[jet] is None:
99  pass
100  else:
101  jet.matchedGenJet = pairs[jet]
102  #Add JER correction for MC jets. Requires gen-jet matching.
103  if self.cfg_comp.isMC and hasattr(self.cfg_ana, 'jerCorr') and self.cfg_ana.jerCorr:
104  self.jerCorrection(jet)
105  #Add JES correction for MC jets.
106  if self.cfg_comp.isMC and hasattr(self.cfg_ana, 'jesCorr'):
107  self.jesCorrection(jet, self.cfg_ana.jesCorr)
108  if self.testJet( jet ):
109  event.jets.append(jet)
110  if self.testBJet(jet):
111  event.bJets.append(jet)
112 
113  self.counters.counter('jets').inc('all events')
114 
115  event.cleanJets, dummy = cleanObjectCollection( event.jets,
116  masks = leptons,
117  deltaRMin = 0.5 )
118  event.cleanBJets, dummy = cleanObjectCollection( event.bJets,
119  masks = leptons,
120  deltaRMin = 0.5 )
121 
122  pairs = matchObjectCollection( leptons, allJets, 0.5*0.5)
123  # associating a jet to each lepton
124  for lepton in leptons:
125  jet = pairs[lepton]
126  if jet is None:
127  lepton.jet = lepton
128  else:
129  lepton.jet = jet
130 
131  # associating a leg to each clean jet
132  invpairs = matchObjectCollection( event.cleanJets, leptons, 99999. )
133  for jet in event.cleanJets:
134  leg = invpairs[jet]
135  jet.leg = leg
136 
137  for jet in event.cleanJets:
138  jet.matchGenParton=999.0
139 
140  if self.cfg_comp.isMC and "BB" in self.cfg_comp.name:
141  genParticles = self.mchandles['genParticles'].product()
142  event.genParticles = map( GenParticle, genParticles)
143  for gen in genParticles:
144  if abs(gen.pdgId())==5 and gen.mother() and abs(gen.mother().pdgId())==21:
145  for jet in event.cleanJets:
146  dR=deltaR2(jet.eta(), jet.phi(), gen.eta(), gen.phi() )
147  if dR<jet.matchGenParton:
148  jet.matchGenParton=dR
149 
150  event.jets30 = [jet for jet in event.jets if jet.pt()>30]
151  event.cleanJets30 = [jet for jet in event.cleanJets if jet.pt()>30]
152  if len( event.jets30 )>=2:
153  self.counters.counter('jets').inc('at least 2 good jets')
154  if len( event.cleanJets30 )>=2:
155  self.counters.counter('jets').inc('at least 2 clean jets')
156  if len(event.cleanBJets)>0:
157  self.counters.counter('jets').inc('at least 1 b jet')
158  if len(event.cleanBJets)>1:
159  self.counters.counter('jets').inc('at least 2 b jets')
160  return True
161 
def testJet(self, jet)
Definition: JetAnalyzer.py:206
def jerCorrection(self, jet)
Definition: JetAnalyzer.py:162
def matchObjectCollection
Definition: deltar.py:152
Definition: Jet.py:1
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
def process(self, event)
Definition: JetAnalyzer.py:69
def jesCorrection(self, jet, scale=0.)
Definition: JetAnalyzer.py:185
def cleanObjectCollection(objects, masks, deltaRMin)
Definition: deltar.py:119
T1 deltaR2(T1 eta1, T2 phi1, T3 eta2, T4 phi2)
Definition: deltaR.h:36
def testBJet(self, jet)
Definition: JetAnalyzer.py:211
def JetAnalyzer.JetAnalyzer.testBJet (   self,
  jet 
)

Definition at line 211 of file JetAnalyzer.py.

References funct.abs(), JetAnalyzer.JetAnalyzer.is2012, and JetAnalyzer.JetAnalyzer.testJetID().

Referenced by JetAnalyzer.JetAnalyzer.process().

211  def testBJet(self, jet):
212  # medium csv working point
213  # https://twiki.cern.ch/twiki/bin/viewauth/CMS/BTagPerformanceOP#B_tagging_Operating_Points_for_3
214  jet.btagMVA = jet.btag("combinedSecondaryVertexBJetTags")
215  jet.btagFlag = self.btagSF.BTagSFcalc.isbtagged(
216  jet.pt(),
217  jet.eta(),
218  jet.btag("combinedSecondaryVertexBJetTags"),
219  abs(jet.partonFlavour()),
220  not self.cfg_comp.isMC,
221  0,0,
222  self.is2012
223  )
224  return jet.pt()>20 and \
225  abs( jet.eta() ) < 2.4 and \
226  self.testJetID(jet)
227 
def testJetID(self, jet)
Definition: JetAnalyzer.py:197
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
def testBJet(self, jet)
Definition: JetAnalyzer.py:211
def JetAnalyzer.JetAnalyzer.testJet (   self,
  jet 
)

Definition at line 206 of file JetAnalyzer.py.

References funct.abs(), and JetAnalyzer.JetAnalyzer.testJetID().

Referenced by JetAnalyzer.JetAnalyzer.process().

206  def testJet( self, jet ):
207  return jet.pt() > self.cfg_ana.jetPt and \
208  abs( jet.eta() ) < self.cfg_ana.jetEta and \
209  self.testJetID(jet)
210 
def testJet(self, jet)
Definition: JetAnalyzer.py:206
def testJetID(self, jet)
Definition: JetAnalyzer.py:197
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
def JetAnalyzer.JetAnalyzer.testJetID (   self,
  jet 
)

Definition at line 197 of file JetAnalyzer.py.

Referenced by JetAnalyzer.JetAnalyzer.testBJet(), and JetAnalyzer.JetAnalyzer.testJet().

197  def testJetID(self, jet):
198  jet.puJetIdPassed = jet.puJetId()
199  jet.pfJetIdPassed = jet.jetID("POG_PFID_Loose")
200  if self.cfg_ana.relaxJetId:
201  return True
202  else:
203  return jet.puJetIdPassed and jet.pfJetIdPassed
204 
205 
def testJetID(self, jet)
Definition: JetAnalyzer.py:197

Member Data Documentation

JetAnalyzer.JetAnalyzer.btagSF

Definition at line 44 of file JetAnalyzer.py.

JetAnalyzer.JetAnalyzer.is2012

Definition at line 45 of file JetAnalyzer.py.

Referenced by JetAnalyzer.JetAnalyzer.testBJet().