NamdCentLB.h

Go to the documentation of this file.

/*****************************************************************************
 * $Source: /home/cvs/namd/cvsroot/namd2/src/NamdCentLB.h,v $
 * $Author: jlai7 $
 * $Date: 2012/11/27 21:13:18 $
 * $Revision: 1.17 $
 *****************************************************************************/

#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();
  NamdCentLB(CkMigrateMessage *);
#if CHARM_VERSION > 60301
  CLBMigrateMsg* Strategy(LDStats* stats);
#else
  CLBMigrateMsg* Strategy(LDStats* stats, int n_pes);
#endif

private:
  CmiBool QueryBalanceNow(int step);
  CmiBool 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_ */

---