DrawGe68EnergySub.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Isolates Ge68 spectrum like EnergyStatsDiff and then isolates smaller peak by subtracting larger gaussian.
# Creats two graphs, one of the fit of the larger peak alone and the other of the resulting smaller peak with fit.
#
#  Usage:
#   nuwa.py DrawEnergySub

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

# Make shortcuts to any ROOT classes you want to use
TF1 = gbl.TF1
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")

        canvas = TCanvas()


        hist=self.stats["/file0/energy/reconEnergy"]
        histnosrc=self.stats["/file1/energy/reconEnergy"]

        hist.Sumw2()
        histnosrc.Sumw2()

        histfinal=hist.Clone()
        histfinal.SetName("nobgEnergyFit")
        histfinal.Add(histnosrc,-0.057)


        peak2 = TF1("m1","gaus",0.8,1.5)
        total = TF1("mtotal", "gaus(0) + gaus(3)", 0, 1.5)
        histfinal.Fit(peak2,"R")
        canvas.SaveAs("nobgEnergyFit1.png")

        histclone=histfinal.Clone()
        for iBin in range(0,250):
          prevContent = histfinal.GetBinContent(iBin)
          prevContent -= peak2.Eval(histfinal.GetBinCenter(iBin))
          histclone.SetBinContent(iBin,prevContent)
        histfinal = histclone
        histfinal.Draw()
        histfinal.Fit("gaus","","",0.0,0.85)

        canvas.SaveAs("nobgEnergyFit2.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_run.root", "file1":"100000StatsnoSRC.root"}
    return

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