In This Package:

WaveDumper.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 WaveDumper 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 WaveDumperAlg(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/Elec/ElecHeader"]
        
        if (hdr.__class__ != gbl.DayaBay.ElecHeader ): 
            self.warning("ElecHeader not found for this event")
            return SUCCESS
        
        crateHeader = hdr.crateHeader()
        hitDt = 12.5
        energyDt = 5
        adcDt = 25
        tdcDt = 25./16
        for det,crate in crateHeader.crates():
            self.debug("Processing " + det.detName())
            crate.__class__=gbl.DayaBay.ElecFeeCrate
            
            boardpath="/file1/event_%d/%s/"%(self.eventNumber,
                                                           det.detName())                
            esumAdc = crate.esumADC()
            energyClock = [tick*energyDt for tick in range(esumAdc.size())]
            print esumAdc
            graph = TGraph(esumAdc.size(),
                           array('f',energyClock),
                           array('f',esumAdc))
            graph.SetMarkerStyle(8)   # large circle
            graph.SetMarkerColor(600) # blue
            graph.SetTitle("esumADC")
            self.stats[boardpath+"esumAdc"] = graph
           
            esumUpper = crate.esumUpper()
            energyClock = [tick*energyDt for tick in range(esumUpper.size())]
            graph = TGraph(esumAdc.size(),
                           array('f',energyClock),
                           array('f',esumUpper))
            graph.SetMarkerStyle(8)   # large circle
            graph.SetMarkerColor(600) # blue
            graph.SetTitle("esumUpper")
            self.stats[boardpath+"esumUpper"] = graph

            esumLower = crate.esumLower()
            energyClock = [tick*energyDt for tick in range(esumLower.size())]
            graph = TGraph(esumLower.size(),
                           array('f',energyClock),
                           array('f',esumLower))
            graph.SetMarkerStyle(8)   # large circle
            graph.SetMarkerColor(600) # blue
            graph.SetTitle("esumLower")
            self.stats[boardpath+"esumLower"] = graph
           
            esumTotal = crate.esumTotal()
            energyClock = [tick*energyDt for tick in range(esumTotal.size())]
            graph = TGraph(esumTotal.size(),
                           array('f',energyClock),
                           array('f',esumTotal))
            graph.SetMarkerStyle(8)   # large circle
            graph.SetMarkerColor(600) # blue
            graph.SetTitle("esumTotal")
            self.stats[boardpath+"esumTotal"] = graph
                           
           
            for chID,channel in crate.channelData():
                path="/file1/"
                path+="event_%d/%s/board_%d/connector_%d/"%(self.eventNumber,
                                                           det.detName(),
                                                           chID.board(),
                                                           chID.connector())
#                self.debug(path)
                # only proceed if the channel was hit
                tdc = channel.tdc()
                if tdc.size() > 0:
                    adcHigh = channel.adcHigh()
                    adcHighClock = [tick*adcDt for tick in range(adcHigh.size())]
                    graph=TGraph(adcHigh.size(),
                                 array('f',adcHighClock),
                                 array('f',adcHigh))
                    graph.SetMarkerStyle(8)   # large circle
                    graph.SetMarkerColor(600) # blue
                    graph.SetTitle("adcHigh")
                    self.stats[path+"adcHigh"] = graph
                   
                    hit = channel.hit()
                    hitClock = [tick*hitDt for tick in range(hit.size())]
                    graph= TGraph(hit.size(),
                                                    array('f',hitClock),
                                                    array('f',hit))
                    graph.SetMarkerStyle(8)   # large circle
                    graph.SetMarkerColor(600) # blue
                    graph.SetTitle("nhit")
                    self.stats[path+"nhit"]=graph
                    
                    adcLow = channel.adcLow()
                    adcLowClock = [tick*adcDt for tick in range(adcLow.size())]
                    graph = TGraph(adcLow.size(),
                                   array('f',adcLowClock),
                                   array('f',adcLow))
                    graph.SetMarkerStyle(8)   # large circle
                    graph.SetMarkerColor(600) # blue
                    graph.SetTitle("adcLow")
                    self.stats[path+"adcLow"] = 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

---