TupleEx1.py

Go to the documentation of this file.

#!/usr/bin/env python
 
"""
*******************************************************************************
*                                                                             *
* Simple example which illustrate the usage of useful                         *
* algorithm  base class for N-Tuple manipulations                             *
*                                                                             *
*******************************************************************************
"""
from __future__ import print_function
 
# =============================================================================
__author__ = "Vanya BELYAEV Ivan.Belyaev@lapp.in2p3.fr"
# =============================================================================
 
import math
 
import GaudiPython
 
SUCCESS = GaudiPython.SUCCESS
 
# random numbewrs
Rndm = GaudiPython.gbl.Rndm
Numbers = Rndm.Numbers
 
from GaudiPython.GaudiAlgs import TupleAlgo, mapvct
 
# =============================================================================
# Primitive function which transform arbitrary sequence  into
#  GaudiPython.Vector ( std::vector<double> )
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2006-11-26
 
 
def vct(sequence):
    """
    Primitive function which transform arbitrary sequence  into
    GaudiPython.Vector ( std::vector<double> )
    """
    result = GaudiPython.gbl.GaudiPython.Vector()
    if hasattr(sequence, "__len__"):
        result.reserve(len(sequence))
    elif hasattr(sequence, "size"):
        result.reserve(sequence.size())
 
    for item in sequence:
        result.push_back(item)
    return result
 
 
# =============================================================================
# @class TupleEx1
#  Simple algorithm which book&fill 3 histograms
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2006-11-26
 
 
class TupleEx1(TupleAlgo):
    """
    Simple algorithm which implicitely book&fill N-Tuples
    """
 
    # the main executiomethod
 
    def execute(self):
        """
        The major method 'execute', it is invoked for each event
        """
 
        rSvc = self.randSvc()
        gauss = Numbers(rSvc, Rndm.Gauss(0.0, 1.0))
        flat = Numbers(rSvc, Rndm.Flat(-10, 10))
        expo = Numbers(rSvc, Rndm.Exponential(1.0))
        breit = Numbers(rSvc, Rndm.BreitWigner(0.0, 1.0))
        poisson = Numbers(rSvc, Rndm.Poisson(2.0))
        binom = Numbers(rSvc, Rndm.Binomial(8, 0.25))
 
        # =====================================================================
        # primitive row-wise n-tuple
        # =====================================================================
        tuple1 = self.nTuple(1, "Trivial Row-Wise Tuple", 42)
 
        # fill N-Tuple with double/float numbers:
        tuple1.column("gauss", gauss())
        tuple1.column("flat", flat())
        tuple1.column("expo", expo())
        tuple1.column("breit", breit())
 
        # fill N-Tuple with integer numbers:
        tuple1.column("poiss1", int(poisson()))
        tuple1.column("binom1", int(binom()))
 
        # fill N-Tuple with long long numbers:
        tuple1.column_ll("poiss2", int(poisson()))
        tuple1.column_ll("binom2", int(binom()))
 
        # fill N-Tuple with unsigned long long numbers:
        tuple1.column_ull("poiss3", int(poisson()))
        tuple1.column_ull("binom3", int(binom()))
 
        # fill N-Tuple with "reduced" integer numbers:
        tuple1.column("poiss4", int(poisson()), 0, 14)
        tuple1.column("binom4", int(binom()), 0, 14)
 
        # fill N-Tuple with "boolean" numbers:
        tuple1.column("poisb", poisson() > 1.0)
 
        # commit the row
        tuple1.write()
 
        # =====================================================================
        # the same n-tuple but column-wise
        # =====================================================================
        tuple2 = self.nTuple(2, "Trivial Column-Wise Tuple")
 
        # fill N-Tuple with double/float numbers:
        tuple2.column("gauss", gauss())
        tuple2.column("flat", flat())
        tuple2.column("expo", expo())
        tuple2.column("breit", breit())
 
        # fill N-Tuple with integer numbers:
        tuple2.column("poiss", int(poisson()))
        tuple2.column("binom", int(binom()))
        # fill N-Tuple with "reduced" integer numbers:
        tuple2.column("poiss", int(poisson()), 0, 14)
        tuple2.column("binom", int(binom()), 0, 14)
 
        # fill N-Tuple with "boolean" numbers:
        tuple2.column("poisb", poisson() > 1.0)
 
        # commit the row
        tuple2.write()
 
        # =====================================================================
        # book and fill Column-wise NTuple with "fixed"-size arrays/vectors
        # =====================================================================
        tuple3 = self.nTuple(3, "Fixed-size arrays/vectors")
 
        tuple3.array("arflat", vct([flat() for i in range(0, 50)]))
        tuple3.array("arexpo", vct([expo() for i in range(0, 62)]))
        tuple3.array("argau", vct([gauss() for i in range(0, 42)]))
        t = tuple([gauss() for i in range(0, 42)])
        tuple3.array("argau2", vct(t))
 
        tuple3.write()
 
        return SUCCESS
 
 
# =============================================================================
# job configuration
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2006-11-26
def configure(gaudi=None):
    """Configuration of the job"""
 
    if not gaudi:
        gaudi = GaudiPython.AppMgr()
 
    gaudi.JobOptionsType = "NONE"
    gaudi.EvtSel = "NONE"
    gaudi.HistogramPersistency = "ROOT"
 
    gaudi.ExtSvc += ["NTupleSvc"]
 
    ntSvc = gaudi.service("NTupleSvc")
    ntSvc.Output = ["MYLUN DATAFILE='TupleEx1.root' OPT='NEW' TYP='ROOT'"]
 
    gaudi.config()
 
    gaudi.DLLs = [
        "GaudiAlg",
        "RootHistCnv",
    ]
 
    alg = TupleEx1("TupleEx1")
    gaudi.setAlgorithms([alg])
 
    # configure the properties
    alg.NTupleLUN = "MYLUN"
 
    return SUCCESS
 
 
# =============================================================================
# The actual job excution
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2006-11-26
if "__main__" == __name__:
    print(__doc__)
    gaudi = GaudiPython.AppMgr()
    configure(gaudi)
    gaudi.run(20)
 
# =============================================================================
# The END
# =============================================================================