예제 #1
0
def findOneSubEvent(waveform, cfdconf, config, ch):
    """
    waveform: numpy array
    cfdconf: cfd.cfdiscConfig
    ch: int
    """

    # calculate expectation in first bin
    #pbin1_fast = config.fastfraction*prob_exp( 0, cfdconf.nspersample, 1.0/config.fastconst )
    #pbin1_slow = config.slowfraction*prob_exp( 0, cfdconf.nspersample, 1.0/config.slowconst )
    #pbin1 = pbin1_fast + pbin1_slow # expected fraction of combined slow-fast exponential in first bin

    # Find peaks
    cfdvec = cfd.runCFdiscriminator(waveform, cfdconf)
    peaks_sorted = cfdvec.getAmpOrderedList(reverse=True)

    # Sort by diff height, get biggest and make a subevent out of it
    if len(peaks_sorted) > 0:
        tstart = peaks_sorted[0].tfire
        maxamp = peaks_sorted[0].maxamp
        tmax = peaks_sorted[0].tmax
        #scale = waveform[tmax]/pbin1
        tend = tstart
        spe_sigma = 4.0 * cfdconf.nspersample

        # python
        #expectation = calcExpectation()

        # cythonized!
        #expectation = cyse.calcScintResponse( np.maximum(0,tmax-20), len(waveform), tmax, spe_sigma, (maxamp-cfdconf.pedestal), config.fastconst, config.slowconst, cfdconf.nspersample )

        # native c++
        expectation = cyse.pyCalcScintResponse(np.maximum(0, tmax - 20),
                                               len(waveform), tmax, spe_sigma,
                                               (maxamp - cfdconf.pedestal),
                                               config.fastconst,
                                               config.slowconst,
                                               cfdconf.nspersample)

        tend = tstart + len(expectation)

        #print expectation[:10]
        #print expectation_old[:10]
        #raw_input()

        return ChannelSubEvent(ch, tstart, tend, tmax, maxamp, expectation)
    return None
예제 #2
0
def findOneSubEvent( waveform, cfdconf, config, ch ):
    """
    waveform: numpy array
    cfdconf: cfd.cfdiscConfig
    ch: int
    """

    # calculate expectation in first bin
    #pbin1_fast = config.fastfraction*prob_exp( 0, cfdconf.nspersample, 1.0/config.fastconst )
    #pbin1_slow = config.slowfraction*prob_exp( 0, cfdconf.nspersample, 1.0/config.slowconst )
    #pbin1 = pbin1_fast + pbin1_slow # expected fraction of combined slow-fast exponential in first bin
    
    # Find peaks
    cfdvec = cfd.runCFdiscriminator( waveform, cfdconf )
    peaks_sorted = cfdvec.getAmpOrderedList( reverse=True )

    # Sort by diff height, get biggest and make a subevent out of it
    if len(peaks_sorted)>0:
        tstart = peaks_sorted[0].tfire
        maxamp = peaks_sorted[0].maxamp
        tmax   = peaks_sorted[0].tmax
        #scale = waveform[tmax]/pbin1
        tend = tstart
        spe_sigma = 4.0*cfdconf.nspersample

        # python
        #expectation = calcExpectation()

        # cythonized!
        #expectation = cyse.calcScintResponse( np.maximum(0,tmax-20), len(waveform), tmax, spe_sigma, (maxamp-cfdconf.pedestal), config.fastconst, config.slowconst, cfdconf.nspersample )

        # native c++
        expectation = cyse.pyCalcScintResponse( np.maximum(0,tmax-20), len(waveform), tmax, spe_sigma, (maxamp-cfdconf.pedestal), config.fastconst, config.slowconst, cfdconf.nspersample )
        
        tend = tstart + len(expectation)

        #print expectation[:10]
        #print expectation_old[:10]
        #raw_input()

        return ChannelSubEvent( ch, tstart, tend, tmax, maxamp, expectation )
    return None
예제 #3
0
def calcEventRates( cfdsettings, 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()
예제 #4
0
def calcEventRates(cfdsettings,
                   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()