python.RootUtils.RootFileDumper Class Reference

Inheritance diagram for python.RootUtils.RootFileDumper:

Collaboration diagram for python.RootUtils.RootFileDumper:

Public Member Functions
def	__init__ (self, fname, tree_name="CollectionTree")
def	dump (self, tree_name, itr_entries, leaves=None, retvecs=False, sortleaves=True)

Public Attributes
	allgood
	root_file
	tree

Private Attributes
	_trees

Detailed Description

A helper class to dump in more or less human readable form the content of
any TTree.

Definition at line 198 of file RootUtils.py.

Constructor & Destructor Documentation

__init__()

def python.RootUtils.RootFileDumper.__init__	(		self,
			fname,
			tree_name = "CollectionTree"
	)

Definition at line 204 of file RootUtils.py.

    def __init__(self, fname, tree_name="CollectionTree"):
        object.__init__(self)
 
        ROOT = import_root()
 
        # remember if an error or problem occurred during the dump
        self.allgood = True
        
        self.root_file = ROOT.TFile.Open(fname)
        if (self.root_file is None or
            not isinstance(self.root_file, ROOT.TFile) or
            not self.root_file.IsOpen()):
            raise IOError('could not open [%s]'% fname)
 
        self.tree = self.root_file.Get(tree_name)
        if self.tree is None or not isinstance(self.tree, ROOT.TTree):
            raise AttributeError('no tree [%s] in file [%s]', tree_name, fname)
 
        if 0:
            self._trees = []
            keys = [k.GetName() for k in self.root_file.GetListOfKeys()]
            for k in keys:
                o = self.root_file.Get(k)
                if isinstance(o, ROOT.TTree):
                    self._trees.append(k)
                    pass
 
        return
 

Member Function Documentation

dump()

def python.RootUtils.RootFileDumper.dump	(		self,
			tree_name,
			itr_entries,
			leaves = None,
			retvecs = False,
			sortleaves = True
	)

Definition at line 233 of file RootUtils.py.

    def dump(self, tree_name, itr_entries, leaves=None, retvecs=False, sortleaves=True):
 
        ROOT = import_root()
        import AthenaPython.PyAthena as PyAthena
        _pythonize = PyAthena.RootUtils.PyROOTInspector.pyroot_inspect2
 
        self.tree = self.root_file.Get(tree_name)
        if self.tree is None or not isinstance(self.tree, ROOT.TTree):
            raise AttributeError('no tree [%s] in file [%s]', tree_name, self.root_file.GetName())
 
        tree = self.tree
        nentries = tree.GetEntries()
        branches = sorted([b.GetName().rstrip('\0') for b in tree.GetListOfBranches()])
        if leaves is None: leaves = branches
        else:              leaves = [str(b).rstrip('\0') for b in leaves]
        
        # handle itr_entries
        if isinstance(itr_entries, str):
            if ':' in itr_entries:
                def toint(s):
                    if s == '':
                        return None
                    try:
                        return int(s)
                    except ValueError:
                        return s
                from itertools import islice
                itr_entries = islice(range(nentries),
                                     *map(toint, itr_entries.split(':')))
            elif ('range' in itr_entries or
                  ',' in itr_entries):
                itr_entries = eval(itr_entries)
            else:
                try:
                    _n = int(itr_entries)
                    itr_entries = range(_n)
                except ValueError:
                    print ("** err ** invalid 'itr_entries' argument. will iterate over all entries.")
                    itr_entries = range(nentries)
        elif isinstance(itr_entries, list):
            itr_entries = itr_entries
        else:
            itr_entries = range(itr_entries)
                
        list_ = list
        map_ = map
        str_ = str
        isinstance_ = isinstance
 
        for ientry in itr_entries:
            hdr = ":: entry [%05i]..." % (ientry,)
            #print (hdr)
            #print (hdr, file=self.fout)
            err = tree.LoadTree(ientry)
            if err < 0:
                print ("**err** loading tree for entry",ientry)
                self.allgood = False
                break
 
            nbytes = tree.GetEntry(ientry)
            if nbytes <= 0:
                print ("**err** reading entry [%s] of tree [%s]" % (ientry, tree_name))
                hdr = ":: entry [%05i]... [ERR]" % (ientry,)
                print (hdr)
                self.allgood = False
                continue
 
            for br_name in leaves:
                hdr = "::  branch [%s]..." % (br_name,)
                #print (hdr)
                #tree.GetBranch(br_name).GetEntry(ientry)
                py_name = [br_name]
 
                br = tree.GetBranch (br_name)
                if br.GetClassName() != '':
                    val = getattr(tree, br_name)
                else:
                    vals = [_getLeaf (l) for l in br.GetListOfLeaves()]
                    if len(vals) == 0:
                        val = None
                    elif len(vals) == 1:
                        val = vals
                    else:
                        val = tuple(vals)
                if not (val is None):
                    #print ("-->",val,br_name)
                    try:
                        vals = _pythonize(val, py_name, True, retvecs)
                    except Exception as err:
                        print ("**err** for branch [%s] val=%s (type=%s)" % (
                            br_name, val, type(val),
                            ))
                        self.allgood = False
                        print (err)
                    if sortleaves:
                        viter = sorted(vals, key = lambda x: '.'.join(s for s in x[0] if isinstance_(s, str_)))
                    else:
                        viter = vals
                    for o in viter:
                        n = list_(map_(str_, o[0]))
                        v = o[1]
                        yield tree_name, ientry, n, v
 
                pass # loop over branch names
            pass # loop over entries

Member Data Documentation

_trees

python.RootUtils.RootFileDumper._trees

private

Definition at line 223 of file RootUtils.py.

allgood

python.RootUtils.RootFileDumper.allgood

Definition at line 210 of file RootUtils.py.

root_file

python.RootUtils.RootFileDumper.root_file

Definition at line 212 of file RootUtils.py.

tree

python.RootUtils.RootFileDumper.tree

Definition at line 218 of file RootUtils.py.

---

The documentation for this class was generated from the following file:

• RootUtils.py