DrawGe68EnergyGeo.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Draws graph of reconstructed energies of different geometries plotted together.
#
#  Usage:
#   nuwa.py DrawEnergyGeo

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

# 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["/file3/energy/reconEnergy"].Draw()
        self.stats["/file3/energy/reconEnergy"].SetLineColor(4)
        self.stats["/file1/energy/reconEnergy"].Draw("same")
        self.stats["/file1/energy/reconEnergy"].SetLineColor(1)
        self.stats["/file0/energy/reconEnergy"].Draw("same")
        self.stats["/file0/energy/reconEnergy"].SetLineColor(2)
        self.stats["/file2/energy/reconEnergy"].Draw("same")
        self.stats["/file2/energy/reconEnergy"].SetLineColor(3)
        self.stats["/file4/energy/reconEnergy"].Draw("same")
        self.stats["/file4/energy/reconEnergy"].SetLineColor(5)
        self.stats["/file5/energy/reconEnergy"].Draw("same")
        self.stats["/file5/energy/reconEnergy"].SetLineColor(6)
        self.stats["/file6/energy/reconEnergy"].Draw("same")
        self.stats["/file6/energy/reconEnergy"].SetLineColor(7)
        self.stats["/file7/energy/reconEnergy"].Draw("same")
        self.stats["/file7/energy/reconEnergy"].SetLineColor(8)
        self.stats["/file8/energy/reconEnergy"].Draw("same")
        self.stats["/file8/energy/reconEnergy"].SetLineColor(9)



        leg=TLegend(0.1,0.65,0.3,0.9);
        leg.AddEntry(self.stats["/file0/energy/reconEnergy"],"No geometry","l");
        leg.AddEntry(self.stats["/file1/energy/reconEnergy"],"With geometry","l");
        leg.AddEntry(self.stats["/file2/energy/reconEnergy"],"Al Vacuum","l");
        leg.AddEntry(self.stats["/file4/energy/reconEnergy"],"SS Vacuum","l");
        leg.AddEntry(self.stats["/file3/energy/reconEnergy"],"Al + SS Vacuum","l");
        leg.AddEntry(self.stats["/file5/energy/reconEnergy"],"Al + SS + Ge Vacuum","l");
        leg.AddEntry(self.stats["/file8/energy/reconEnergy"],"Al + SS + Acr Vacuum","l");
        leg.AddEntry(self.stats["/file7/energy/reconEnergy"],"Ge + Acr Vacuum","l");
        leg.AddEntry(self.stats["/file6/energy/reconEnergy"],"Al + SS + Ge + Acr Vacuum","l");
        leg.Draw();


        canvas.SaveAs("reconEnergyDiff.png")

        status = DybPythonAlg.finalize(self)
        return status


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

def configure():
    from StatisticsSvc.StatisticsSvcConf import StatisticsSvc
    statsSvc = StatisticsSvc()
    statsSvc.Input={"file0":"stats_runnoGEO0.root", 
        "file1":"stats_runGEO0.root","file2":"stats_runGEO0Alvacuum.root","file3":"stats_runGEO0AlSSvacuum.root","file4":"stats_runGEO0SSvacuum.root", 
         "file5":"stats_runGEO0AlSSGEvacuum.root","file6":"stats_runGEO0AlSSGEAcrvacuum.root","file7":"stats_runGeAcrvacuum.root",
        "file8":"stats_runGEOAlSSAcrvacuum.root"}
    return

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