DiLeptonAnalyzer.DiLeptonAnalyzer Class Reference

Inheritance diagram for DiLeptonAnalyzer.DiLeptonAnalyzer:

Public Member Functions
def	beginLoop (self, setup)
def	bestDiLepton (self, diLeptons)
def	buildDiLeptons (self, cmgDiLeptons, event)
def	buildLeptons (self, cmgLeptons, event)
def	buildOtherLeptons (self, cmgLeptons, event)
def	declareHandles (self)
def	leptonAccept (self, leptons)
def	process (self, iEvent, event)
def	selectionSequence (self, event, fillCounter, leg1IsoCut=None, leg2IsoCut=None)
def	shiftEnergyScale (self, event)
def	testDeltaR (self, diLepton)
def	testLeg1 (self, leg, isocut=None)
def	testLeg1ID (self, leg)
def	testLeg1Iso (self, leg, isocut)
def	testLeg2 (self, leg, isocut=None)
def	testLeg2ID (self, leg)
def	testLeg2Iso (self, leg, isocut)
def	testLegKine (self, leg, ptcut, etacut)
def	testMass (self, diLepton)
def	thirdLeptonVeto (self, leptons, otherLeptons, isoCut=0.3)
def	trigMatched (self, event, leg, legName)

Static Public Attributes
	DiObjectClass
	LeptonClass
	OtherLeptonClass

Detailed Description

Generic analyzer for Di-Leptons.
See ZMuMuAnalyzer for a concrete case.

Example configuration, and list of parameters:
#O means optional

ana = cfg.Analyzer(
  'DiLeptonAnalyzer',
  scaleShift1 = eScaleShift,  #O shift factor for leg 1 energy scale
  scaleShift2 = tauScaleShift,#O same for leg 2
  pt1 = 20,   # pt, eta, iso cuts for leg 1
  eta1 = 2.3,
  iso1 = None,
  pt2 = 20,   # same for leg 2
  eta2 = 2.1,
  iso2 = 0.1,
  m_min = 10,    # mass range
  m_max = 99999,
  dR_min = 0.5,  #O min delta R between the two legs
  triggerMap = pathsAndFilters, #O, necessary for trigger matching
  verbose = False               #from base Analyzer class
  )

COLIN: need to specify what is needed in the event.
COLIN: need to make delta R non optional.
COLIN: make the dR_min parameter non optional

Definition at line 10 of file DiLeptonAnalyzer.py.

Member Function Documentation

beginLoop()

def DiLeptonAnalyzer.DiLeptonAnalyzer.beginLoop	(		self,
			setup
	)

Definition at line 49 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana, CSCSPEvent.counters(), cms::cuda::AtomicPairCounter::Atomic2.counters, NMaxPerLumi.counters, analyzer.Analyzer.counters, HistogramManager.counters, counter.Counters.counters, and cscdqm::ChamberCounterKeyType.counters.

    def beginLoop(self, setup):
        super(DiLeptonAnalyzer,self).beginLoop(setup)
        self.counters.addCounter('DiLepton')
        count = self.counters.counter('DiLepton')
        count.register('all events')
        count.register('> 0 di-lepton')
        # count.register('di-lepton cut string ok')
        count.register('lepton accept')
        count.register('third lepton veto')
        count.register('leg1 offline cuts passed')
        count.register('leg1 trig matched')
        count.register('leg2 offline cuts passed')
        count.register('leg2 trig matched')
        count.register('{min:3.1f} < m < {max:3.1f}'.format( min = self.cfg_ana.m_min,
                                                             max = self.cfg_ana.m_max ))
        if hasattr(self.cfg_ana, 'dR_min'):
            count.register('dR > {min:3.1f}'.format( min = self.cfg_ana.dR_min))
                           
        count.register('exactly 1 di-lepton')

bestDiLepton()

def DiLeptonAnalyzer.DiLeptonAnalyzer.bestDiLepton	(		self,
			diLeptons
	)

Returns the best diLepton (the one with highest pt1 + pt2).

Definition at line 293 of file DiLeptonAnalyzer.py.

References SiStripPI.max.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def bestDiLepton(self, diLeptons):
        '''Returns the best diLepton (the one with highest pt1 + pt2).'''
        return max( diLeptons, key=operator.methodcaller( 'sumPt' ) )
    

buildDiLeptons()

