PETreeMerger.h

#ifndef PE_TREE_MERGER_H
#define PE_TREE_MERGER_H
 
#include "ParallelGravity.h"


class PETreeMerger : public CBase_PETreeMerger {
 
  CkVec<GenericTreeNode*> submittedRoots;
  CkVec<TreePiece*> submittedTreePieces;
 
  NonEmptyTreePieceCounter submittingTreePieceCounter;
 
  public:
  PETreeMerger() {}
  PETreeMerger(CkMigrateMessage *m) : CBase_PETreeMerger(m) {}
  void pup(PUP::er &p) { CBase_PETreeMerger::pup(p); }
 
  void mergeNonLocalRequests(GenericTreeNode *root, TreePiece *treePiece);
  void freeTree();
 
  private:
 
  void mergeWalk(CkVec<GenericTreeNode*> &mergeList, CkVec<TreePiece*> &treePieceList);
  void requestNonLocalMoments(GenericTreeNode *pickedNode, TreePiece *pickedTreePiece);
 
};

#endif