--- trunk/src/restraints/RestraintForceManager.cpp	2012/08/22 02:28:28	1782
+++ trunk/src/restraints/RestraintForceManager.cpp	2014/09/22 19:18:35	2020
@@ -35,11 +35,15 @@
  *                                                                      
  * [1]  Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).             
  * [2]  Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).          
- * [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008).          
+ * [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 234107 (2008).          
  * [4]  Kuang & Gezelter,  J. Chem. Phys. 133, 164101 (2010).
  * [5]  Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
  */
  
+#ifdef IS_MPI
+#include <mpi.h>
+#endif
+
 #include "config.h"
 #include <cmath>
 
@@ -52,11 +56,7 @@
 #include "utils/StringUtils.hpp"
 #include "selection/SelectionEvaluator.hpp"
 #include "selection/SelectionManager.hpp"
-#ifdef IS_MPI
-#include <mpi.h>
-#endif
 
-
 namespace OpenMD {
 
   RestraintForceManager::RestraintForceManager(SimInfo* info): ForceManager(info) {
@@ -101,7 +101,6 @@ namespace OpenMD {
       if (myType.compare("MOLECULAR")==0){
 
         int molIndex;
-        std::vector<Vector3d> ref;
         Vector3d refCom;
 
         if (!stamp[i]->haveMolIndex()) {
@@ -137,7 +136,9 @@ namespace OpenMD {
           // this proc doesn't have the molecule.  Do a quick check to
           // make sure another processor is supposed to have it.
          
-          int myrank = MPI::COMM_WORLD.Get_rank();
+          int myrank;
+          MPI_Comm_rank( MPI_COMM_WORLD, &myrank);
+
           if (info_->getMolToProc(molIndex) == myrank) {
          
             // If we were supposed to have it but got a null, then freak out.
@@ -159,7 +160,8 @@ namespace OpenMD {
 #ifdef IS_MPI
         // only handle this molecular restraint if this processor owns the
         // molecule
-        int myrank = MPI::COMM_WORLD.Get_rank();
+        int myrank;
+        MPI_Comm_rank( MPI_COMM_WORLD, &myrank);
         if (info_->getMolToProc(molIndex) == myrank) {
 
 #endif
@@ -223,12 +225,18 @@ namespace OpenMD {
         evaluator.loadScriptString(objectSelection);
         seleMan.setSelectionSet(evaluator.evaluate());        
         int selectionCount = seleMan.getSelectionCount();
-        
+
+#ifdef IS_MPI    
+        MPI_Allreduce(MPI_IN_PLACE, &selectionCount, 1, MPI_INT, MPI_SUM, 
+                      MPI_COMM_WORLD);
+#endif
+                
         sprintf(painCave.errMsg,
                 "Restraint Info: The specified restraint objectSelection,\n"
                 "\t\t%s\n"
                 "\twill result in %d integrable objects being\n"
                 "\trestrained.\n", objectSelection.c_str(), selectionCount);
+        painCave.severity = OPENMD_INFO;
         painCave.isFatal = 0;
         simError();                     
 
@@ -311,16 +319,15 @@ namespace OpenMD {
   void RestraintForceManager::calcForces(){
 
     ForceManager::calcForces();    
-    RealType restPot_local, restPot;
+    RealType restPot(0.0);
 
-    restPot_local = doRestraints(1.0);
+    restPot = doRestraints(1.0);
 
 #ifdef IS_MPI    
-    MPI::COMM_WORLD.Allreduce(&restPot_local, &restPot, 1, 
-                              MPI::REALTYPE, MPI::SUM);
-#else
-    restPot = restPot_local;
+    MPI_Allreduce(MPI_IN_PLACE, &restPot, 1, MPI_REALTYPE, MPI_SUM, 
+                  MPI_COMM_WORLD);
 #endif
+
     currSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot();
     RealType pot = currSnapshot_->getLongRangePotential();
     pot += restPot;
@@ -389,6 +396,7 @@ namespace OpenMD {
 
       std::vector<Vector3d> struc;
       std::vector<Vector3d> forces;
+
       
       for(sd = (*rm)->beginIntegrableObject(ioi); sd != NULL; 
           sd = (*rm)->nextIntegrableObject(ioi)) {
@@ -407,12 +415,11 @@ namespace OpenMD {
         index++;
       }
       
-      unscaledPotential_ += mRest->getUnscaledPotential();
+      unscaledPotential_ += mRest->getUnscaledPotential();      
 
-      restInfo = mRest->getRestraintInfo();
-
       // only collect data on restraints that we're going to print:
       if (mRest->getPrintRestraint()) 
+        restInfo = mRest->getRestraintInfo();
         restInfo_.push_back(restInfo);
     }
 
@@ -449,7 +456,6 @@ namespace OpenMD {
       
       // phew.  At this point, we should have the pointer to the
       // correct Object restraint in the variable oRest.
-
       oRest->setScaleFactor(scalingFactor);
       
       Vector3d pos = (*ro)->getPos();
@@ -473,10 +479,9 @@ namespace OpenMD {
 
       unscaledPotential_ += oRest->getUnscaledPotential();
 
-      restInfo = oRest->getRestraintInfo();
-
       // only collect data on restraints that we're going to print:
       if (oRest->getPrintRestraint()) 
+        restInfo = oRest->getRestraintInfo();      
         restInfo_.push_back(restInfo);
     }