TTreeHelper.TTreeSelector Class Reference

Inheritance diagram for TTreeHelper.TTreeSelector:

Collaboration diagram for TTreeHelper.TTreeSelector:

Public Member Functions
def	__init__ (self, *args)
def	loadTrees (self, arg)

Public Attributes
	file
	hashlist
	entries

Detailed Description

TreePlotter helper class in python

Definition at line 12 of file TTreeHelper.py.

Constructor & Destructor Documentation

__init__()

def TTreeHelper.TTreeSelector.__init__	(		self,
		*	args
	)

Definition at line 14 of file TTreeHelper.py.

    def __init__(self, *args):
        super(TTreeSelector, self).__init__()
        # open the file
        try :
            self.file = TFile.Open(args[0])
        except IndexError :
            print ('[!] You need to specify a file name in the argument list')
        # get the hashlist and entries
        self.hashlist = self.file.GetListOfKeys()
        self.entries  = self.hashlist.GetEntries()
        print ('[>] File [',args[0],'] sucessfully loaded')
 

Member Function Documentation

loadTrees()

def TTreeHelper.TTreeSelector.loadTrees	(		self,
			arg
	)

load Trees method: loads trees that match one of the arguments in name

Definition at line 26 of file TTreeHelper.py.

    def loadTrees(self, arg):
        """load Trees method: loads trees that match one of the arguments in name"""
        treedict = {}
        for itree in range(0,self.entries):
            treename = self.hashlist.At(itree).GetName()
            if treename.find(arg) >= 0 :
                tree = self.file.Get(treename)
                treedict[treename] = tree
        print ('[>] ',len(treedict),'Trees loaded matching the selection:', arg)
        return treedict

Member Data Documentation

entries

TTreeHelper.TTreeSelector.entries

Definition at line 23 of file TTreeHelper.py.

file

TTreeHelper.TTreeSelector.file

Definition at line 18 of file TTreeHelper.py.

hashlist

TTreeHelper.TTreeSelector.hashlist

Definition at line 22 of file TTreeHelper.py.

---

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

• TTreeHelper.py