Inheritance diagram for jetCollectionTools.GenJetAdder:

Public Member Functions
def	__init__ (self)

def	addGenJetCollection (self, proc, jet, inputCollection="", genName="", minPt=5.)

def	getSequence (self, proc)

Public Attributes
	gpLabel

	main

	prerequisites

Detailed Description

Tool to schedule modules for building a genjet collection with input MiniAODs

Definition at line 62 of file jetCollectionTools.py.

Constructor & Destructor Documentation

◆ init()

def jetCollectionTools.GenJetAdder.__init__ ( self )

Definition at line 66 of file jetCollectionTools.py.

   def __init__(self):
     self.prerequisites = []
     self.main = []
     self.gpLabel = "prunedGenParticles"
  

Member Function Documentation

◆ addGenJetCollection()

def jetCollectionTools.GenJetAdder.addGenJetCollection	(	self,
		proc,
		jet,
		inputCollection = `""`,
		genName = `""`,
		minPt = `5.`
	)

Definition at line 82 of file jetCollectionTools.py.

   def addGenJetCollection(self,
       proc,
       jet,
       inputCollection    = "",
       genName            = "",
       minPt              = 5.,
     ):
     print("jetCollectionTools::GenJetAdder::addGenJetCollection: Adding Gen Jet Collection: {}".format(jet))
     currentTasks = []
  
     #
     # Decide which jet collection we're dealing with
     #
     jetLower = jet.lower()
     jetUpper = jet.upper()
     tagName  = jetUpper
     genJetInfo = GenJetInfo(jet,inputCollection)
  
     #=======================================================
     #
     # If gen jet collection in MiniAOD is not 
     # specified, build the genjet collection.
     #
     #========================================================
     if not inputCollection:
       print("jetCollectionTools::GenJetAdder::addGenJetCollection: inputCollection not specified. Building genjet collection now")
       #
       # Setup GenParticles
       #
       packedGenPartNoNu = "packedGenParticlesForJetsNoNu"
       if packedGenPartNoNu not in self.prerequisites:
         setattr(proc, packedGenPartNoNu, cms.EDFilter("CandPtrSelector",
             src = cms.InputTag("packedGenParticles"),
             cut = cms.string("abs(pdgId) != 12 && abs(pdgId) != 14 && abs(pdgId) != 16"),
           )
         )
         self.prerequisites.append(packedGenPartNoNu)    
       #
       # Create the GenJet collection
       #
       genJetsCollection = "{}{}{}".format(genJetInfo.jetAlgo.upper(), genJetInfo.jetSize, 'GenJetsNoNu')
       setattr(proc, genJetsCollection, ak4GenJets.clone(
           src           = packedGenPartNoNu,
           jetAlgorithm  = cms.string(supportedJetAlgos[genJetInfo.jetAlgo]),
           rParam        = cms.double(genJetInfo.jetSizeNr),
         )
       )
       self.prerequisites.append(genJetsCollection)
     #
     # GenJet Flavour Labelling
     #
     genFlavour = "{}Flavour".format(genJetInfo.jetTagName)
     setattr(proc, genFlavour, patJetFlavourAssociation.clone(
         jets         = cms.InputTag(genJetsCollection),
         jetAlgorithm = cms.string(supportedJetAlgos[genJetInfo.jetAlgo]),
         rParam       = cms.double(genJetInfo.jetSizeNr),
       )
     )
  
     currentTasks.append(genFlavour)
     self.main.extend(currentTasks)
  
     return genJetInfo
  
 #============================================
 #
 # RecoJetInfo
 #
 #============================================

References mps_setup.append, jetCollectionTools.GenJetAdder.main, custom_jme_cff.TableGenJetAdder.main, custom_jme_cff.TableRecoJetAdder.main, jetCollectionTools.GenJetAdder.prerequisites, and edm.print().

◆ getSequence()

def jetCollectionTools.GenJetAdder.getSequence	(	self,
		proc
	)

Definition at line 71 of file jetCollectionTools.py.

   def getSequence(self, proc):
     tasks = self.prerequisites + self.main
  
     resultSequence = cms.Sequence()
     for idx, task in enumerate(tasks):
       if idx == 0:
         resultSequence = cms.Sequence(getattr(proc, task))
       else:
         resultSequence.insert(idx, getattr(proc, task))
     return resultSequence
   