NGnThreadData.h

#ifndef Ndmspc_NHnSparseThreadData_H
#define Ndmspc_NHnSparseThreadData_H
// #include "NBinning.h"
// #include "NStorageTree.h"
#include "NGnTree.h"
#include "NThreadData.h"
#include <unordered_set>
 
namespace Ndmspc {

class NGnThreadData : public NThreadData {
  public:
  NGnThreadData();

  virtual ~NGnThreadData();

  bool Init(size_t id, NGnProcessFuncPtr func, NGnBeginFuncPtr beginFunc, NGnEndFuncPtr endFunc, NGnTree * ngnt,
            NBinning * binningIn, NGnTree * input = nullptr, const std::string & filename = "",
            const std::string & treename = "ngnt");

  void SetProcessFunc(NGnProcessFuncPtr func) { fProcessFunc = func; }

  bool InitStorage(NStorageTree * ts, const std::string & filename = "", const std::string & treename = "ngnt");

  NGnTree * GetHnSparseBase() const { return fHnSparseBase; }

  Long64_t GetNProcessed() const { return fNProcessed; }

  void SetNProcessed(Long64_t n) { fNProcessed = n; }

  void SetCfg(const json & cfg) { fCfg = cfg; }

  json GetCfg() const { return fCfg; }

  void SetResultsFilename(const std::string & filename) { fResultsFilename = filename; }

  const std::string & GetResultsFilename() const { return fResultsFilename; }

  bool IsPureCopy() const { return fIsPureCopy; }

  void SetIsPureCopy(bool val) { fIsPureCopy = val; }

  virtual void Process(const std::vector<int> & coords);
 
  void SetCurrentDefinitionName(const std::string & name);
  void SyncCurrentDefinitionIds(const std::vector<Long64_t> & ids);
 
  void ExecuteBeginFunction();
  void ExecuteEndFunction();

  void FlushDeferredDeletes();

  virtual Long64_t Merge(TCollection * list);
 
  private:
  NGnProcessFuncPtr          fProcessFunc{nullptr};  
  NGnBeginFuncPtr            fBeginFunc{nullptr};    
  NGnEndFuncPtr              fEndFunc{nullptr};      
  NGnTree *                  fHnSparseBase{nullptr}; 
  Long64_t                   fNProcessed{0};         
  NBinning *                 fBiningSource{nullptr}; 
  json                       fCfg{};                 
  bool                       fIsPureCopy{false};     
  std::string                fResultsFilename{};     
  std::unordered_set<Long64_t> fProcessedBinIds{};  
  std::vector<TObject *>     fDeferredDeletes;       

  ClassDef(NGnThreadData, 1);
};
} // namespace Ndmspc
#endif