examples/charm++/cell/hello/hello.C

   1 #include <stdio.h>
   2 #include "hello.decl.h"
   3
   4 /*readonly*/ CProxy_Main mainProxy;
   5 /*readonly*/ int nElements;
   6
   7
   8 class Main : public CBase_Main {
   9
  10   public:
  11     Main(CkArgMsg* m) {
  12
  13       //Process command-line arguments
  14       nElements = 5;
  15       if (m->argc > 1) nElements = atoi(m->argv[1]);
  16       delete m;
  17
  18       //Start the computation
  19       CkPrintf("Running Hello on %d processors for %d elements\n", CkNumPes(), nElements);
  20       mainProxy = thisProxy;
  21
  22       CProxy_Hello arr = CProxy_Hello::ckNew(nElements);
  23       char *msg = "Hello from Main";
  24       arr[0].saySomething(strlen(msg) + 1, msg, -1);
  25     };
  26
  27     void done(void) {
  28       CkPrintf("All done\n");
  29       CkExit();
  30     };
  31 };
  32
  33
  34 class Hello : public CBase_Hello {
  35
  36   // Declare the CkIndex_Hello class as a friend of this class so that the accelerated
  37   //   entry methods can access the member variables of this class
  38   friend class CkIndex_Hello;
  39
  40   public:
  41
  42     Hello() { }
  43     Hello(CkMigrateMessage *m) {}
  44     ~Hello() { }
  45
  46     void saySomething_callback() {
  47       if (thisIndex < nElements - 1) {
  48         char msgBuf[128];
  49         int msgLen = sprintf(msgBuf, "Hello from %d", thisIndex) + 1;
  50         thisProxy[thisIndex+1].saySomething(msgLen, msgBuf, thisIndex);
  51       } else {
  52         mainProxy.done();
  53       }
  54     }
  55 };
  56
  57
  58 #include "hello.def.h"