def runChSubEventTest( opdata, opdisplay ):
ch_subevents = {}
products = []
if not chbych:
chsubevtdict, chpostwfms = runSubEventDisc( opdata, config, retpostwfm=True, maxch=32 )
ch_subevents = chsubevtdict
for ch,subevents in chsubevtdict.items():
for subevent in subevents:
tx = []
ty = []
thresh = []
for (x,y) in subevent.expectation:
gy = subevent.gainfactor*y
tx.append( x )
ty.append( gy )
sig = np.sqrt( gy/20.0 )
thresh.append( gy + 3.0*sig*20.0 )
y = np.array( ty, dtype=np.float )
x = np.array( tx, dtype=np.float )
th = np.array( thresh, dtype=np.float )
if drawchsubevents:
pexp = pg.PlotCurveItem()
pexp.setData( x=x, y=y, pen=(255,0,0,255) )
pthresh = pg.PlotCurveItem()
pthresh.setData( x=x, y=th, pen=(0,255,0,255) )
#opdisplay.addUserWaveformItem( pexp, ch=ch )
#opdisplay.addUserWaveformItem( pthresh, ch=ch )
products.append( { "plotitem":pexp,"femch":ch,"screen":"waveform"} )
products.append( { "plotitem":pthresh,"femch":ch,"screen":"waveform"} )
pwfm = pg.PlotCurveItem()
postwfm = chpostwfms[ch]
pwfm.setData( x=range(0,len(postwfm)), y=postwfm-ped.getpedestal(postwfm, 10, 10), pen=(255,255,0,255) )
#opdisplay.addUserWaveformItem( pwfm, ch=ch )
products.append( { "plotitem":pwfm, "femch":ch, "screen":"waveform"} )
else:
print "runChSubEventTest Event-by-Event!"
opdisplay.setOverlayMode( True )
for ch in xrange(0,32):
print "CHANNEL ",ch
opdisplay.selectChannels( [ch] )
chwfm = opdata.getData(slot=5)[:,ch]
if np.max(chwfm)>=4094:
print "channel ",ch," switch to low gain!"
chwfm = opdata.getData(slot=6)[:,ch]
subevents,wfm = runSubEventDiscChannel( chwfm, config, femch, retpostwfm=True )
ch_subevents[ ch ] = subevents
for subevent in subevents:
tx = []
ty = []
thresh = []
for (x,y) in subevent.expectation:
tx.append( x )
ty.append( y )
sig = np.sqrt( y/20.0 )
thresh.append( y + 3.0*sig*20.0 )
y = np.array( ty, dtype=np.float )
x = np.array( tx, dtype=np.float )
th = np.array( thresh, dtype=np.float )
if drawchsubevents:
pexp = pg.PlotCurveItem()
pexp.setData( x=x, y=y, pen=(255,0,0,255) )
pthresh = pg.PlotCurveItem()
pthresh.setData( x=x, y=th, pen=(0,255,0,255) )
opdisplay.addUserWaveformItem( pexp, ch=ch )
opdisplay.addUserWaveformItem( pthresh, ch=ch )
print subevent.expectation
products.append( { "plotitem":pexp,"femch":ch,"screen":"waveform"} )
products.append( { "plotitem":pthresh,"femch":ch,"screen":"waveform"} )
pwfm = pg.PlotCurveItem()
postwfm = wfm
pwfm.setData( x=range(0,len(postwfm)), y=postwfm-ped.getpedestal(postwfm, 10, 10), pen=(255,255,0,255) )
opdisplay.addUserWaveformItem( pwfm, ch=ch )
opdisplay.plotData()
#raw_input()
opdisplay.clearUserWaveformItem()
products.append( { "plotitem":pwfm,"femch":ch,"screen":"waveform"} )
# merger of events
merger_gatehalfwidth = 1
pethresh = 5.0
nchthresh = 3.0
hitacc, peacc = makeSubEventAccumulators( ch_subevents, merger_gatehalfwidth, opdata, pmtspe )
# defining subevents
print np.argmax(hitacc),hitacc
print np.argmax(peacc),peacc
phitacc = pg.PlotCurveItem()
phitacc.setData( x=range(0,len(hitacc)), y=hitacc, pen=(255,255,0,255) )
ppeacc = pg.PlotCurveItem()
ppeacc.setData( x=range(0,len(peacc)), y=peacc, pen=(0,125,255,255) )
products.append( { "plotitem":phitacc,"femch":None,"screen":"waveform"} )
products.append( { "plotitem":ppeacc,"femch":None,"screen":"waveform"} )
return products
def calc_rates( inputfile, nevents, outfile, wffile=False, rawdigitfile=False, VISUALIZE=False ):
# check that the data file type was selected
if wffile==False and rawdigitfile==False:
print "Select either wffile or rawdigitfile"
return
# load data
if wffile==True:
opdata = WFOpData( inputfile )
elif rawdigitfile==True:
opdata = RawDigitsOpData( inputfile )
out = rt.TFile( outfile, "RECREATE" )
# Event tree
eventtree = rt.TTree("eventtree","PMT Rates")
event = array('i',[0])
samples = array('i',[0])
nfires = array('i',[0]*NCHANS)
echmax = array('f',[0])
eventtree.Branch( 'event', event,'event/I' )
eventtree.Branch( 'samples', samples, 'samples/I' )
eventtree.Branch( 'nfires', nfires, 'nfires[%d]/I'%(NCHANS) )
eventtree.Branch( 'chmax', echmax, 'chmax/F' )
# Pulse Tree
pulsetree = rt.TTree("pulsetree","PMT Rates")
pch = array('i',[0])
pt = array( 'f',[0] )
pwindt = array( 'f',[0] )
pchdt = array( 'f',[0] )
pmaxamp = array('f',[0])
ped = array('f',[0])
pchmaxamp = array('f',[0])
parea = array('f',[0])
pulsetree.Branch( 'event', event, 'event/I' )
pulsetree.Branch( 'ch',pch,'ch/I' )
pulsetree.Branch( 't',pt,'t/F' )
pulsetree.Branch( 'windt',pwindt,'windt/F' )
pulsetree.Branch( 'chdt',pchdt,'chdt/F' )
pulsetree.Branch( 'amp',pmaxamp,'amp/F' )
pulsetree.Branch( 'area',parea,'area/F' )
pulsetree.Branch( "ped",ped,"ped/F")
pulsetree.Branch( 'chmaxamp',pchmaxamp,'chmaxamp/F' )
first_event = opdata.first_event
if VISUALIZE and PYQTGRAPH:
opdisplay = OpDetDisplay( opdata )
opdisplay.show()
more = opdata.getEvent( first_event )
ievent = first_event
#for ievent in range(first_event,first_event+nevents+1):
while more:
if ievent%50==0:
print "Event: ",ievent
event[0] = ievent
if VISUALIZE:
more = opdisplay.gotoEvent( ievent )
# for each event gather disc fires and sort by time
event_times = []
event_discs = {}
echmax[0] = 0
chpedestals = {}
chmaxamps = {}
for femch in range(0,NCHANS):
wfm = opdata.getData(slot=5)[:,femch]
discs = runCFdiscriminator( wfm, cfdsettings )
theped = getpedestal( wfm, 10, 2 )
if theped is None:
ped[0] = 2048.0
else:
ped[0] = theped
chpedestals[femch] = ped[0]
nfires[femch] = len(discs)
pch[0] = femch
pchmaxamp[0] = np.max( wfm - ped[0] )
chmaxamps[femch] = pchmaxamp[0]
if echmax[0]<pchmaxamp[0]:
echmax[0] = pchmaxamp[0]
samples[0] = len(wfm)
pchdt[0] = -1
pmaxamp[0] = 0
if len(discs)==0:
print "Discriminator found zero pusles in channel=",femch," event=",event[0]
for idisc,disc in enumerate(discs):
disc.ch = femch
t = disc.tfire + 0.01*disc.ch # the channel number is just to keep a unique tag
event_times.append( t )
event_discs[t] = disc
tdisc = disc.tfire
disc.pmaxamp = np.max( wfm[tdisc:tdisc+cfdsettings.deadtime]-ped[0] )
pped = getpedestal( wfm[ np.maximum(0,tdisc-20):tdisc ] , 5, 2 )
if pped is None:
pped = ped[0]
disc.parea = np.sum( wfm[tdisc:tdisc+cfdsettings.deadtime] ) - cfdsettings.deadtime*pped
if idisc>0:
disc.pchdt = tdisc - discs[idisc-1].tfire
else:
disc.pchdt = -1
if VISUALIZE and PYQTGRAPH:
discfire = pg.PlotCurveItem()
x = np.linspace( 15.625*(tdisc-5), 15.625*(tdisc+5+cfdsettings.deadtime), cfdsettings.deadtime+10 )
y = np.ones( cfdsettings.deadtime+10 )*femch
y[5:5+cfdsettings.deadtime] += 2
discfire.setData( x=x, y=y, pen=(255,0,0,255) )
opdisplay.addUserWaveformItem( discfire, femch )
event_times.sort()
for idisc,t in enumerate(event_times):
disc = event_discs[t]
pch[0] = disc.ch
pt[0] = disc.tfire
pmaxamp[0] = disc.pmaxamp
pchdt[0] = disc.pchdt
ped[0] = chpedestals[disc.ch]
pchmaxamp[0] = chmaxamps[disc.ch]
parea[0] = disc.parea
if idisc==0:
pwindt[0] = -1
else:
pwindt[0] = disc.tfire-event_discs[ event_times[idisc-1] ].tfire
pulsetree.Fill()
if VISUALIZE:
print "please enjoy plot"
raw_input()
eventtree.Fill()
more = opdata.getNextEvent()
ievent = opdata.current_event
if not more:
print "no more"
break
eventtree.Write()
pulsetree.Write()
def runChSubEventTest(opdata, opdisplay):
ch_subevents = {}
products = []
if not chbych:
chsubevtdict, chpostwfms = runSubEventDisc(opdata,
config,
retpostwfm=True,
maxch=32)
ch_subevents = chsubevtdict
for ch, subevents in chsubevtdict.items():
for subevent in subevents:
tx = []
ty = []
thresh = []
for (x, y) in subevent.expectation:
gy = subevent.gainfactor * y
tx.append(x)
ty.append(gy)
sig = np.sqrt(gy / 20.0)
thresh.append(gy + 3.0 * sig * 20.0)
y = np.array(ty, dtype=np.float)
x = np.array(tx, dtype=np.float)
th = np.array(thresh, dtype=np.float)
if drawchsubevents:
pexp = pg.PlotCurveItem()
pexp.setData(x=x, y=y, pen=(255, 0, 0, 255))
pthresh = pg.PlotCurveItem()
pthresh.setData(x=x, y=th, pen=(0, 255, 0, 255))
#opdisplay.addUserWaveformItem( pexp, ch=ch )
#opdisplay.addUserWaveformItem( pthresh, ch=ch )
products.append({
"plotitem": pexp,
"femch": ch,
"screen": "waveform"
})
products.append({
"plotitem": pthresh,
"femch": ch,
"screen": "waveform"
})
pwfm = pg.PlotCurveItem()
postwfm = chpostwfms[ch]
pwfm.setData(x=range(0, len(postwfm)),
y=postwfm - ped.getpedestal(postwfm, 10, 10),
pen=(255, 255, 0, 255))
#opdisplay.addUserWaveformItem( pwfm, ch=ch )
products.append({
"plotitem": pwfm,
"femch": ch,
"screen": "waveform"
})
else:
print "runChSubEventTest Event-by-Event!"
opdisplay.setOverlayMode(True)
for ch in xrange(0, 32):
print "CHANNEL ", ch
opdisplay.selectChannels([ch])
chwfm = opdata.getData(slot=5)[:, ch]
if np.max(chwfm) >= 4094:
print "channel ", ch, " switch to low gain!"
chwfm = opdata.getData(slot=6)[:, ch]
subevents, wfm = runSubEventDiscChannel(chwfm,
config,
femch,
retpostwfm=True)
ch_subevents[ch] = subevents
for subevent in subevents:
tx = []
ty = []
thresh = []
for (x, y) in subevent.expectation:
tx.append(x)
ty.append(y)
sig = np.sqrt(y / 20.0)
thresh.append(y + 3.0 * sig * 20.0)
y = np.array(ty, dtype=np.float)
x = np.array(tx, dtype=np.float)
th = np.array(thresh, dtype=np.float)
if drawchsubevents:
pexp = pg.PlotCurveItem()
pexp.setData(x=x, y=y, pen=(255, 0, 0, 255))
pthresh = pg.PlotCurveItem()
pthresh.setData(x=x, y=th, pen=(0, 255, 0, 255))
opdisplay.addUserWaveformItem(pexp, ch=ch)
opdisplay.addUserWaveformItem(pthresh, ch=ch)
print subevent.expectation
products.append({
"plotitem": pexp,
"femch": ch,
"screen": "waveform"
})
products.append({
"plotitem": pthresh,
"femch": ch,
"screen": "waveform"
})
pwfm = pg.PlotCurveItem()
postwfm = wfm
pwfm.setData(x=range(0, len(postwfm)),
y=postwfm - ped.getpedestal(postwfm, 10, 10),
pen=(255, 255, 0, 255))
opdisplay.addUserWaveformItem(pwfm, ch=ch)
opdisplay.plotData()
#raw_input()
opdisplay.clearUserWaveformItem()
products.append({
"plotitem": pwfm,
"femch": ch,
"screen": "waveform"
})
# merger of events
merger_gatehalfwidth = 1
pethresh = 5.0
nchthresh = 3.0
hitacc, peacc = makeSubEventAccumulators(ch_subevents,
merger_gatehalfwidth, opdata,
pmtspe)
# defining subevents
print np.argmax(hitacc), hitacc
print np.argmax(peacc), peacc
phitacc = pg.PlotCurveItem()
phitacc.setData(x=range(0, len(hitacc)), y=hitacc, pen=(255, 255, 0, 255))
ppeacc = pg.PlotCurveItem()
ppeacc.setData(x=range(0, len(peacc)), y=peacc, pen=(0, 125, 255, 255))
products.append({"plotitem": phitacc, "femch": None, "screen": "waveform"})
products.append({"plotitem": ppeacc, "femch": None, "screen": "waveform"})
return products