File indexing completed on 2024-11-25 02:29:49
0001 from PhysicsTools.Heppy.analyzers.core.TreeAnalyzerNumpy import TreeAnalyzerNumpy
0002 from . import ntuple
0003
0004 class ZJetsTreeAnalyzer(TreeAnalyzerNumpy):
0005
0006 def beginLoop(self, setup):
0007 super(ZJetsTreeAnalyzer, self).beginLoop(setup)
0008 ntuple.bookJet('jet1')
0009 ntuple.bookJet('jet2')
0010 ntuple.bookZ('dimuon')
0011
0012
0013 def process(self, event):
0014 if len(event.jets)>0:
0015 ntuple.fillJet('jet1', event.jets[0])
0016 if len(event.jets)>1:
0017 ntuple.fillJet('jet2', event.jets[1])
0018 if len(event.dimuons>1):
0019 ntuple.fillZ('dimuon', event.dimuons[0])
0020 ntuple.tree.tree.Fill()
0021
0022