MCustomWaveform.cc

Go to the documentation of this file.

#include "MCustomWaveform.hh"
#include "QEvent.hh"
#include "QEventAssembler.hh"
#include "QPulse.hh"
#include "QPulseInfo.hh"
#include "QHeader.hh"
#include "QRunDataHandle.hh"
#include "QPulseFiller.hh"
#include <cmath>
 
 
REGISTER_MODULE(MCustomWaveform)
 
using namespace Diana;
 
QDriver::Action MCustomWaveform::Init(QEventAssembler& eva)
{
    fInputOwner = GetString("InputOwner","DAQ");
    fmSecLeft = GetInt("mSecLeft",0);
    fmSecRight = GetInt("mSecRight",0);
    fOverWritePulse = GetBool("Overwrite",true);
    QEvent& ev = eva.GetEvent();
    ev.Require<QPulse>(fInputOwner.c_str(),"Pulse");
    ev.Require<QPulseInfo>(fInputOwner.c_str(),"PulseInfo");
    if(!fOverWritePulse) {
        ev.Add<QPulse>("Pulse");
        ev.Add<QPulseInfo>("PulseInfo");
    }
    fRun = -1;
    fRunDuration = 0;
    fChannelInfos.clear();
     return ACT_NEXTEV;
}
 
QDriver::Action MCustomWaveform::Do(QEventAssembler& eva)
{
    QEvent& ev = eva.GetEvent();
    const QPulse& inputPulse = ev.Get<QPulse>(fInputOwner.c_str(),"Pulse");
    const QPulseFiller* filler = inputPulse.GetPulseFiller();
    if(!filler || !filler->IsValid()) Panic("QPulse does not contain a valid QPulseFiller, access to continuous data not possible.");
 
    const QHeader& header = ev.Get<QHeader>(fInputOwner.c_str(),"Header");
    const int run = header.GetRun();
 
    if(fRun != run) {
        fChannelInfos.clear();
        QRunDataHandle rHandle(run);
        GlobalData().Get<QRunData>("",&rHandle,"");
        fRunDuration = rHandle.Get().fDuration*1e9;
        fRun = run;
        const std::map<int,QChannelRunData*>& chanMap = rHandle.Get().GetChannelsRunData();
        std::map<int,QChannelRunData*>::const_iterator  iter = chanMap.begin();
        while(iter != chanMap.end()) {
            const int channel = iter->first;
            const QChannelRunData& channelRunData = *(iter->second);
 
 
            ChannelInfo& chInfo = fChannelInfos[channel];
            chInfo.fdt = (Long64_t)1e9/channelRunData.fSamplingFrequency;
 
            // round to sample size
            if(fmSecLeft != 0) chInfo.fCorrectednSecLeft = llround(fmSecLeft*1e6/chInfo.fdt)*chInfo.fdt;
            else chInfo.fCorrectednSecLeft = 0LL;
            if(fmSecRight != 0) chInfo.fCorrectednSecRight = llround(fmSecRight*1e6/chInfo.fdt)*chInfo.fdt;
            else chInfo.fCorrectednSecRight = 0LL;
            chInfo.fBlackSheep = false;
 
            // force even number of samples
            int numberOfSamples = channelRunData.fNumberOfSamples + (chInfo.fCorrectednSecLeft + chInfo.fCorrectednSecRight) / chInfo.fdt;
            if(numberOfSamples%2 != 0) {
                if(fmSecLeft != 0) chInfo.fCorrectednSecLeft -= chInfo.fdt;
                else if(fmSecRight != 0) chInfo.fCorrectednSecRight -= chInfo.fdt;
                numberOfSamples = channelRunData.fNumberOfSamples + (chInfo.fCorrectednSecLeft + chInfo.fCorrectednSecRight) / chInfo.fdt;
            }
 
            // overwrite QRunData. This hack is something you should not teach to younger generations.
            if(fOverWritePulse) {
                // void* cast to break the 'const' wall.
                void* rdv = (void*)&channelRunData;
                QChannelRunData* rd = (QChannelRunData*)rdv;
                rd->fNumberOfSamples = numberOfSamples;
            }
            iter++;
        }
        GlobalData().Set(&rHandle,"CurrentWriter");
    }
 
    const QPulseInfo& inputPI = ev.Get<QPulseInfo>(fInputOwner.c_str(),"PulseInfo");
    const int channel = inputPI.GetChannelId();
 
    if(fChannelInfos.find(channel) == fChannelInfos.end()) {
        Warn("Channel %d was not found in QRunData and will not be customized",channel);
        ChannelInfo& chInfo = fChannelInfos[channel];
        chInfo.fBlackSheep = true;
 
    }
 
    const ChannelInfo& chInfo = fChannelInfos[channel];
    if(chInfo.fBlackSheep) return ACT_NEXTEV;
 
    Long64_t start = filler->GetStartTime()-chInfo.fCorrectednSecLeft;
    Long64_t stop = filler->GetStopTime()+chInfo.fCorrectednSecRight;
 
    QPulse outputPulse;
    QPulseInfo outputPI;
    if(start >= 0 && stop <= fRunDuration) {
        outputPulse.SetPulseFiller(*filler);
        outputPulse.GetPulseFiller()->SetStartTime(start); 
        outputPulse.GetPulseFiller()->SetStopTime(stop); 
        outputPulse.Validate();
        outputPI = inputPI;
        QSampleInfo& MS = outputPI.GetMasterSample();
        MS.SetSampleIndex(MS.GetSampleIndex()+chInfo.fCorrectednSecLeft/chInfo.fdt);
        std::vector<QSampleInfo>& SI = outputPI.GetSamplesInfo();
        for(size_t s = 0; s < SI.size(); s++) 
            SI[s].SetSampleIndex(SI[s].GetSampleIndex()+chInfo.fCorrectednSecLeft/chInfo.fdt);
        outputPI.Validate();
    } else {
        Warn("Event number %d (Channel %d) is skipped since there are not enough data to build the waveform",header.GetEventNumber(),channel);
        return ACT_SKIP;
    }
    if(fOverWritePulse) {
        eva.Get<QPulse>(fInputOwner.c_str(),"Pulse") = outputPulse;
        eva.Get<QPulseInfo>(fInputOwner.c_str(),"PulseInfo") = outputPI;
    } else {
        ev.Get<QPulse>("Pulse") = outputPulse  ;
        ev.Get<QPulseInfo>("PulseInfo") = outputPI ;
    }
    return ACT_NEXTEV;
 
}
 
QDriver::Action MCustomWaveform::Done()
{
    return ACT_NEXTEV;
}