In This Package:

Viewer.py

Go to the documentation of this file.

#!/usr/bin/env python


from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl
from array import array

TH1F = gbl.TH1F
TGraph = gbl.TGraph
TCanvas = gbl.TCanvas
TFile = gbl.TFile
DayaBay = gbl.DayaBay
gStyle = gbl.gStyle
gStyle.SetPalette(1)
gStyle.SetMarkerStyle(8)
gStyle.SetMarkerColor(4)

class EsViewerAlg(GaudiAlgo):
    "An algorithm for generating figures of ElecSim Data Objects"
    def __init__(self,name):
        GaudiAlgo.__init__(self,name)
        print "Making EsViewerAlg",name
        self.plot = {}

    def initialize(self):
        status = GaudiAlgo.initialize(self)
        print "Init EsViewerAlg",self.name()
        if status.isFailure(): return status
        #self.cableSvc = self.svc('ICableSvc','StaticCableSvc')
        #self.testH = TH1F("testH","Test Histogram",100,0.0,2.0)
        return SUCCESS

    def execute(self):
        print "Executing EsViewerAlg",self.name()
        evt = self.evtSvc()
        hdr = evt["/Event/Elec/ElecHeader"]
        crateHdr = hdr.crateHeader()
        crates = crateHdr.crates()
        crate = crates[DayaBay.Detector(0x01,1)]
        # Signal clock ticks [ns]
        hitDt = 12.5
        energyDt = 1
        adcDt = 25
        tdcDt = 25./16.
        # Draw entire crate nhit signal
        nhitBoard = crate.nhit()
        nhitSignal = None
        for board in range(1,13):
            boardId = DayaBay.FeeChannelId(board,0,0x01,1)
            boardNhit = nhitBoard[boardId]
            if nhitSignal == None:
                nhitSignal = [0] * boardNhit.size()
            for cycle in range(boardNhit.size()):
                nhitSignal[cycle] += boardNhit[cycle]
        nhitClock = [tick*hitDt for tick in range(boardNhit.size())]
        self.plot["nhit"] = TGraph(boardNhit.size(),
                                   array('f',nhitClock),
                                   array('f',nhitSignal))
        # Draw an example channel
        channel = None
        tdc = None
        for connector in range(1,17):
            # Look for a channel with hits
            channelId = DayaBay.FeeChannelId(1,connector,0x01,1)
            channel = crate.channel(channelId)
            tdc = channel.tdc()
            if tdc.size() > 0:
                print "Found a hit channel"
                break
        energy = channel.energy()
        energyClock = [tick*energyDt for tick in range(energy.size())]
        self.plot["energy"] = TGraph(energy.size(),
                                     array('f',energyClock),
                                     array('f',energy))
        hit = channel.hit()
        hitClock = [tick*hitDt for tick in range(hit.size())]
        self.plot["hit"] = TGraph(hit.size(),
                                  array('f',hitClock),
                                  array('f',hit))
        adcLow = channel.adcLow()
        adcLowClock = [tick*adcDt for tick in range(adcLow.size())]
        self.plot["adcLow"] = TGraph(adcLow.size(),
                                     array('f',adcLowClock),
                                     array('f',adcLow))
        adcHigh = channel.adcHigh()
        adcHighClock = [tick*adcDt for tick in range(adcHigh.size())]
        self.plot["adcHigh"] = TGraph(adcHigh.size(),
                                      array('f',adcHighClock),
                                      array('f',adcHigh))
        print nhitSignal
        return SUCCESS

    def finalize(self):
        print "Finalizing EsViewerAlg",self.name()
        outFile = TFile("elecSimPlots.root","RECREATE")
        for key in self.plot.keys():
            self.plot[key].SetTitle(key)
            self.plot[key].SetName(key)
            self.plot[key].GetXaxis().SetTitle("time [ns]")
            self.plot[key].Write()
            #canvas = TCanvas()
            #self.plot[key].Draw("APL")
            #canvas.SaveAs(key+".root")
        status = GaudiAlgo.finalize(self)
        return status

def configure():
    return

def run(app):
    '''
    Configure and add an algorithm to generate figures of ElecSim Data Objects
    '''
    viewer = EsViewerAlg("EsViewer")
    app.addAlgorithm(viewer)
    pass

---