MRawDataFilter.cc

Go to the documentation of this file.

#include "MRawDataFilter.hh"
#include "QEvent.hh"
#include "QRawEvent.hh"
#include "QPulse.hh"
#include "QVector.hh"
#include "QMatrix.hh"
#include "QRunDataHandle.hh"
 
using namespace Diana;
using std::vector;
 
REGISTER_MODULE(MRawDataFilter)
 
 
void MRawDataFilter::GetCuts(const QEvent& ev)
{
    fKeepHeater = GetBool("KeepHeater",false);
    fKeepLED = GetBool("KeepLED",false);
    fKeepSignal = GetBool("KeepSignal",false);
    fKeepNoise  = GetBool("KeepNoise",false);
    fKeepSide  = GetBool("KeepSide",false, false);
    fKeepReversed  = GetBool("KeepReversed",false, false);
    fKeepVeto  = GetBool("KeepVetos",false);
    fKeepThermometers = GetBool("KeepThermometers",false);
    fKeepBolometers = GetBool("KeepBolometers",true,false);
    fKeepAllChannelTypes = GetBool("KeepAllChannelTypes",false,false);
    
    fKeepRunDataChannels = GetBool("KeepOnlyRunDataChannels",false,false);
    fLoadEventList = GetBool("LoadEventList",false,false);
    fMinEventNumber = GetInt("MinEventNumber",-1,false);
    fMaxEventNumber = GetInt("MaxEventNumber",-1,false);
 
 
    vector<int> empty;
    fKeepChannels = GetVectorInt("KeepChannels",empty,false); 
    fRejectChannels = GetVectorInt("RejectChannels",empty,false); 
    if(!fKeepChannels.empty() && !fRejectChannels.empty()) {
        Panic("Options KeepChannels and RejectChannels cannot be used at the same time");
    }
 
    if(fLoadEventList) {
        std::string fileName = GetString("EventList",Q_STRING_DEFAULT); 
        const QMatrix& eventChannel = GlobalData().Get<QMatrix>("EventList",fileName);
            QVector eventsFromFile = eventChannel.GetCol(1);
            for ( size_t ev = 0; ev < eventsFromFile.Size(); ev++) 
                fEventList.insert((unsigned int)eventsFromFile[ev]);
    }
    fRunData.Clear();
    ev.Require<QHeader>("DAQ","Header");
    ev.Require<QPulseInfo>("DAQ","PulseInfo");
}
 
bool MRawDataFilter::Filter(const QEvent& ev)
{
    const QHeader& header = ev.Get<QHeader>("DAQ","Header");
 
    if(fLoadEventList) {
        unsigned int evnum = header.GetEventNumber();
        if(!fEventList.count(evnum))  return false;
    }
    if(fMinEventNumber != -1 && header.GetEventNumber() < fMinEventNumber) return false;
    if(fMaxEventNumber != -1 && header.GetEventNumber() > fMaxEventNumber) return false;
 
    const int run = header.GetRun();
    if(run != fRunData.fNumber) {
       QRunDataHandle rHandle(run);
       GlobalData().Get<QRunData>("",&rHandle,"");
       fRunData = rHandle.Get();
    }
    const QPulseInfo& pulseInfo = ev.Get<QPulseInfo>("DAQ","PulseInfo");
    const int channel = pulseInfo.GetChannelId();
 
    bool result = true;
    if(fRunData.GetChannelRunData(channel).fIsThermometer) result = fKeepThermometers;
    else if(fRunData.GetChannelRunData(channel).fIsBolometer) result = fKeepBolometers;
    if(!result && fKeepAllChannelTypes) result = true;
    if(!result) return false;
 
    result = false;
    if(fKeepHeater && pulseInfo.GetIsPulser()) result = true;
    if(fKeepLED && pulseInfo.GetIsLED()) result = true;
    if(fKeepNoise && pulseInfo.GetIsNoise()) result = true;
    if(fKeepSignal && pulseInfo.GetIsSignal()) result = true;
    if(fKeepVeto && pulseInfo.GetIsVeto()) result = true;
    if(fKeepSide && pulseInfo.GetIsSide()) result = true;
    if(fKeepReversed && pulseInfo.GetIsReversed()) result = true;
    if(!result) return false;
 
    if(!fKeepChannels.empty() 
            && find(fKeepChannels.begin(),fKeepChannels.end(),channel) == fKeepChannels.end()) 
        return false;
 
    if(!fRejectChannels.empty()
            && find(fRejectChannels.begin(),fRejectChannels.end(),channel) != fRejectChannels.end()) 
        return false;
 
    if(fKeepRunDataChannels
            && std::find(fRunData.fThermalDetectorChannels.begin(),fRunData.fThermalDetectorChannels.end(),channel)
            == fRunData.fThermalDetectorChannels.end()) 
        return false;
 
    return true;
}