tauTools::AddTauCollection Class Reference

Inherits FWCore::GuiBrowsers::ConfigToolBase::ConfigToolBase.

List of all members.

Public Member Functions
def	__call__
def	__init__
def	getDefaultParameters
def	toolCode
Private Attributes
	_comment
	_parameters
Static Private Attributes
tuple	_defaultParameters = dicttypes.SortedKeysDict()
string	_label = 'addTauCollection'

---

Detailed Description

Add a new collection of taus. Takes the configuration from the
already configured standard tau collection as starting point;
replaces before calling addTauCollection will also affect the
new tau collections

Definition at line 280 of file tauTools.py.

---

Constructor & Destructor Documentation

def tauTools::AddTauCollection::__init__

(

self

)

Definition at line 289 of file tauTools.py.

00290                       :
00291         ConfigToolBase.__init__(self)
00292         self.addParameter(self._defaultParameters, 'tauCollection',
00293                           self._defaultValue, 'Input tau collection', cms.InputTag)
00294         self.addParameter(self._defaultParameters, 'algoLabel',
00295                           self._defaultValue, "label to indicate the tau algorithm (e.g.'shrinkingCone')", str)
00296         self.addParameter(self._defaultParameters, 'typeLabel',
00297                           self._defaultValue, "label to indicate the type of constituents (either 'PFTau' or 'Tau')", str)
00298         self.addParameter(self._defaultParameters, 'doPFIsoDeposits',
00299                           True, "run sequence for computing particle-flow based IsoDeposits")
00300         self.addParameter(self._defaultParameters, 'standardAlgo',
00301                           "shrinkingCone", "standard algorithm label of the collection from which the clones " \
00302                          + "for the new tau collection will be taken from " \
00303                          + "(note that this tau collection has to be available in the event before hand)")
00304         self.addParameter(self._defaultParameters, 'standardType',
00305                           "PFTau", "standard constituent type label of the collection from which the clones " \
00306                          + " for the new tau collection will be taken from "\
00307                          + "(note that this tau collection has to be available in the event before hand)")
00308         
00309         self._parameters=copy.deepcopy(self._defaultParameters)
00310         self._comment = ""
        

---

Member Function Documentation

def tauTools::AddTauCollection::__call__	(		self,
			process,
			tauCollection = None,
			algoLabel = None,
			typeLabel = None,
			doPFIsoDeposits = None,
			standardAlgo = None,
			standardType = None
	)

Definition at line 314 of file tauTools.py.

00321                                            :
00322 
00323         if tauCollection is None:
00324             tauCollection = self._defaultParameters['tauCollection'].value
00325         if algoLabel is None:
00326             algoLabel = self._defaultParameters['algoLabel'].value
00327         if typeLabel is None:
00328             typeLabel = self._defaultParameters['typeLabel'].value
00329         if doPFIsoDeposits is None:
00330             doPFIsoDeposits = self._defaultParameters['doPFIsoDeposits'].value
00331         if standardAlgo is None:
00332             standardAlgo = self._defaultParameters['standardAlgo'].value
00333         if standardType is None:
00334             standardType = self._defaultParameters['standardType'].value
00335 
00336         self.setParameter('tauCollection', tauCollection)
00337         self.setParameter('algoLabel', algoLabel)
00338         self.setParameter('typeLabel', typeLabel)
00339         self.setParameter('doPFIsoDeposits', doPFIsoDeposits)
00340         self.setParameter('standardAlgo', standardAlgo)
00341         self.setParameter('standardType', standardType)
00342    
00343         self.apply(process) 
        

def tauTools::AddTauCollection::getDefaultParameters

(

self

)

Definition at line 311 of file tauTools.py.

                                  :
        return self._defaultParameters

def tauTools::AddTauCollection::toolCode	(		self,
			process
	)

Definition at line 344 of file tauTools.py.

                               :        
        tauCollection = self._parameters['tauCollection'].value
        algoLabel = self._parameters['algoLabel'].value
        typeLabel = self._parameters['typeLabel'].value
        doPFIsoDeposits = self._parameters['doPFIsoDeposits'].value
        standardAlgo = self._parameters['standardAlgo'].value
        standardType = self._parameters['standardType'].value

        ## disable computation of particle-flow based IsoDeposits
        ## in case tau is of CaloTau type
        if typeLabel == 'Tau':
