trackTools::MakePATTrackCandidates 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 = 'makePATTrackCandidates'

---

Detailed Description

Create pat track candidates from AOD track collections:

Definition at line 68 of file trackTools.py.

---

Constructor & Destructor Documentation

def trackTools::MakePATTrackCandidates::__init__

(

self

)

Definition at line 75 of file trackTools.py.

                      :
        ConfigToolBase.__init__(self)
        self.addParameter(self._defaultParameters,'label','TrackCands', "output will be 'all/selectedLayer1'+label")
        self.addParameter(self._defaultParameters,'input',cms.InputTag('patAODTrackCands'), 'name of the input collection')
        self.addParameter(self._defaultParameters,'selection','pt > 10', 'selection on PAT Layer 1 objects')
        self.addParameter(self._defaultParameters,'isolation',{'tracker':0.3, 'ecalTowers':0.3, 'hcalTowers':0.3}, "solation to use (as 'source': value of dR)\ntracker     : as muon iso from tracks\necalTowers  : as muon iso from calo tower\nhcalTowers  : as muon iso from calo towers",allowedValues=['tracker','ecalTowers','hcalTowers'])
        self.addParameter(self._defaultParameters,'isoDeposits',['tracker','ecalTowers','hcalTowers'], 'iso deposits')
        self.addParameter(self._defaultParameters,'mcAs',None, "eplicate mc match as the one used by PAT on this AOD collection (None=no mc match); choose 'photon', 'electron', 'muon', 'tau','jet', 'met' as input string",Type=str, allowedValues=['photon', 'electron', 'muon', 'tau','jet', 'met', None], acceptNoneValue = True) 
        
        self._parameters=copy.deepcopy(self._defaultParameters)
        self._comment = ""

---

Member Function Documentation

def trackTools::MakePATTrackCandidates::__call__	(		self,
			process,
			label = None,
			input = None,
			selection = None,
			isolation = None,
			isoDeposits = None,
			mcAs = None
	)

Definition at line 90 of file trackTools.py.

00097                                      :
00098         if label  is None:
00099             label=self._defaultParameters['label'].value
00100         if input is None:
00101             input=self._defaultParameters['input'].value
00102         if selection is None:
00103             selection=self._defaultParameters['selection'].value
00104         if isolation is None:
00105             isolation=self._defaultParameters['isolation'].value
00106         if isoDeposits is None:
00107             isoDeposits=self._defaultParameters['isoDeposits'].value
00108         if mcAs is None:
00109             mcAs=self._defaultParameters['mcAs'].value    
00110         self.setParameter('label',label)
00111         self.setParameter('input',input)
00112         self.setParameter('selection',selection)
00113         self.setParameter('isolation',isolation)
00114         self.setParameter('isoDeposits',isoDeposits)
00115         self.setParameter('mcAs',mcAs,True)                                                                                              
00116         self.apply(process) 
        

def trackTools::MakePATTrackCandidates::getDefaultParameters

(

self

)

Definition at line 87 of file trackTools.py.

                                  :
        return self._defaultParameters

def trackTools::MakePATTrackCandidates::toolCode	(		self,
			process
	)

