Aida2RootEx.py

Go to the documentation of this file.

#!/usr/bin/env python2.7
 
" Simple example to illustrate the usage of aida2root converter "
from __future__ import print_function
# =============================================================================
__author__ = "Vanya BELYAEV ibelyaev@phys.syr.edu"
# =============================================================================
import os
import sys
import ROOT
if "-b" in sys.argv:
    # Force batch mode
    ROOT.gROOT.SetBatch(True)
 
from GaudiPython.GaudiAlgs import HistoAlgo, aida2root, SUCCESS
 
# list of booked histograms
paths = ('HistoEx/ 1D histo ', 'HistoEx/ 2D histo ', 'HistoEx/ 3D histo ',
         'HistoEx1/ 1D histo ', 'HistoEx1/ 2D histo ', 'HistoEx1/ 3D histo ',
         'HistoEx2/ x vs y    ', 'HistoEx2/ x vs y+3x ',
         'HistoEx2/ x vs y-3x ', 'HistoEx2/ x vs y    (profile)',
         'HistoEx2/ x vs y+3x (profile)', 'HistoEx2/ x vs y-3x (profile)')
 
# =============================================================================
# @class Aida2RootEx1
#  Simple algorithm which used aida2root utility
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2007-01-24
 
 
class Aida2RootEx1(HistoAlgo):
    # Standard Constructor
    def __init__(self, name="Aida2RootEx1"):
        """ Standard Constructor """
        HistoAlgo.__init__(self, name)
 
    # the main execution method
    def execute(self):
        " The main execution method "
 
        # list of booked histograms
        for path in paths:
            self.Print("AIDA object: '%s'" % path)
            # get AIDA pointer
            aida = self.histoSvc(path)
            if not aida:
                return self.Error("Invalid AIDA at '%s'" % path)
            # explicitly convert to ROOT
            root = aida2root(aida)
            if not root:
                return self.Error("Invalid conversion to ROOT '%s'" % path)
            # use the native ROOT printout
            root.Print()
 
        return SUCCESS
 
 
# =============================================================================
 
# =============================================================================
# @class Aida2RootEx2
#  Simple algorithm which uses aida2root utility
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2007-01-24
 
 
class Aida2RootEx2(HistoAlgo):
    # Standard Constructor
    def __init__(self, name="Aida2RootEx2"):
        """ Standard Constructor """
        HistoAlgo.__init__(self, name)
 
    # the main execution method
    def execute(self):
        " The main execution method "
 
        # get the service itself
        s = self.histoSvc()
 
        for path in paths:
            self.Print("AIDA object: '%s'" % path)
            root = s.getAsROOT(path)
            if not root:
                return self.Error("Invalid conversion to ROOT '%s'" % path)
            # use the native ROOT printout
            root.Print()
 
        return SUCCESS
 
 
# =============================================================================
 
 
# =============================================================================
# The main configuration method
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2007-01-24
def configure(gaudi=None):
    """ the main configuration method """
 
    if not gaudi:
        from GaudiPython.Bindings import AppMgr
        gaudi = AppMgr()
 
    # reuse the previous example
    import HistoEx2
    HistoEx2.configure(gaudi)
 
    # create the algorithms
    alg1 = Aida2RootEx1()
    alg2 = Aida2RootEx2()
    # append them to the list of Top-Level algorithms
    gaudi.addAlgorithm(alg1)
    gaudi.addAlgorithm(alg2)
 
    return SUCCESS
 
 
# =============================================================================
# The third way to convert AIDA histograms into ROOT
#  @author Vanya BELYAEV ibelyaev@physics.syr.edu
#  @date 2007-01-24
def useScript(histos):
    "  the third way to convert AIDA histograms into ROOT "
 
    from GaudiPython.Bindings import AppMgr
    g = AppMgr()
 
    hsvc = g.histsvc()
 
    i = 0
    for histo in histos:
        root = hsvc.getAsROOT(histo)
        if not root:
            print("ERROR in access the histogram '%s' " % histo)
            continue
        canvas = ROOT.TCanvas('canvas_%d' % i, histo, 250, 250)
        root.Draw()
        name = histo.replace('/', '_')
        name = name.replace('\\', '_')
        name = name.replace('"', '_')
        name = name.replace("'", '_')
        name = name.replace("'", '_')
        name = name.replace(" ", '_')
        name = name.replace(os.sep, '_') + '.png'
        if os.path.exists(name):
            # strictly speaking, it is not needed, but avoids a message on the stderr
            os.remove(name)
        canvas.Print(name)
        print("The file name is '%s'" % name)
        i += 1
 
 
# =============================================================================
 
# =============================================================================
# The actual job execution
# =============================================================================
if '__main__' == __name__:
    print(__doc__, __author__)
 
    from GaudiPython.Bindings import AppMgr
 
    gaudi = AppMgr()
    configure(gaudi)
    gaudi.run(5)
 
    # use the scripts
    useScript(paths)
 
# =============================================================================
# The END
# =============================================================================