Revision a00246c779936b353f49762420d6f56227c21221 authored by Collin J. Sutton on 21 February 2014, 22:27:27 UTC, committed by Tom Quinn on 02 September 2016, 22:12:38 UTC
Needed for similar reason as described in commit @c9a512e: ChaNGa's
dependency generation assumes that CkCache.h is a local include, and here
the files _it_ includes were being listed without path in Makefile.dep.

Change-Id: Ie9086cfd66e0174d7f9027eed5112a1add6e3851
1 parent e0d7d68
Raw File
Orb3dLB.h
/**
 * \addtogroup CkLdb
*/
/*@{*/

#ifndef _ORB3DLB_H_
#define _ORB3DLB_H_

#include "CentralLB.h"
#include "MapStructures.h"
#include "Orb3dLB.decl.h"
#include "TaggedVector3D.h"
#include <queue>

void CreateOrb3dLB();
BaseLB * AllocateOrb3dLB();

class Orb3dLB : public CBase_Orb3dLB {
  friend class MultistepLB;
private:
  bool firstRound;
  bool centroidsAllocated;
  ComparatorFn compares[NDIMS];
  ComparatorFn pc[NDIMS];
  // pointer to stats->to_proc
  CkVec<int> *mapping;

  int procsPerNode;

  CkVec<TPObject> tps;
  CkVec<Node> nodes;

  void init();
  bool QueryBalanceNow(int step);

  void printData(BaseLB::LDStats &stats, int phase, int *revObjMap);

public:
  Orb3dLB(const CkLBOptions &);
  Orb3dLB(CkMigrateMessage *m) : CBase_Orb3dLB(m) {init();}
  void work(BaseLB::LDStats* stats);
  void directMap(int tpstart, int tpend, int nodestart, int nodeend);
  void map(int tpstart, int tpend, int nodestart, int nodeend, int xs, int ys, int zs, int dim);
  int nextDim(int dim, int xs, int ys, int zs);
  void partitionEvenLoad(int tpstart, int tpend, int &tpmid);
  void halveNodes(int nodestart, int nodeend, int &nodemid);


};

#endif /* _ORB3DLB_H_ */

/*@}*/
back to top