o815/o815.cpp

   1 #include "o815.h"
   2
   3 #include <sstream>
   4 #include <iomanip>
   5
   6 #include "latlib/progress.h"
   7
   8 using namespace std;
   9
  10 extern int opterr;
  11
  12 o815::o815(int argc, char **argv, const string& _programid, comoption specOps[], void (*helpHeader)()) {
  13   long timestamp;
  14
  15   programid = _programid;
  16
  17   comargs.nmeas = 100;
  18   comargs.nskip = 10;
  19   comargs.nequi = 100;
  20   comargs.lsize[0] = 4;
  21   comargs.lsize[1] = 4;
  22   comargs.obscache = make_pair("",0);
  23   comargs.confcache = make_pair("",0);
  24   comargs.outdir="";
  25   comargs.idonly = false;
  26   comargs.showjobnum = false;
  27
  28   MPI_Init(&argc, &argv);
  29   MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
  30   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  31
  32   paraQ = new paraq(numprocs, rank);
  33
  34   addComOption("lsize",  required_argument, NULL,                'L', "define lattice size",                   "xsize:tsize");
  35   addComOption("nmeas",  required_argument, NULL,                'N', "set number of measurements",            "nmeas");
  36   addComOption("skip",   required_argument, NULL,                'S', "set number of skips between configs",   "nskip");
  37   addComOption("nequi",  required_argument, NULL,                'E', "set number of equilibrations",          "nequi");
  38   addComOption("oro",    required_argument, NULL,                'o', "read-only observable-store directory",  "dir");
  39   addComOption("orw",    required_argument, NULL,                'O', "read-write observable-store directory", "dir");
  40   addComOption("cro",    required_argument, NULL,                'c', "read-only config-store directory",      "dir");
  41   addComOption("crw",    required_argument, NULL,                'C', "read-write config-store directory",     "dir");
  42   addComOption("write",  required_argument, NULL,                'w', "data writeout directory",               "dir");
  43   addComOption("idonly", no_argument,       &comargs.idonly,     'i', "show output-id only",                   "");
  44   addComOption("jobnum", no_argument,       &comargs.showjobnum, 'j', "show jobnumber only",                   "");
  45
  46   if (specOps != NULL)
  47     for (int ispecop = 0; specOps[ispecop].name != ""; ispecop++)
  48       comOptions.push_back(specOps[ispecop]);
  49
  50   if (argc > 1)
  51     parseArgs(argc, argv, specOps);
  52   else {
  53     int longestCombinedLong=0;
  54
  55     if (helpHeader != NULL)
  56       helpHeader();
  57
  58     cout << "Options:" << endl;
  59
  60     for (vector<comoption>::iterator opit = comOptions.begin(); opit != comOptions.end(); ++opit) {
  61       stringstream combinedLong;
  62
  63       combinedLong << opit->name;
  64       if (opit->has_arg == required_argument)
  65         combinedLong << "=" << opit->argdesc;
  66       else if (opit->has_arg == optional_argument)
  67         combinedLong << "[=" << opit->argdesc << "]";
  68
  69       if (longestCombinedLong < combinedLong.str().size())
  70         longestCombinedLong = combinedLong.str().size();
  71     }
  72
  73     for (vector<comoption>::iterator opit = comOptions.begin(); opit != comOptions.end(); ++opit) {
  74       stringstream combinedLong;
  75
  76       cout << "  -" << char(opit->val) << ", --";
  77
  78       combinedLong << opit->name;
  79       if (opit->has_arg == required_argument)
  80         combinedLong << "=" << opit->argdesc;
  81       else if (opit->has_arg == optional_argument)
  82         combinedLong << "[=" << opit->argdesc << "]";
  83
  84       cout << setw(longestCombinedLong) << setiosflags(ios::left) << combinedLong.str();
  85
  86       cout << "\t" << opit->optdesc;
  87
  88       cout << endl;
  89     }
  90     exit(0);
  91   }
  92
  93   if(comargs.idonly) {
  94     cout << programid << headMaster() << endl << flush;
  95     exit(0);
  96   }
  97
  98   if(rank==0) {
  99     timestamp = time(NULL);
 100     for(int idest=1; idest<numprocs; idest++)
 101       MPI_Send(&timestamp, 1, MPI_LONG, idest, 123, MPI_COMM_WORLD);
 102   }
 103   else if(rank>0)
 104     MPI_Recv(&timestamp, 1, MPI_LONG, 0, 123, MPI_COMM_WORLD, &mpiStatus);
 105
 106   out = new writeout(comargs.outdir, programid+headMaster(), rank, numprocs, timestamp);
 107 }
 108
 109 void o815::mainLoop() {
 110   if( comargs.showjobnum ) {
 111     for( int i=1; i<=paraQ->getTotalJobs(); i++ ) {
 112       if( paraQ->getTotalJobs()%i == 0 ) cout << paraQ->getTotalJobs()/i << "@" << i << " ";
 113     }
 114     cout << endl;
 115     exit(0);
 116   }
 117
 118   *out->log << "OBS: Starting main loop." << endl;
 119
 120   for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit)
 121     (*obsit)->start();
 122
 123   while( nextParas() ) {
 124     *out->log << endl << "O815: < ";
 125     for (vector<string>::iterator parait = paraQ->allParaIds.begin(); parait != paraQ->allParaIds.end(); ++parait)
 126       *out->log << *parait << "=" << (*paraQ)[*parait] << " ";
 127     *out->log << ">" << endl << flush;
 128
 129     Sim->_newParas();
 130
 131     progress measProg(comargs.nmeas);
 132
 133     *out->log << "OBS: Starting measurement of observables:";
 134     for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit)
 135       *out->log << " " << (*obsit)->obsid;
 136     *out->log << endl << flush;
 137
 138     Sim->nequileft = comargs.nequi;
 139
 140     for( int imeas=0; imeas<comargs.nmeas; imeas++ ) {
 141       bool nextAlready = false;
 142       for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit) {
 143         bool readnewObs;
 144         int nequileftNewObs;
 145
 146         hypercache::readO( (*obsit)->ocid, readnewObs, nequileftNewObs );
 147
 148         if( ! readnewObs ) {
 149           if (!nextAlready) {
 150             Sim->nextConfig();
 151             nextAlready = true;
 152           }
 153           (*obsit)->meas(false);
 154           hypercache::writeO( (*obsit)->ocid, comargs.nmeas );
 155         }
 156         else
 157           (*obsit)->meas(true);
 158       }
 159       while( measProg.madeStep(imeas) )
 160         *out->log << "O815: " << measProg.done()*100 << "% of measurements done." << endl << flush;
 161     }
 162     for (vector<obs*>::iterator obsit = observables.begin(); obsit != observables.end(); ++obsit)
 163       (*obsit)->finish();
 164   }
 165 }
 166
 167 void o815::addComOption(const char* name, int has_arg, int *flag, int val, const char* optdesc, const char* argdesc) {
 168   comoption newop = { name, has_arg, flag, val, optdesc, argdesc };
 169   comOptions.push_back(newop);
 170 }
 171
 172 void o815::parseArgs(int argc, char **argv, comoption specOps[]) {
 173   int opt;
 174   int indexptr=0;
 175   stringstream optargstr;
 176   option allOptions[comOptions.size()+1];
 177
 178   for (int iop=0; iop < comOptions.size(); iop++) {
 179     allOptions[iop].name = comOptions[iop].name.c_str();
 180     allOptions[iop].has_arg = comOptions[iop].has_arg;
 181     //allOptions[iop].flag = NULL;
 182     allOptions[iop].flag = comOptions[iop].flag;
 183     allOptions[iop].val = comOptions[iop].val;
 184
 185     optargstr << char(allOptions[iop].val);
 186     if (allOptions[iop].has_arg == required_argument)
 187       optargstr << ":";
 188     else if (allOptions[iop].has_arg == optional_argument)
 189       optargstr << "::";
 190   }
 191
 192   allOptions[comOptions.size()].name = 0;
 193   allOptions[comOptions.size()].has_arg = 0;
 194   allOptions[comOptions.size()].flag = 0;
 195   allOptions[comOptions.size()].val = 0;
 196
 197   while((opt = getopt_long( argc, argv, optargstr.str().c_str(), allOptions, &indexptr )) != -1)
 198     switch(opt) {
 199     case 'L':
 200       listArg(comargs.lsize, 2, optarg);
 201       break;
 202     case 'N':
 203       comargs.nmeas = atoi(optarg);
 204       break;
 205     case 'S':
 206       comargs.nskip = atoi(optarg);
 207       break;
 208     case 'E':
 209       comargs.nequi = atoi(optarg);
 210       break;
 211     case 'o':
 212       comargs.obscache.first = optarg;
 213       comargs.obscache.second = 1;
 214       break;
 215     case 'O':
 216       comargs.obscache.first = optarg;
 217       comargs.obscache.second = 2;
 218       break;
 219     case 'c':
 220       comargs.confcache.first = optarg;
 221       comargs.confcache.second = 1;
 222       break;
 223     case 'C':
 224       comargs.confcache.first = optarg;
 225       comargs.confcache.second = 2;
 226       break;
 227     case 'w':
 228       comargs.outdir = optarg;
 229       break;
 230     default:
 231       if ( opt != 0) {
 232         comoption* thisop = getOptionByVal(opt);
 233         if (thisop->flag != 0)
 234           *thisop->flag = thisop->val;
 235         else
 236           parsedSpecOps.push_back( pair<int,char*>(thisop->val, optarg) );
 237       }
 238       break;
 239     }
 240
 241   for (int ilon=0; optind+ilon < argc; ilon++)
 242     lonelyArgs.push_back(argv[optind+ilon]);
 243 }
 244
 245 o815::comoption* o815::getOptionByVal(int val) {
 246   for (vector<comoption>::iterator opit = comOptions.begin(); opit != comOptions.end(); ++opit)
 247     if ( opit->val == val )
 248       return &(*opit);
 249   exit(1);
 250 }
 251
 252 void o815::listArg(int *target, int tlen, char *listarg) {
 253   int nargs=0;
 254
 255   for( int pos=0; pos<strlen(listarg); pos++ )
 256     if( listarg[pos] == ':' ) nargs++;
 257
 258   if(nargs==0)
 259     for(int i=0; i<tlen; i++) target[i] = atoi(listarg);
 260   else
 261     {
 262       target[0] = atoi(strtok(listarg, ":"));
 263       for(int i=0; i<nargs; i++)
 264         target[i+1] = atoi(strtok(NULL, ":"));
 265     }
 266 }
 267
 268 string o815::headMaster()
 269 {
 270   stringstream hm;
 271
 272   hm << "-L" << comargs.lsize[0] << "_" << comargs.lsize[1] << "-E" << comargs.nequi << "-S" << comargs.nskip << "-N" << comargs.nmeas
 273      << paraQ->rangeString();
 274
 275   return hm.str();
 276 }
 277
 278 o815::~o815() {
 279   if(comargs.outdir=="") {
 280     MPI_Barrier(MPI_COMM_WORLD);
 281     if(rank==0)
 282       cout << "#end" << endl << flush;
 283   }
 284   hypercache::finalize();
 285   delete out;
 286   MPI_Finalize();
 287 }
 288
 289 int o815::nextParas()
 290 {
 291   if( paraQ->nextParas() ) {
 292     for (vector<string>::iterator parait = paraQ->allParaIds.begin(); parait != paraQ->allParaIds.end(); ++parait)
 293       hypercache::setPara(*parait, (*paraQ)[*parait]);
 294     return 1;
 295   }
 296   else
 297     return 0;
 298 }
 299
 300 void o815::addPara(const string& paraid, const double& paraDefault) {
 301   hypercache::addPara(paraid);
 302   paraQ->setDefault(paraid, paraDefault);
 303 }