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