EventFile.h

#ifndef FRAMEWORK_EVENT_EVENTFILE_H_
#define FRAMEWORK_EVENT_EVENTFILE_H_
 
//---< C++ >---//
#include <map>
#include <string>
#include <vector>
 
//---< Framework >---//
#include "Framework/Configure/Parameters.h"
#include "Framework/Event.h"
 
//---< ROOT >---//
#include "TFile.h"
#include "TTree.h"
 
namespace ldmx {
class RunHeader;
}
 
namespace framework {
 
class EventFile {
 public:
  EventFile(const framework::config::Parameters &params,
            const std::string &filename, EventFile *parent, bool isOutputFile,
            bool isSingleOutput, bool isLoopable);
 
  EventFile(const framework::config::Parameters &param,
            const std::string &fileName, bool isLoopable);
 
  EventFile(const framework::config::Parameters &param,
            const std::string &fileName, EventFile *parent,
            bool isSingleOutput = false);
 
  EventFile(const framework::config::Parameters &params,
            const std::string &fileName);
 
  ~EventFile();
 
  bool isCorrupted() const;
 
  void addDrop(const std::string &rule);
 
  void setupEvent(Event *evt);
 
  void updateParent(EventFile *parent);
 
  Event *getEvent() { return event_; };
 
  bool nextEvent(bool storeCurrentEvent = true);
 
  int skipToEvent(int offset);
 
  void writeRunHeader(ldmx::RunHeader &runHeader);
 
  void writeRunTree();
 
  ldmx::RunHeader *getRunHeaderPtr(int runNumber);
 
  ldmx::RunHeader &getRunHeader(int runNumber);
 
  const std::string &getFileName() { return file_name_; }
 
 private:
  void importRunHeaders();
 
 private:
  Long64_t entries_{-1};
 
  Long64_t ientry_{-1};
 
  std::string file_name_;
 
  bool is_output_file_;
 
  bool is_single_output_;
 
  bool is_loopable_{false};
 
  TFile *file_{nullptr};
 
  TTree *tree_{nullptr};
 
  EventFile *parent_{nullptr};
 
  Event *event_{nullptr};
 
  std::vector<std::pair<std::string, bool>> pre_clone_rules_;
 
  std::vector<std::string> reactivate_rules_;
 
  std::map<int, std::pair<bool, ldmx::RunHeader *>> run_map_;
};
}  // namespace framework
 
#endif  // FRAMEWORK_EVENT_EVENTFILE_H