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   return masterdefault;
 118 }
 119
 120 int paraq::getTotalJobs() {
 121   initJobList();
 122   return jobList.size();
 123 }
 124
 125 void paraq::uniqueAllParaIdsAdd(const string& paraid) {
 126   for (vector<string>::iterator allparait = allParaIds.begin(); allparait != allParaIds.end(); ++allparait)
 127     if ( *allparait == paraid )
 128       return;
 129
 130   allParaIds.push_back(paraid);
 131 }