DrawGe68EnergySpec.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Plots resconstructed energy spectra (from EnergyStatsDiff) from all positions of one ACU axis on one graph.
#
#  Usage:
#   nuwa.py DrawEnergySpec

# Load DybPython
from DybPython.DybPythonAlg import DybPythonAlg
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl
from decimal import *

# Make shortcuts to any ROOT classes you want to use
TCanvas = gbl.TCanvas
TLegend = gbl.TLegend
gbl.gStyle.SetPalette(1)
gbl.gStyle.SetOptFit(1)

# Make your algorithm
class DrawEnergyFigsAlg(DybPythonAlg):
    "Algorithm to draw figures from the energy statistics file"
    def __init__(self,name):
        DybPythonAlg.__init__(self,name)
        return

    def initialize(self):
        status = DybPythonAlg.initialize(self)
        if status.isFailure(): return status
        self.info("initializing")

        return SUCCESS

    def execute(self):
        self.info("executing")
        
        return SUCCESS
        
    def finalize(self):
        self.info("finalizing")

        # Make energy histograms
        canvas = TCanvas()

        self.stats["/file1/energy/nobgEnergy"].Draw()
        self.stats["/file1/energy/nobgEnergy"].SetLineColor(2)

        self.stats["/file0/energy/nobgEnergy"].Draw("same")
        self.stats["/file0/energy/nobgEnergy"].SetLineColor(1)
        self.stats["/file2/energy/nobgEnergy"].Draw("same")
        self.stats["/file2/energy/nobgEnergy"].SetLineColor(3)
        self.stats["/file3/energy/nobgEnergy"].Draw("same")
        self.stats["/file3/energy/nobgEnergy"].SetLineColor(4)
        self.stats["/file4/energy/nobgEnergy"].Draw("same")
        self.stats["/file4/energy/nobgEnergy"].SetLineColor(5)
        self.stats["/file5/energy/nobgEnergy"].Draw("same")
        self.stats["/file5/energy/nobgEnergy"].SetLineColor(6)
        self.stats["/file6/energy/nobgEnergy"].Draw("same")
        self.stats["/file6/energy/nobgEnergy"].SetLineColor(7)
        self.stats["/file8/energy/nobgEnergy"].Draw("same")
        self.stats["/file8/energy/nobgEnergy"].SetLineColor(8)

        leg=TLegend(0.5,0.5,0.6,0.9);
        leg.SetHeader("Legend");
        leg.AddEntry(self.stats["/file5/energy/nobgEnergy"],"-125.0","l");
        leg.AddEntry(self.stats["/file3/energy/nobgEnergy"],"-75.0","l");
        leg.AddEntry(self.stats["/file1/energy/nobgEnergy"],"-25.0","l");
        leg.AddEntry(self.stats["/file0/energy/nobgEnergy"],"0.0","l");
        leg.AddEntry(self.stats["/file2/energy/nobgEnergy"],"25.0","l");
        leg.AddEntry(self.stats["/file4/energy/nobgEnergy"],"75.0","l");
        leg.AddEntry(self.stats["/file6/energy/nobgEnergy"],"125.0","l");
        leg.AddEntry(self.stats["/file8/energy/nobgEnergy"],"175.0","l");
        leg.Draw();

        canvas.SaveAs("A_nobgEnergySpec.png")
        
        status = DybPythonAlg.finalize(self)
        return status


#####  Job Configuration for nuwa.py ########################################

def configure():
    from StatisticsSvc.StatisticsSvcConf import StatisticsSvc
    statsSvc = StatisticsSvc()
    statsSvc.Input ={"file0":"10000energyStatsDiff_10.root",
                     "file1":"10000energyStatsDiff_11.root",
                     "file2":"10000energyStatsDiff_12.root",
                     "file3":"10000energyStatsDiff_13.root",
                     "file4":"10000energyStatsDiff_14.root",
                     "file5":"10000energyStatsDiff_15.root",
                     "file6":"10000energyStatsDiff_16.root",
                     "file8":"10000energyStatsDiff_18.root"
                    }
    return

def run(app):
    '''
    Configure and add an algorithm to job
    '''
    app.ExtSvc += ["StatisticsSvc"]
    energyFigsAlg = DrawEnergyFigsAlg("MyEnergyFigs")
    app.addAlgorithm(energyFigsAlg)
    pass