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

Constructor & Destructor Documentation

def custom_jme_cff.TableRecoJetAdder.__init__

(

self

)

Definition at line 271 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 285 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,
        )
    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)

    tightJetIdLepVeto = "tightJetIdLepVeto{}".format(recoJetInfo.jetTagName)
    if not recoJetInfo.skipUserData:
      altTasks = copy.deepcopy(currentTasks)
      for idx, task in enumerate(altTasks):
        if task == tightJetIdLepVeto:
          altTasks[idx] = looseJetId
      for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
        modifier.toReplaceWith(currentTasks, altTasks)
    self.main.extend(currentTasks)

def custom_jme_cff.TableRecoJetAdder.getSequence	(		self,
			proc
	)

Definition at line 274 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 272 of file custom_jme_cff.py.

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