changa/html/Orb3dLB_8h_source.html

#ifndef _ORB3DLB_H_

#define _ORB3DLB_H_


#include "CentralLB.h"

#include "MapStructures.h"

#include "Orb3dLB.decl.h"

#include "TaggedVector3D.h"

#include <queue>


class Orb3dLB : public CBase_Orb3dLB {

  friend class MultistepLB;

private:

  bool firstRound;

  bool centroidsAllocated;

  ComparatorFn compares[NDIMS];

  ComparatorFn pc[NDIMS];

  // pointer to stats->to_proc

  decltype(LDStats::to_proc) *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_ */


MapStructures.h