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