RdChannelNoiseImporter_AERA.h

Go to the documentation of this file.

#ifndef _RdChannelNoiseImporter_AERA_RdChannelNoiseImporter_AERA_h_
#define _RdChannelNoiseImporter_AERA_RdChannelNoiseImporter_AERA_h_

#include <fwk/VModule.h>

#include <evt/Event.h>
#include <revt/Channel.h>

#include <utl/AugerUnits.h>
#include <utl/RandomEngine.h>
#include <utl/TimeStamp.h>
#include <utl/UTCDateTime.h>

#include <rdet/Channel.h>

#include <TBranch.h>
#include <TTree.h>

#include <RadioFileIO.h>
#include <string>

namespace evt {
  class Event;
}

namespace revt {
  class Channel;
}

namespace RdChannelNoiseImporter_AERA {

  class RdChannelNoiseImporter_AERA : public fwk::VModule {

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

  private:
    // switches controlling noise file selection
    std::string fNoiseFileSelection;
    std::string fNoiseFilePath;
    bool fExpectSubDirectories = false;
    std::string fFilenamePrefix;
    std::string fFilenameSuffix;
    std::string fTimeIntervalInFile;
    bool fRejectStationsWithoutNoiseInformation = true;
    bool fReplaceDataWithNoise = false;
    utl::TimeStamp fRandomNoiseTimeIntervalStart = 0;
    utl::TimeStamp fRandomNoiseTimeIntervalStop = 0;

    // switches controlling the reading of noise from a selected file
    std::string fEvtSelInNoiseFile;
    int fNoiseEvtNr = -1;
    unsigned int fNumberOfRejectedStationsWithoutNoise = 0;
    double fMaxTimeDifference = 10 * utl::minute;

    utl::RandomEngine::RandomEngineType* fRandomEngine = nullptr;

    fwk::VModule::ResultFlag checkSetup(); //check the setup given by the init-file for consistency

    std::string GetNoiseFileName(const utl::UTCDateTime& eventTime, utl::TimeStamp& noiseTimeStamp);
    bool GetNoiseFileNameAtTime(const utl::UTCDateTime& dateTime,
                                const utl::UTCDateTime& startingDay,
                                std::string& NoiseImportfilename);
    int GetNoiseEventNumber(evt::Event& event, RadioFileIO* RadioFileIORead,
                            TBranch* const evtSecBranch, TTree* const IOLibTree,
                            unsigned int NumberEvents, utl::UTCDateTime eventTime);
    bool CheckMatchingChannel(const rdet::Channel& detChannel, const rdet::Channel& replacementChannel);

    void RemovePedestal(revt::ChannelADCTimeSeries& timeSeries);
    void RemovePedestal(std::vector<int>& timeSeries);

    REGISTER_MODULE("RdChannelNoiseImporter_AERA", RdChannelNoiseImporter_AERA);
  };

}

#endif