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
	self.addParameter(self._defaultParameters, 'jetCorrLabel', (pfTauJECpayloadMapping, ['L2Relative', 'L3Absolute']), "payload and list of new jet correction labels", tuple, acceptNoneValue = True)
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 348 of file tauTools.py.

---

Constructor & Destructor Documentation

def tauTools::AddTauCollection::__init__

(

self

)

Definition at line 357 of file tauTools.py.

                      :
        ConfigToolBase.__init__(self)
        self.addParameter(self._defaultParameters, 'tauCollection',
                          self._defaultValue, 'Input tau collection', cms.InputTag)
        self.addParameter(self._defaultParameters, 'algoLabel',
                          self._defaultValue, "label to indicate the tau algorithm (e.g.'hps')", str)
        self.addParameter(self._defaultParameters, 'typeLabel',
                          self._defaultValue, "label to indicate the type of constituents (either 'PFTau' or 'Tau')", str)
        self.addParameter(self._defaultParameters, 'doPFIsoDeposits',
                          True, "run sequence for computing particle-flow based IsoDeposits")
        ##self.addParameter(self._defaultParameters, 'jetCorrLabel',
        ##                  (pfTauJECpayloadMapping, ['L2Relative', 'L3Absolute']),
        ##                  "payload and list of new jet correction labels", tuple, acceptNoneValue = True)
        self.addParameter(self._defaultParameters, 'jetCorrLabel',
                          None, "payload and list of new jet correction labels", tuple, acceptNoneValue = True)
        self.addParameter(self._defaultParameters, 'standardAlgo',
                          "hps", "standard algorithm label of the collection from which the clones " \
                         + "for the new tau collection will be taken from " \
                         + "(note that this tau collection has to be available in the event before hand)")
        self.addParameter(self._defaultParameters, 'standardType',
                          "PFTau", "standard constituent type label of the collection from which the clones " \
                         + " for the new tau collection will be taken from "\
                         + "(note that this tau collection has to be available in the event before hand)")

        self._parameters=copy.deepcopy(self._defaultParameters)
        self._comment = ""

---

Member Function Documentation

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

Definition at line 387 of file tauTools.py.

                                           :

        if tauCollection is None:
            tauCollection = self._defaultParameters['tauCollection'].value
        if algoLabel is None:
            algoLabel = self._defaultParameters['algoLabel'].value
        if typeLabel is None:
            typeLabel = self._defaultParameters['typeLabel'].value
        if doPFIsoDeposits is None:
            doPFIsoDeposits = self._defaultParameters['doPFIsoDeposits'].value
        if jetCorrLabel is None:
            jetCorrLabel = self._defaultParameters['jetCorrLabel'].value
        if standardAlgo is None:
            standardAlgo = self._defaultParameters['standardAlgo'].value
        if standardType is None:
            standardType = self._defaultParameters['standardType'].value

        self.setParameter('tauCollection', tauCollection)
        self.setParameter('algoLabel', algoLabel)
        self.setParameter('typeLabel', typeLabel)
        self.setParameter('doPFIsoDeposits', doPFIsoDeposits)
        self.setParameter('jetCorrLabel', jetCorrLabel)
        self.setParameter('standardAlgo', standardAlgo)
        self.setParameter('standardType', standardType)

        self.apply(process)

def tauTools::AddTauCollection::getDefaultParameters

(

self

)

Definition at line 384 of file tauTools.py.

                                  :
        return self._defaultParameters

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

Definition at line 421 of file tauTools.py.

