Revision 801f5a06219a6458a5b55105dd1430fd74dfae67 authored by Tom Quinn on 09 January 2016, 05:02:47 UTC, committed by Michael Robson on 12 January 2016, 21:58:37 UTC
Change-Id: I7cc96fc7b61bc40e9d6c5ce8a122cd2e41c72824
1 parent 84ab0fe
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 ...