def DiLeptonAnalyzer.DiLeptonAnalyzer.buildDiLeptons	(		self,
			cmgDiLeptons,
			event
	)

Creates python DiLeptons from the di-leptons read from the disk.
to be overloaded if needed.

Definition at line 70 of file DiLeptonAnalyzer.py.

References __class__< T >.__class__(), and genParticles_cff.map.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.process().

    def buildDiLeptons(self, cmgDiLeptons, event):
        '''Creates python DiLeptons from the di-leptons read from the disk.
        to be overloaded if needed.'''
        return map( self.__class__.DiObjectClass, cmgDiLeptons )

buildLeptons()

def DiLeptonAnalyzer.DiLeptonAnalyzer.buildLeptons	(		self,
			cmgLeptons,
			event
	)

Creates python Leptons from the leptons read from the disk.
to be overloaded if needed.

Definition at line 76 of file DiLeptonAnalyzer.py.

References __class__< T >.__class__(), and genParticles_cff.map.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.process().

    def buildLeptons(self, cmgLeptons, event):
        '''Creates python Leptons from the leptons read from the disk.
        to be overloaded if needed.'''
        return map( self.__class__.LeptonClass, cmgLeptons )

buildOtherLeptons()

def DiLeptonAnalyzer.DiLeptonAnalyzer.buildOtherLeptons	(		self,
			cmgLeptons,
			event
	)

Creates python Leptons from the leptons read from the disk.
to be overloaded if needed.

Definition at line 82 of file DiLeptonAnalyzer.py.

References __class__< T >.__class__(), and genParticles_cff.map.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.process().

    def buildOtherLeptons(self, cmgLeptons, event):
        '''Creates python Leptons from the leptons read from the disk.
        to be overloaded if needed.'''
        return map( self.__class__.LeptonClass, cmgLeptons )


        

declareHandles()

def DiLeptonAnalyzer.DiLeptonAnalyzer.declareHandles

(

self

)

Definition at line 215 of file DiLeptonAnalyzer.py.

    def declareHandles(self):        
        super(DiLeptonAnalyzer, self).declareHandles()
        # self.handles['cmgTriggerObjectSel'] =  AutoHandle(
        #     'cmgTriggerObjectSel',
        #     'std::vector<cmg::TriggerObject>'
        #     )
    

leptonAccept()

def DiLeptonAnalyzer.DiLeptonAnalyzer.leptonAccept	(		self,
			leptons
	)

Should implement a default version running on event.leptons.

Definition at line 222 of file DiLeptonAnalyzer.py.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def leptonAccept(self, leptons):
        '''Should implement a default version running on event.leptons.'''
        return True
    

process()

def DiLeptonAnalyzer.DiLeptonAnalyzer.process	(		self,
			iEvent,
			event
	)

Definition at line 89 of file DiLeptonAnalyzer.py.

References DiLeptonAnalyzer.DiLeptonAnalyzer.buildDiLeptons(), DiLeptonAnalyzer.DiLeptonAnalyzer.buildLeptons(), DiLeptonAnalyzer.DiLeptonAnalyzer.buildOtherLeptons(), analyzer.Analyzer.cfg_ana, core.AutoHandle.AutoHandle.handles, core.Analyzer.Analyzer.handles, HTTP.RequestManager.handles, core.Analyzer.Analyzer.readCollections(), DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence(), and DiLeptonAnalyzer.DiLeptonAnalyzer.shiftEnergyScale().

    def process(self, iEvent, event):
        self.readCollections( iEvent )
        event.diLeptons = self.buildDiLeptons( self.handles['diLeptons'].product(), event )
        event.leptons = self.buildLeptons( self.handles['leptons'].product(), event )
        event.otherLeptons = self.buildOtherLeptons( self.handles['otherLeptons'].product(), event )
        self.shiftEnergyScale(event)
        return self.selectionSequence(event, fillCounter=True,
                                      leg1IsoCut=self.cfg_ana.iso1,
                                      leg2IsoCut=self.cfg_ana.iso2)

selectionSequence()

def DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence	(		self,
			event,
			fillCounter,
			leg1IsoCut = None,
			leg2IsoCut = None
	)

Definition at line 115 of file DiLeptonAnalyzer.py.

