CMS 3D CMS Logo
applySubstructure_cff.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
2 
3 from PhysicsTools.PatAlgos.tools.helpers import getPatAlgosToolsTask, addToProcessAndTask
4 
5 def applySubstructure( process, postfix="" ) :
6 
7  task = getPatAlgosToolsTask(process)
8 
9  from PhysicsTools.PatAlgos.tools.jetTools import addJetCollection
10 
11 
12  from PhysicsTools.PatAlgos.producersLayer1.jetProducer_cfi import _patJets as patJetsDefault
13 
14 
15  # Configure the RECO jets
16  from RecoJets.JetProducers.ak8PFJets_cfi import ak8PFJetsPuppi, ak8PFJetsPuppiSoftDrop, ak8PFJetsPuppiConstituents
17  setattr(process,'ak8PFJetsPuppi'+postfix,ak8PFJetsPuppi.clone())
18  setattr(process,'ak8PFJetsPuppiConstituents'+postfix, ak8PFJetsPuppiConstituents.clone())
19  setattr(process,'ak8PFJetsPuppiSoftDrop'+postfix, ak8PFJetsPuppiSoftDrop.clone( src = 'ak8PFJetsPuppiConstituents'+postfix+':constituents' ))
20  from RecoJets.JetProducers.ak8PFJetsPuppi_groomingValueMaps_cfi import ak8PFJetsPuppiSoftDropMass
21  setattr(process,'ak8PFJetsPuppiSoftDropMass'+postfix, ak8PFJetsPuppiSoftDropMass.clone())
22  from Configuration.Eras.Modifier_run2_miniAOD_80XLegacy_cff import run2_miniAOD_80XLegacy
23  from Configuration.Eras.Modifier_run2_miniAOD_94XFall17_cff import run2_miniAOD_94XFall17
24  from Configuration.ProcessModifiers.run2_miniAOD_UL_cff import run2_miniAOD_UL
25  _run2_miniAOD_ANY = (run2_miniAOD_80XLegacy | run2_miniAOD_94XFall17 | run2_miniAOD_UL)
26  from Configuration.Eras.Modifier_pA_2016_cff import pA_2016
27  from Configuration.Eras.Modifier_run3_miniAOD_12X_cff import run3_miniAOD_12X
28  if postfix=='':
29  # Avoid recomputing the PUPPI collections that are present in AOD
30  _rerun_puppijets_task = task.copy()
31  _rerun_puppijets_task.add(getattr(process,'ak8PFJetsPuppi'),
32  getattr(process,'ak8PFJetsPuppiConstituents'),
33  getattr(process,'ak8PFJetsPuppiSoftDrop'),
34  getattr(process,'ak8PFJetsPuppiSoftDropMass'))
35  (_run2_miniAOD_ANY | pA_2016 | run3_miniAOD_12X ).toReplaceWith(task, _rerun_puppijets_task)
36  (_run2_miniAOD_ANY | pA_2016 | run3_miniAOD_12X ).toModify(getattr(process,'ak8PFJetsPuppiConstituents'+postfix),
37  cut = cms.string('pt > 170.0 && abs(rapidity()) < 2.4'))
38  else:
39  task.add(getattr(process,'ak8PFJetsPuppi'+postfix),
40  getattr(process,'ak8PFJetsPuppiConstituents'+postfix),
41  getattr(process,'ak8PFJetsPuppiSoftDrop'+postfix),
42  getattr(process,'ak8PFJetsPuppiSoftDropMass'+postfix))
43 
44  from RecoJets.JetProducers.ak8GenJets_cfi import ak8GenJets, ak8GenJetsSoftDrop, ak8GenJetsConstituents
45  addToProcessAndTask('ak8GenJetsNoNuConstituents'+postfix, ak8GenJetsConstituents.clone(src='ak8GenJetsNoNu'), process, task )
46  addToProcessAndTask('ak8GenJetsNoNuSoftDrop'+postfix,ak8GenJetsSoftDrop.clone(src=cms.InputTag('ak8GenJetsNoNuConstituents'+postfix, 'constituents')),process,task)
47  addToProcessAndTask('slimmedGenJetsAK8SoftDropSubJets'+postfix,
48  cms.EDProducer("PATGenJetSlimmer",
49  src = cms.InputTag("ak8GenJetsNoNuSoftDrop"+postfix, "SubJets"),
50  packedGenParticles = cms.InputTag("packedGenParticles"),
51  cut = cms.string(""),
52  cutLoose = cms.string(""),
53  nLoose = cms.uint32(0),
54  clearDaughters = cms.bool(False), #False means rekeying
55  dropSpecific = cms.bool(True), # Save space
56  ), process, task )
57 
58 
59  addJetCollection(
60  process,
61  postfix=postfix,
62  labelName = 'AK8PFPuppiSoftDrop' + postfix,
63  jetSource = cms.InputTag('ak8PFJetsPuppiSoftDrop'+postfix),
64  btagDiscriminators = ['None'],
65  genJetCollection = cms.InputTag('slimmedGenJetsAK8'),
66  jetCorrections = ('AK8PFPuppi', ['L2Relative', 'L3Absolute'], 'None'),
67  getJetMCFlavour = False # jet flavor disabled
68  )
69 
70  addJetCollection(
71  process,
72  postfix=postfix,
73  labelName = 'AK8PFPuppiSoftDropSubjets',
74  jetSource = cms.InputTag('ak8PFJetsPuppiSoftDrop'+postfix,'SubJets'),
75  algo = 'ak', # needed for subjet flavor clustering
76  rParam = 0.8, # needed for subjet flavor clustering
77  btagDiscriminators = ['pfDeepCSVJetTags:probb', 'pfDeepCSVJetTags:probbb', 'pfCombinedInclusiveSecondaryVertexV2BJetTags','pfCombinedMVAV2BJetTags'],
78  jetCorrections = ('AK4PFPuppi', ['L2Relative', 'L3Absolute'], 'None'),
79  explicitJTA = True, # needed for subjet b tagging
80  svClustering = True, # needed for subjet b tagging
81  genJetCollection = cms.InputTag('slimmedGenJetsAK8SoftDropSubJets'),
82  fatJets=cms.InputTag('ak8PFJetsPuppi'), # needed for subjet flavor clustering
83  groomedFatJets=cms.InputTag('ak8PFJetsPuppiSoftDrop') # needed for subjet flavor clustering
84  )
85 
86  from Configuration.Eras.Modifier_run3_common_cff import run3_common
87  run3_common.toModify(process.patJetsAK8PFPuppiSoftDropSubjets,
88  discriminatorSources = cms.VInputTag(
89  cms.InputTag("pfDeepCSVJetTagsAK8PFPuppiSoftDropSubjets","probb"),
90  cms.InputTag("pfDeepCSVJetTagsAK8PFPuppiSoftDropSubjets","probbb")
91  )
92  )
93 
94  # add groomed ECFs and N-subjettiness to soft dropped pat::Jets for fat jets and subjets
95  process.load('RecoJets.JetProducers.ECF_cff')
96  addToProcessAndTask('nb1AK8PuppiSoftDrop'+postfix, process.ecfNbeta1.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix), cuts = cms.vstring('', '', 'pt > 250')), process, task)
97  addToProcessAndTask('nb2AK8PuppiSoftDrop'+postfix, process.ecfNbeta2.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix), cuts = cms.vstring('', '', 'pt > 250')), process, task)
98 
99  #too slow now ==> disable
100  from Configuration.Eras.Modifier_pp_on_XeXe_2017_cff import pp_on_XeXe_2017
101 
102  for e in [pp_on_XeXe_2017]:
103  e.toModify(getattr(process,'nb1AK8PuppiSoftDrop'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
104  e.toModify(getattr(process,'nb2AK8PuppiSoftDrop'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
105 
106  from RecoJets.JetProducers.nJettinessAdder_cfi import Njettiness
107  addToProcessAndTask('NjettinessAK8Subjets'+postfix, Njettiness.clone(), process, task)
108  getattr(process,"NjettinessAK8Subjets"+postfix).src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix, "SubJets")
109  getattr(process,"patJetsAK8PFPuppiSoftDrop").userData.userFloats.src += ['nb1AK8PuppiSoftDrop'+postfix+':ecfN2','nb1AK8PuppiSoftDrop'+postfix+':ecfN3']
110  getattr(process,"patJetsAK8PFPuppiSoftDrop").userData.userFloats.src += ['nb2AK8PuppiSoftDrop'+postfix+':ecfN2','nb2AK8PuppiSoftDrop'+postfix+':ecfN3']
111  addToProcessAndTask('nb1AK8PuppiSoftDropSubjets'+postfix, process.ecfNbeta1.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix, "SubJets")), process, task)
112  addToProcessAndTask('nb2AK8PuppiSoftDropSubjets'+postfix, process.ecfNbeta2.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix, "SubJets")), process, task)
113  getattr(process,"patJetsAK8PFPuppiSoftDropSubjets"+postfix).userData.userFloats.src += ['nb1AK8PuppiSoftDropSubjets'+postfix+':ecfN2','nb1AK8PuppiSoftDropSubjets'+postfix+':ecfN3']
114  getattr(process,"patJetsAK8PFPuppiSoftDropSubjets"+postfix).userData.userFloats.src += ['nb2AK8PuppiSoftDropSubjets'+postfix+':ecfN2','nb2AK8PuppiSoftDropSubjets'+postfix+':ecfN3']
115  getattr(process,"patJetsAK8PFPuppiSoftDropSubjets"+postfix).userData.userFloats.src += ['NjettinessAK8Subjets'+postfix+':tau1','NjettinessAK8Subjets'+postfix+':tau2','NjettinessAK8Subjets'+postfix+':tau3','NjettinessAK8Subjets'+postfix+':tau4']
116 
117  for e in [pp_on_XeXe_2017]:
118  e.toModify(getattr(process,'nb1AK8PuppiSoftDropSubjets'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
119  e.toModify(getattr(process,'nb2AK8PuppiSoftDropSubjets'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
120 
121 
122  # Patify AK8 PF PUPPI
123  addJetCollection(process, postfix=postfix, labelName = 'AK8Puppi',
124  jetSource = cms.InputTag('ak8PFJetsPuppi'+postfix),
125  algo= 'AK', rParam = 0.8,
126  jetCorrections = ('AK8PFPuppi', cms.vstring(['L2Relative', 'L3Absolute']), 'None'),
127  btagDiscriminators = None,
128  genJetCollection = cms.InputTag('slimmedGenJetsAK8')
129  )
130  getattr(process,"patJetsAK8Puppi"+postfix).userData.userFloats.src = [] # start with empty list of user floats
131  getattr(process,"selectedPatJetsAK8Puppi"+postfix).cut = cms.string("pt > 100")
132  getattr(process,"selectedPatJetsAK8Puppi"+postfix).cutLoose = cms.string("pt > 30")
133  getattr(process,"selectedPatJetsAK8Puppi"+postfix).nLoose = cms.uint32(3)
134 
135  from RecoJets.JetAssociationProducers.j2tParametersVX_cfi import j2tParametersVX
136  addToProcessAndTask('ak8PFJetsPuppiTracksAssociatorAtVertex'+postfix, cms.EDProducer("JetTracksAssociatorAtVertex",
137  j2tParametersVX.clone( coneSize = cms.double(0.8) ),
138  jets = cms.InputTag("ak8PFJetsPuppi") ),
139  process, task)
140  addToProcessAndTask('patJetAK8PuppiCharge'+postfix, cms.EDProducer("JetChargeProducer",
141  src = cms.InputTag("ak8PFJetsPuppiTracksAssociatorAtVertex"),
142  var = cms.string('Pt'),
143  exp = cms.double(1.0) ),
144  process, task)
145 
146 
147  getattr(process,"patJetsAK8Puppi"+postfix).userData.userFloats.src += ['ak8PFJetsPuppiSoftDropMass'+postfix]
148  getattr(process,"patJetsAK8Puppi"+postfix).addTagInfos = cms.bool(False)
149 
150 
151  # add PUPPI Njetiness
152  addToProcessAndTask('NjettinessAK8Puppi'+postfix, Njettiness.clone(), process, task)
153  getattr(process,"NjettinessAK8Puppi"+postfix).src = cms.InputTag("ak8PFJetsPuppi"+postfix)
154  getattr(process,"patJetsAK8Puppi").userData.userFloats.src += ['NjettinessAK8Puppi'+postfix+':tau1','NjettinessAK8Puppi'+postfix+':tau2','NjettinessAK8Puppi'+postfix+':tau3','NjettinessAK8Puppi'+postfix+':tau4']
155 
156 
157  addToProcessAndTask("slimmedJetsAK8PFPuppiSoftDropSubjets"+postfix,
158  cms.EDProducer("PATJetSlimmer",
159  src = cms.InputTag("selectedPatJetsAK8PFPuppiSoftDropSubjets"),
160  packedPFCandidates = cms.InputTag("packedPFCandidates"),
161  dropJetVars = cms.string("1"),
162  dropDaughters = cms.string("0"),
163  rekeyDaughters = cms.string("1"),
164  dropTrackRefs = cms.string("1"),
165  dropSpecific = cms.string("1"),
166  dropTagInfos = cms.string("1"),
167  modifyJets = cms.bool(True),
168  mixedDaughters = cms.bool(False),
169  modifierConfig = cms.PSet( modifications = cms.VPSet() )
170  ),
171  process, task)
172 
173 
174 
175  addToProcessAndTask("slimmedJetsAK8PFPuppiSoftDropPacked"+postfix,
176  cms.EDProducer("BoostedJetMerger",
177  jetSrc=cms.InputTag("selectedPatJetsAK8PFPuppiSoftDrop"),
178  subjetSrc=cms.InputTag("slimmedJetsAK8PFPuppiSoftDropSubjets")
179  ),
180  process, task )
181 
182 
183  addToProcessAndTask("packedPatJetsAK8"+postfix, cms.EDProducer("JetSubstructurePacker",
184  jetSrc = cms.InputTag("selectedPatJetsAK8Puppi"+postfix),
185  distMax = cms.double(0.8),
186  algoTags = cms.VInputTag(
187  cms.InputTag("slimmedJetsAK8PFPuppiSoftDropPacked"+postfix)
188  ),
189  algoLabels = cms.vstring(
190  'SoftDropPuppi'
191  ),
192  fixDaughters = cms.bool(True),
193  packedPFCandidates = cms.InputTag("packedPFCandidates"+postfix),
194  ),
195  process, task)
196 
197  # switch off daughter re-keying since it's done in the JetSubstructurePacker (and can't be done afterwards)
198  process.slimmedJetsAK8.rekeyDaughters = "0"
199  # Reconfigure the slimmedAK8 jet information to keep
200  process.slimmedJetsAK8.dropDaughters = cms.string("pt < 170")
201  process.slimmedJetsAK8.dropSpecific = cms.string("pt < 170")
202  process.slimmedJetsAK8.dropTagInfos = cms.string("pt < 170")
applySubstructure_cff.applySubstructure
def applySubstructure(process, postfix="")
Definition: applySubstructure_cff.py:5
helpers.addToProcessAndTask
def addToProcessAndTask(label, module, process, task)
Definition: helpers.py:28
ak8PFJets_cfi
j2tParametersVX_cfi
jetProducer_cfi
helpers.getPatAlgosToolsTask
def getPatAlgosToolsTask(process)
Definition: helpers.py:13
Generated for CMSSW Reference Manual by doxygen 1.8.14