OfflineROOTFile.h

Go to the documentation of this file.

#ifndef _OfflineROOTFile_h_
#define _OfflineROOTFile_h_

#include <io/VROOTFile.h>
#include <io/StreamerSwitch.h>


class TBranch;

namespace evt {
  class Event;
}

namespace utl {
  class Branch;
}

namespace io {

  class Event_ROOT;
  class Config_ROOT;


  class OfflineROOTFile : public VROOTFile {

  private:
    // set default switch (constructed with all streamers on)
    static const io::StreamerSwitch fgDefaultSwitch;

  public:
    OfflineROOTFile() = default;

    OfflineROOTFile(const std::string& filename, const Mode mode = eRead,
                    utl::Branch* const b = nullptr,
                    const io::StreamerSwitch& streamerSwitch = fgDefaultSwitch);

    virtual ~OfflineROOTFile();

    void Open(const std::string& filename, const Mode mode, utl::Branch* const b = nullptr) override;

    void Close() override;
    
    void Write(const evt::Event& event) override;

    Status Read(evt::Event& event) override;

    Status FindEvent(const unsigned int eventId) override;

    Status GotoPosition(const unsigned int evNo) override;

    int GetNEvents() override;


    std::string GetConfig();

    static const std::string kOfflineTreeName;
    static const std::string kOfflineBranchName;
    static const std::string kConfigBranchName;

  private:
    Status GetEventBranch();

    unsigned int fCurEventNo = 0;

    TBranch* fBranch = nullptr;
    TBranch* fConfigBranch = nullptr;
    io::Event_ROOT* fCurEvent = nullptr;
    io::Config_ROOT* fConfig = nullptr;
    io::StreamerSwitch fStreamerSwitch;
    
    std::ofstream* fConfigOut = nullptr;
  };

}


#endif