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 26 of file cmstools.py.

def all(container):

  # loop over ROOT::TTree and similar
  if hasattr(container,'GetEntries'):
    try:
      entries = container.GetEntries()
      for entry in xrange(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 xrange(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 54 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 47 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