python.cmstools Namespace Reference

Classes
class	cmserror
class	Event
class	EventBranch
class	EventTree

Functions
def	all (container)
	workaround iterator generators for ROOT classes More...
def	createBranchBuffer (branch)
	auto branch types (Chris Jones) More...
def	loop (begin, end)

Detailed Description

Python helper tools for CMS FWLite

benedikt.hegner@cern.ch

Function Documentation

def python.cmstools.all

(

container

)

workaround iterator generators for ROOT classes

Definition at line 27 of file cmstools.py.

def all(container):

  # loop over ROOT::TTree and similar
  if hasattr(container,'GetEntries'):
    try:
      entries = container.GetEntries()
      for entry in range(entries):
        yield entry
    except:
        raise cmserror("Looping of %s failed" %container) 

  # loop over std::vectors and similar
  elif hasattr(container, 'size'):
    try:
      entries = container.size()
      for entry in range(entries):
        yield container[entry]
    except:
      pass

 # loop over containers with begin and end iterators

def python.cmstools.createBranchBuffer

(

branch

)

auto branch types (Chris Jones)

Definition at line 55 of file cmstools.py.

def createBranchBuffer(branch):
    reColons = re.compile(r'::')
    reCloseTemplate =re.compile(r'>')
    reOpenTemplate =re.compile(r'<')
    branchType = ROOT.branchToClass(branch)
    #buffer = eval ('ROOT.'+reColons.sub(".",reOpenTemplate.sub("(ROOT.",reCloseTemplate.sub(")",branchType.GetName())))+'()')
    buffer = ROOT.MakeRootClass(branchType.GetName()) ()
    if( branch.GetName()[-1] != '.') and (branch.GetName()!="EventAuxiliary"):
        branch.SetAddress(buffer)
    else:
        branch.SetAddress(ROOT.AddressOf(buffer))
    return buffer

def python.cmstools.loop	(		begin,
			end
	)

Convert a pair of C++ iterators into a python generator

Definition at line 48 of file cmstools.py.

def loop(begin, end):
    """Convert a pair of C++ iterators into a python generator"""
    while (begin != end):
        yield begin.__deref__()  #*b
        begin.__preinc__()       #++b