custom_jme_cff.TableGenJetAdder Class Reference

Inheritance diagram for custom_jme_cff.TableGenJetAdder:

Public Member Functions
def	__init__ (self)
def	addTable (self, proc, genJetInfo)
def	getSequence (self, proc)

Public Attributes
	main

Detailed Description

Tool to store gen jet variables in NanoAOD for customized
gen jet collections.

Definition at line 191 of file custom_jme_cff.py.

Constructor & Destructor Documentation

def custom_jme_cff.TableGenJetAdder.__init__

(

self

)

Definition at line 196 of file custom_jme_cff.py.

  def __init__(self):
    self.main = []

Member Function Documentation

def custom_jme_cff.TableGenJetAdder.addTable	(		self,
			proc,
			genJetInfo
	)

Definition at line 213 of file custom_jme_cff.py.

References custom_jme_cff.TableGenJetAdder.main, and edm.print().

  def addTable(self, proc, genJetInfo):
    currentTasks = []
    
    print("custom_jme_cff::TableGenJetAdder::addTable: Adding Table for GenJet Collection: {}".format(genJetInfo.jet))
    
    name = nanoInfo_genjets[genJetInfo.jet]["name"]
    doc  = nanoInfo_genjets[genJetInfo.jet]["doc"]
    
    if name in genjetNameInNano:
      raise RuntimeError('GenJet collection name (%s) taken in NanoAOD for %s' %(name, genJetInfo.jet))

    #
    # GenJet Table
    #
    table = "{}Table".format(genJetInfo.jetTagName)
    genJetsCollection = "{}{}{}".format(genJetInfo.jetAlgo.upper(), genJetInfo.jetSize, 'GenJetsNoNu')
    setattr(proc, table, cms.EDProducer("SimpleCandidateFlatTableProducer",
        src       = cms.InputTag(genJetsCollection),
        cut       = cms.string(""),
        name      = cms.string(name),
        doc       = cms.string('{} (generator level)'.format(doc)),
        singleton = cms.bool(False),
        extension = cms.bool(False),
        variables = cms.PSet(P4Vars,
        area      = jetTable.variables.area,
        ),
      )
    )    
    currentTasks.append(table)

    #
    # GenJet Flavour Table
    #
    genFlavour = "{}Flavour".format(genJetInfo.jetTagName)
    genFlavourTable = "{}Table".format(genFlavour)
    if genFlavourTable in self.main:
      raise ValueError("Step '%s' already implemented" % genFlavourTable)
    setattr(proc, genFlavourTable, cms.EDProducer("GenJetFlavourTableProducer",
        name            = cms.string(name),
        src             = cms.InputTag(genJetsCollection),
        cut             = cms.string(""),
        deltaR          = cms.double(0.1),
        jetFlavourInfos = cms.InputTag(genFlavour),
      )
    )
    currentTasks.append(genFlavourTable)
    self.main.extend(currentTasks)

#============================================
#
# TableRecoJetAdder
#
#============================================

def custom_jme_cff.TableGenJetAdder.getSequence	(		self,
			proc
	)

Tool to add 

Definition at line 199 of file custom_jme_cff.py.

References custom_jme_cff.TableGenJetAdder.main.

  def getSequence(self, proc):
    """
    Tool to add 
    """
    tasks = 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

custom_jme_cff.TableGenJetAdder.main

Definition at line 197 of file custom_jme_cff.py.

Referenced by jetCollectionTools.RecoJetAdder.addRecoJetCollection(), custom_jme_cff.TableGenJetAdder.addTable(), jetCollectionTools.GenJetAdder.getSequence(), custom_jme_cff.TableGenJetAdder.getSequence(), jetCollectionTools.RecoJetAdder.getSequence(), and custom_jme_cff.TableRecoJetAdder.getSequence().