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 188 of file custom_jme_cff.py.

Constructor & Destructor Documentation

__init__()

def custom_jme_cff.TableGenJetAdder.__init__

(

self

)

Definition at line 193 of file custom_jme_cff.py.

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

Member Function Documentation

addTable()

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

Definition at line 210 of file custom_jme_cff.py.

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

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

getSequence()

def custom_jme_cff.TableGenJetAdder.getSequence	(		self,
			proc
	)

Tool to add 

Definition at line 196 of file custom_jme_cff.py.

  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
 

References custom_jme_cff.TableGenJetAdder.main.

Member Data Documentation

main

custom_jme_cff.TableGenJetAdder.main

Definition at line 194 of file custom_jme_cff.py.

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