NTreeBranch.h

#ifndef Ndmspc_NTreeBranch_H
#define Ndmspc_NTreeBranch_H
#include <TObject.h>
#include <TBranch.h>
#include <string>
 
namespace Ndmspc {
 
class NTreeBranch : public TObject {
  public:
  NTreeBranch(TTree * tree = nullptr, const std::string & name = "", void * address = nullptr,
              const std::string & objClassName = "TObject");
 
  virtual ~NTreeBranch();
 
  virtual void Print(Option_t * option = "") const;
 
  void SetName(const std::string & name) { fName = name; }
 
  void SetObjectClassName(const std::string & objClassName) { fObjectClassName = objClassName; }
 
  void SetBranch(TBranch * branch) { fBranch = branch; }
 
  void SetObject(TObject * obj) { fObject = obj; }
 
  TBranch * GetBranch() { return fBranch; }
 
  int GetBranchStatus() const { return fBranchStatus; }
 
  TObject * GetObject() const { return fObject; }
 
  std::string GetObjectClassName() const { return fObjectClassName; }
 
  TBranch * Branch(TTree * tree, void * address);
 
  void SetAddress(void * address, bool deleteExisting = false);
 
  void SetBranchAddress(TTree * tree);
 
  void SetBranchStatus(int status) { fBranchStatus = status; }
 
  Long64_t GetEntry(TTree * tree, Long64_t entry);
 
  void SaveEntry(NTreeBranch * hnstIn, bool useProjection = false, const std::string projOpt = "OE");
 
  private:
  std::string fName{""};                   
  int         fBranchStatus{1};            
  TBranch *   fBranch{nullptr};            
  TObject *   fObject{nullptr};            
  std::string fObjectClassName{"TObject"}; 
 
  ClassDef(NTreeBranch, 1);
};
} // namespace Ndmspc
#endif