NStorageTree.h

#ifndef Ndmspc_NStorageTree_H
#define Ndmspc_NStorageTree_H
#include <TObject.h>
#include <TTree.h>
#include <TFile.h>
#include "NBinning.h"
#include "NBinningPoint.h"
#include "NTreeBranch.h"
 
namespace Ndmspc {
 
class NStorageTree : public TObject {
  public:
  NStorageTree(NBinning * binning = nullptr);
 
  virtual ~NStorageTree();
 
  virtual void Print(Option_t * option = "") const;
 
  virtual void Clear(Option_t * option = "");
 
 
  bool SetFileTree(TFile * file, TTree * tree);
 
  bool InitTree(const std::string & filename = "", const std::string & treename = "ngnt");
 
  Long64_t GetEntries() const { return fTree ? fTree->GetEntries() : 0; }
 
  Long64_t GetEntry(Long64_t entry, NBinningPoint * point = nullptr, bool checkBinningDef = false);
 
  // Int_t    Fill(NBinningPoint * point);
 
  Int_t Fill(NBinningPoint * point, NStorageTree * hnstIn = nullptr, bool ignoreFilledCheck = false,
             std::vector<std::vector<int>> ranges = {}, bool useProjection = false);
 
  bool Close(bool write = false, std::map<std::string, TList *> outputs = {});
 
  TFile * GetFile() const { return fFile; }
 
  Long64_t Merge(TCollection * list);
 
 
  std::vector<std::string> GetBrancheNames(bool onlyEnabled = false) const;
 
  std::map<std::string, NTreeBranch> GetBranchesMap() const { return fBranchesMap; }
 
  void SetBranchesMap(std::map<std::string, NTreeBranch> map) { fBranchesMap = map; }
 
  bool AddBranch(const std::string & name, void * address, const std::string & className);
 
  NTreeBranch * GetBranch(const std::string & name);
 
  TObject * GetBranchObject(const std::string & name);
 
  void SetEnabledBranches(std::vector<std::string> branches, int status = 1);
 
  void SetBranchAddresses();
 
  NBinning * GetBinning() const { return fBinning; }
 
  void SetBinning(NBinning * binning) { fBinning = binning; }
 
  TTree * GetTree() const { return fTree; }
 
  std::string GetFileName() const { return fFileName; }
 
  std::string GetPrefix() const { return fPrefix; }
 
  std::string GetPostfix() const { return fPostfix; }
 
  void SetOutputs(TMap * outputs) { fOutputs = outputs; }
 
  protected:
  std::string                        fFileName{"ngnt.root"}; 
  TFile *                            fFile{nullptr};         
  TTree *                            fTree{nullptr};         
  std::string                        fPrefix{""};            
  std::string                        fPostfix{""};           
  std::map<std::string, NTreeBranch> fBranchesMap;           
  TMap *                             fOutputs;               
  NBinning *                         fBinning{nullptr};      
 
  ClassDef(NStorageTree, 1);
};
} // namespace Ndmspc
#endif