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

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

20 строк662 B

Shahzad Malik Muzaffar

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

22 ноя 2024, 20:16

22 ноя 2024, 20:168d027cc

from PhysicsTools.Heppy.analyzers.core.TreeAnalyzerNumpy import TreeAnalyzerNumpy
from . import ntuple

class ZJetsTreeAnalyzer(TreeAnalyzerNumpy):
    
    def beginLoop(self, setup):
        super(ZJetsTreeAnalyzer, self).beginLoop(setup)
        ntuple.bookJet('jet1')
        ntuple.bookJet('jet2')
        ntuple.bookZ('dimuon')

        
    def process(self, event):
        if len(event.jets)>0:
            ntuple.fillJet('jet1', event.jets[0])
        if len(event.jets)>1:
            ntuple.fillJet('jet2', event.jets[1])
        if len(event.dimuons>1):
            ntuple.fillZ('dimuon', event.dimuons[0])
        ntuple.tree.tree.Fill()