#!/usr/bin/env python
# =============================================================================
# $Id: HistoEx1.py,v 1.6 2007/10/29 14:29:47 mato Exp $
# =============================================================================
"""
*******************************************************************************
*                                                                             *
* Simple example which illustrate the usage of useful algorithm  base class   *
* HistoAlgo (python version of C++ GaudiHistoAlg) for "easy" histogramming.   *
* It is an extension of HistoEx module, it provides similar algorithm, but    *
* with explicit histogram manipulation trhrough explicit book and fill        *
*                                                                             *
*******************************************************************************
"""
# =============================================================================
__author__ = 'Vanya BELYAEV Ivan.Belyaev@lapp.in2p3.fr'
# =============================================================================

from   GaudiPython.GaudiAlgs   import HistoAlgo, SUCCESS

# =============================================================================
# Simple algorithm which book&fill 3 histograms
# =============================================================================
class HistoEx1 (HistoAlgo) :
    """ Simple algorithm which explicitely book&fill three histograms """
    def __init__ ( self , name ) :
        """ Constructor """
        HistoAlgo.__init__( self , name )

    def initialize ( self ) :
        """ Initialization, initialize the base class and book histograms """
        status = HistoAlgo.initialize( self )
        if status.isFailure() : return status

        self.h1 = self.book1D ( ' 1D histo '
                                , 0 , 20 , 20 )
        self.h2 = self.book2D ( ' 2D histo '
                                , 0 , 20 , 20
                                , 0 , 20 , 20 )
        self.h3 = self.book3D ( ' 3D histo '
                                , 0 , 20 , 20
                                , 0 , 20 , 20
                                , 0 , 20 , 20 )

        return SUCCESS

    def execute( self ) :
        """ The major method 'execute', it is invoked for each event """

        for i in range(0,10) :
            self.h1.fill( i , 0.166 )
            for j in range(0,10) :
                self.h2.fill( i   , j )
                for k in range(0,10) :
                    self.h3.fill( i , j , k )


        return SUCCESS


# =============================================================================
# job configuration
# =============================================================================
def configure( gaudi = None  ) :
    """ Configuration of the job """

    import HistoEx
    
    if not gaudi :
        from GaudiPython.Bindings import AppMgr
        gaudi = AppMgr()

    HistoEx.configure( gaudi )

    alg = HistoEx1('HistoEx1')
    gaudi.addAlgorithm( alg )

    alg.HistoPrint = True

    return SUCCESS


# =============================================================================
# The actual job excution
# =============================================================================
if '__main__' == __name__ :
    print __doc__ , __author__
    
    from GaudiPython.Bindings import AppMgr
    gaudi = AppMgr()
    
    configure( gaudi )
    
    gaudi.run(20)
    import GaudiPython.HistoUtils
    
    alg = gaudi.algorithm( 'HistoEx1' )
    histos = alg.Histos()
    for key in histos :
        histo = histos[ key ]
        if hasattr ( histo , 'dump' ) :
            print histo.dump ( 80 , 20 , True )

# =============================================================================