paraq.cpp

   1 #include "paraq.h"
   2
   3 #include <algorithm>
   4 #include <iostream>
   5 #include <math.h>
   6 #include <string.h>
   7 #include <sstream>
   8
   9 using namespace std;
  10
  11 paraq::paraq(int _nprocs, const int _rank)
  12 {
  13   thisjob = -1;
  14   nprocs = _nprocs;
  15   rank = _rank;
  16   masterdefault = 0;
  17   jobListInitialized = false;
  18 }
  19
  20 void paraq::addRange(const string& paraid, char *range)
  21 {
  22   int nargs=0;
  23   double goodrange[3];
  24
  25   for( int pos=0; pos<strlen(range); pos++ )
  26     if( range[pos] == ':' ) nargs++;
  27   if(nargs==0)
  28     for(int i=0; i<3; i++) goodrange[i] = strtod(range, NULL);
  29   else{
  30     goodrange[0] = strtod(strtok(range, ":"), NULL);
  31     for(int i=0; i<nargs; i++) goodrange[i+1] = strtod(strtok(NULL, ":"), NULL);
  32   }
  33
  34   addRange(paraid, goodrange[0], goodrange[1], goodrange[2]);
  35 }
  36
  37 string paraq::rangeString()
  38 {
  39   stringstream rangestring;
  40
  41   for( map< string, vector<double*> >::iterator paraIt = rangeMap.begin(); paraIt != rangeMap.end(); ++paraIt )
  42     if( paraIt->second.size() > 0 )
  43       for( vector<double*>::iterator rangeIt = paraIt->second.begin(); rangeIt != paraIt->second.end(); ++rangeIt)
  44         rangestring << "-" << paraIt->first << (*rangeIt)[0] << "_" << (*rangeIt)[1] << "_" << (*rangeIt)[2];
  45
  46   for( map<string,double>::iterator defIt = defaultPara.begin(); defIt != defaultPara.end(); ++defIt )
  47     if( rangeMap.find(defIt->first) == rangeMap.end() )
  48       rangestring << "-" << defIt->first << defIt->second;
  49
  50   return rangestring.str();
  51 }
  52
  53 void paraq::addRange(const string& paraid, double min, double max, double step)
  54 {
  55   double *newrange = new double[3];
  56   newrange[0] = min;
  57   newrange[1] = max;
  58   newrange[2] = step;
  59   rangeMap[paraid].push_back( newrange );
  60
  61   for( double paraval = min; paraval <= max+0.5*step; paraval += step+(step==0) )
  62     if( ! inParas(paraMap[paraid], paraval) ) paraMap[paraid].push_back( paraval );
  63
  64   uniqueAllParaIdsAdd(paraid);
  65 }
  66
  67 bool paraq::inParas( vector<double>& paraVec, double& tofind )
  68 {
  69   for( vector<double>::iterator it = paraVec.begin(); it < paraVec.end(); ++it)
  70     if( fabs(*it - tofind) < 0.0000000001 ) return true;
  71   return false;
  72 }
  73
  74 void paraq::initJobList()
  75 {
  76   if( ! jobListInitialized ) {
  77     map<string,double> jobZero;
  78     jobList.push_back( jobZero );
  79
  80     for( map< string, vector< double > >::iterator paraIt = paraMap.begin(); paraIt != paraMap.end(); ++paraIt )
  81       while( jobList.begin()->find(paraIt->first) == jobList.begin()->end() ) {
  82         for( vector<double>::iterator valIt = paraIt->second.begin(); valIt != paraIt->second.end(); ++valIt ) {
  83           jobList.push_back( *jobList.begin() );
  84           jobList.back()[paraIt->first] = *valIt;
  85         }
  86         jobList.erase( jobList.begin() );
  87       }
  88     jobListInitialized = true;
  89   }
  90 }
  91
  92 int paraq::nextParas()
  93 {
  94   if( thisjob == -1 ) {
  95     initJobList();
  96     thisjob = rank;
  97   }
  98   else
  99     thisjob += nprocs;
 100
 101   if( jobList.begin()->size() == 0 ) return 0;
 102
 103   if( thisjob < jobList.size() ) return 1;
 104   else
 105     return 0;
 106 }
 107
 108 double& paraq::operator[] (string paraid) {
 109   if( jobList[thisjob].find(paraid) != jobList[thisjob].end() ) {
 110     return jobList[thisjob][paraid];
 111   }
 112
 113   if( defaultPara.find(paraid) != defaultPara.end() ) {
 114     return defaultPara[paraid];
 115   }
 116
 117   cerr << "PARAQ: Parameter " << paraid << " does not exist!" << endl << flush;
 118   exit(1);
 119 }
 120
 121 int paraq::getTotalJobs() {
 122   initJobList();
 123   return jobList.size();
 124 }
 125
 126 void paraq::uniqueAllParaIdsAdd(const string& paraid) {
 127   for (vector<string>::iterator allparait = allParaIds.begin(); allparait != allParaIds.end(); ++allparait)
 128     if ( *allparait == paraid )
 129       return;
 130
 131   allParaIds.push_back(paraid);
 132 }
 133
 134 string paraq::getParaNames() {
 135   stringstream paranames;
 136
 137   for (vector<string>::iterator parait = allParaIds.begin(); parait != allParaIds.end(); ++parait)
 138     paranames << *parait << ":";
 139
 140   return paranames.str();
 141 }
 142
 143 string paraq::getParaVals() {
 144   stringstream paravals;
 145
 146   for (vector<string>::iterator parait = allParaIds.begin(); parait != allParaIds.end(); ++parait) {
 147     if ( parait != allParaIds.begin() )
 148       paravals << "\t";
 149     paravals << (*this)[*parait];
 150   }
 151
 152   return paravals.str();
 153 }