References DiLeptonAnalyzer.DiLeptonAnalyzer.bestDiLepton(), analyzer.Analyzer.cfg_ana, analyzer.Analyzer.cfg_comp, looper.Looper.cfg_comp, CSCSPEvent.counters(), cms::cuda::AtomicPairCounter::Atomic2.counters, NMaxPerLumi.counters, analyzer.Analyzer.counters, HistogramManager.counters, counter.Counters.counters, cscdqm::ChamberCounterKeyType.counters, DiLeptonAnalyzer.DiLeptonAnalyzer.leptonAccept(), DiLeptonAnalyzer.DiLeptonAnalyzer.testDeltaR(), DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1(), DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2(), DiLeptonAnalyzer.DiLeptonAnalyzer.testMass(), DiLeptonAnalyzer.DiLeptonAnalyzer.thirdLeptonVeto(), and DiLeptonAnalyzer.DiLeptonAnalyzer.trigMatched().

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.process().

    def selectionSequence(self, event, fillCounter, leg1IsoCut=None, leg2IsoCut=None):

        if fillCounter: self.counters.counter('DiLepton').inc('all events')
        # if not self.triggerList.triggerPassed(event.triggerObject):
        #    return False
        # self.counters.counter('DiLepton').inc('trigger passed ')

        # if event.eventId == 155035:
        #    import pdb; pdb.set_trace()
            
        # import pdb; pdb.set_trace()
        if len(event.diLeptons) == 0:
            return False
        if fillCounter: self.counters.counter('DiLepton').inc('> 0 di-lepton')

        # import pdb; pdb.set_trace()
        # testing di-lepton itself
        selDiLeptons = event.diLeptons
        # selDiLeptons = self.selectDiLeptons( selDiLeptons ) 

        event.leptonAccept = False
        if self.leptonAccept( event.leptons ):
            if fillCounter: self.counters.counter('DiLepton').inc('lepton accept')
            event.leptonAccept = True

        event.thirdLeptonVeto = False
        if self.thirdLeptonVeto(event.leptons, event.otherLeptons):
            if fillCounter: self.counters.counter('DiLepton').inc('third lepton veto')
            event.thirdLeptonVeto = True

        # testing leg1
        selDiLeptons = [ diL for diL in selDiLeptons if \
                         self.testLeg1( diL.leg1(), leg1IsoCut ) ]
        if len(selDiLeptons) == 0:
            return False
        else:
            if fillCounter: self.counters.counter('DiLepton').inc('leg1 offline cuts passed')

        if len(self.cfg_comp.triggers)>0:
            # trigger matching leg1
            selDiLeptons = [diL for diL in selDiLeptons if \
                            self.trigMatched(event, diL.leg1(), 'leg1')]
            if len(selDiLeptons) == 0:
                return False
            else:
                if fillCounter: self.counters.counter('DiLepton').inc('leg1 trig matched')

        # testing leg2 
        selDiLeptons = [ diL for diL in selDiLeptons if \
                         self.testLeg2( diL.leg2(), leg2IsoCut ) ]
        if len(selDiLeptons) == 0:
            return False
        else:
            if fillCounter: self.counters.counter('DiLepton').inc('leg2 offline cuts passed')

        if len(self.cfg_comp.triggers)>0:
            # trigger matching leg2
            selDiLeptons = [diL for diL in selDiLeptons if \
                            self.trigMatched(event, diL.leg2(), 'leg2')]
            if len(selDiLeptons) == 0:
                return False
            else:
                if fillCounter: self.counters.counter('DiLepton').inc('leg2 trig matched')

        # mass cut 
        selDiLeptons = [ diL for diL in selDiLeptons if \
                         self.testMass(diL) ]
        if len(selDiLeptons)==0:
            return False
        else:
            if fillCounter: self.counters.counter('DiLepton').inc(
                '{min:3.1f} < m < {max:3.1f}'.format( min = self.cfg_ana.m_min,
                                                      max = self.cfg_ana.m_max )
                )

        # delta R cut
        if hasattr(self.cfg_ana, 'dR_min'):
            selDiLeptons = [ diL for diL in selDiLeptons if \
                             self.testDeltaR(diL) ]
            if len(selDiLeptons)==0:
                return False
            else:
                if fillCounter: self.counters.counter('DiLepton').inc(
                    'dR > {min:3.1f}'.format( min = self.cfg_ana.dR_min )
                )

        # exactly one? 
        if len(selDiLeptons)==0:
            return False
        elif len(selDiLeptons)==1:
            if fillCounter: self.counters.counter('DiLepton').inc('exactly 1 di-lepton')
        
        event.diLepton = self.bestDiLepton( selDiLeptons )
        event.leg1 = event.diLepton.leg1()
        event.leg2 = event.diLepton.leg2()
        event.selectedLeptons = [event.leg1, event.leg2]

        return True
    

