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 (vector< pair<string,string> >::iterator linkit = linkedParas.begin(); linkit != linkedParas.end(); ++linkit)
  47     rangestring << "-" << linkit->first << "->" << linkit->second;
  48
  49   for( map<string,double>::iterator defIt = defaultPara.begin(); defIt != defaultPara.end(); ++defIt )
  50     if( rangeMap.find(defIt->first) == rangeMap.end() ) {
  51       bool linkedpara = false;
  52
  53       for (vector< pair<string,string> >::iterator linkit = linkedParas.begin(); linkit != linkedParas.end(); ++linkit)
  54         if (linkit->first == defIt->first) {
  55           linkedpara = true;
  56           break;
  57         }
  58
  59       if (!linkedpara)
  60         rangestring << "-" << defIt->first << defIt->second;
  61     }
  62
  63   return rangestring.str();
  64 }
  65
  66 void paraq::addRange(const string& paraid, double min, double max, double step)
  67 {
  68   double *newrange = new double[3];
  69   newrange[0] = min;
  70   newrange[1] = max;
  71   newrange[2] = step;
  72   rangeMap[paraid].push_back( newrange );
  73
  74   for( double paraval = min; paraval <= max+0.5*step; paraval += step+(step==0) )
  75     if( ! inParas(paraMap[paraid], paraval) ) paraMap[paraid].push_back( paraval );
  76
  77   uniqueAllParaIdsAdd(paraid);
  78 }
  79
  80 bool paraq::inParas( vector<double>& paraVec, double& tofind )
  81 {
  82   for( vector<double>::iterator it = paraVec.begin(); it < paraVec.end(); ++it)
  83     if( fabs(*it - tofind) < 0.0000000001 ) return true;
  84   return false;
  85 }
  86
  87 void paraq::initJobList()
  88 {
  89   if( ! jobListInitialized ) {
  90     map<string,double> jobZero;
  91     jobList.push_back( jobZero );
  92
  93     for( map< string, vector< double > >::iterator paraIt = paraMap.begin(); paraIt != paraMap.end(); ++paraIt )
  94       while( jobList.begin()->find(paraIt->first) == jobList.begin()->end() ) {
  95         for( vector<double>::iterator valIt = paraIt->second.begin(); valIt != paraIt->second.end(); ++valIt ) {
  96           jobList.push_back( *jobList.begin() );
  97
  98           jobList.back()[paraIt->first] = *valIt;
  99
 100           /* also set linked parameters */
 101           for (vector< pair<string,string> >::iterator linkit = linkedParas.begin(); linkit != linkedParas.end(); ++linkit)
 102             if (linkit->second == paraIt->first)
 103               jobList.back()[linkit->first] = *valIt;
 104         }
 105         jobList.erase( jobList.begin() );
 106       }
 107     jobListInitialized = true;
 108   }
 109 }
 110
 111 int paraq::nextParas()
 112 {
 113   if( thisjob == -1 ) {
 114     initJobList();
 115     thisjob = rank;
 116   }
 117   else
 118     thisjob += nprocs;
 119
 120   if( jobList.begin()->size() == 0 ) return 0;
 121
 122   if( thisjob < jobList.size() ) return 1;
 123   else
 124     return 0;
 125 }
 126
 127 double& paraq::operator[] (string paraid) {
 128   if( jobList[thisjob].find(paraid) != jobList[thisjob].end() ) {
 129     return jobList[thisjob][paraid];
 130   }
 131
 132   if( defaultPara.find(paraid) != defaultPara.end() ) {
 133     return defaultPara[paraid];
 134   }
 135
 136   cerr << "PARAQ: Parameter " << paraid << " does not exist!" << endl << flush;
 137   exit(1);
 138 }
 139
 140 int paraq::getTotalJobs() {
 141   initJobList();
 142   return jobList.size();
 143 }
 144
 145 void paraq::uniqueAllParaIdsAdd(const string& paraid) {
 146   for (vector<string>::iterator allparait = allParaIds.begin(); allparait != allParaIds.end(); ++allparait)
 147     if ( *allparait == paraid )
 148       return;
 149
 150   allParaIds.push_back(paraid);
 151 }
 152
 153 string paraq::getParaNames() {
 154   stringstream paranames;
 155
 156   for (vector<string>::iterator parait = allParaIds.begin(); parait != allParaIds.end(); ++parait)
 157     paranames << *parait << ":";
 158
 159   return paranames.str();
 160 }
 161
 162 string paraq::getParaVals() {
 163   stringstream paravals;
 164
 165   for (vector<string>::iterator parait = allParaIds.begin(); parait != allParaIds.end(); ++parait) {
 166     if ( parait != allParaIds.begin() )
 167       paravals << "\t";
 168     paravals << (*this)[*parait];
 169   }
 170
 171   return paravals.str();
 172 }