In This Package:

Ge68EnergyStats.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Makes reconstructed energy and energy bias histograms and outputs physical time elapsed
#
#  Usage:
#   nuwa.py -n -1 EnergyStats reconAndSimData.root

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

# Make shortcuts to any ROOT classes you want to use
TH1F = gbl.TH1F
TH2F = gbl.TH2F

# Make shortcuts to any ROOT classes you want to use

loaddict("libCLHEPRflx")
loaddict("libHepMCRflx")
Detector = gbl.DayaBay.Detector
AdPmtSensor = gbl.DayaBay.AdPmtSensor
ServiceMode = gbl.ServiceMode
ReconStatus = gbl.ReconStatus

# Make your algorithm
class EnergyStatsAlg(DybPythonAlg):
    "Algorithm to make 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")

        self.cableSvc = self.svc('ICableSvc','StaticCableSvc')
        if self.cableSvc == None:
            self.error("Failed to get StaticCableSvc")
            return FAILURE

        # Make energy histograms

        hist = TH1F("EnergyBias","Energy Bias",500,-2.0,2.5)
        hist.GetXaxis().SetTitle("Energy Difference [MeV]")
        hist.GetYaxis().SetTitle("Generated Particles")
        hist.SetLineColor(4)
        self.stats["/file0/energy/EnergyBias"] = hist

        hist = TH1F("reconEnergy","Reconstructed Energy",500,0.0,3.0)
        hist.GetXaxis().SetTitle("Reconstructed Visible Energy [E_rec]")
        hist.GetYaxis().SetTitle("Triggered Readouts")
        hist.SetLineColor(4)
        self.stats["/file0/energy/reconEnergy"] = hist
        return SUCCESS

    def execute(self):
        self.info("executing")
        
        evt = self.evtSvc()


        # reconstructed energy
        recHdr = evt["/Event/Rec/RecHeader"]


        roHdr = evt["/Event/Readout/ReadoutHeader"]
        readout = roHdr.readout()
        if readout != None:
           trigTime = readout.triggerTime()
           print "seconds=", trigTime.GetSec()
           print "nanoseconds=", trigTime.GetNanoSec()
         

        if recHdr == None:
            self.error("Failed to get RecHeader")
            return FAILURE
        recResults = recHdr.recResults()
        recTrigger = recResults["AdSimple"]
        if recTrigger.energyStatus() == ReconStatus.kGood:
            reconEnergy = recTrigger.energy()
            self.stats["/file0/energy/reconEnergy"].Fill(reconEnergy)
            self.stats["/file0/energy/EnergyBias"].Fill(reconEnergy-1.022)
        return SUCCESS

    def finalize(self):
        self.info("finalizing")

        status = DybPythonAlg.finalize(self)
        return status


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

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

def run(app):
    '''
    Configure and add an algorithm to job
    '''
    app.ExtSvc += ["StaticCableSvc", "StatisticsSvc"]
    energyStatsAlg = EnergyStatsAlg("MyEnergyStats")
    app.addAlgorithm(energyStatsAlg)
    pass

---