MAdvancedPeakFinder.cc

Go to the documentation of this file.

#include "MAdvancedPeakFinder.hh"
#include "QEvent.hh"
#include "QEventList.hh"
#include "QRawEvent.hh"
#include "QBaseModule.hh"
#include "QSampleInfo.hh"
#include "QRunDataHandle.hh"
#include "QBaseType.hh"
#include "QBaselineData.hh"
#include <TMath.h>
#include "QCountPulsesData.hh"
 
REGISTER_MODULE(MAdvancedPeakFinder)
 
using namespace Diana;
 
void MAdvancedPeakFinder::Init(QEvent& ev)
{
    //Getting Arguments from cfg
    fPulseLabel = GetString("PulseLabel","DAQ@Pulse",false);
    TriggerAlgo = GetString("Algorithm","BULLDAQ",false);
    fDebounce = GetDouble("Debounce",0,false);
    fSaveFiltered = GetBool("SaveFiltered",false,false);
    fSaveHeight = GetBool("SaveHeight", false, false);
    if(fDebounce >=1){
        Panic("Choose a value between 0 and 1 excluded!");
    }
    //adding members to .root file
    if(fSaveFiltered){
        ev.Add<QPulse>("Peaks");
    }
    if(fSaveHeight){
        ev.Add<QDouble>("HighestTrigger");
    }
    ev.Add<QCountPulsesData>("CountPulsesData");
 
    
    
    //Require
    ev.RequireByLabel<QPulse>(fPulseLabel);
    ev.Require<QPulseInfo>("DAQ","PulseInfo");
    ev.Require<QHeader>("DAQ","Header");
}
 
void MAdvancedPeakFinder::Do(QEvent& ev)
{
 
    const QPulseInfo& pulseInfo = ev.Get<QPulseInfo>("DAQ","PulseInfo");
    const QHeader& header = ev.Get<QHeader>("DAQ","Header");
    const int chan = pulseInfo.GetChannelId();
 
    QRunDataHandle rHandle(header.GetRun());
    GlobalData().Get("",&rHandle,"");
 
    const QRunData& runData = rHandle.Get();
    const QChannelRunData& channelRunData = runData.GetChannelRunData(chan);
    
    
    const QPulse& rawPulse = ev.GetByLabel<QPulse>(fPulseLabel);
    QPulse fff = rawPulse;
    QVector ppp =  rawPulse.GetSamples();
    
    it = channel2trigger.find(chan);
    if(it==channel2trigger.end()){
        Info("Adding trigger for channel %d",chan);
        QTrigger* Trigger = QTriggerFactory(TriggerAlgo);
        channel2trigger.insert(std::pair<int,QTrigger*>(chan,Trigger));
        channel2trigger[chan]->SetDebounce((int)(fDebounce*ppp.Size()));
        channel2trigger[chan]->SetReader(this);
        channel2trigger[chan]->SetSamplingFrequency(channelRunData.fSamplingFrequency);
        channel2trigger[chan]->SetWindow(ppp.Size());
        channel2trigger[chan]->SetChannel(chan);
    }
    
    vector<double> trace (ppp.Size());
    for(size_t i =0; i <ppp.Size();i++){
        trace[i] = ppp[i];
    }
    
    channel2trigger[chan]->TriggerStream(trace);
    vector<int> trigs = channel2trigger[chan]->GetTriggers();
    QVector Intervals;
    if(trigs.size()>1){
        Intervals.Resize(trigs.size()-1);
        for(size_t i =1; i <trigs.size();i++){
            Intervals[i-1] = trigs[i]-trigs[i-1];
        }
        Intervals/=channelRunData.fSamplingFrequency;
    }else{
        Intervals.Resize(1);
        Intervals.Initialize(0);
    }
    if(fSaveFiltered){
        vector<double> filt_stream = channel2trigger[chan]->GetFilteredStream();
    
        QVector stream(filt_stream.size());
        for(size_t i =0; i <filt_stream.size();i++){
            stream[i] = filt_stream[i];
        }
 
        fff.SetSamplesADC(stream);
    
        ev.Get<QPulse>("Peaks") =fff;
    }
    if(fSaveHeight){
        vector<double> THeights = channel2trigger[chan]->GetTriggerHeights();
        std::set<double> Triggers (THeights.begin(), THeights.end());
        ev.Get<QDouble>("HighestTrigger") = *Triggers.rbegin();
    }
    ev.Get<QCountPulsesData>("CountPulsesData").fNumberOfPulses =(int)trigs.size();
    ev.Get<QCountPulsesData>("CountPulsesData").fTimeIntervals = Intervals;
}  
 
 
void MAdvancedPeakFinder::Done()
{
    /* This method is called at the end of the event loop. 
     * Here you can:
     *
     * 1) Operate on the sequence execution parameters(see QBaseModule.hh).
     *
     * 2) Read/Write global data.
     */
}