NamdCentLB.h

Go to the documentation of this file.

/*****************************************************************************
 * $Source: /home/cvs/namd/cvsroot/namd2/src/NamdCentLB.h,v $
 * $Author: jim $
 * $Date: 2013/08/22 15:17:18 $
 * $Revision: 1.19 $
 *****************************************************************************/

#ifndef _NAMDCENTLB_H_
#define _NAMDCENTLB_H_

#include <CentralLB.h>
#include "NamdCentLB.decl.h"

#include "Node.h"
#include "PatchMap.h"
#include "SimParameters.h"
#include "RefineOnly.h"
#include "Alg7.h"
#include "AlgRecBisection.h"
#include "InfoStream.h"
#include "TorusLB.h"
#include "RefineTorusLB.h"

void CreateNamdCentLB();
NamdCentLB *AllocateNamdCentLB();

class NamdCentLB : public CentralLB {

public:
  NamdCentLB(const CkLBOptions& opt);
  NamdCentLB(CkMigrateMessage *);
  CLBMigrateMsg* Strategy(LDStats* stats);

private:
  bool QueryBalanceNow(int step);
  bool QueryDumpData();
  int buildData(LDStats* stats);
  int requiredProxies(PatchID id, int neighborNodes[]);
#if USE_TOPOMAP 
  int requiredProxiesOnProcGrid(PatchID id, int neighborNodes[]);
#endif
  void dumpDataASCII(char *file, int numProcessors, int numPatches,
                int numComputes);
  void loadDataASCII(char *file, int &numProcessors, int &numPatches,
                int &numComputes);

  computeInfo *computeArray;
  patchInfo *patchArray;
  processorInfo *processorArray;
};

#endif /* _NAMDCENTLB_H_ */