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
	skipUserData

Detailed Description

Class to hold information of a recojet collection

Definition at line 129 of file jetCollectionTools.py.

Constructor & Destructor Documentation

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

Definition at line 133 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()
      
#============================================
#
# RecoJetAdder
#
#============================================

Member Data Documentation

jetCollectionTools.RecoJetInfo.doCalo

Definition at line 163 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.doCS

Definition at line 166 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.doPF

Definition at line 164 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.inputCollection

Definition at line 138 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jet

Definition at line 134 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetAlgo

Definition at line 156 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetCorrPayload

Definition at line 169 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetLower

Definition at line 135 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetPUMethod

Definition at line 159 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetReco

Definition at line 158 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetSize

Definition at line 157 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetSizeNr

Definition at line 161 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetTagName

Definition at line 137 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetUpper

Definition at line 136 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.skipUserData

Definition at line 167 of file jetCollectionTools.py.