jetCollectionTools.GenJetAdder Class Reference

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

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

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

Definition at line 88 of file jetCollectionTools.py.

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

Referenced by jetCollectionTools.GenJetAdder.getSequence().

    ):
    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
#
#============================================

def jetCollectionTools.GenJetAdder.getSequence	(		self,
			proc
	)

Definition at line 71 of file jetCollectionTools.py.

References jetCollectionTools.GenJetAdder.addGenJetCollection(), jetCollectionTools.GenJetAdder.main, custom_jme_cff.TableGenJetAdder.main, custom_jme_cff.TableRecoJetAdder.main, and jetCollectionTools.GenJetAdder.prerequisites.

  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
  

Member Data Documentation

jetCollectionTools.GenJetAdder.gpLabel

Definition at line 69 of file jetCollectionTools.py.

Referenced by jetCollectionTools.RecoJetAdder.addRecoJetCollection().

jetCollectionTools.GenJetAdder.main

Definition at line 68 of file jetCollectionTools.py.

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

jetCollectionTools.GenJetAdder.prerequisites

Definition at line 67 of file jetCollectionTools.py.

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