Back to home page

Project CMSSW displayed by LXR

 
 

    


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