In This Package:

DrawFigs.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Example module to draw figures from the energy stats file
#
#  Usage:
#   nuwa.py UnderstandingEnergy.DrawFigs

# 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
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["/file0/energy/genEnergy"].Draw()
        canvas.SaveAs("genEnergy.png")
        
        self.stats["/file0/energy/genKineticEnergy"].Draw()
        canvas.SaveAs("genKineticEnergy.png")

        self.stats["/file0/energy/simScintEnergy"].Draw()
        canvas.SaveAs("simScintEnergy.png")

        self.stats["/file0/energy/simQuenchedEnergy"].Draw()
        canvas.SaveAs("simQuenchedEnergy.png")

        self.stats["/file0/energy/simQuenching"].Draw("colz")
        canvas.SaveAs("simQuenching.png")

        self.stats["/file0/energy/simHits"].Draw()
        canvas.SaveAs("simHits.png")

        self.stats["/file0/energy/simHitsVsQE"].Draw("colz")
        canvas.SaveAs("simHitsVsQE.png")

        self.stats["/file0/energy/adcSumVsSimHits"].Draw("colz")
        canvas.SaveAs("adcSumVsSimHits.png")

        self.stats["/file0/energy/calibAdcSum"].Draw()
        canvas.SaveAs("calibAdcSum.png")

        self.stats["/file0/energy/calibAdcSumVsSimHits"].Draw("colz")
        canvas.SaveAs("calibAdcSumVsSimHits.png")

        self.stats["/file0/energy/reconEnergy"].Draw()
        self.stats["/file0/energy/reconEnergy"].Fit("gaus","","",7.0,10.0)
        canvas.SaveAs("reconEnergy.png")

        self.stats["/file0/energy/calibAdcSumVsReconEnergy"].Draw("colz")
        canvas.SaveAs("calibAdcSumVsReconEnergy.png")

        self.stats["/file0/energy/reconEnergyVsQE"].Draw("colz")
        canvas.SaveAs("reconEnergyVsQE.png")

        status = DybPythonAlg.finalize(self)
        return status


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

def configure():
    from StatisticsSvc.StatisticsSvcConf import StatisticsSvc
    statsSvc = StatisticsSvc()
    statsSvc.Input ={"file0":"energyStats.root"}
    return

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

---