Functions
def	applySubstructure (process, postfix="")
Function Documentation

◆ applySubstructure()

def applySubstructure_cff.applySubstructure	(	process,
		postfix = `""`
	)
Definition at line 5 of file applySubstructure_cff.py.
References helpers.addToProcessAndTask(), and helpers.getPatAlgosToolsTask().
Referenced by miniAOD_tools.miniAOD_customizeCommon().
 def applySubstructure( process, postfix="" ) :
 
     task = getPatAlgosToolsTask(process)
 
     from PhysicsTools.PatAlgos.tools.jetTools import addJetCollection
 
 
     from PhysicsTools.PatAlgos.producersLayer1.jetProducer_cfi import _patJets as patJetsDefault
 
 
     # Configure the RECO jets
     from RecoJets.JetProducers.ak8PFJets_cfi import ak8PFJetsPuppi, ak8PFJetsPuppiSoftDrop, ak8PFJetsPuppiConstituents
     setattr(process,'ak8PFJetsPuppi'+postfix,ak8PFJetsPuppi.clone())
     setattr(process,'ak8PFJetsPuppiConstituents'+postfix, ak8PFJetsPuppiConstituents.clone(cut = cms.string('pt > 170.0 && abs(rapidity()) < 2.4') ))
     setattr(process,'ak8PFJetsPuppiSoftDrop'+postfix, ak8PFJetsPuppiSoftDrop.clone( src = 'ak8PFJetsPuppiConstituents'+postfix+':constituents' ))
     from RecoJets.JetProducers.ak8PFJetsPuppi_groomingValueMaps_cfi import ak8PFJetsPuppiSoftDropMass
     setattr(process,'ak8PFJetsPuppiSoftDropMass'+postfix, ak8PFJetsPuppiSoftDropMass.clone())
     from Configuration.Eras.Modifier_run2_miniAOD_80XLegacy_cff import run2_miniAOD_80XLegacy
     from Configuration.Eras.Modifier_run2_miniAOD_94XFall17_cff import run2_miniAOD_94XFall17
     from Configuration.ProcessModifiers.run2_miniAOD_UL_cff import run2_miniAOD_UL
     _run2_miniAOD_ANY = (run2_miniAOD_80XLegacy | run2_miniAOD_94XFall17 | run2_miniAOD_UL)
     from Configuration.Eras.Modifier_pA_2016_cff import pA_2016
     from Configuration.Eras.Modifier_run3_miniAOD_12X_cff import run3_miniAOD_12X
     if postfix=='':
       # Avoid recomputing the PUPPI collections that are present in AOD
       _rerun_puppijets_task = task.copy()
       _rerun_puppijets_task.add(getattr(process,'ak8PFJetsPuppi'),
                                 getattr(process,'ak8PFJetsPuppiConstituents'),
                                 getattr(process,'ak8PFJetsPuppiSoftDrop'),
                                 getattr(process,'ak8PFJetsPuppiSoftDropMass'))
       (_run2_miniAOD_ANY | pA_2016 | run3_miniAOD_12X ).toReplaceWith(task, _rerun_puppijets_task)
       (_run2_miniAOD_ANY | pA_2016 | run3_miniAOD_12X ).toModify(getattr(process,'ak8PFJetsPuppiConstituents'+postfix),
                                                                  cut = cms.string('pt > 170.0 && abs(rapidity()) < 2.4'))
     else:
       task.add(getattr(process,'ak8PFJetsPuppi'+postfix),
                getattr(process,'ak8PFJetsPuppiConstituents'+postfix),
                getattr(process,'ak8PFJetsPuppiSoftDrop'+postfix),
                getattr(process,'ak8PFJetsPuppiSoftDropMass'+postfix))
 
     from RecoJets.JetProducers.ak8GenJets_cfi import ak8GenJets, ak8GenJetsSoftDrop, ak8GenJetsConstituents
     addToProcessAndTask('ak8GenJetsNoNuConstituents'+postfix, ak8GenJetsConstituents.clone(src='ak8GenJetsNoNu'), process, task )
     addToProcessAndTask('ak8GenJetsNoNuSoftDrop'+postfix,ak8GenJetsSoftDrop.clone(src=cms.InputTag('ak8GenJetsNoNuConstituents'+postfix, 'constituents')),process,task)
     addToProcessAndTask('slimmedGenJetsAK8SoftDropSubJets'+postfix,
                             cms.EDProducer("PATGenJetSlimmer",
                                                src = cms.InputTag("ak8GenJetsNoNuSoftDrop"+postfix, "SubJets"),
                                                packedGenParticles = cms.InputTag("packedGenParticles"),
                                                cut = cms.string(""),
                                                cutLoose = cms.string(""),
                                                nLoose = cms.uint32(0),
                                                clearDaughters = cms.bool(False), #False means rekeying
                                                dropSpecific = cms.bool(True),  # Save space
                                                ), process, task )
 
     
     addJetCollection(
         process,
         postfix=postfix,
         labelName = 'AK8PFPuppiSoftDrop' + postfix,
         jetSource = cms.InputTag('ak8PFJetsPuppiSoftDrop'+postfix),
         btagDiscriminators = ['None'],
         genJetCollection = cms.InputTag('slimmedGenJetsAK8'), 
         jetCorrections = ('AK8PFPuppi', ['L2Relative', 'L3Absolute'], 'None'),
         getJetMCFlavour = False # jet flavor disabled
     )
     
     addJetCollection(
         process,
         postfix=postfix,
         labelName = 'AK8PFPuppiSoftDropSubjets',
         jetSource = cms.InputTag('ak8PFJetsPuppiSoftDrop'+postfix,'SubJets'),
         algo = 'ak',  # needed for subjet flavor clustering
         rParam = 0.8, # needed for subjet flavor clustering
         btagDiscriminators = ['pfDeepCSVJetTags:probb', 'pfDeepCSVJetTags:probbb', 'pfCombinedInclusiveSecondaryVertexV2BJetTags','pfCombinedMVAV2BJetTags'],
         jetCorrections = ('AK4PFPuppi', ['L2Relative', 'L3Absolute'], 'None'),
         explicitJTA = True,  # needed for subjet b tagging
         svClustering = True, # needed for subjet b tagging
         genJetCollection = cms.InputTag('slimmedGenJetsAK8SoftDropSubJets'), 
         fatJets=cms.InputTag('ak8PFJetsPuppi'),             # needed for subjet flavor clustering
         groomedFatJets=cms.InputTag('ak8PFJetsPuppiSoftDrop') # needed for subjet flavor clustering
     )
 
     from Configuration.Eras.Modifier_run3_common_cff import run3_common
     run3_common.toModify(process.patJetsAK8PFPuppiSoftDropSubjets,
                          discriminatorSources = cms.VInputTag(
                             cms.InputTag("pfDeepCSVJetTagsAK8PFPuppiSoftDropSubjets","probb"),
                             cms.InputTag("pfDeepCSVJetTagsAK8PFPuppiSoftDropSubjets","probbb")
                          )
     )
 
     # add groomed ECFs and N-subjettiness to soft dropped pat::Jets for fat jets and subjets
     process.load('RecoJets.JetProducers.ECF_cff')
     addToProcessAndTask('nb1AK8PuppiSoftDrop'+postfix, process.ecfNbeta1.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix), cuts = cms.vstring('', '', 'pt > 250')), process, task)
     addToProcessAndTask('nb2AK8PuppiSoftDrop'+postfix, process.ecfNbeta2.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix), cuts = cms.vstring('', '', 'pt > 250')), process, task)
 
     #too slow now ==> disable
     from Configuration.Eras.Modifier_pp_on_XeXe_2017_cff import pp_on_XeXe_2017
     from Configuration.Eras.Modifier_phase2_common_cff import phase2_common
 
     for e in [pp_on_XeXe_2017, phase2_common]:
         e.toModify(getattr(process,'nb1AK8PuppiSoftDrop'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
         e.toModify(getattr(process,'nb2AK8PuppiSoftDrop'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
 
     from RecoJets.JetProducers.nJettinessAdder_cfi import Njettiness
     addToProcessAndTask('NjettinessAK8Subjets'+postfix, Njettiness.clone(), process, task)
     getattr(process,"NjettinessAK8Subjets"+postfix).src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix, "SubJets")
     getattr(process,"patJetsAK8PFPuppiSoftDrop").userData.userFloats.src += ['nb1AK8PuppiSoftDrop'+postfix+':ecfN2','nb1AK8PuppiSoftDrop'+postfix+':ecfN3']
     getattr(process,"patJetsAK8PFPuppiSoftDrop").userData.userFloats.src += ['nb2AK8PuppiSoftDrop'+postfix+':ecfN2','nb2AK8PuppiSoftDrop'+postfix+':ecfN3']
     addToProcessAndTask('nb1AK8PuppiSoftDropSubjets'+postfix, process.ecfNbeta1.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix, "SubJets")), process, task)
     addToProcessAndTask('nb2AK8PuppiSoftDropSubjets'+postfix, process.ecfNbeta2.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix, "SubJets")), process, task)
     getattr(process,"patJetsAK8PFPuppiSoftDropSubjets"+postfix).userData.userFloats.src += ['nb1AK8PuppiSoftDropSubjets'+postfix+':ecfN2','nb1AK8PuppiSoftDropSubjets'+postfix+':ecfN3']
     getattr(process,"patJetsAK8PFPuppiSoftDropSubjets"+postfix).userData.userFloats.src += ['nb2AK8PuppiSoftDropSubjets'+postfix+':ecfN2','nb2AK8PuppiSoftDropSubjets'+postfix+':ecfN3']
     getattr(process,"patJetsAK8PFPuppiSoftDropSubjets"+postfix).userData.userFloats.src += ['NjettinessAK8Subjets'+postfix+':tau1','NjettinessAK8Subjets'+postfix+':tau2','NjettinessAK8Subjets'+postfix+':tau3','NjettinessAK8Subjets'+postfix+':tau4']
 
     for e in [pp_on_XeXe_2017, phase2_common]:
         e.toModify(getattr(process,'nb1AK8PuppiSoftDropSubjets'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
         e.toModify(getattr(process,'nb2AK8PuppiSoftDropSubjets'+postfix), cuts = ['pt > 999999', 'pt > 999999', 'pt > 999999'] )
 
         
     # Patify AK8 PF PUPPI
     addJetCollection(process, postfix=postfix, labelName = 'AK8Puppi',
                      jetSource = cms.InputTag('ak8PFJetsPuppi'+postfix),
                      algo= 'AK', rParam = 0.8,
                      jetCorrections = ('AK8PFPuppi', cms.vstring(['L2Relative', 'L3Absolute']), 'None'),
                      btagDiscriminators = ([
                          'pfCombinedSecondaryVertexV2BJetTags',
                          'pfCombinedInclusiveSecondaryVertexV2BJetTags',
                          'pfCombinedMVAV2BJetTags',
                          'pfDeepCSVJetTags:probb',
                          'pfDeepCSVJetTags:probc',
                          'pfDeepCSVJetTags:probudsg',
                          'pfDeepCSVJetTags:probbb',
                          'pfBoostedDoubleSecondaryVertexAK8BJetTags']),
                      genJetCollection = cms.InputTag('slimmedGenJetsAK8')
                      )
     getattr(process,"patJetsAK8Puppi"+postfix).userData.userFloats.src = [] # start with empty list of user floats
     getattr(process,"selectedPatJetsAK8Puppi"+postfix).cut = cms.string("pt > 100")
     getattr(process,"selectedPatJetsAK8Puppi"+postfix).cutLoose = cms.string("pt > 30")
     getattr(process,"selectedPatJetsAK8Puppi"+postfix).nLoose = cms.uint32(3)
 
     from Configuration.Eras.Modifier_run3_common_cff import run3_common
     run3_common.toModify(process.patJetsAK8Puppi,
                          discriminatorSources = cms.VInputTag(
                             cms.InputTag("pfDeepCSVJetTagsAK8Puppi","probb"),
                             cms.InputTag("pfDeepCSVJetTagsAK8Puppi","probc"),
                             cms.InputTag("pfDeepCSVJetTagsAK8Puppi","probudsg"),
                             cms.InputTag("pfDeepCSVJetTagsAK8Puppi","probbb")
                          )
     )
 
     from RecoJets.JetAssociationProducers.j2tParametersVX_cfi import j2tParametersVX
     addToProcessAndTask('ak8PFJetsPuppiTracksAssociatorAtVertex'+postfix, cms.EDProducer("JetTracksAssociatorAtVertex",
                                       j2tParametersVX.clone( coneSize = cms.double(0.8) ),
                                       jets = cms.InputTag("ak8PFJetsPuppi") ),
                         process, task)
     addToProcessAndTask('patJetAK8PuppiCharge'+postfix, cms.EDProducer("JetChargeProducer",
                                      src = cms.InputTag("ak8PFJetsPuppiTracksAssociatorAtVertex"),
                                      var = cms.string('Pt'),
                                      exp = cms.double(1.0) ), 
                         process, task)
 
     
     getattr(process,"patJetsAK8Puppi"+postfix).userData.userFloats.src += ['ak8PFJetsPuppiSoftDropMass'+postfix]
     getattr(process,"patJetsAK8Puppi"+postfix).addTagInfos = cms.bool(False)
 
 
     # add PUPPI Njetiness    
     addToProcessAndTask('NjettinessAK8Puppi'+postfix, Njettiness.clone(), process, task)
     getattr(process,"NjettinessAK8Puppi"+postfix).src = cms.InputTag("ak8PFJetsPuppi"+postfix)
     getattr(process,"patJetsAK8Puppi").userData.userFloats.src += ['NjettinessAK8Puppi'+postfix+':tau1','NjettinessAK8Puppi'+postfix+':tau2','NjettinessAK8Puppi'+postfix+':tau3','NjettinessAK8Puppi'+postfix+':tau4']
 
     
     addToProcessAndTask("slimmedJetsAK8PFPuppiSoftDropSubjets"+postfix,
                         cms.EDProducer("PATJetSlimmer",
                              src = cms.InputTag("selectedPatJetsAK8PFPuppiSoftDropSubjets"),
                              packedPFCandidates = cms.InputTag("packedPFCandidates"),
                              dropJetVars = cms.string("1"),
                              dropDaughters = cms.string("0"),
                              rekeyDaughters = cms.string("1"),
                              dropTrackRefs = cms.string("1"),
                              dropSpecific = cms.string("1"),
                              dropTagInfos = cms.string("1"),
                              modifyJets = cms.bool(True),
                              mixedDaughters = cms.bool(False),
                              modifierConfig = cms.PSet( modifications = cms.VPSet() )
                                        ),
                         process, task)
 
     
     
     addToProcessAndTask("slimmedJetsAK8PFPuppiSoftDropPacked"+postfix,
                         cms.EDProducer("BoostedJetMerger",
                                jetSrc=cms.InputTag("selectedPatJetsAK8PFPuppiSoftDrop"),
                                subjetSrc=cms.InputTag("slimmedJetsAK8PFPuppiSoftDropSubjets")
                                        ),
                         process, task )
 
     
     addToProcessAndTask("packedPatJetsAK8"+postfix, cms.EDProducer("JetSubstructurePacker",
                                            jetSrc = cms.InputTag("selectedPatJetsAK8Puppi"+postfix),
                                            distMax = cms.double(0.8),
                                            algoTags = cms.VInputTag(
                                                cms.InputTag("slimmedJetsAK8PFPuppiSoftDropPacked"+postfix)
                                            ),
                                            algoLabels = cms.vstring(
                                                'SoftDropPuppi'
                                            ),
                                           fixDaughters = cms.bool(True),
                                           packedPFCandidates = cms.InputTag("packedPFCandidates"+postfix),
                                                                    ),
                         process, task)
 
     # switch off daughter re-keying since it's done in the JetSubstructurePacker (and can't be done afterwards)
     process.slimmedJetsAK8.rekeyDaughters = "0"
     # Reconfigure the slimmedAK8 jet information to keep 
     process.slimmedJetsAK8.dropDaughters = cms.string("pt < 170")
     process.slimmedJetsAK8.dropSpecific = cms.string("pt < 170")
     process.slimmedJetsAK8.dropTagInfos = cms.string("pt < 170")
Functions

Function Documentation

◆ applySubstructure()
