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
  65 bool paraq::inParas( vector<double>& paraVec, double& tofind )
  66 {
  67   for( vector<double>::iterator it = paraVec.begin(); it < paraVec.end(); ++it)
  68     if( fabs(*it - tofind) < 0.0000000001 ) return true;
  69   return false;
  70 }
  71
  72 void paraq::initJobList()
  73 {
  74   if( ! jobListInitialized ) {
  75     map<string,double> jobZero;
  76     jobList.push_back( jobZero );
  77
  78     for( map< string, vector< double > >::iterator paraIt = paraMap.begin(); paraIt != paraMap.end(); ++paraIt )
  79       while( jobList.begin()->find(paraIt->first) == jobList.begin()->end() ) {
  80         for( vector<double>::iterator valIt = paraIt->second.begin(); valIt != paraIt->second.end(); ++valIt ) {
  81           jobList.push_back( *jobList.begin() );
  82           jobList.back()[paraIt->first] = *valIt;
  83         }
  84         jobList.erase( jobList.begin() );
  85       }
  86     jobListInitialized = true;
  87   }
  88 }
  89
  90 int paraq::nextParas()
  91 {
  92   if( thisjob == -1 ) {
  93     initJobList();
  94     thisjob = rank;
  95   }
  96   else
  97     thisjob += nprocs;
  98
  99   if( jobList.begin()->size() == 0 ) return 0;
 100
 101   if( thisjob < jobList.size() ) return 1;
 102   else
 103     return 0;
 104 }
 105
 106 double& paraq::operator[] (string paraid) {
 107   if( jobList[thisjob].find(paraid) != jobList[thisjob].end() ) {
 108     return jobList[thisjob][paraid];
 109   }
 110
 111   if( defaultPara.find(paraid) != defaultPara.end() ) {
 112     return defaultPara[paraid];
 113   }
 114
 115   return masterdefault;
 116 }
 117
 118 int paraq::getTotalJobs() {
 119   initJobList();
 120   return jobList.size();
 121 }