00422                                :
00423         tauCollection = self._parameters['tauCollection'].value
00424         algoLabel = self._parameters['algoLabel'].value
00425         typeLabel = self._parameters['typeLabel'].value
00426         doPFIsoDeposits = self._parameters['doPFIsoDeposits'].value
00427         jetCorrLabel = self._parameters['jetCorrLabel'].value
00428         standardAlgo = self._parameters['standardAlgo'].value
00429         standardType = self._parameters['standardType'].value
00430 
00431         ## disable computation of particle-flow based IsoDeposits
00432         ## in case tau is of CaloTau type
00433         if typeLabel == 'Tau':
00434 #            print "NO PF Isolation will be computed for CaloTau (this could be improved later)"
00435             doPFIsoDeposits = False
00436 
00437         ## create old module label from standardAlgo
00438         ## and standardType and return
00439         def oldLabel(prefix = ''):
00440             if prefix == '':
00441                 return "patTaus"
00442             else:
00443                 return prefix + "PatTaus"
00444 
00445         ## capitalize first character of appended part
00446         ## when creating new module label
00447         ## (giving e.g. "patTausCaloRecoTau")
00448         def capitalize(label):
00449             return label[0].capitalize() + label[1:]
00450 
00451         ## create new module label from old module
00452         ## label and return
00453         def newLabel(oldLabel):
00454             newLabel = oldLabel
00455             if ( oldLabel.find(standardAlgo) >= 0 and oldLabel.find(standardType) >= 0 ):
00456                 oldLabel = oldLabel.replace(standardAlgo, algoLabel).replace(standardType, typeLabel)
00457             else:
00458                 oldLabel = oldLabel + capitalize(algoLabel + typeLabel)
00459             return oldLabel
00460 
00461         ## clone module and add it to the patDefaultSequence
00462         def addClone(hook, **replaceStatements):
00463             ## create a clone of the hook with corresponding
00464             ## parameter replacements
00465             newModule = getattr(process, hook).clone(**replaceStatements)
00466             ## add the module to the sequence
00467             addModuleToSequence(hook, newModule)
00468 
00469         ## clone module for computing particle-flow IsoDeposits
00470         def addPFIsoDepositClone(hook, **replaceStatements):
00471             newModule = getattr(process, hook).clone(**replaceStatements)
00472             newModuleIsoDepositExtractor = getattr(newModule, "ExtractorPSet")
00473             setattr(newModuleIsoDepositExtractor, "tauSource", getattr(newModule, "src"))
00474             addModuleToSequence(hook, newModule)
00475 
00476         ## add module to the patDefaultSequence
00477         def addModuleToSequence(hook, newModule):
00478             hookModule = getattr(process, hook)
00479             ## add the new module with standardAlgo &
00480             ## standardType replaced in module label
00481             setattr(process, newLabel(hook), newModule)
00482             ## add new module to default sequence
00483             ## just behind the hookModule
00484             process.patDefaultSequence.replace( hookModule, hookModule*newModule )
00485 
00486         ## add a clone of patTaus
00487         addClone(oldLabel(), tauSource = tauCollection)
00488 
00489         ## add a clone of selectedPatTaus
00490         addClone(oldLabel('selected'), src = cms.InputTag(newLabel(oldLabel())))
00491 
00492         ## add a clone of cleanPatTaus
00493         addClone(oldLabel('clean'), src=cms.InputTag(newLabel(oldLabel('selected'))))
00494 
00495         ## get attributes of new module
00496         newTaus = getattr(process, newLabel(oldLabel()))
00497 
00498         ## add a clone of gen tau matching
00499         addClone('tauMatch', src = tauCollection)
00500         addClone('tauGenJetMatch', src = tauCollection)
00501 
00502         ## add a clone of IsoDeposits computed based on particle-flow
00503         if doPFIsoDeposits:
00504             addPFIsoDepositClone('tauIsoDepositPFCandidates', src = tauCollection)
00505             addPFIsoDepositClone('tauIsoDepositPFChargedHadrons', src = tauCollection)
00506             addPFIsoDepositClone('tauIsoDepositPFNeutralHadrons', src = tauCollection)
00507             addPFIsoDepositClone('tauIsoDepositPFGammas', src = tauCollection)
00508 
00509         if jetCorrLabel:
00510             addClone('patTauJetCorrFactors', src = tauCollection)
00511             getattr(process,newLabel('patTauJetCorrFactors')).payload = jetCorrLabel[0]
00512             getattr(process,newLabel('patTauJetCorrFactors')).levels = jetCorrLabel[1]
00513             getattr(process, newLabel('patTaus')).tauJetCorrFactorsSource = cms.VInputTag(cms.InputTag(newLabel('patTauJetCorrFactors')))
00514 
00515         ## fix label for input tag
00516         def fixInputTag(x): x.setModuleLabel(newLabel(x.moduleLabel))
00517 
00518         ## provide patTau inputs with individual labels
00519         fixInputTag(newTaus.genParticleMatch)
00520         fixInputTag(newTaus.genJetMatch)
00521         fixInputTag(newTaus.isoDeposits.pfAllParticles)
00522         fixInputTag(newTaus.isoDeposits.pfNeutralHadron)
00523         fixInputTag(newTaus.isoDeposits.pfChargedHadron)
00524         fixInputTag(newTaus.isoDeposits.pfGamma)
00525         fixInputTag(newTaus.userIsolation.pfAllParticles.src)
00526         fixInputTag(newTaus.userIsolation.pfNeutralHadron.src)
00527         fixInputTag(newTaus.userIsolation.pfChargedHadron.src)
00528         fixInputTag(newTaus.userIsolation.pfGamma.src)
00529 
00530         ## set discriminators
00531         ## (using switchTauCollection functions)
00532         oldTaus = getattr(process, oldLabel())
00533 #        if typeLabel == 'Tau':
00534 #            switchToCaloTau(process,
00535 #                            pfTauLabel = getattr(oldTaus, "tauSource"),
00536 #                            caloTauLabel = getattr(newTaus, "tauSource"),
00537 #                            patTauLabel = capitalize(algoLabel + typeLabel))
00538 #        else:
00539         switchToPFTauByType(process, pfTauType = algoLabel + typeLabel,
00540                                 pfTauLabelNew = getattr(newTaus, "tauSource"),
00541                                 pfTauLabelOld = getattr(oldTaus, "tauSource"),
00542                                 patTauLabel = capitalize(algoLabel + typeLabel))
00543 
addTauCollection=AddTauCollection()

---

Member Data Documentation

tauTools::AddTauCollection::_comment [private]

Definition at line 360 of file tauTools.py.

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

Definition at line 356 of file tauTools.py.

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

Definition at line 355 of file tauTools.py.

tauTools::AddTauCollection::_parameters [private]

self.addParameter(self._defaultParameters, 'jetCorrLabel', (pfTauJECpayloadMapping, ['L2Relative', 'L3Absolute']), "payload and list of new jet correction labels", tuple, acceptNoneValue = True)

Definition at line 360 of file tauTools.py.