In This Package:

DrawACUNeutronCapturePosition.py

Go to the documentation of this file.

#!/usr/bin/env python
#
#Usage:
#       nuwa.py DrawCapturePosition

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

TCanvas = gbl.TCanvas
gbl.gStyle.SetPalette(1)
gbl.gStyle.SetOptFit(1)

class DrawCapturePositionAlg(DybPythonAlg):
    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 position histograms
        canvas = TCanvas()
        self.stats["/file0/position/CapturePositionAll"].Draw("colz")
        canvas.SaveAs("CapturePositionAll.png")
      

        self.stats["/file0/position/CapturePosition"].Draw("colz")
        canvas.SaveAs("CapturePosition.png")

        self.stats["/file0/position/CapturePosition6MeV"].Draw("colz")
        canvas.SaveAs("CapturePosition6MeV.png")
        
        self.stats["/file0/position/ReconPosition1"].Draw()
        canvas.SaveAs("ReconPosition1.png")

        self.stats["/file0/position/ReconPosition2"].Draw()
        canvas.SaveAs("ReconPosition2.png")

        self.stats["/file0/position/ReconPosition3"].Draw()
        canvas.SaveAs("ReconPosition3.png")

        self.stats["/file0/position/ReconPosition4"].Draw()
        canvas.SaveAs("ReconPosition4.png")

        self.stats["/file0/position/ReconPosition5"].Draw()
        canvas.SaveAs("ReconPosition5.png")
       
        self.stats["/file0/energy/ReconEnergy"].Draw()
        canvas.SaveAs("ReconEnergy.png")
 
        status = DybPythonAlg.finalize(self)
        return status


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

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

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

---