HistoEx2.py

Go to the documentation of this file.

#!/usr/bin/env python3
 
"""
*******************************************************************************
*                                                                             *
* Simple example which illustrate the usage of useful algorithm  base class   *
* HistoAlgo (python version of C++ GaudiHistoAlg) for 'easy' histogramming.   *
*                                                                             *
* This example illustrates the usage of 1D and 2D profile histograms          *
*                                                                             *
*******************************************************************************
"""
from __future__ import print_function
 
# =============================================================================
__author__ = "Vanya BELYAEV ibelyaev@physics.syr.edu"
# =============================================================================
import os
 
from GaudiPython.Bindings import gbl as cpp
from GaudiPython.GaudiAlgs import SUCCESS, HistoAlgo
 
Rndm = cpp.Rndm
 
# =============================================================================
# Simple algorithm which book&fill two profile histograms
# =============================================================================
 
 
class HistoEx2(HistoAlgo):
    """Simple algorithm which explicitly book&fill profile histograms"""
 
    def __init__(self, name="HistoEx2"):
        """Constructor"""
        HistoAlgo.__init__(self, name)
 
    def execute(self):
        """The major method 'execute', it is invoked for each event"""
        gauss = Rndm.Numbers(self.randSvc(), Rndm.Gauss(0, 1))
 
        for i in range(0, 1000):
            x = gauss()
            y = gauss()
            self.plot2D(x, y, " x vs y    ", -2, 2, -4, 4)
            self.plot2D(x, y + 3 * x, " x vs y+3x ", -2, 2, -4, 4)
            self.plot2D(x, y - 3 * x, " x vs y-3x ", -2, 2, -4, 4)
            self.profile1D(x, y, " x vs y    (profile)", -2, 2)
            self.profile1D(x, y + 3 * x, " x vs y+3x (profile)", -2, 2)
            self.profile1D(x, y - 3 * x, " x vs y-3x (profile)", -2, 2)
        return SUCCESS
 
 
# =============================================================================
# job configuration
# =============================================================================
def configure(gaudi=None):
    """Configuration of the job"""
 
    import HistoEx1
 
    if not gaudi:
        from GaudiPython.Bindings import AppMgr
 
        gaudi = AppMgr()
 
    HistoEx1.configure(gaudi)
 
    hsvc = gaudi.service("HistogramPersistencySvc")
    hsvc.OutputFile = "histoex2.root"
 
    alg = HistoEx2("HistoEx2")
    gaudi.addAlgorithm(alg)
 
    alg.HistoPrint = True
 
    return SUCCESS
 
 
# =============================================================================
# The actual job execution
# =============================================================================
if "__main__" == __name__:
    print(__doc__ + __author__)
 
    import GaudiPython.HistoUtils
    from GaudiPython.Bindings import AppMgr
 
    gaudi = AppMgr()
    configure(gaudi)
 
    gaudi.run(20)
 
    # Skip the next part when running within QMTest
    for alg in ("HistoEx", "HistoEx1", "HistoEx2"):
        alg = gaudi.algorithm(alg)
        histos = alg.Histos()
        for key in sorted(histos):
            histo = histos[key]
            print(" Alg='%s', ID='%s' , Histo=%s " % (alg.name(), key, histo))
            if hasattr(histo, "dump"):
                print(histo.dump(60, 20, True))