shiftEnergyScale()

def DiLeptonAnalyzer.DiLeptonAnalyzer.shiftEnergyScale	(		self,
			event
	)

Definition at line 100 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana, and genParticles_cff.map.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.process().

    def shiftEnergyScale(self, event):
        scaleShift1 = None
        scaleShift2 = None
        if hasattr( self.cfg_ana, 'scaleShift1'):
            scaleShift1 = self.cfg_ana.scaleShift1
        if hasattr( self.cfg_ana, 'scaleShift2'):
            scaleShift2 = self.cfg_ana.scaleShift2
        if scaleShift1:
            # import pdb; pdb.set_trace()
            map( lambda x: x.leg1().scaleEnergy(scaleShift1), event.diLeptons )
        if scaleShift2:
            map( lambda x: x.leg2().scaleEnergy(scaleShift2), event.diLeptons )
            map( lambda x: x.scaleEnergy(scaleShift2), event.leptons )
        

testDeltaR()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testDeltaR	(		self,
			diLepton
	)

returns True if the two diLepton.leg1() and .leg2() have a delta R larger than the dR_min parameter.

Definition at line 286 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana, PbPb_ZMuSkimMuonDPG_cff.deltaR, and PVValHelper.eta.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def testDeltaR(self, diLepton):
        '''returns True if the two diLepton.leg1() and .leg2() have a delta R larger than the dR_min parameter.'''
        dR = deltaR( diLepton.leg1().eta(), diLepton.leg1().phi(),
                     diLepton.leg2().eta(), diLepton.leg2().phi())
        return dR > self.cfg_ana.dR_min

    

testLeg1()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1	(		self,
			leg,
			isocut = None
	)

returns testLeg1ID && testLeg1Iso && testLegKine for leg1

Definition at line 232 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana, ZMuMuAnalyzer.ZMuMuAnalyzer.testLeg1ID, DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1ID(), ZMuMuAnalyzer.ZMuMuAnalyzer.testLeg1Iso, DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1Iso(), and DiLeptonAnalyzer.DiLeptonAnalyzer.testLegKine().

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def testLeg1(self, leg, isocut=None):
        '''returns testLeg1ID && testLeg1Iso && testLegKine for leg1'''
        return self.testLeg1ID(leg) and \
               self.testLeg1Iso(leg, isocut) and \
               self.testLegKine(leg, self.cfg_ana.pt1, self.cfg_ana.eta1)

testLeg1ID()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1ID	(		self,
			leg
	)

Always return true by default, overload in your subclass

Definition at line 252 of file DiLeptonAnalyzer.py.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1().

    def testLeg1ID(self, leg):
        '''Always return true by default, overload in your subclass'''
        return True

    

testLeg1Iso()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1Iso	(		self,
			leg,
			isocut
	)

If isocut is None, the iso value is taken from the iso1 parameter.
Checks the standard dbeta corrected isolation.

Definition at line 257 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1().

    def testLeg1Iso(self, leg, isocut):
        '''If isocut is None, the iso value is taken from the iso1 parameter.
        Checks the standard dbeta corrected isolation.
        '''
        if isocut is None:
            isocut = self.cfg_ana.iso1
        return leg.relIso(0.5) < isocut

    

testLeg2()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2	(		self,
			leg,
			isocut = None
	)

returns testLeg2ID && testLeg2Iso && testLegKine for leg2

Definition at line 239 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana, ZMuMuAnalyzer.ZMuMuAnalyzer.testLeg2ID, DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2ID(), ZMuMuAnalyzer.ZMuMuAnalyzer.testLeg2Iso, DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2Iso(), and DiLeptonAnalyzer.DiLeptonAnalyzer.testLegKine().

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def testLeg2(self, leg, isocut=None):
        '''returns testLeg2ID && testLeg2Iso && testLegKine for leg2'''
        return self.testLeg2ID(leg) and \
               self.testLeg2Iso(leg, isocut) and \
               self.testLegKine(leg, self.cfg_ana.pt2, self.cfg_ana.eta2)

