examples/cxx/basic/Master.cxx

   1 #include "Master.hpp"\r
   2 #include "BasicTask.hpp"\r
   3 #include "FinalizeTask.hpp"\r
   4 #include <Host.hpp>\r
   5 #include <HostNotFoundException.hpp>\r
   6 \r
   7 #include <Msg.hpp>\r
   8 \r
   9 #include <iostream>\r
  10 using namespace std;\r
  11 \r
  12 \r
  13 #ifndef BUFFMAX\r
  14 #define BUFFMAX 260\r
  15 #endif\r
  16 \r
  17 MSG_IMPLEMENT_DYNAMIC(Master, Process);\r
  18 \r
  19 int Master::main(int argc, char** argv)\r
  20 {\r
  21         int channel = 0;\r
  22         char buff[BUFFMAX + 1] = {0};\r
  23         int numberOfTasks;              \r
  24         double taskComputeSize;         \r
  25         double taskCommunicateSize;\r
  26 \r
  27         if (argc < 3) \r
  28         {\r
  29                 cerr <<"Master needs 3 arguments" << endl;\r
  30                 exit(1);\r
  31         }\r
  32         \r
  33         cout << "Hello I'm " << getName() << " on " << getHost().getName() << "!" << endl;\r
  34         \r
  35         int slaveCount = 0;\r
  36         Host* slaves = NULL;\r
  37         \r
  38         cout << "argc=" <<  argc << endl;\r
  39         \r
  40         for (int i = 0; i< argc; i++)       \r
  41                 cout << "argv:" << argv[i] << endl;\r
  42         \r
  43         sscanf(argv[0],"%d", &numberOfTasks);\r
  44         \r
  45         sscanf(argv[1],"%lg", &taskComputeSize);\r
  46         \r
  47         sscanf(argv[2],"%lg", &taskCommunicateSize);\r
  48         \r
  49         BasicTaskPtr* todo = new BasicTaskPtr[numberOfTasks];\r
  50         \r
  51         for (int i = 0; i < numberOfTasks; i++) \r
  52         {\r
  53                 sprintf(buff,"Task_%d",i);\r
  54                 todo[i] = new BasicTask(buff, taskComputeSize, taskCommunicateSize);\r
  55                 memset(buff, 0 , BUFFMAX + 1); \r
  56         }\r
  57         \r
  58         slaveCount = argc - 3;\r
  59         slaves = new Host[slaveCount];\r
  60         \r
  61         for(int i = 3; i < argc ; i++)  \r
  62         {\r
  63                 try \r
  64                 {\r
  65                         slaves[i-3] = Host::getByName(argv[i]);\r
  66                 }\r
  67                 \r
  68                 catch(HostNotFoundException e) \r
  69                 {\r
  70                         cerr << e.toString() <<". Stopping Now!" << endl;\r
  71                         exit(1);\r
  72                 }\r
  73         }\r
  74         \r
  75         cout <<"[" << getName() << ":" << getHost().getName() << "] " << "Got slave(s) :"  << slaveCount << endl;\r
  76         \r
  77         for (int i = 0; i < slaveCount; i++)\r
  78                 cout <<"[" << getName() << ":" << getHost().getName() << "] " << "\t" << slaves[i].getName() << endl;\r
  79         \r
  80         cout <<"[" << getName() << ":" << getHost().getName() << "] " << "Got " << numberOfTasks << " task to process." << endl;\r
  81         \r
  82         for (int i = 0; i < numberOfTasks; i++) \r
  83         {\r
  84                 cout <<"[" << getName() << ":" << getHost().getName() << "] " << "Sending \"" << todo[i]->getName() << "\" to \"" << slaves[i % slaveCount].getName() << "\"" << endl;\r
  85         \r
  86                 if(!strcmp(Host::currentHost().getName(), slaves[i % slaveCount].getName())) \r
  87                         cout <<"[" <<  getName() << ":" << getHost().getName() << "] " << "Hey ! It's me ! ";\r
  88 \r
  89                 slaves[i % slaveCount].put(channel, todo[i]);\r
  90         }\r
  91         \r
  92         cout <<"[" << getName() << ":" << getHost().getName() << "] " << "Send completed" << endl;\r
  93         \r
  94         cout <<"[" << getName() << ":" << getHost().getName() << "] " << "All tasks have been dispatched. Let's tell everybody the computation is over." << endl;\r
  95         \r
  96         for (int i = 0; i < slaveCount; i++) \r
  97                         slaves[i].put(channel, new FinalizeTask());\r
  98 \r
  99         delete[] todo;\r
 100         delete[] slaves;\r
 101         \r
 102         cout <<"[" << getName() << ":" << getHost().getName() << "] " << "Goodbye now!" << endl;\r
 103 \r
 104         delete this;\r
 105 \r
 106         return 0;\r
 107 }