Definition at line 117 of file trackTools.py.

                               :                
        label=self._parameters['label'].value
        input=self._parameters['input'].value
        selection=self._parameters['selection'].value
        isolation=self._parameters['isolation'].value
        isoDeposits=self._parameters['isoDeposits'].value
        mcAs=self._parameters['mcAs'].value
                                                
        ## add patTracks to the process
        from PhysicsTools.PatAlgos.producersLayer1.genericParticleProducer_cfi import patGenericParticles
        setattr(process, 'pat' + label, patGenericParticles.clone(src = input))
        ## add selectedPatTracks to the process
        setattr(process, 'selectedPat' + label, cms.EDFilter("PATGenericParticleSelector",
                                                             src = cms.InputTag("pat"+label),
                                                             cut = cms.string(selection) 
                                                             ) 
                )
        ## add cleanPatTracks to the process
        from PhysicsTools.PatAlgos.cleaningLayer1.genericTrackCleaner_cfi import cleanPatTracks
        setattr(process, 'cleanPat' + label, cleanPatTracks.clone(src = cms.InputTag('selectedPat' + label)))
        
        ## get them as variables, so we can put them in the sequences and/or configure them
        l1cands         = getattr(process, 'pat' + label)
        selectedL1cands = getattr(process, 'selectedPat' + label)
        cleanL1cands    = getattr(process, 'cleanPat' + label)
        
        ## insert them in sequence, after the electrons
        process.patCandidates.replace(process.patElectrons, l1cands + process.patElectrons)
        process.selectedPatCandidates.replace(process.selectedPatElectrons, process.selectedPatElectrons + selectedL1cands)
        process.cleanPatCandidates.replace(process.cleanPatElectrons, process.cleanPatElectrons + cleanL1cands)
    
        ## add them to the Summary Tables
        process.patCandidateSummary.candidates += [ cms.InputTag("allPat"+label) ]
        process.selectedPatCandidateSummary.candidates += [ cms.InputTag("selectedPat"+label) ]
        process.cleanPatCandidateSummary.candidates += [ cms.InputTag("cleanPat"+label) ]
    
        ## isolation: start with empty config
        if(isolation or isoDeposits):
            process.load("TrackPropagation.SteppingHelixPropagator.SteppingHelixPropagatorAlong_cfi")
            process.load("TrackPropagation.SteppingHelixPropagator.SteppingHelixPropagatorOpposite_cfi")
            process.load("TrackPropagation.SteppingHelixPropagator.SteppingHelixPropagatorAny_cfi")
        isoModules = []
        runIsoDeps = {'tracker':False, 'caloTowers':False}
    
        for source,deltaR in isolation.items():
            ## loop items in isolation
            if(source == 'tracker'):
                runIsoDeps['tracker'] = True
                l1cands.userIsolation.tracker = cms.PSet(
                    src    = cms.InputTag('pat'+label+'IsoDepositTracks'),
                    deltaR = cms.double(deltaR),
                    )
            elif(source == 'ecalTowers'):
                runIsoDeps['caloTowers'] = True
                l1cands.userIsolation.ecal = cms.PSet(
                    src    = cms.InputTag('pat'+label+'IsoDepositCaloTowers', 'ecal'),
                    deltaR = cms.double(deltaR),
                    )
            elif(source == 'hcalTowers'):
                runIsoDeps['caloTowers'] = True
                l1cands.userIsolation.hcal = cms.PSet(
                    src    = cms.InputTag('pat'+label+'IsoDepositCaloTowers', 'hcal'),
                    deltaR = cms.double(deltaR),
                    )
            
        for source in isoDeposits:
            ## loop items in isoDeposits
            if(source == 'tracker'):
                runIsoDeps['tracker'] = True
                l1cands.isoDeposits.tracker = cms.InputTag('pat'+label+'IsoDepositTracks') 
            elif(source == 'ecalTowers'):
                runIsoDeps['caloTowers'] = True
                l1cands.isoDeposits.ecal = cms.InputTag('pat'+label+'IsoDepositCaloTowers', 'ecal') 
            elif(source == 'hcalTowers'):
                runIsoDeps['caloTowers'] = True
                l1cands.isoDeposits.hcal = cms.InputTag('pat'+label+'IsoDepositCaloTowers', 'hcal')
            
        for dep in [ dep for dep,runme in runIsoDeps.items() if runme == True ]:
            if(dep == 'tracker'):
                from RecoMuon.MuonIsolationProducers.trackExtractorBlocks_cff import MIsoTrackExtractorCtfBlock
                setattr(process, 'pat'+label+'IsoDepositTracks',
                        cms.EDProducer("CandIsoDepositProducer",
                                       src                  = input,
                                       trackType            = cms.string('best'),
                                       MultipleDepositsFlag = cms.bool(False),
                                       ExtractorPSet        = cms.PSet( MIsoTrackExtractorCtfBlock )
                                       )
                        )
                isoModules.append( getattr(process, 'pat'+label+'IsoDepositTracks') )
            elif(dep == 'caloTowers'):
                from RecoMuon.MuonIsolationProducers.caloExtractorByAssociatorBlocks_cff import MIsoCaloExtractorByAssociatorTowersBlock
                setattr(process, 'pat'+label+'IsoDepositCaloTowers',
                        cms.EDProducer("CandIsoDepositProducer",
                                       src                  = input,
                                       trackType            = cms.string('best'),
                                       MultipleDepositsFlag = cms.bool(True),
                                       ExtractorPSet        = cms.PSet( MIsoCaloExtractorByAssociatorTowersBlock )
                                       )
                        )
                isoModules.append( getattr(process, 'pat'+label+'IsoDepositCaloTowers') )
        for m in isoModules:
            process.patDefaultSequence.replace(l1cands, m * l1cands)
        # ES
        process.load( 'TrackingTools.TrackAssociator.DetIdAssociatorESProducer_cff' ) 
        # MC
        from PhysicsTools.PatAlgos.tools.helpers import MassSearchParamVisitor
        if(type(mcAs) != type(None)):
            findMatch= []
            findMatch.append(getattr(process, mcAs+'Match'))
            
            ## clone mc matchiong module of object mcAs and add it to the path
            setattr(process, 'pat'+label+'MCMatch', findMatch[0].clone(src = input))
            process.patDefaultSequence.replace( l1cands, getattr(process, 'pat'+label+'MCMatch') * l1cands)
            l1cands.addGenMatch = True
            l1cands.genParticleMatch = cms.InputTag('pat'+label+'MCMatch')

      
makePATTrackCandidates=MakePATTrackCandidates()

---

Member Data Documentation

trackTools::MakePATTrackCandidates::_comment [private]

Definition at line 75 of file trackTools.py.

tuple trackTools::MakePATTrackCandidates::_defaultParameters = dicttypes.SortedKeysDict() [static, private]

Definition at line 73 of file trackTools.py.

string trackTools::MakePATTrackCandidates::_label = 'makePATTrackCandidates' [static, private]

Definition at line 72 of file trackTools.py.

trackTools::MakePATTrackCandidates::_parameters [private]

Definition at line 75 of file trackTools.py.