#ifndef PE_TREE_MERGER_H
#define PE_TREE_MERGER_H

#include "ParallelGravity.h"

/// @addtogroup TreeBuild
/// @{

/// @brief Group to coordinate requests for remote moments during the
/// tree build
///
/// This group is used if MERGE_REMOTE_REQUESTS is defined.
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