EventFile.h

#ifndef FRAMEWORK_EVENT_EVENTFILE_H_
#define FRAMEWORK_EVENT_EVENTFILE_H_
 
//---< C++ >---//
#include <map>
#include <memory>
#include <string>
#include <vector>
 
//---< Framework >---//
#include "Framework/Configure/Parameters.h"
#include "Framework/Event.h"
#include "Framework/Logger.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(std::shared_ptr<ldmx::RunHeader> runHeader);
 
  void writeRunTree(bool completed = false);
 
  ldmx::RunHeader* getRunHeaderPtr(int runNumber);
 
  ldmx::RunHeader& getRunHeader(int runNumber);
 
  const std::string& getFileName() { return file_name_; }
 
  std::vector<int> getIncompleteRuns() const;
 
 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::shared_ptr<ldmx::RunHeader>> run_map_;
 
  TTree* run_tree_{nullptr};
 
  bool run_headers_have_completeness_{false};
 
  enableLogging("EventFile")
};
}  // namespace framework
 
#endif  // FRAMEWORK_EVENT_EVENTFILE_H