RdChannelNoiseImporter_RD.h

Go to the documentation of this file.

#ifndef _RdChannelNoiseImporter_RD_RdChannelNoiseImporter_RD_h_
#define _RdChannelNoiseImporter_RD_RdChannelNoiseImporter_RD_h_


#include <fwk/VModule.h>
#include <utl/RandomEngine.h>
#include <vector>
#include <revt/Channel.h>


namespace RdChannelNoiseImporter_RD {

  class RdChannelNoiseImporter_RD : public fwk::VModule {

  public:
    fwk::VModule::ResultFlag Init() override;
    fwk::VModule::ResultFlag Run(evt::Event& event) override;
    fwk::VModule::ResultFlag Finish() override;

  struct EventN {
    int traceCh0[2048] = {0};
    int traceCh1[2048] = {0};
    int stationId = 0;
    int sec = 0;
    int nanosec = 0;
    int evtno = 0;
  };

  private:
    fwk::VModule::ResultFlag ReadFromRootFile();

    // switches controlling noise file selection
    std::string fNoiseFilePath;
    bool fAddNoiseFromSameTimeToBothChannels = true;
    bool fTraceShifting = false;
    bool fReplaceDataWithNoise = false;

    utl::RandomEngine::RandomEngineType *fRandomEngine;

    // preliminary. Has to be revisited once there is a common strategy to get noise traces from event data
    std::pair<std::vector<revt::ChannelADCTimeSeries>, std::vector<revt::ChannelADCTimeSeries>> fNoiseTraces;

    REGISTER_MODULE("RdChannelNoiseImporter_RD", RdChannelNoiseImporter_RD);
  };
}


#endif