...

[phys/latlib.git] / paraq.cpp

diff --git a/paraq.cpp b/paraq.cpp
index e66f13860ff44a8880319a0224fd5458356a489d..1bd57b3275f428a38bb597a451b8da900f070713 100644 (file)
--- a/paraq.cpp
+++ b/paraq.cpp
@@ -12,6 +12,7 @@ paraq::paraq(int _nprocs, const int _rank)
   nprocs = _nprocs;
   rank = _rank;
   masterdefault = 0;
+  jobListInitialized = false;
 }
 
 void paraq::addRange(const string& paraid, double min, double max, double step)
@@ -27,12 +28,12 @@ bool paraq::inParas( vector<double>& paraVec, double& tofind )
   return false;
 }
 
-int paraq::nextParas()
+void paraq::initJobList()
 {
-  if( thisjob == -1 ) {   
+  if( ! jobListInitialized ) {
     map<string,double> jobZero;
     jobList.push_back( jobZero );
-
+    
     for( map< string, vector< double > >::iterator paraIt = paraMap.begin(); paraIt != paraMap.end(); ++paraIt )
       while( jobList.begin()->find(paraIt->first) == jobList.begin()->end() ) {
        for( vector<double>::iterator valIt = paraIt->second.begin(); valIt != paraIt->second.end(); ++valIt ) {
@@ -41,12 +42,21 @@ int paraq::nextParas()
        }       
        jobList.erase( jobList.begin() );
       }
+    jobListInitialized = true;
+  }
+}
 
+int paraq::nextParas()
+{
+  if( thisjob == -1 ) {   
+    initJobList();
     thisjob = rank;
   }
   else
     thisjob += nprocs;
 
+  if( jobList.begin()->size() == 0 ) return 0;
+
   if( thisjob < jobList.size() ) return 1;
   else 
     return 0;
@@ -63,3 +73,8 @@ double& paraq::operator[] (string paraid) {
 
   return masterdefault;
 }
+
+int paraq::getTotalJobs() { 
+  initJobList();
+  return jobList.size(); 
+}