]> git.treefish.org Git - phys/latlib.git/blob - paraq.cpp
...
[phys/latlib.git] / 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 }