source: fact/tools/pyscripts/sandbox/kraehenb/plotevents.py@ 14962

#!/usr/bin/python -tt
# ********************************
# Plot the calibrated events of a file
# 
# written by Thomas Kraehenbuehl, ETH Zurich
# tpk@phys.ethz.ch, +41 44 633 3973
# September 2012
# ********************************

datafile = '/fact/raw/2012/06/08/20120608_009.fits.gz'
calibfile = '/fact/raw/2012/06/08/20120608_007.drs.fits.gz'
pixelnr = 0

import numpy as np
from fileplotter import FilePlotter
import os

datafilename = os.path.basename(datafile)
calibfilename = os.path.basename(calibfile)

from ROOT import gSystem
gSystem.Load("calfactfits_h.so")
from ROOT import *
data = CalFactFits(datafile,calibfile)
npcalevent  = np.empty( data.npix * data.nroi, np.float64) #.reshape(data.npix ,data.nroi)
data.SetNpcaldataPtr(npcalevent)
nroi = data.data_nroi

gROOT.SetStyle("Plain")
#gStyle.SetOptTitle(0)
gStyle.SetOptStat(0)

#Create a canvas with automatic save-function
#Note: most options of Fileplotter (legend, title etc. do not work here)
fp = FilePlotter(out_name="EventDisplay")
fp.canvas()

title = "Data: %s, DRS: %s, Px %i Ev %i"%(datafile,calibfile,pixelnr,0)
pixProfile = TProfile("pix", title, nroi, -0.5, nroi-0.5)
pixProfile.GetXaxis().SetTitle("Slice @ 2 GHz")
pixProfile.GetYaxis().SetTitle("Amplitude (mV)")

print "\nPress"
print "-ENTER to continue"
print "-'a' to abort"
print "-an integer to jump to this event"
print "-'s' to save and continue.\n"

nextEvent = 0

while data.GetCalEvent():
        if nextEvent>data.event_id:
                continue
        
        print data.event_id, hex(data.event_triggertype),
        
        pixProfile.Reset()
        title = "Data: %s, DRS: %s, ContHardID %i Ev %i"%(datafilename,calibfilename,pixelnr,data.event_id)
        pixProfile.SetTitle(title)
        
        for k in range(3,nroi):
                pixProfile.Fill(k,npcalevent[pixelnr*nroi+k]);
        
        pixProfile.Draw()
        fp.canv.Modified()
        fp.canv.Update()
        
        try:
                temp = raw_input()
        except Exception, err:
                print "Caught exception:", err
        except KeyboardInterrupt, err:
                print "Caught keyboard interrupt"
        
        try:
                nextEvent = int(temp)
        except ValueError, err:
                nextEvent = 0
        
        if temp=='s':
                fp.save()
                
        if temp=='a': break

del fp,data