TTreeHelper.TTreeSelector Class Reference

Inheritance diagram for TTreeHelper.TTreeSelector:

Collaboration diagram for TTreeHelper.TTreeSelector:

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

Public Attributes
	file = TFile.Open(args[0])
	hashlist = self.file.GetListOfKeys()
	entries = self.hashlist.GetEntries()

Detailed Description

TreePlotter helper class in python

Definition at line 12 of file TTreeHelper.py.

Constructor & Destructor Documentation

__init__()

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()

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 = self.hashlist.GetEntries()

Definition at line 23 of file TTreeHelper.py.

file

TTreeHelper.TTreeSelector.file = TFile.Open(args[0])

Definition at line 18 of file TTreeHelper.py.

hashlist

TTreeHelper.TTreeSelector.hashlist = self.file.GetListOfKeys()

Definition at line 22 of file TTreeHelper.py.

---

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

• TTreeHelper.py