In This Package:

Eff.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Example of doing simple efficiency with sim15 output


This assumes you made a file with somthing like:

  nuwa.py -n1000 -m Sim15.FullChainSimple -o s15fcs.root

Then run like

  nuwa.py -n -1 -m Sim15.Eff s15fcs.root
  ...
  Importing module Sim15.Eff
  ...
  

'''

from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE, Bindings, gbl
from DybPython.Util import irange

def visitInputHeaders(obj,visitor):
    '''
    Recursively visit obj's inputHeaders and their inputHeaders.
    The visitor should be callable with a HeaderObject.
    '''
    visitor(obj)
    ihs = obj.inputHeaders()
    siz = ihs.size();
    for ind in range(siz):
        ih = ihs[ind]
        visitInputHeaders(ih,visitor)
        continue
    return

class Sim15Eff(GaudiAlgo):
    '''
    Give an example of calculating some efficiencies
    '''

    def __init__(self,name='Sim15Eff'):
        GaudiAlgo.__init__(self,name)
        self.genCounts = {}     # count GenHeaders generated
        self.trigCounts = {}    # count GenHeaders found via Readouts
        return

    def execute(self):
        tes = self.evtSvc()

        rs = tes['/Event/RegistrationSequence']

        regs = rs.registrations()
        nregs = regs.size()
        #print 'Found %d HeaderObjects'%regs.size()
        for ireg in range(0,nregs):
            reg = regs[ireg]
            loc = reg.path()
            obj = reg.object()  # this auto casts to right HeaderObject type

            #print loc, obj.name()
            if loc == '/Event/Gen/GenHeader':
                self._add_generated(obj)
                pass
            
            if loc == '/Event/Readout/ReadoutHeader':
                ghs = self._get_gen_headers(obj)
                #print 'Found %d GenHeaders in this readout'%len(ghs)
                for gh in ghs:
                    self._add_triggered(gh)
                pass

            continue
        return SUCCESS

    def finalize(self):
        '''
        Print efficiencies. 
        '''

        for nam in self.genCounts.keys():
            if nam not in self.trigCounts.keys():
                self.trigCounts[nam] = 0
            print 'Kin type: %s has %d/%d = %f'\
                %(nam,self.trigCounts[nam],self.genCounts[nam],
                  (self.trigCounts[nam]*1.0)/(self.genCounts[nam]*1.0))
            continue
        return SUCCESS

    def _get_gen_headers(self,obj):
        '''
        Descend through obj's inputHeaders and build up a list of GenHeaders
        '''
        class CollectGH:
            def __init__(self):
                self.gh_list = []
                return
            def __call__(self,obj):
                if obj.name() == '/GenHeader':
                    self.gh_list.append(obj)
                return
            pass

        cgh = CollectGH()
        visitInputHeaders(obj,cgh)
        return cgh.gh_list

    def _add_generated(self,obj):
        '''
        Count a generated event
        '''
        nam = obj.generatorName()
        try:
            self.genCounts[nam] += 1
        except KeyError:
            self.genCounts[nam] = 1
            pass
        return

    def _add_triggered(self,obj):
        '''
        Count a triggered event
        '''
        nam = obj.generatorName()
        try:
            self.trigCounts[nam] += 1
        except KeyError:
            self.trigCounts[nam] = 1
            pass
        return
    

def configure(argv=None):
    return

def run(app):
    alg = Sim15Eff()
    app.addAlgorithm(alg)
    return

---