jetCollectionTools.RecoJetInfo Class Reference

Inheritance diagram for jetCollectionTools.RecoJetInfo:

Public Member Functions
def	__init__ (self, jet, inputCollection)

Public Attributes
	doCalo
	doCS
	doPF
	inputCollection
	jet
	jetAlgo
	jetCorrPayload
	jetLower
	jetPUMethod
	jetReco
	jetSize
	jetSizeNr
	jetTagName
	jetUpper
	patJetFinalCollection
	skipUserData

Detailed Description

Class to hold information of a recojet collection

Definition at line 133 of file jetCollectionTools.py.

Constructor & Destructor Documentation

def jetCollectionTools.RecoJetInfo.__init__	(		self,
			jet,
			inputCollection
	)

Definition at line 137 of file jetCollectionTools.py.

  def __init__(self, jet, inputCollection):
    self.jet = jet
    self.jetLower   = jet.lower()
    self.jetUpper   = jet.upper()
    self.jetTagName = self.jetUpper
    self.inputCollection = inputCollection
    algoKey     = 'algo'
    sizeKey     = 'size'
    recoKey     = 'reco'
    puMethodKey = 'puMethod'
    jetRegex = re.compile(
      r'(?P<{algo}>({algoList}))(?P<{size}>[0-9]+)(?P<{reco}>(pf|calo))(?P<{puMethod}>(chs|puppi|sk|cs|))'.format(
        algo     = algoKey,
        algoList = '|'.join(supportedJetAlgos.keys()),
        size     = sizeKey,
        reco     = recoKey,
        puMethod = puMethodKey,
      )
    )
    jetMatch = jetRegex.match(jet.lower())
    if not jetMatch:
      raise RuntimeError('Invalid jet collection: %s' % jet)

    self.jetAlgo     = jetMatch.group(algoKey)
    self.jetSize     = jetMatch.group(sizeKey)
    self.jetReco     = jetMatch.group(recoKey)
    self.jetPUMethod = jetMatch.group(puMethodKey)

    self.jetSizeNr = float(self.jetSize) / 10.

    self.doCalo = self.jetReco == "calo"
    self.doPF   = self.jetReco == "pf"

    self.doCS   = self.jetPUMethod == "cs"
    self.skipUserData = self.doCalo or (self.jetPUMethod in [ "puppi", "sk" ] and inputCollection == "")
    
    self.jetCorrPayload = "{}{}{}".format(
      self.jetAlgo.upper(), self.jetSize, "Calo" if self.doCalo else self.jetReco.upper()
    )

    if self.jetPUMethod == "puppi":
      self.jetCorrPayload += "Puppi"
    elif self.jetPUMethod in [ "cs", "sk" ]:
      self.jetCorrPayload += "chs"
    else:
      self.jetCorrPayload += self.jetPUMethod.lower()

    self.patJetFinalCollection = ""
      
#============================================
#
# RecoJetAdder
#
#============================================

Member Data Documentation

jetCollectionTools.RecoJetInfo.doCalo

Definition at line 167 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.doCS

Definition at line 170 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.doPF

Definition at line 168 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.inputCollection

Definition at line 142 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jet

Definition at line 138 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetAlgo

Definition at line 160 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetCorrPayload

Definition at line 173 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetLower

Definition at line 139 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetPUMethod

Definition at line 163 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetReco

Definition at line 162 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetSize

Definition at line 161 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetSizeNr

Definition at line 165 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetTagName

Definition at line 141 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetUpper

Definition at line 140 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.patJetFinalCollection

Definition at line 184 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.skipUserData

Definition at line 171 of file jetCollectionTools.py.