examples/charm++/cell/md/patch.h

   1 #ifndef __PATCH_H__
   2 #define __PATCH_H__
   3
   4
   5 #include "md_config.h"
   6 #include "patch.decl.h"
   7 #include "pairCompute.decl.h"
   8
   9
  10 class Patch : public CBase_Patch {
  11
  12   // Declare CkIndex_Patch as a friend so accelerated entry methods can access
  13   //   the member variables of the object they execute on
  14   friend class CkIndex_Patch;
  15
  16   private:
  17
  18     /// Member Variables ///
  19     int remainingForceCheckIns;
  20     int remainingIterations;
  21
  22     int numParticles;
  23     MD_FLOAT* particleX;  // x-coordinate
  24     MD_FLOAT* particleY;  // y-coordinate
  25     MD_FLOAT* particleZ;  // z-coordinate
  26     MD_FLOAT* particleQ;  // charge
  27     MD_FLOAT* particleM;  // mass
  28     MD_FLOAT* velocityX;  // velocity x-component
  29     MD_FLOAT* velocityY;  // velocity y-component
  30     MD_FLOAT* velocityZ;  // velocity z-component
  31     MD_FLOAT* forceSumX;  // Buffer to sum of force x components from computes
  32     MD_FLOAT* forceSumY;  // Buffer to sum of force y components from computes
  33     MD_FLOAT* forceSumZ;  // Buffer to sum of force z components from computes
  34
  35     #if USE_ARRAY_SECTIONS != 0
  36       CProxySection_PairCompute pairComputeArraySection_lower;
  37       CProxySection_PairCompute pairComputeArraySection_upper;
  38     #endif
  39
  40     #if USE_PROXY_PATCHES != 0
  41       CProxy_ProxyPatch proxyPatchProxy;
  42       int* proxyPatchPresentFlags;
  43     #endif
  44
  45
  46     // DMK - DEBUG
  47     unsigned int localFlopCount;
  48
  49
  50     /// Member Functions ///
  51     void randomizeParticles();
  52     MD_FLOAT randf();
  53     void startIteration_common(int numIters);
  54
  55   public:
  56
  57     /// Constructor(s) \ Destructor ///
  58     Patch();
  59     Patch(CkMigrateMessage* msg);
  60     ~Patch();
  61
  62     /// Entry Methods ///
  63     void init(int numParticles);
  64     void init(int numParticles, CProxy_ProxyPatch proxyPatchProxy);
  65     void init_common(int numParticles);
  66     void startIteration();
  67     void startIterations(int numIters);
  68     void forceCheckIn(int numParticles, MD_FLOAT* forceX, MD_FLOAT* forceY, MD_FLOAT* forceZ);
  69     void forceCheckIn(int numParticles, MD_FLOAT* forceX, MD_FLOAT* forceY, MD_FLOAT* forceZ, int numForceCheckIns);
  70     void integrate_callback();
  71
  72 };
  73
  74
  75 class ProxyPatch : public CBase_ProxyPatch {
  76
  77   // Declare CkIndex_ProxyPatch as a friend so accelerated entry methods can access
  78   //   the member variables of the object they execute on
  79   friend class CkIndex_ProxyPatch;
  80
  81   private:
  82
  83     /// Member Variables ///
  84     int numParticles;
  85     MD_FLOAT* particleX;
  86     MD_FLOAT* particleY;
  87     MD_FLOAT* particleZ;
  88     MD_FLOAT* particleQ;
  89     MD_FLOAT* forceSumX;
  90     MD_FLOAT* forceSumY;
  91     MD_FLOAT* forceSumZ;
  92
  93     int patchIndex;
  94
  95     int checkInCount;
  96     int numToCheckIn;
  97
  98   public:
  99
 100     /// Constructor(s) \ Destructor ///
 101     ProxyPatch(int proxyIndex);
 102     ProxyPatch(CkMigrateMessage *msg);
 103     ~ProxyPatch();
 104
 105     /// Entry Methods ///
 106     void init(int numParticles);
 107     void patchData(int numParticles, MD_FLOAT* particleX, MD_FLOAT* particleY, MD_FLOAT* particleZ, MD_FLOAT* particleQ);
 108     void forceCheckIn(int numParticles, MD_FLOAT* forceX, MD_FLOAT* forceY, MD_FLOAT* forceZ);
 109
 110 };
 111
 112 #endif //__PATCH_H__