o815/o815.cpp

   1 #include "o815.h"
   2
   3 #include <sstream>
   4
   5 o815::o815(int argc, char **argv, const string& _programid) {
   6   long timestamp;
   7
   8   programid = _programid;
   9
  10   comargs.nmeas = 100;
  11   comargs.nskip = 10;
  12   comargs.nequi = 100;
  13   comargs.lsize[0] = 4;
  14   comargs.lsize[1] = 4;
  15   comargs.obscache = make_pair("",0);
  16   comargs.confcache = make_pair("",0);
  17   comargs.outdir="";
  18   comargs.idonly = false;
  19   comargs.showjobnum = false;
  20
  21   MPI_Init(&argc, &argv);
  22   MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
  23   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  24
  25   paraQ = new paraq(numprocs, rank);
  26
  27   parseArgs(argc, argv);
  28
  29   if(comargs.idonly) {
  30     cout << programid << headMaster() << endl << flush;
  31     exit(0);
  32   }
  33
  34   if( comargs.showjobnum ) {
  35     for( int i=1; i<=paraQ->getTotalJobs(); i++ ) {
  36       if( paraQ->getTotalJobs()%i == 0 ) cout << paraQ->getTotalJobs()/i << "@" << i << " ";
  37     }
  38     cout << endl;
  39     exit(0);
  40   }
  41
  42   if(rank==0) {
  43     timestamp = time(NULL);
  44     for(int idest=1; idest<numprocs; idest++)
  45       MPI_Send(&timestamp, 1, MPI_LONG, idest, 123, MPI_COMM_WORLD);
  46   }
  47   else if(rank>0)
  48     MPI_Recv(&timestamp, 1, MPI_LONG, 0, 123, MPI_COMM_WORLD, &mpiStatus);
  49
  50   out = new writeout(comargs.outdir, programid+headMaster(), rank, numprocs, timestamp);
  51 }
  52
  53 void o815::mainLoop() {
  54   *out->log << "OBS: Starting main loop." << endl;
  55
  56   for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit) {
  57     (*obsit)->start();
  58   }
  59
  60   //while( nextParas() ) {
  61   //}
  62
  63   /*
  64   for (int i=0; i<100; i++) {
  65     for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit) {
  66       (*obsit)->meas();
  67     }
  68   }
  69   */
  70
  71   for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit) {
  72     (*obsit)->finish();
  73   }
  74 }
  75
  76 void o815::parseArgs(int argc, char **argv) {
  77   int opt = 0;
  78
  79   while( (opt = getopt(argc, argv, "L:N:S:E:o:O:c:C:w:i:j:")) != -1 )
  80     switch(opt) {
  81     case 'L':
  82       listArg(comargs.lsize, 2, optarg);
  83       break;
  84     case 'N':
  85       comargs.nmeas = atoi(optarg);
  86       break;
  87     case 'S':
  88       comargs.nskip = atoi(optarg);
  89       break;
  90     case 'E':
  91       comargs.nequi = atoi(optarg);
  92       break;
  93     case 'o':
  94       comargs.obscache.first = optarg;
  95       comargs.obscache.second = 1;
  96       break;
  97     case 'O':
  98       comargs.obscache.first = optarg;
  99       comargs.obscache.second = 2;
 100       break;
 101     case 'c':
 102       comargs.confcache.first = optarg;
 103       comargs.confcache.second = 1;
 104       break;
 105     case 'C':
 106       comargs.confcache.first = optarg;
 107       comargs.confcache.second = 2;
 108       break;
 109     case 'w':
 110       comargs.outdir = optarg;
 111       break;
 112     case 'i':
 113       comargs.idonly = atoi(optarg);
 114       break;
 115     case 'j':
 116       comargs.showjobnum = atoi(optarg);
 117       break;
 118     }
 119 }
 120
 121 void o815::listArg(int *target, int tlen, char *listarg) {
 122   int nargs=0;
 123
 124   for( int pos=0; pos<strlen(listarg); pos++ )
 125     if( listarg[pos] == ':' ) nargs++;
 126
 127   if(nargs==0)
 128     for(int i=0; i<tlen; i++) target[i] = atoi(listarg);
 129   else
 130     {
 131       target[0] = atoi(strtok(listarg, ":"));
 132       for(int i=0; i<nargs; i++)
 133         target[i+1] = atoi(strtok(NULL, ":"));
 134     }
 135 }
 136
 137 string o815::headMaster()
 138 {
 139   stringstream hm;
 140
 141   hm << "-L" << comargs.lsize[0] << "_" << comargs.lsize[1] << "-E" << comargs.nequi << "-S" << comargs.nskip << "-N" << comargs.nmeas
 142      << paraQ->rangeString();
 143
 144   return hm.str();
 145 }
 146
 147 o815::~o815() {
 148   if(comargs.outdir=="") {
 149     MPI_Barrier(MPI_COMM_WORLD);
 150     if(rank==0)
 151       cout << "#end" << endl << flush;
 152   }
 153   delete out;
 154   MPI_Finalize();
 155 }
 156
 157 int o815::nextParas()
 158 {
 159   if( paraQ->nextParas() ) {
 160     /*
 161     setB( (*paraQ)["beta"] );
 162     setK( (*paraQ)["kappa"] );
 163     setL( (*paraQ)["lambda"] );
 164     setM1( (*paraQ)["mu1"] );
 165     setM2( (*paraQ)["mu2"] );
 166     */
 167
 168     return 1;
 169   }
 170   else
 171     return 0;
 172 }