MRunDataLoader.cc

Go to the documentation of this file.

#include "MRunDataLoader.hh"
#include "QEvent.hh"
#include "QRawEvent.hh"
#include "QError.hh"
#include "QBaseType.hh"
#include "QRunDataHandle.hh"
 
 
using namespace Diana;
 
REGISTER_MODULE(MRunDataLoader)
 
MRunDataLoader::MRunDataLoader()
{
}
 
MRunDataLoader::~MRunDataLoader() 
{
}
 
void MRunDataLoader::Init(QEvent& ev) 
{
    fIsOnline = GetBool("IsOnline",false,false);
    fRunDataOwner = GetString("RunDataOwner","DAQ",false);
    fRunDataInput = GetString("Input" ,"DB");
    fRunDataOutputFile = GetString("Output","",false);
    fWriteIntoEvent = GetBool("WriteIntoEvent",false,false);
    if(fWriteIntoEvent) {
        ev.Add<QChannelRunData>("ChannelRunData");
    }
    fSaveRunData = (fRunDataOutputFile != "");
    if(fSaveRunData) {
        fRunDataOutputFile2 = GetString("Output2","",false);
    }
    fThisRun = -9999999;
    ev.Require<QHeader>("DAQ","Header");
    int dataset = GetInt("Dataset",-1,false);
    if(dataset > 0) {
        if(fWriteIntoEvent) {
            ev.Add<QInt>("Dataset");
        }
        GlobalHandle<QInt> dHandle("Dataset");
        dHandle.Set(dataset);
        GlobalData().Set(&dHandle,"");
    }
    fRunsInput = GetString("RunsInput",fRunDataInput,false);
}
 
void MRunDataLoader::Do(QEvent& ev) 
{
    const QHeader& raw = ev.Get<QHeader>("DAQ","Header");
    const int currRun = raw.GetRun();
 
    bool load = false;
    if(currRun != fThisRun) {
        fThisRun = currRun;
        fRunData.Reset();
        load = true;
    }
 
    if(load) {
        // get rundata
        QRunDataHandle rdh(fThisRun);
        rdh.SetIsOnline(fIsOnline);
        GlobalData().Get(fRunDataOwner,&rdh,fRunDataInput);
        if(!rdh.IsValid()) {
            Error("RunData is not valid for run %d. Error: %s",fThisRun,rdh.GetError().GetDescription().c_str());
            load = false;
        } else {
            fRunData = rdh.Get();
 
            // save rundata to files 
            if(fSaveRunData) {
                GlobalData().Set(&rdh,fRunDataOutputFile);
                if(fRunDataOutputFile2 != "") 
                    GlobalData().Set(&rdh,fRunDataOutputFile2);
            }
            // print rundata 
            Info("RunData:");
            std::stringstream dump;
            rdh.Get().Dump(dump);
            Info("%s",dump.str().c_str());
        }
        GlobalHandle<QInt> dHandle("Dataset");
        GlobalData().Get(&dHandle,"");
        if(dHandle.IsValid() && fRunsInput != "NONE") {
            // MV FIXME: check that runnumber is in dataset
        } // dataset
    }
 
     if(fWriteIntoEvent) {
        // save into event channel based information
        const QPulseInfo& pi = ev.Get<QPulseInfo>("DAQ","PulseInfo");
        const int channel = pi.GetChannelId();
 
        if(fRunData.IsValid()) {
            const QChannelRunData& channelRunData = fRunData.GetChannelRunData(channel);
            ev.Get<QChannelRunData>("ChannelRunData") = channelRunData;
        }
        GlobalHandle<QInt> dHandle("Dataset");
        GlobalData().Get(&dHandle,"");
        if(dHandle.IsValid()) {
            ev.Get<QInt>("Dataset") = dHandle.Get();
        }
   }
 
    return;
}
 
void MRunDataLoader::Done() 
{
}