PhysicsTools/HeppyCore/python/framework/eventstfile.py - master - githubmirror/cmssw

PhysicsTools/HeppyCore/python/framework/eventstfile.py

29 строк895 B

Colin

unittest for eventstfile

05 янв 2016, 13:48

05 янв 2016, 13:481947e0a

# Copyright (C) 2014 Colin Bernet
# https://github.com/cbernet/heppy/blob/master/LICENSE

from ROOT import TFile

class Events(object):
    '''Event list from a tree in a root file.
    '''
    def __init__(self, filename, treename, options=None):
        self.file = TFile(filename)
        if self.file.IsZombie():
            raise ValueError('file {fnam} does not exist'.format(fnam=filename))
        self.tree = self.file.Get(treename)
        if self.tree == None: # is None would not work
            raise ValueError('tree {tree} does not exist in file {fnam}'.format(
                tree = treename,
                fnam = filename
                ))

    def size(self):
        return self.tree.GetEntries()

    def to(self, iEv):
        '''navigate to event iEv.'''
        self.tree.GetEntry(iEv)
        return self.tree

    def __iter__(self):
        return iter(self.tree)