testLeg2ID()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2ID	(		self,
			leg
	)

Always return true by default, overload in your subclass

Definition at line 266 of file DiLeptonAnalyzer.py.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2().

    def testLeg2ID(self, leg):
        '''Always return true by default, overload in your subclass'''
        return True

    

testLeg2Iso()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2Iso	(		self,
			leg,
			isocut
	)

If isocut is None, the iso value is taken from the iso2 parameter.
Checks the standard dbeta corrected isolation.

Definition at line 271 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2().

    def testLeg2Iso(self, leg, isocut):
        '''If isocut is None, the iso value is taken from the iso2 parameter.
        Checks the standard dbeta corrected isolation.
        '''
        if isocut is None:
            isocut = self.cfg_ana.iso2
        return leg.relIso(0.5) < isocut

testLegKine()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testLegKine	(		self,
			leg,
			ptcut,
			etacut
	)

Tests pt and eta.

Definition at line 246 of file DiLeptonAnalyzer.py.

References funct.abs().

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg1(), and DiLeptonAnalyzer.DiLeptonAnalyzer.testLeg2().

    def testLegKine(self, leg, ptcut, etacut ):
        '''Tests pt and eta.'''
        return leg.pt() > ptcut and \
               abs(leg.eta()) < etacut 

    

testMass()

def DiLeptonAnalyzer.DiLeptonAnalyzer.testMass	(		self,
			diLepton
	)

returns True if the mass of the dilepton is between the m_min and m_max parameters

Definition at line 280 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def testMass(self, diLepton):
        '''returns True if the mass of the dilepton is between the m_min and m_max parameters'''
        mass = diLepton.mass()
        return self.cfg_ana.m_min < mass and mass < self.cfg_ana.m_max

thirdLeptonVeto()

def DiLeptonAnalyzer.DiLeptonAnalyzer.thirdLeptonVeto	(		self,
			leptons,
			otherLeptons,
			isoCut = 0.3
	)

Should implement a default version running on event.leptons.

Definition at line 227 of file DiLeptonAnalyzer.py.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def thirdLeptonVeto(self, leptons, otherLeptons, isoCut = 0.3) :
        '''Should implement a default version running on event.leptons.'''
        return True

trigMatched()

def DiLeptonAnalyzer.DiLeptonAnalyzer.trigMatched	(		self,
			event,
			leg,
			legName
	)

Returns true if the leg is matched to a trigger object as defined in the
triggerMap parameter

Definition at line 298 of file DiLeptonAnalyzer.py.

References analyzer.Analyzer.cfg_ana.

Referenced by DiLeptonAnalyzer.DiLeptonAnalyzer.selectionSequence().

    def trigMatched(self, event, leg, legName):
        '''Returns true if the leg is matched to a trigger object as defined in the
        triggerMap parameter'''
        if not hasattr( self.cfg_ana, 'triggerMap'):
            return True
        path = event.hltPath
        triggerObjects = event.triggerObjects
        filters = self.cfg_ana.triggerMap[ path ]
        filter = None
        if legName == 'leg1':
            filter = filters[0]
        elif legName == 'leg2':
            filter = filters[1]
        else:
            raise ValueError( 'legName should be leg1 or leg2, not {leg}'.format(
                leg=legName )  )

        # JAN: Need a hack for the embedded samples: No trigger matching in that case
        if filter == '':
            return True

        # the dR2Max value is 0.3^2
        pdgIds = None
        if len(filter) == 2:
            filter, pdgIds = filter[0], filter[1]
        return triggerMatched(leg, triggerObjects, path, filter,
                              # dR2Max=0.089999,
                              dR2Max=0.25,
                              pdgIds=pdgIds )

Member Data Documentation

DiObjectClass

DiLeptonAnalyzer.DiLeptonAnalyzer.DiObjectClass

static

Definition at line 45 of file DiLeptonAnalyzer.py.

LeptonClass

DiLeptonAnalyzer.DiLeptonAnalyzer.LeptonClass

static

Definition at line 46 of file DiLeptonAnalyzer.py.

OtherLeptonClass

DiLeptonAnalyzer.DiLeptonAnalyzer.OtherLeptonClass

static

Definition at line 47 of file DiLeptonAnalyzer.py.

---

Generated for CMSSW Reference Manual by  1.8.14