MTutorialAveragePulse.cc

Go to the documentation of this file.

 
#include "MTutorialAveragePulse.hh"
#include "QEvent.hh"
#include "QEventList.hh"
#include "QPulseInfo.hh"
#include "QPulse.hh"
 
REGISTER_MODULE(MTutorialAveragePulse)
 
using namespace Diana;
 
// Init method is called before the event loop 
void MTutorialAveragePulse::Init(QEvent& ev)
{
    // Get output file name from config file
    fOutput = GetString("Output","avg/averagepulses.root");
}
 
// Do method is called on each event which passes
// filter selection
void MTutorialAveragePulse::Do(QEvent& ev)
{
    // Get channel number and raw samples from the event
    const QPulseInfo& pulseInfo = ev.Get<QPulseInfo>("DAQ","PulseInfo");
    const int chan = pulseInfo.GetChannelId();
    const QPulse& pulse = ev.Get<QPulse>("DAQ","Pulse");
    const QVector& samples = pulse.GetSamples();
 
    // if this is the first event for this channel, initialize to zero the variables for the average.
    if(fAveragePulses.find(chan) == fAveragePulses.end()) {
        fAveragePulses[chan].fNumberOfEvents = 0;
        fAveragePulses[chan].fSamples.Resize(samples.Size());
        fAveragePulses[chan].fSamples.Initialize(0.);
    }
 
    // sum up QVectors and record the number of calls for each channel
    fAveragePulses[chan].fSamples += samples;
    fAveragePulses[chan].fNumberOfEvents++;
}
 
// Done method is called at the end of the event loop
void MTutorialAveragePulse::Done()
{
    std::map<int,ChannelInfo>::const_iterator iter = fAveragePulses.begin();
    while(iter != fAveragePulses.end()) {
        // divide the sum by the number of calls
        const int chan = iter->first;
        QVector ap = iter->second.fSamples;
        int nev = iter->second.fNumberOfEvents;
        ap /= nev;
 
        // store average QVector on output file with name Samples, and set the channel number as key for this object.
        // The label of this object will be: TutorialAveragePulse@Samples_chan%04d
        // (In official data processing the module used is MAveragePulses, 
        // which saves objects of type QAveragePulse via the QAveragePulseHandle which implements the I/O with the DB.
        GlobalHandle<QVector> apHandle("Samples");
        apHandle.SetChannel(chan);
        apHandle.Set(ap);
        GlobalData().Set(&apHandle,fOutput);
        iter++;
    }
}