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.

References _switchToPFTau(), and createfilelist.int.

Referenced by _switchToPFTau().

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

_switchToPFTau()

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

private

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

Definition at line 38 of file tauTools.py.

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

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

                   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)

applyPostfix()

def tauTools.applyPostfix	(		process,
			label,
			postfix
	)

Definition at line 10 of file tauTools.py.

Referenced by _switchToPFTau(), and switchToPFTauHPS().

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

switchToPFTauByType()

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

Definition at line 204 of file tauTools.py.

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

                        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)

switchToPFTauFixedCone()

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

Definition at line 164 of file tauTools.py.

References _switchToPFTau(), and switchToPFTauShrinkingCone().

                           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)

switchToPFTauHPS()

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

Definition at line 187 of file tauTools.py.

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

Referenced by switchToPFTauShrinkingCone().

                     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

switchToPFTauShrinkingCone()

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

Definition at line 175 of file tauTools.py.

References _switchToPFTau(), and switchToPFTauHPS().

Referenced by switchToPFTauFixedCone().

                               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

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.