tauTools Namespace Reference

Classes
class	AddTauCollection

Functions
def	_buildIDSourcePSet (tauType, idSources, postfix="")
def	_switchToPFTau (process, tauSource, pfTauType, idSources, patTauLabel="", postfix="")
def	applyPostfix (process, label, postfix)
def	switchToPFTauByType (process, pfTauType=None, tauSource=cms.InputTag('hpsPFTauProducer'), patTauLabel="", postfix="")
def	switchToPFTauFixedCone (process, tauSource=cms.InputTag('fixedConePFTauProducer'), patTauLabel="", postfix="")
def	switchToPFTauHPS (process, tauSource=cms.InputTag('hpsPFTauProducer'), patTauLabel="", jecLevels=[], postfix="")
def	switchToPFTauShrinkingCone (process, tauSource=cms.InputTag('shrinkingConePFTauProducer'), patTauLabel="", postfix="")

Variables
	classicPFTauIDSources
	classicTauIDSources
	hpsTauIDSources

Function Documentation

_buildIDSourcePSet()

def tauTools._buildIDSourcePSet (   tauType,   idSources,   postfix = ""  )

private

Build a PSet defining the tau ID sources to embed into the pat::Tau 

Definition at line 18 of file tauTools.py.

def _buildIDSourcePSet(tauType, idSources, postfix =""):
    """ Build a PSet defining the tau ID sources to embed into the pat::Tau """
    output = cms.PSet()
    for label, discriminator in idSources:
        wp = -99
        if ";" in discriminator: #read from PFTauDiscriminatorContainer
            discriminator, wp_str = discriminator.split(";")
            wp = int(wp_str)
        id = cms.PSet()
        id.inputTag = cms.InputTag(discriminator.replace(":", postfix + ":"))
        id.provenanceConfigLabel=cms.string("")
        id.idLabel=cms.string("")
        setattr(output, label, id)
    return output
 

References createfilelist.int.

Referenced by _switchToPFTau().

_switchToPFTau()

def tauTools._switchToPFTau (   process,   tauSource,   pfTauType,   idSources,   patTauLabel = "",   postfix = ""  )

private

internal auxiliary function to switch to **any** PFTau collection

Definition at line 33 of file tauTools.py.

def _switchToPFTau(process,
                   tauSource,
                   pfTauType,
                   idSources,
                   patTauLabel = "",
                   postfix = ""):
    """internal auxiliary function to switch to **any** PFTau collection"""
    print(' switching PAT Tau input to: ', tauSource)
 
    applyPostfix(process, "tauMatch" + patTauLabel, postfix).src = tauSource
    applyPostfix(process, "tauGenJetMatch" + patTauLabel, postfix).src = tauSource
    
    applyPostfix(process, "tauIsoDepositPFCandidates" + patTauLabel, postfix).src = tauSource
    applyPostfix(process, "tauIsoDepositPFCandidates" + patTauLabel, postfix).ExtractorPSet.tauSource = tauSource
    applyPostfix(process, "tauIsoDepositPFChargedHadrons" + patTauLabel, postfix).src = tauSource
    applyPostfix(process, "tauIsoDepositPFChargedHadrons" + patTauLabel, postfix).ExtractorPSet.tauSource = tauSource
    applyPostfix(process, "tauIsoDepositPFNeutralHadrons" + patTauLabel, postfix).src = tauSource
    applyPostfix(process, "tauIsoDepositPFNeutralHadrons" + patTauLabel, postfix).ExtractorPSet.tauSource = tauSource
    applyPostfix(process, "tauIsoDepositPFGammas" + patTauLabel, postfix).src = tauSource
    applyPostfix(process, "tauIsoDepositPFGammas" + patTauLabel, postfix).ExtractorPSet.tauSource = tauSource
    
    applyPostfix(process, "patTaus" + patTauLabel, postfix).tauSource = tauSource
    # CV: reconstruction of tau lifetime information not enabled for tau collections other than 'hpsPFTauProducer' yet
    applyPostfix(process, "patTaus" + patTauLabel, postfix).tauTransverseImpactParameterSource = ""
    applyPostfix(process, "patTaus" + patTauLabel, postfix).tauIDSources = _buildIDSourcePSet(pfTauType, idSources, postfix)
 
    if hasattr(process, "cleanPatTaus" + patTauLabel + postfix):
        getattr(process, "cleanPatTaus" + patTauLabel + postfix).preselection = preselection
 
# Name mapping for classic tau ID sources (present for fixed and shrinkingCones)

References _buildIDSourcePSet(), applyPostfix(), and print().

Referenced by switchToPFTauFixedCone(), switchToPFTauHPS(), and switchToPFTauShrinkingCone().

applyPostfix()

def tauTools.applyPostfix	(		process,
			label,
			postfix
	)