#            print "NO PF Isolation will be computed for CaloTau (this could be improved later)"
            doPFIsoDeposits = False
 
        ## create old module label from standardAlgo
        ## and standardType and return
        def oldLabel(prefix = ''):
            if prefix == '':
                return "patTaus"
            else:
                return prefix + "PatTaus"

        ## capitalize first character of appended part
        ## when creating new module label
        ## (giving e.g. "patTausCaloRecoTau")
        def capitalize(label):
            return label[0].capitalize() + label[1:]    

        ## create new module label from old module
        ## label and return
        def newLabel(oldLabel):
            newLabel = oldLabel
            if ( oldLabel.find(standardAlgo) >= 0 and oldLabel.find(standardType) >= 0 ):
                oldLabel = oldLabel.replace(standardAlgo, algoLabel).replace(standardType, typeLabel)
            else:
                oldLabel = oldLabel + capitalize(algoLabel + typeLabel)
            return oldLabel

        ## clone module and add it to the patDefaultSequence
        def addClone(hook, **replaceStatements):
            ## create a clone of the hook with corresponding
            ## parameter replacements
            newModule = getattr(process, hook).clone(**replaceStatements)
            ## add the module to the sequence
            addModuleToSequence(hook, newModule)

        ## clone module for computing particle-flow IsoDeposits
        def addPFIsoDepositClone(hook, **replaceStatements):
            newModule = getattr(process, hook).clone(**replaceStatements)
            newModuleIsoDepositExtractor = getattr(newModule, "ExtractorPSet")
            setattr(newModuleIsoDepositExtractor, "tauSource", getattr(newModule, "src"))
            addModuleToSequence(hook, newModule)
            
        ## add module to the patDefaultSequence
        def addModuleToSequence(hook, newModule):
            hookModule = getattr(process, hook)
            ## add the new module with standardAlgo &
            ## standardType replaced in module label
            setattr(process, newLabel(hook), newModule)
            ## add new module to default sequence
            ## just behind the hookModule
            process.patDefaultSequence.replace( hookModule, hookModule*newModule )        

        ## add a clone of patTaus
        addClone(oldLabel(), tauSource = tauCollection)
        
        ## add a clone of selectedPatTaus    
        addClone(oldLabel('selected'), src = cms.InputTag(newLabel(oldLabel())))
        
        ## add a clone of cleanPatTaus    
        addClone(oldLabel('clean'), src=cms.InputTag(newLabel(oldLabel('selected'))))

        ## get attributes of new module
        newTaus = getattr(process, newLabel(oldLabel()))

        ## add a clone of gen tau matching
        addClone('tauMatch', src = tauCollection)
        addClone('tauGenJetMatch', src = tauCollection)

        ## add a clone of IsoDeposits computed based on particle-flow
        if doPFIsoDeposits:
            addPFIsoDepositClone('tauIsoDepositPFCandidates', src = tauCollection)
            addPFIsoDepositClone('tauIsoDepositPFChargedHadrons', src = tauCollection)
            addPFIsoDepositClone('tauIsoDepositPFNeutralHadrons', src = tauCollection)
            addPFIsoDepositClone('tauIsoDepositPFGammas', src = tauCollection)

        ## fix label for input tag
        def fixInputTag(x): x.setModuleLabel(newLabel(x.moduleLabel))

        ## provide patTau inputs with individual labels
        fixInputTag(newTaus.genParticleMatch)
        fixInputTag(newTaus.genJetMatch)
        fixInputTag(newTaus.isoDeposits.pfAllParticles)
        fixInputTag(newTaus.isoDeposits.pfNeutralHadron)
        fixInputTag(newTaus.isoDeposits.pfChargedHadron)
        fixInputTag(newTaus.isoDeposits.pfGamma)
        fixInputTag(newTaus.userIsolation.pfAllParticles.src)
        fixInputTag(newTaus.userIsolation.pfNeutralHadron.src)
        fixInputTag(newTaus.userIsolation.pfChargedHadron.src)
        fixInputTag(newTaus.userIsolation.pfGamma.src)
        
        ## set discriminators
        ## (using switchTauCollection functions)
        oldTaus = getattr(process, oldLabel())
#        if typeLabel == 'Tau':
#            switchToCaloTau(process,
#                            pfTauLabel = getattr(oldTaus, "tauSource"),
#                            caloTauLabel = getattr(newTaus, "tauSource"),
#                            patTauLabel = capitalize(algoLabel + typeLabel))
#        else:
        switchToPFTauByType(process, pfTauType = algoLabel + typeLabel,
                                pfTauLabelNew = getattr(newTaus, "tauSource"),
                                pfTauLabelOld = getattr(oldTaus, "tauSource"),
                                patTauLabel = capitalize(algoLabel + typeLabel))
       
addTauCollection=AddTauCollection()

---

Member Data Documentation

tauTools::AddTauCollection::_comment [private]

Definition at line 289 of file tauTools.py.

tuple tauTools::AddTauCollection::_defaultParameters = dicttypes.SortedKeysDict() [static, private]

Definition at line 288 of file tauTools.py.

string tauTools::AddTauCollection::_label = 'addTauCollection' [static, private]

Definition at line 287 of file tauTools.py.

tauTools::AddTauCollection::_parameters [private]

Definition at line 289 of file tauTools.py.