In This Package:

ReadoutDumper.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Sctipt to dump waveforms to a Root File
#
#  Usage:
#   nuwa.py -n -1 ReadoutDumper simulationData.root

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

# Make shortcuts to ROOT classes
TH1F = gbl.TH1F
TFile = gbl.TFile
TGraph = gbl.TGraph

# Make your algorithm
class ReadoutDumperAlg(DybPythonAlg):
    "Example Python Algorithm"
    def __init__(self,name):
        DybPythonAlg.__init__(self,name)
        self.plot={}
        return

    def initialize(self):
        status = DybPythonAlg.initialize(self)
        if status.isFailure(): return status
        self.info("initializing")
        
        self.eventNumber = 0

        # Initialize services
        #self.cableSvc = self.svc('ICableSvc','StaticCableSvc')

        #self.stats.defaultPath = "/file1/waves/"
        # Make the trigger time histogram
        #self.stats["waveform"] = TGraph

        return SUCCESS

    def execute(self):
        self.info("executing")
        
        self.eventNumber+=1

        evt = self.evtSvc()
        hdr = evt["/Event/Readout/ReadoutHeader"]
        
        if (hdr.__class__ != gbl.DayaBay.ReadoutHeader ): 
            self.warning("Readout Header not found for this event")
            return SUCCESS
        
        readout = hdr.readout()
        #readout.__class__=gbl.DayaBay.ReadoutPmtCrate
        self.debug(str(readout.channelReadout()))
        det = readout.detector()
        
        # This works 
        
        #for channel,ro in readout.channelReadout():
        #    self.verbose("channel: "+str(channel))
        #    self.debug("size="+str(ro.size())) 

        dt = 1
        
        for channel,fadcRo in readout.fadcReadout():
            self.debug("FADC channel: " + str(channel.connector()) )
            tmp="["
            for adc in fadcRo: tmp=tmp+str(adc)+","
            self.verbose(tmp[:-1]+"]")

            path="/file1/event_%d/%s/"%(self.eventNumber, det.detName())

            # fadc = fadcRo
            clock = [tick*dt for tick in range(fadcRo.size())]
            print fadcRo
            graph = TGraph(fadcRo.size(), array('f',clock), array('f',fadcRo))
            graph.SetMarkerStyle(8)
            graph.SetMarkerColor(600)
            graph.SetTitle("FADC")
            graph.GetXaxis().SetTitle("Time (ns)")
            graph.GetYaxis().SetTitle("FADC Value")
            self.verbose("array size" + str(len(clock)))
            #graph.GetXaxis().SetLimits(200,600)
            self.stats[path+"connector_%d/fadc"%channel.connector()] = graph
               
        return SUCCESS
        
    def finalize(self):        
        self.info("finalizing")
        
        status = DybPythonAlg.finalize(self)
        return status


def configure():
    #from StatisticsSvc.StatisticsSvcConf import StatisticsSvc
    #statsSvc = StatisticsSvc()
    #statsSvc.Output ={"file1":"waveforms.root"}
    return

---