Revision 8696ae81c1c3f47e312611d3e310616e4dd21238 authored by Tom Quinn on 09 January 2016, 04:36:28 UTC, committed by Tom Quinn on 09 January 2016, 04:36:28 UTC
Change-Id: Ib688f8b91bc36d72c3abfa62f4f24b6d3d5390c3
1 parent 40365aa
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_ */
/*@}*/
Computing file changes ...