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