jetCollectionTools.RecoJetInfo Class Reference

Inheritance diagram for jetCollectionTools.RecoJetInfo:

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

Public Attributes
	doCalo
	doCS
	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 151 of file jetCollectionTools.py.

Constructor & Destructor Documentation

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

Definition at line 155 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.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 185 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.doCS

Definition at line 186 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.inputCollection

Definition at line 160 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jet

Definition at line 156 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetAlgo

Definition at line 178 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetCorrPayload

Definition at line 189 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetLower

Definition at line 157 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetPUMethod

Definition at line 181 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetReco

Definition at line 180 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetSize

Definition at line 179 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetSizeNr

Definition at line 183 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetTagName

Definition at line 159 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.jetUpper

Definition at line 158 of file jetCollectionTools.py.

jetCollectionTools.RecoJetInfo.skipUserData

Definition at line 187 of file jetCollectionTools.py.