Inheritance diagram for python.cmstools.EventTree:

Public Member Functions
def	__getattr__ (self, name)

def	__getitem__ (self, key)

def	__init__ (self, obj)

def	__iter__ (self)

def	branch (self, name)

def	cppCode (self, name)

def	getListOfAliases (self)

def	index (self)

def	SetAlias (self, alias, fullName)

def	tree (self)

Private Member Functions
def	__setBranchIndicies (self)

Private Attributes
	_aliases

	_index

	_tree

	_usedBranches

Detailed Description

Definition at line 68 of file cmstools.py.

Constructor & Destructor Documentation

def python.cmstools.EventTree.__init__	(	self,
		obj
	)

Definition at line 69 of file cmstools.py.

       def __init__(self,obj):
           sys.stderr.write ("WARNING: This package has been deprecated and will be removed in the near future.\nPlease switch to using FWLite.Python (https://twiki.cern.ch/twiki/bin/viewauth/CMS/WorkBookFWLitePython)\n")
           treeName = 'Events'
           if isinstance(obj, ROOT.TTree):
               self._tree = obj
           elif isinstance(obj, ROOT.TFile):
               self._tree = obj.Get(treeName)
           elif isinstance(obj, str):
               self._tree = ROOT.TFile.Open(obj).Get(treeName)
           else:
               raise cmserror("EventTree accepts only TTrees, TFiles and filenames")
           self._usedBranches = dict()
           self._index = -1
           self._aliases = self._tree.GetListOfAliases()

Member Function Documentation

def python.cmstools.EventTree.__getattr__	(	self,
		name
	)

Definition at line 120 of file cmstools.py.

References TableOutputBranches.branch(), SummaryTableOutputBranches::NamedBranchPtr.branch, edm::BranchToCount.branch, TriggerOutputBranches::NamedBranchPtr.branch, TableOutputBranches::NamedBranchPtr.branch, NanoAODOutputModule::CommonEventBranches.branch(), python.cmstools.EventTree.branch(), NanoAODOutputModule::CommonLumiBranches.branch(), and NanoAODOutputModule::CommonRunBranches.branch().

120 def __getattr__(self, name):

121 return self.branch(name)

python.cmstools.EventTree.__getattr__

def __getattr__(self, name)

Definition: cmstools.py:120

python.cmstools.EventTree.branch

def branch(self, name)

Definition: cmstools.py:83

def python.cmstools.EventTree.__getitem__	(	self,
		key
	)

Definition at line 122 of file cmstools.py.

References python.cmstools.EventTree.__setBranchIndicies(), SeedingNode< DATA >._index, Vispa.Share.FindAlgorithm.FindAlgorithm._index, RPCMuonExtraStruct._index, and python.cmstools.EventTree._index.

       def __getitem__(self,key):
           if key <0 or key > self._tree.GetEntries():
               raise IndexError
           self._index = key
           self.__setBranchIndicies()
           self._tree.GetEntry(self._index,0)
           return Event(self)

def python.cmstools.EventTree.__iter__ ( self )

Definition at line 129 of file cmstools.py.

References python.cmstools.EventTree.__setBranchIndicies(), SeedingNode< DATA >._index, Vispa.Share.FindAlgorithm.FindAlgorithm._index, RPCMuonExtraStruct._index, and python.cmstools.EventTree._index.

       def __iter__(self):
           # flushing/initializing the root buffers 
           entry = 0
           self._index = entry
           self.__setBranchIndicies()
           self._tree.GetEntry(self._index,0)
           # the real loop
           for entry in xrange(self._tree.GetEntries()):
               self._index = entry
               self.__setBranchIndicies()
               self._tree.GetEntry(self._index,0)
               yield Event(self)    # TODO: don't return a new object but update the old one 
               
 

def python.cmstools.EventTree.__setBranchIndicies ( self )

private

Definition at line 117 of file cmstools.py.

References SeedingNode< DATA >._index, Vispa.Share.FindAlgorithm.FindAlgorithm._index, RPCMuonExtraStruct._index, python.cmstools.EventTree._index, and python.cmstools.EventTree._usedBranches.

Referenced by python.cmstools.EventTree.__getitem__(), and python.cmstools.EventTree.__iter__().

       def __setBranchIndicies(self):
           for branch in six.itervalues(self._usedBranches):
               branch.setIndex(self._index)

def python.cmstools.EventTree.branch	(	self,
		name
	)

Definition at line 83 of file cmstools.py.

References python.cmstools.EventTree._usedBranches.

Referenced by python.cmstools.EventTree.__getattr__().

       def branch(self,name):
           # support for aliases
           alias = self._tree.GetAlias(name)
           if alias != '': name = alias 
           # access the branch in ttree
           if name in self._usedBranches:
               return self._usedBranches[name]
           self._usedBranches[name]=EventBranch(self,name)
           return self._usedBranches[name]

def python.cmstools.EventTree.cppCode	(	self,
		name
	)

C++ code for accessing the product inside the full framework

Definition at line 92 of file cmstools.py.

       def cppCode(self, name):
           """C++ code for accessing the product inside the full framework"""
           alias = self._tree.GetAlias(name)
           if alias != '': name = alias
           tmpBranch = self._tree.GetBranch(name)
           typeString = ROOT.branchToClass(tmpBranch).GetName()
           if "edm::Wrapper" in typeString:
               typeString = typeString.replace("<edm::Wrapper","")
               typeString = typeString.rstrip(">")
           nameParts = name.split("_")
           if nameParts[2] == "":
               cppCode = 'edm::Handle<%s > dummy;\nevent.getByLabel("%s", dummy);'\
                         %(typeString, nameParts[1])
           else:
               cppCode = 'edm::Handle<%s > dummy;\nevent.getByLabel("%s", "%s", dummy);'\
                         %(typeString, nameParts[1], nameParts[2])
           return cppCode