o815/o815.cpp

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