Aida2RootEx.py

Go to the documentation of this file.

#!/usr/bin/env python3
 
" 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 SUCCESS, HistoAlgo, aida2root
 
# 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)
 
    hsvc = gaudi.service("HistogramPersistencySvc")
    hsvc.OutputFile = "aida2rootex.root"
 
    # 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
# =============================================================================