CMS 3D CMS Logo
runJetUncertainties.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
2 
3 from PhysicsTools.PatAlgos.tools.ConfigToolBase import *
4 
5 import PhysicsTools.PatAlgos.tools.helpers as configtools
6 from PhysicsTools.PatAlgos.tools.trigTools import _addEventContent
7 from PhysicsTools.PatUtils.tools.jmeUncertaintyTools import JetMEtUncertaintyTools
8 
9 from PhysicsTools.PatUtils.patPFMETCorrections_cff import *
10 import RecoMET.METProducers.METSigParams_cfi as jetResolutions
11 from PhysicsTools.PatAlgos.producersLayer1.metProducer_cfi import patMETs
12 
13 class RunJetUncertainties(JetMEtUncertaintyTools):
14 
15  """ Produce collection of pat::Jets with jet energy and resolution shifted up/down,
16  in order to estimate effect of systematic uncertainties
17  """
18  _label='runJetUncertainties'
19  _defaultParameters = dicttypes.SortedKeysDict()
20  def __init__(self):
21  JetMEtUncertaintyTools.__init__(self)
22  self.addParameter(self._defaultParameters, 'dRjetCleaning', 0.5,
23  "Eta-phi distance for extra jet cleaning", Type=float)
24  self._parameters = copy.deepcopy(self._defaultParameters)
25  self._comment = ""
26 
27  def __call__(self, process,
28  electronCollection = None,
29  photonCollection = None,
30  muonCollection = None,
31  tauCollection = None,
32  jetCollection = None,
33  dRjetCleaning = None,
34  jetCorrLabel = None,
35  doSmearJets = None,
36  jetSmearFileName = None,
37  jetSmearHistogram = None,
38  jetCorrPayloadName = None,
39  jetCorrLabelUpToL3 = None,
40  jetCorrLabelUpToL3Res = None,
41  jecUncertaintyFile = None,
42  jecUncertaintyTag = None,
43  varyByNsigmas = None,
44  addToPatDefaultSequence = None,
45  outputModule = None,
46  postfix = None):
47  JetMEtUncertaintyTools.__call__(
48  self, process,
49  electronCollection = electronCollection,
50  photonCollection = photonCollection,
51  muonCollection = muonCollection,
52  tauCollection = tauCollection,
53  jetCollection = jetCollection,
54  jetCorrLabel = jetCorrLabel,
55  doSmearJets = doSmearJets,
56  jetSmearFileName = jetSmearFileName,
57  jetSmearHistogram = jetSmearHistogram,
58  jetCorrPayloadName = jetCorrPayloadName,
59  jetCorrLabelUpToL3 = jetCorrLabelUpToL3,
60  jetCorrLabelUpToL3Res = jetCorrLabelUpToL3Res,
61  jecUncertaintyFile = jecUncertaintyFile,
62  jecUncertaintyTag = jecUncertaintyTag,
63  varyByNsigmas = varyByNsigmas,
64  addToPatDefaultSequence = addToPatDefaultSequence,
65  outputModule = outputModule,
66  postfix = postfix)
67  if dRjetCleaning is None:
68  dRjetCleaning = self._defaultParameters['dRjetCleaning'].value
69 
70  self.setParameter('dRjetCleaning', dRjetCleaning)
71 
72  self.apply(process)
73 
74  def toolCode(self, process):
75  electronCollection = self._parameters['electronCollection'].value
76  photonCollection = self._parameters['photonCollection'].value
77  muonCollection = self._parameters['muonCollection'].value
78  tauCollection = self._parameters['tauCollection'].value
79  jetCollection = self._parameters['jetCollection'].value
80  jetCorrLabel = self._parameters['jetCorrLabel'].value
81  dRjetCleaning = self._parameters['dRjetCleaning'].value
82  doSmearJets = self._parameters['doSmearJets'].value
83  jetSmearFileName = self._parameters['jetSmearFileName'].value
84  jetSmearHistogram = self._parameters['jetSmearHistogram'].value
85  jetCorrPayloadName = self._parameters['jetCorrPayloadName'].value
86  jetCorrLabelUpToL3 = self._parameters['jetCorrLabelUpToL3'].value
87  jetCorrLabelUpToL3Res = self._parameters['jetCorrLabelUpToL3Res'].value
88  jecUncertaintyFile = self._parameters['jecUncertaintyFile'].value
89  jecUncertaintyTag = self._parameters['jecUncertaintyTag'].value
90  varyByNsigmas = self._parameters['varyByNsigmas'].value
91  addToPatDefaultSequence = self._parameters['addToPatDefaultSequence'].value
92  outputModule = self._parameters['outputModule'].value
93  postfix = self._parameters['postfix'].value
94 
95  if not hasattr(process, "jetUncertaintySequence" + postfix):
96  jetUncertaintySequence = cms.Sequence()
97  setattr(process, "jetUncertaintySequence" + postfix, jetUncertaintySequence)
98  jetUncertaintySequence = getattr(process, "jetUncertaintySequence" + postfix)
99 
100  collectionsToKeep = []
101 
102  # produce collection of jets not overlapping with reconstructed
103  # electrons/photons, muons and tau-jet candidates
104  lastJetCollection, cleanedJetCollection = \
105  self._addCleanedJets(process, jetCollection,
106  electronCollection, photonCollection, muonCollection, tauCollection,
107  jetUncertaintySequence, postfix)
108 
109  # smear jet energies to account for difference in jet resolutions between MC and Data
110  # (cf. JME-10-014 PAS)
111  jetCollectionResUp = None
112  jetCollectionResDown = None
113  if doSmearJets:
114  lastJetCollection = \
115  self._addSmearedJets(process, cleanedJetCollection, [ "smeared", jetCollection.value() ],
116  jetSmearFileName, jetSmearHistogram, varyByNsigmas,
117  uncertaintySequence = jetUncertaintySequence, postfix = postfix)
118  jetCollectionResUp = \
119  self._addSmearedJets(process, cleanedJetCollection, [ "smeared", jetCollection.value(), "ResUp" ],
120  jetSmearFileName, jetSmearHistogram, varyByNsigmas, -1.,
121  uncertaintySequence = jetUncertaintySequence, postfix = postfix)
122  collectionsToKeep.append(jetCollectionResUp)
123  jetCollectionResDown = \
124  self._addSmearedJets(process, cleanedJetCollection, [ "smeared", jetCollection.value(), "ResDown" ],
125  jetSmearFileName, jetSmearHistogram, varyByNsigmas, +1.,
126  uncertaintySequence = jetUncertaintySequence, postfix = postfix)
127  collectionsToKeep.append(jetCollectionResDown)
128 
129  collectionsToKeep.append(lastJetCollection)
130 
131  #--------------------------------------------------------------------------------------------
132  # produce collection of electrons/photons, muons, tau-jet candidates and jets
133  # shifted up/down in energy by their respective energy uncertainties
134  #--------------------------------------------------------------------------------------------
135 
136  shiftedParticleSequence, shiftedParticleCollections, addCollectionsToKeep = \
137  self._addShiftedParticleCollections(process,
138  None,
139  None,
140  None,
141  None,
142  jetCollection, cleanedJetCollection, lastJetCollection,
143  jetCollectionResUp, jetCollectionResDown,
144  jetCorrLabelUpToL3, jetCorrLabelUpToL3Res,
145  jecUncertaintyFile, jecUncertaintyTag,
146  varyByNsigmas,
147  postfix)
148  setattr(process, "shiftedParticlesForJetUncertainties" + postfix, shiftedParticleSequence)
149  jetUncertaintySequence += getattr(process, "shiftedParticlesForJetUncertainties" + postfix)
150  collectionsToKeep.extend(addCollectionsToKeep)
151 
152  # insert metUncertaintySequence into patDefaultSequence
153  if addToPatDefaultSequence:
154  if not hasattr(process, "patDefaultSequence"):
155  raise ValueError("PAT default sequence is not defined !!")
156  process.patDefaultSequence += jetUncertaintySequence
157 
158  # add shifted + unshifted collections pf pat::Electrons/Photons,
159  # Muons, Taus, Jets and MET to PAT-tuple event content
160  if outputModule is not None and hasattr(process, outputModule):
161  getattr(process, outputModule).outputCommands = _addEventContent(
162  getattr(process, outputModule).outputCommands,
163  [ 'keep *_%s_*_%s' % (collectionToKeep, process.name_()) for collectionToKeep in collectionsToKeep ])
164 
165 runJetUncertainties = RunJetUncertainties()
runJetUncertainties.RunJetUncertainties.toolCode
def toolCode(self, process)
Definition: runJetUncertainties.py:74
runJetUncertainties.RunJetUncertainties._defaultParameters
_defaultParameters
Definition: runJetUncertainties.py:19
trigTools._addEventContent
def _addEventContent(outputCommands, eventContent)
Definition: trigTools.py:31
runJetUncertainties.RunJetUncertainties.__init__
def __init__(self)
Definition: runJetUncertainties.py:20
runJetUncertainties.RunJetUncertainties
Definition: runJetUncertainties.py:13
metProducer_cfi
runJetUncertainties.RunJetUncertainties.__call__
def __call__(self, process, electronCollection=None, photonCollection=None, muonCollection=None, tauCollection=None, jetCollection=None, dRjetCleaning=None, jetCorrLabel=None, doSmearJets=None, jetSmearFileName=None, jetSmearHistogram=None, jetCorrPayloadName=None, jetCorrLabelUpToL3=None, jetCorrLabelUpToL3Res=None, jecUncertaintyFile=None, jecUncertaintyTag=None, varyByNsigmas=None, addToPatDefaultSequence=None, outputModule=None, postfix=None)
Definition: runJetUncertainties.py:27
helpers
runJetUncertainties.RunJetUncertainties._parameters
_parameters
Definition: runJetUncertainties.py:24
METSigParams_cfi
runJetUncertainties.RunJetUncertainties._comment
_comment
Definition: runJetUncertainties.py:25
patPFMETCorrections_cff
ConfigToolBase
Generated for CMSSW Reference Manual by doxygen 1.8.16