Definition at line 10 of file tauTools.py.

def applyPostfix(process, label, postfix):
    result = None
    if hasattr(process, label+postfix):
        result = getattr(process, label + postfix)
    else:
        raise ValueError("Error in <applyPostfix>: No module of name = %s attached to process !!" % (label + postfix))
    return result
 

Referenced by _switchToPFTau(), and switchToPFTauHPS().

switchToPFTauByType()

def tauTools.switchToPFTauByType	(		process,
			pfTauType = None,
			tauSource = cms.InputTag('hpsPFTauProducer'),
			patTauLabel = "",
			postfix = ""
	)

Definition at line 200 of file tauTools.py.

def switchToPFTauByType(process,
                        pfTauType = None,
                        tauSource = cms.InputTag('hpsPFTauProducer'),
                        patTauLabel = "",
                        postfix = "" ):
    mapping = {
        'shrinkingConePFTau' : switchToPFTauShrinkingCone,
        'fixedConePFTau'     : switchToPFTauFixedCone,
        'hpsPFTau'           : switchToPFTauHPS
    }
    if not pfTauType in mapping.keys():
        raise ValueError("Error in <switchToPFTauByType>: Undefined pfTauType = %s !!" % pfTauType)
    
    mapping[pfTauType](process, tauSource = tauSource,
                       patTauLabel = patTauLabel, postfix = postfix)
 

Referenced by pfTools.adaptPFTaus(), and tauTools.AddTauCollection.toolCode().

switchToPFTauFixedCone()

def tauTools.switchToPFTauFixedCone	(		process,
			tauSource = cms.InputTag('fixedConePFTauProducer'),
			patTauLabel = "",
			postfix = ""
	)

Definition at line 161 of file tauTools.py.

def switchToPFTauFixedCone(process,
                           tauSource = cms.InputTag('fixedConePFTauProducer'),
                           patTauLabel = "",
                           postfix = ""):
    fixedConeIDSources = copy.copy(classicTauIDSources)
    fixedConeIDSources.extend(classicPFTauIDSources)
 
    _switchToPFTau(process, tauSource, 'fixedConePFTau', fixedConeIDSources,
                   patTauLabel = patTauLabel, postfix = postfix)
 
# switch to PFTau collection produced for shrinking signal cone of size dR = 5.0/Et(PFTau)

References _switchToPFTau().

switchToPFTauHPS()

def tauTools.switchToPFTauHPS	(		process,
			tauSource = cms.InputTag('hpsPFTauProducer'),
			patTauLabel = "",
			jecLevels = [],
			postfix = ""
	)

Definition at line 183 of file tauTools.py.

def switchToPFTauHPS(process,
                     tauSource = cms.InputTag('hpsPFTauProducer'),
                     patTauLabel = "",
                     jecLevels = [],
                     postfix = ""):
 
    _switchToPFTau(process, tauSource, 'hpsPFTau', hpsTauIDSources,
                   patTauLabel = patTauLabel, postfix = postfix)
 
    # CV: enable tau lifetime information for HPS PFTaus
    applyPostfix(process, "patTaus" + patTauLabel, postfix).tauTransverseImpactParameterSource = tauSource.value().replace("Producer", "TransverseImpactParameters")
 
    
    if hasattr(process, "cleanPatTaus" + patTauLabel + postfix):
        getattr(process, "cleanPatTaus" + patTauLabel + postfix).preselection = preselection
 
# Select switcher by string

References _switchToPFTau(), applyPostfix(), and python.rootplot.root2matplotlib.replace().

switchToPFTauShrinkingCone()

def tauTools.switchToPFTauShrinkingCone	(		process,
			tauSource = cms.InputTag('shrinkingConePFTauProducer'),
			patTauLabel = "",
			postfix = ""
	)

Definition at line 172 of file tauTools.py.

def switchToPFTauShrinkingCone(process,
                               tauSource = cms.InputTag('shrinkingConePFTauProducer'),
                               patTauLabel = "",
                               postfix = ""):
    shrinkingIDSources = copy.copy(classicTauIDSources)
    shrinkingIDSources.extend(classicPFTauIDSources)
 
    _switchToPFTau(process, tauSource, 'shrinkingConePFTau', shrinkingIDSources,
                   patTauLabel = patTauLabel, postfix = postfix)
 
# switch to hadron-plus-strip(s) (HPS) PFTau collection

References _switchToPFTau().

Variable Documentation

classicPFTauIDSources

tauTools.classicPFTauIDSources

Definition at line 73 of file tauTools.py.

classicTauIDSources

tauTools.classicTauIDSources

Definition at line 63 of file tauTools.py.

hpsTauIDSources

tauTools.hpsTauIDSources

Definition at line 81 of file tauTools.py.