jetCollectionTools.GenJetAdder Class Reference

Inheritance diagram for jetCollectionTools.GenJetAdder:

Public Member Functions
def	__init__ (self)
def	addGenJetCollection (self, proc, jet, inputCollection="", minPtFastjet=None)
def	addProcessAndTask (self, proc, label, module)

Public Attributes
	gpLabel
	main
	prerequisites

Detailed Description

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

Definition at line 63 of file jetCollectionTools.py.

Constructor & Destructor Documentation

def jetCollectionTools.GenJetAdder.__init__

(

self

)

Definition at line 67 of file jetCollectionTools.py.

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

Member Function Documentation

def jetCollectionTools.GenJetAdder.addGenJetCollection	(		self,
			proc,
			jet,
			inputCollection = "",
			minPtFastjet = None
	)

Definition at line 81 of file jetCollectionTools.py.

References jetCollectionTools.GenJetAdder.addProcessAndTask(), jetCollectionTools.GenJetAdder.prerequisites, and edm.print().

Referenced by jetCollectionTools.GenJetAdder.addProcessAndTask().

    ):
    print("jetCollectionTools::GenJetAdder::addGenJetCollection: Adding Gen Jet Collection: {}".format(jet))

    #
    # Decide which genJet collection we are dealing with
    #
    genJetInfo = GenJetInfo(jet,inputCollection)
    jetLower = genJetInfo.jetLower
    jetUpper = genJetInfo.jetUpper

    #=======================================================
    #
    # If genJet 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:
        self.addProcessAndTask(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')
      self.addProcessAndTask(proc, genJetsCollection, ak4GenJets.clone(
          src           = packedGenPartNoNu,
          jetAlgorithm  = cms.string(supportedJetAlgos[genJetInfo.jetAlgo]),
          rParam        = cms.double(genJetInfo.jetSizeNr),
        )
      )
      #
      # Set minimum pt threshold of gen jets to be saved after fastjet clustering
      #
      if minPtFastjet != None:
        getattr(proc, genJetsCollection).jetPtMin = minPtFastjet
      self.prerequisites.append(genJetsCollection)

    return genJetInfo
#============================================
#
# RecoJetInfo
#
#============================================

def jetCollectionTools.GenJetAdder.addProcessAndTask	(		self,
			proc,
			label,
			module
	)

Definition at line 72 of file jetCollectionTools.py.

References jetCollectionTools.GenJetAdder.addGenJetCollection(), helpers.addToProcessAndTask(), and helpers.getPatAlgosToolsTask().

Referenced by jetCollectionTools.GenJetAdder.addGenJetCollection(), and jetCollectionTools.RecoJetAdder.addRecoJetCollection().

  def addProcessAndTask(self, proc, label, module):
    task = getPatAlgosToolsTask(proc)
    addToProcessAndTask(label, module, proc, task)

Member Data Documentation

jetCollectionTools.GenJetAdder.gpLabel

Definition at line 70 of file jetCollectionTools.py.

Referenced by jetCollectionTools.RecoJetAdder.addRecoJetCollection().

jetCollectionTools.GenJetAdder.main

Definition at line 69 of file jetCollectionTools.py.

Referenced by jetCollectionTools.RecoJetAdder.addRecoJetCollection().

jetCollectionTools.GenJetAdder.prerequisites

Definition at line 68 of file jetCollectionTools.py.

Referenced by jetCollectionTools.GenJetAdder.addGenJetCollection(), and jetCollectionTools.RecoJetAdder.addRecoJetCollection().