custom_jme_cff.TableRecoJetAdder Class Reference

Inheritance diagram for custom_jme_cff.TableRecoJetAdder:

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

Public Attributes
	main

Detailed Description

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

Definition at line 261 of file custom_jme_cff.py.

Constructor & Destructor Documentation

def custom_jme_cff.TableRecoJetAdder.__init__

(

self

)

Definition at line 266 of file custom_jme_cff.py.

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

Member Function Documentation

def custom_jme_cff.TableRecoJetAdder.addTable	(		self,
			proc,
			recoJetInfo
	)

Definition at line 280 of file custom_jme_cff.py.

References edm.print(), and common_cff.Var().

  def addTable(self, proc, recoJetInfo):

    currentTasks = []

    print("custom_jme_cff::TableRecoJetAdder::addTable: Adding Table for Reco Jet Collection: {}".format(recoJetInfo.jet))

    name = nanoInfo_recojets[recoJetInfo.jet]["name"]
    doc  = nanoInfo_recojets[recoJetInfo.jet]["doc"]

    if name in recojetNameInNano:
      raise RuntimeError('RecoJet collection name (%s) taken in NanoAOD for %s' %(name, recoJetInfo.jet))

    table = "{}Table".format(recoJetInfo.jetTagName)
    if recoJetInfo.skipUserData:
      if recoJetInfo.doCalo:
        tableContents = cms.PSet(
          P4Vars,
          area      = jetTable.variables.area,
          rawFactor = jetTable.variables.rawFactor,
          emf       = Var("emEnergyFraction()", float, doc = "electromagnetic energy fraction", precision = 10),
        )
      else:
        tableContents = cms.PSet(
          P4Vars,
          area      = jetTable.variables.area,
          rawFactor = jetTable.variables.rawFactor,
        )
    elif "puppi" in recoJetInfo.jet:
      tableContents = JETVARS.clone(
        puppiMultiplicity = Var("userFloat('patPuppiJetSpecificProducer:puppiMultiplicity')",float,doc="Sum of PUPPI weights of particles in the jet", precision= 6),
        neutralPuppiMultiplicity = Var("userFloat('patPuppiJetSpecificProducer:neutralPuppiMultiplicity')",float,doc="Sum of PUPPI weights of neutral particles in the jet", precision= 6)
      )
    else:
      tableContents = JETVARS.clone()
    
    updatedJets = "updatedJets{}".format(recoJetInfo.jetTagName)
    setattr(proc, table, cms.EDProducer("SimpleCandidateFlatTableProducer",
        src       = cms.InputTag(updatedJets),
        cut       = cms.string(""),
        name      = cms.string(name),
        doc       = cms.string(doc),
        singleton = cms.bool(False),
        extension = cms.bool(False),
        variables = tableContents,
      )
    )
    currentTasks.append(table)

    self.main.extend(currentTasks)

def custom_jme_cff.TableRecoJetAdder.getSequence	(		self,
			proc
	)

Definition at line 269 of file custom_jme_cff.py.

References custom_jme_cff.TableGenJetAdder.main, and custom_jme_cff.TableRecoJetAdder.main.

  def getSequence(self, proc):
    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.TableRecoJetAdder.main

Definition at line 267 of file custom_jme_cff.py.

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