PhysicsTools/Heppy/python/analyzers/examples/ZEleEleAnalyzer.py - master - githubmirror/cmssw

PhysicsTools/Heppy/python/analyzers/examples/ZEleEleAnalyzer.py

30 строк1 KB

Shahzad Malik Muzaffar

[ANALYSIS] py2/3 compatibility:drop use of __future__

22 ноя 2024, 20:16

22 ноя 2024, 20:168d027cc

from PhysicsTools.Heppy.analyzers.DiLeptonAnalyzer import DiLeptonAnalyzer
from PhysicsTools.Heppy.analyzers.AutoHandle import AutoHandle
from PhysicsTools.Heppy.physicsobjects.DiObject import DiElectron
from PhysicsTools.Heppy.physicsobjects.PhysicsObjects import Electron


class ZEleEleAnalyzer( DiLeptonAnalyzer ):

    DiObjectClass = DiElectron
    LeptonClass = Electron

    def declareHandles(self):
        super(ZEleEleAnalyzer, self).declareHandles()
        print('ZEleEleAnalyzer.declareHandles')
        self.handles['diLeptons'] = AutoHandle(
            'cmgDiElectronSelStdLep',
            'std::vector<cmg::DiObject<cmg::Electron,cmg::Electron>>'
            )
        self.handles['leptons'] = AutoHandle(
            'cmgElectronSelStdLep',
            'std::vector<cmg::Electron>'
            )

    def testLeg1(self, leg):
        return self.testElectron(leg) and \
               super( ZEleEleAnalyzer, self).testLeg1( leg )

    def testLeg2(self, leg):
        return self.testElectron(leg) and \
               super( ZEleEleAnalyzer, self).testLeg2( leg )