In This Package:

AdViewer.py

Go to the documentation of this file.

#!/usr/bin/env python

class AdViewer:
    '''
    Display output plots in ROOT
    '''
    def __init__(self):
        self.canvases = []
        self.files = []

    def drawPlots(self,filename):
        from ROOT import TFile, TCanvas, TH1F, TH2F, gStyle
        # set some style info
        gStyle.SetPalette(1)
        
        # Open the file
        plotFile = TFile(filename)
        if not plotFile.IsOpen():
            print "drawPlots: Can't open file ",filename
        self.files.append(plotFile)
        path = "adplotter/DayaBayAD1/"
        
        # Draw hit pmts
        hitPmts = plotFile.Get(path+"DayaBayAD1_HitPmts")
        c1 = TCanvas("c1")
        hitPmts.Draw()
        self.canvases.append(c1)
        # Draw time between readouts
        dtReadout = plotFile.Get(path+"DayaBayAD1_DtReadout")
        c2 = TCanvas("c2")
        dtReadout.Draw()
        self.canvases.append(c2)
        # Close-up of short times between readouts
        dtReadoutShort = plotFile.Get(path+"DayaBayAD1_DtReadoutShort")
        c3 = TCanvas("c3")
        dtReadoutShort.Draw()
        self.canvases.append(c3)
        # ADC values from all channels
        adc = plotFile.Get(path+"DayaBayAD1_AdcSpectrum")
        c4 = TCanvas("c4")
        adc.Draw()
        self.canvases.append(c4)
        # TDC values from all channels
        tdc = plotFile.Get(path+"DayaBayAD1_TdcSpectrum")
        c5 = TCanvas("c5")
        tdc.Draw()
        self.canvases.append(c5)
        # Sum of all adc charge in each readout
        chargeSum = plotFile.Get(path+"DayaBayAD1_ChargeSum")
        c6 = TCanvas("c6")
        chargeSum.Draw()
        self.canvases.append(c6)
        # PMTs hit vs. total charge
        hitVsCharge = plotFile.Get(path+"DayaBayAD1_HitPmtVsChargeSum")
        c7 = TCanvas("c7")
        hitVsCharge.Draw("colz")
        self.canvases.append(c7)
        # Position of generated vertex
        vertexZRho = plotFile.Get(path+"DayaBayAD1_GenVertexZRho")
        c8 = TCanvas("c8")
        vertexZRho.Draw("colz")
        self.canvases.append(c8)
        return

    def drawCoincidence(self,filename):
        from ROOT import TFile, TCanvas, TH1F, TH2F, gStyle, TLine
        # set some style info
        gStyle.SetPalette(1)
        
        # Open the file
        plotFile = TFile(filename)
        if not plotFile.IsOpen():
            print "drawCoincidence: Can't open file ",filename
        self.files.append(plotFile)
        path = "adplotter/DayaBayAD1/"
        
        # Sum of all adc charge in each readout
        chargeSum = plotFile.Get(path+"DayaBayAD1_ChargeSum")
        c1 = TCanvas("c1")
        chargeSum.SetTitle("AD charge singles spectrum for DayaBayAD1")
        chargeSum.Draw()
        self.canvases.append(c1)
        
        # Draw neutron capture time
        dtNeutron = plotFile.Get(path+"DayaBayAD1_DtNeutronCapture")
        c2 = TCanvas("c2")
        dtNeutron.Draw()
        dtNeutron.Fit("expo","LL")
        self.canvases.append(c2)

        # Draw prompt spectrum
        promptE = plotFile.Get(path+"DayaBayAD1_PromptEnergy")
        c3 = TCanvas("c3")
        promptE.Draw()
        self.canvases.append(c3)

        # Draw delayed spectrum
        delayedE = plotFile.Get(path+"DayaBayAD1_DelayedEnergy")
        c4 = TCanvas("c4")
        delayedE.Draw()
        self.canvases.append(c4)

        # Draw delayed vs. prompt spectrum
        delayedVsPromptE = plotFile.Get(path+"DayaBayAD1_DelayedVsPromptEnergy")
        c5 = TCanvas("c5")
        self.canvases.append(c5)
        delayedVsPromptE.SetMarkerStyle(8)
        delayedVsPromptE.SetMarkerColor(4)
        delayedVsPromptE.SetMarkerSize(0.5)
        delayedVsPromptE.Draw()
        # Draw signal box
        line = TLine()
        line.SetLineWidth(2)
        line.SetLineColor(2)
        line.SetLineStyle(2)
        line.DrawLine(0.7,6.,0.7,10.)
        line.DrawLine(8.0,6.,8.0,10.)
        line.DrawLine(0.7,6.,8.0,6.)
        line.DrawLine(0.7,10.,8.0,10.)

---