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