RecDataWriter.h

Go to the documentation of this file.

#ifndef _RecDataWriter_h_
#define _RecDataWriter_h_

#include <fwk/VModule.h>
#include <fwk/CentralConfig.h>
#include <utl/CoordinateSystemPtr.h>
#include <utl/config.h>
#include <sevt/Station.h>

#include <adst/FdRecLevel.h>
#include <adst/SdRecLevel.h>
#include <adst/TraceType.h>
#include <adst/RecEventFile.h>

#include <string>
#include <fstream>


class RecEvent;
class GenShower;
class FDEvent;
class SDEvent;
class Detector;
class FileInfo;
class DetectorGeometry;
class RdEvent;

namespace utl {
  class Branch;
};

namespace evt {
  class Event;
  class ShowerFRecData;
}

namespace fevt {
  class Eye;
  class Pixel;
  class EyeRecData;
  class Telescope;
  class TelescopeRecData;
}

#ifdef INCLUDE_HAS
namespace HASTools {
  class MuonProfile;
  class VTankResponse;
  class EMComponent;
}
#endif

namespace revt {
  class REvent;
}

namespace ADST {
  class AsciiConverter;
};

namespace otoa {
  class Offline2ADST;
};


namespace RecDataWriterNG {

  class FOVCalculator;


  class RecDataWriter : public fwk::VModule {

  public:
    RecDataWriter();
    virtual ~RecDataWriter();

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

  private:
    // Methods related to configuration reading
    bool ReadConfig();  //< Does all the accessing of CentralConfig
    bool ReadRootConfig(utl::Branch& topB);  //< Reads just the config for ROOT streaming
    bool ReadAsciiConfig(utl::Branch& topB);  //< Reads just the config for ASCII streaming

    // Methods related to initializing the converters / output related members
    bool PrepareAsciiFile();
    bool PrepareRecEventAndRootFile();

    // Data and configuration related to ADST ROOT writing
    std::string fRootOutputFileName;
    RecEventFile::Mode fRootOutputFileMode = RecEventFile::eWrite;
    bool fWriteROOT = false;
    RecEventFile* fRecEventFile = nullptr;
    otoa::Offline2ADST* fADSTConverter = nullptr;

    // Data and configuration related to ADST ASCII writing
    std::string fAsciiOutputFileName;
    RecEventFile::Mode fAsciiOutputFileMode = RecEventFile::eWrite;
    bool fWriteASCII = false;
    std::string fAsciiDataMode;
    ADST::AsciiConverter* fAsciiConverter = nullptr;
    int fAsciiMinFdRecLevel = 0;  //< minimum FD reconstruction level to store data in ASCII
    int fAsciiMinSdRecLevel = 0;  //< minimum SD reconstruction level to store data in ASCII

    int fVerbosity = 0;

    int fNWritten = 0;
    int fNWrittenStereo = 0;
    int fNWrittenTriple = 0;
    int fNWrittenQuadruple = 0;

    RecEvent* fRecEvent = nullptr;

    REGISTER_MODULE("RecDataWriterNG", RecDataWriter);

  };

}


#endif