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<paralink>::iterator linkit = linkedParas.begin(); linkit != linkedParas.end(); ++linkit)
  47     rangestring << "_" << linkit->first << "--" << linkit->linktype << "--" << 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<paralink>::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<paralink>::iterator linkit = linkedParas.begin(); linkit != linkedParas.end(); ++linkit)
 102             if (linkit->second == paraIt->first)
 103               jobList.back()[linkit->first] = linkTypeFunction(*valIt, linkit->linktype);
 104         }
 105         jobList.erase( jobList.begin() );
 106       }
 107     jobListInitialized = true;
 108   }
 109 }
 110
 111 double paraq::linkTypeFunction (double linkval, int linktype) {
 112   switch (linktype) {
 113
 114   case LINK_1TO1:
 115     return linkval;
 116
 117   case LINK_NEGATIVE:
 118     return -linkval;
 119
 120   }
 121 }
 122
 123 int paraq::nextParas()
 124 {
 125   if( thisjob == -1 ) {
 126     initJobList();
 127     thisjob = rank;
 128   }
 129   else
 130     thisjob += nprocs;
 131
 132   if( jobList.begin()->size() == 0 ) return 0;
 133
 134   if( thisjob < jobList.size() ) return 1;
 135   else
 136     return 0;
 137 }
 138
 139 double& paraq::operator[] (string paraid) {
 140   if( jobList[thisjob].find(paraid) != jobList[thisjob].end() ) {
 141     return jobList[thisjob][paraid];
 142   }
 143
 144   if( defaultPara.find(paraid) != defaultPara.end() ) {
 145     return defaultPara[paraid];
 146   }
 147
 148   cerr << "PARAQ: Parameter " << paraid << " does not exist!" << endl << flush;
 149   exit(1);
 150 }
 151
 152 int paraq::getTotalJobs() {
 153   initJobList();
 154   return jobList.size();
 155 }
 156
 157 void paraq::uniqueAllParaIdsAdd(const string& paraid) {
 158   for (vector<string>::iterator allparait = allParaIds.begin(); allparait != allParaIds.end(); ++allparait)
 159     if ( *allparait == paraid )
 160       return;
 161
 162   allParaIds.push_back(paraid);
 163 }
 164
 165 string paraq::getParaNames() {
 166   stringstream paranames;
 167
 168   for (vector<string>::iterator parait = allParaIds.begin(); parait != allParaIds.end(); ++parait)
 169     paranames << *parait << ":";
 170
 171   return paranames.str();
 172 }
 173
 174 string paraq::getParaVals() {
 175   stringstream paravals;
 176
 177   for (vector<string>::iterator parait = allParaIds.begin(); parait != allParaIds.end(); ++parait) {
 178     if ( parait != allParaIds.begin() )
 179       paravals << "\t";
 180     paravals << (*this)[*parait];
 181   }
 182
 183   return paravals.str();
 184 }