--- trunk/OOPSE/libmdtools/ZConstraint.cpp	2003/08/14 16:16:39	696
+++ trunk/OOPSE/libmdtools/ZConstraint.cpp	2003/08/22 20:04:39	711
@@ -2,8 +2,8 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
 #include "simError.h"
 #include <cmath>
 template<typename T> ZConstraint<T>::ZConstraint(SimInfo* theInfo, ForceFields* the_ff) 
-                                    : T(theInfo, the_ff), fz(NULL), 
-				      indexOfZConsMols(NULL)
+                                    : T(theInfo, the_ff), fz(NULL), curZPos(NULL),
+                                 indexOfZConsMols(NULL), forcePolicy(NULL), curZconsTime(0)
 {
 
   //get properties from SimInfo
@@ -11,6 +11,7 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
   ZConsParaData* zConsParaData;
   DoubleData* sampleTime;
   DoubleData* tolerance;
+  StringData* policy;
   StringData* filename; 
   double COM[3];
 
@@ -26,6 +27,45 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
   double halfOfLargestBox = max(info->boxL[0], max(info->boxL[1], info->boxL[2])) /2;
   zForceConst = Kb * info->target_temp /(halfOfLargestBox * halfOfLargestBox);
 
+  //creat force substraction policy
+  data = info->getProperty(ZCONSFORCEPOLICY_ID);
+  if(!data){
+    sprintf( painCave.errMsg,
+               "ZConstraint Warning: User does not set force substraction policy, "
+               "average force substraction policy is used\n");
+    painCave.isFatal = 0;
+    simError();      
+
+    forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this);
+  }
+  else{
+    policy = dynamic_cast<StringData*>(data);
+    
+    if(!policy){
+      sprintf( painCave.errMsg,
+                 "ZConstraint Error: Convertion from GenericData to StringData failure, "
+                 "average force substraction policy is used\n");
+      painCave.isFatal = 0;
+      simError();      
+
+      forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this);
+    }
+    else{
+      if(policy->getData() == "BYNUMBER")
+        forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this);
+      else if(policy->getData() == "BYMASS")
+        forcePolicy = (ForceSubstractionPolicy*) new PolicyByMass(this);
+      else{
+        sprintf( painCave.errMsg,
+                  "ZConstraint Warning: unknown force substraction policy, "
+                  "average force substraction policy is used\n");
+        painCave.isFatal = 0;
+        simError();      
+      }   
+    }
+  }
+  
+  
   //retrieve sample time of z-contraint 
   data = info->getProperty(ZCONSTIME_ID);
   
@@ -69,7 +109,7 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
   }
   else{
 
-     filename = dynamic_cast<StringData*>(data);
+    filename = dynamic_cast<StringData*>(data);
     
     if(!filename){
 
@@ -83,7 +123,6 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
       this->zconsOutput = filename->getData();
     }
     
-
   }
 
   //retrieve tolerance for z-constraint molecuels
@@ -113,7 +152,7 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
     }
 
   }
-	
+  
   //retrieve index of z-constraint molecules
   data = info->getProperty(ZCONSPARADATA_ID);
   if(!data) {
@@ -154,7 +193,7 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
                "ZConstraint error: index is out of range\n");
         painCave.isFatal = 1;
         simError(); 
-      	}
+        }
 
       maxIndex = (*parameters)[parameters->size() - 1].zconsIndex;
 
@@ -175,50 +214,49 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
       //its initial z coordinate  will be used as default
       for(int i = 0; i < parameters->size(); i++){
 
-	      if(!(*parameters)[i].havingZPos){
-
+        if(!(*parameters)[i].havingZPos){
 #ifndef IS_MPI
-            for(int j = 0; j < nMols; j++){
-              if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
-                 molecules[j].getCOM(COM);
-		          break; 
-              }
+          for(int j = 0; j < nMols; j++){
+            if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
+              molecules[j].getCOM(COM);
+        break; 
             }
+          }
 #else
             //query which processor current zconstraint molecule belongs to
-           int *MolToProcMap;
-           int whichNode;
-			 double initZPos;
-           MolToProcMap = mpiSim->getMolToProcMap();
-           whichNode = MolToProcMap[(*parameters)[i].zconsIndex];
-			  
-           //broadcast the zpos of current z-contraint molecule
-           //the node which contain this 
+          int *MolToProcMap;
+          int whichNode;
+    double initZPos;
+          MolToProcMap = mpiSim->getMolToProcMap();
+          whichNode = MolToProcMap[(*parameters)[i].zconsIndex];
+        
+          //broadcast the zpos of current z-contraint molecule
+          //the node which contain this 
            
-           if (worldRank == whichNode ){
-					 	
-             for(int j = 0; j < nMols; j++)
-               if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
-                 molecules[j].getCOM(COM);
-					 break; 
-               }
-				 
-           }
+          if (worldRank == whichNode ){
+            
+            for(int j = 0; j < nMols; j++)
+              if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
+                molecules[j].getCOM(COM);
+          break; 
+              }
+         
+          }
 
-            MPI_Bcast(&COM[whichDirection], 1, MPI_DOUBLE_PRECISION, whichNode, MPI_COMM_WORLD);			  
+          MPI_Bcast(&COM[whichDirection], 1, MPI_DOUBLE_PRECISION, whichNode, MPI_COMM_WORLD);        
 #endif
             
-	         (*parameters)[i].zPos = COM[whichDirection];
+          (*parameters)[i].zPos = COM[whichDirection];
 
-            sprintf( painCave.errMsg,
+          sprintf( painCave.errMsg,
                      "ZConstraint warningr: Does not specify zpos for z-constraint molecule "
                      "initial z coornidate will be used \n");
-            painCave.isFatal = 0;
-            simError();  
-	  
-	      }
-	    }
-			
+           painCave.isFatal = 0;
+           simError();  
+    
+  }
+      }
+      
     }//end if (!zConsParaData)
   }//end  if (!data)
              
@@ -238,7 +276,9 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
       massOfZConsMols.push_back(molecules[i].getTotalMass());  
 
       zPos.push_back((*parameters)[searchResult].zPos);
-	   kz.push_back((*parameters)[searchResult]. kRatio * zForceConst);
+      cout << "index: "<< (*parameters)[searchResult].zconsIndex 
+             <<"\tzPos = " << (*parameters)[searchResult].zPos << endl; 
+      kz.push_back((*parameters)[searchResult]. kRatio * zForceConst);
       
       molecules[i].getCOM(COM);
     }
@@ -252,9 +292,10 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
   }
 
   fz = new double[zconsMols.size()];
+  curZPos = new double[zconsMols.size()]; 
   indexOfZConsMols = new int [zconsMols.size()];
 
-  if(!fz || !indexOfZConsMols){
+  if(!fz || !curZPos || !indexOfZConsMols){
     sprintf( painCave.errMsg,
              "Memory allocation failure in class Zconstraint\n");
     painCave.isFatal = 1;
@@ -265,11 +306,11 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
   for(int i = 0; i < zconsMols.size(); i++){
     indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex();
 
-	 zconsMols[i]->getCOM(COM);
+    zconsMols[i]->getCOM(COM);
     if (fabs(zPos[i] - COM[whichDirection]) < zconsTol)
-		states.push_back(zcsFixed);
-	 else
-		states.push_back(zcsMoving);
+      states.push_back(zcsFixed);
+    else
+      states.push_back(zcsMoving);
   }
   
 #endif 
@@ -284,7 +325,8 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
 #ifndef IS_MPI
   totalMassOfUncons = totalMassOfUncons_local;
 #else
-  MPI_Allreduce(&totalMassOfUncons_local, &totalMassOfUncons, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);  
+  MPI_Allreduce(&totalMassOfUncons_local, &totalMassOfUncons, 1,
+                      MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);  
 #endif
 
 
@@ -297,11 +339,12 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
 #ifndef IS_MPI
   totNumOfUnconsAtoms = nUnconsAtoms_local;
 #else
-  MPI_Allreduce(&nUnconsAtoms_local, &totNumOfUnconsAtoms, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);  
+  MPI_Allreduce(&nUnconsAtoms_local, &totNumOfUnconsAtoms, 1, 
+                      MPI_INT,MPI_SUM, MPI_COMM_WORLD);  
 #endif  
 
   // creat zconsWriter  
-  fzOut = new ZConsWriter(zconsOutput.c_str());   
+  fzOut = new ZConsWriter(zconsOutput.c_str(), parameters);   
   
   if(!fzOut){
     sprintf( painCave.errMsg,
@@ -309,21 +352,33 @@ template<typename T> ZConstraint<T>::ZConstraint(SimIn
     painCave.isFatal = 1;
     simError();
   }
-  
+
+  forcePolicy->update();
 }
 
 template<typename T> ZConstraint<T>::~ZConstraint()
 {
   if(fz)
     delete[] fz;
+
+  if(curZPos)
+    delete[] curZPos;
   
   if(indexOfZConsMols)
     delete[] indexOfZConsMols;
   
   if(fzOut)
     delete fzOut;
+  
+  if(forcePolicy)
+    delete forcePolicy;
 }
 
+
+/**
+ * 
+ */
+
 #ifdef IS_MPI
 template<typename T> void ZConstraint<T>::update()
 {
@@ -348,8 +403,8 @@ template<typename T> void ZConstraint<T>::update()
     
       zconsMols.push_back(&molecules[i]);      
       zPos.push_back((*parameters)[index].zPos);
-    	kz.push_back((*parameters)[index].kRatio * zForceConst);
-			
+      kz.push_back((*parameters)[index].kRatio * zForceConst);
+      
       massOfZConsMols.push_back(molecules[i].getTotalMass());  
        
       molecules[i].getCOM(COM);
@@ -365,11 +420,11 @@ template<typename T> void ZConstraint<T>::update()
 
   //determine the states of z-constraint molecules
   for(int i = 0; i < zconsMols.size(); i++){
-	   zconsMols[i]->getCOM(COM);
+     zconsMols[i]->getCOM(COM);
       if (fabs(zPos[i] - COM[whichDirection]) < zconsTol)
-		  states.push_back(zcsFixed);
-	   else
-		  states.push_back(zcsMoving);
+      states.push_back(zcsFixed);
+     else
+      states.push_back(zcsMoving);
   }
 
     
@@ -377,15 +432,19 @@ template<typename T> void ZConstraint<T>::update()
   // that we want to make the MPI communication simple
   if(fz)
     delete[] fz;
+  
+  if(curZPos)
+    delete[] curZPos;
     
   if(indexOfZConsMols)
     delete[] indexOfZConsMols;
     
   if (zconsMols.size() > 0){
     fz = new double[zconsMols.size()];
+   curZPos = new double[zconsMols.size()];
     indexOfZConsMols =  new int[zconsMols.size()];
     
-    if(!fz || !indexOfZConsMols){
+    if(!fz || !curZPos || !indexOfZConsMols){
       sprintf( painCave.errMsg,
                "Memory allocation failure in class Zconstraint\n");
       painCave.isFatal = 1;
@@ -399,19 +458,24 @@ template<typename T> void ZConstraint<T>::update()
   }
   else{
     fz = NULL;
+   curZPos = NULL;
     indexOfZConsMols = NULL;
   }
   
+  //
+  forcePolicy->update();
+  
 }
 
 #endif
 
-/**  Function Name: isZConstraintMol
- **  Parameter
- **    Molecule* mol
- **  Return value:
- **    -1, if the molecule is not z-constraint molecule, 
- **    other non-negative values, its index in indexOfAllZConsMols vector
+/**
+ *  Function Name: isZConstraintMol
+ *  Parameter
+ *    Molecule* mol
+ *  Return value:
+ *    -1, if the molecule is not z-constraint molecule, 
+ *    other non-negative values, its index in indexOfAllZConsMols vector
  */
 
 template<typename T> int ZConstraint<T>::isZConstraintMol(Molecule* mol)
@@ -445,6 +509,8 @@ template<typename T> void ZConstraint<T>::integrate(){
   //zero out the velocities of center of mass of unconstrained molecules 
   //and the velocities of center of mass of every single z-constrained molecueles
   zeroOutVel();
+
+  curZconsTime = zconsTime + info->getTime();
   
   T::integrate();
 
@@ -459,20 +525,55 @@ template<typename T> void ZConstraint<T>::calcForce(in
  */ 
 template<typename T> void ZConstraint<T>::calcForce(int calcPot, int calcStress){
   double zsys;
+  double COM[3];
+  double force[3];
+  double zSysCOMVel;
 
   T::calcForce(calcPot, calcStress);
 
-  if (checkZConsState())
+  if (checkZConsState()){
+    
+#ifdef IS_MPI
+    if(worldRank == 0){
+#endif
+      std::cerr << "\n"
+		<< "*******************************************\n"
+		<< " about to call zeroOutVel()\n"
+		<< "*******************************************\n"
+		<< "\n";
+#ifdef IS_MPI
+    }
+#endif
     zeroOutVel();
+
+#ifdef IS_MPI
+    if(worldRank == 0){
+#endif
+      std::cerr << "\n"
+		<< "*******************************************\n"
+		<< " finished zeroOutVel()\n"
+		<< "*******************************************\n"
+		<< "\n";
+#ifdef IS_MPI
+    }
+#endif
+    
+    forcePolicy->update();
+  }  
   
   zsys = calcZSys();
-  cout << "---------------------------------------------------------------------" <<endl;
-  cout << "current time: " << info->getTime() << endl;
-  cout << "center of mass at z: " << zsys << endl;   	 
-  //cout << "before calcForce, the COMVel of moving molecules is " << calcMovingMolsCOMVel() <<endl;
-  cout << "before calcForce, the COMVel of system is " << calcSysCOMVel() <<endl;
+  zSysCOMVel = calcSysCOMVel();
+#ifdef IS_MPI
+  if(worldRank == 0){
+#endif
+    cout << "---------------------------------------------------------------------" <<endl;
+    cout << "current time: " << info->getTime() << endl;
+    cout << "center of mass at z: " << zsys << endl;     
+    cout << "before calcForce, the COMVel of system is " << zSysCOMVel <<endl;
 
-  //cout <<	 "before doZConstraintForce, totalForce is " << calcTotalForce() << endl;
+#ifdef IS_MPI
+  }
+#endif
 
   //do zconstraint force; 
   if (haveFixedZMols())
@@ -482,12 +583,42 @@ template<typename T> void ZConstraint<T>::calcForce(in
   if (haveMovingZMols())
     this->doHarmonic();
 
-  //cout <<	 "after doHarmonic, totalForce is " << calcTotalForce() << endl;
+  //write out forces and current positions of z-constraint molecules
+  if(info->getTime() >= curZconsTime){    
+   for(int i = 0; i < zconsMols.size(); i++){
+      zconsMols[i]->getCOM(COM);
+    curZPos[i] = COM[whichDirection];
 
-  fzOut->writeFZ(info->getTime(), zconsMols.size(),indexOfZConsMols, fz);
-  //cout << "after calcForce, the COMVel of moving molecules is " << calcMovingMolsCOMVel() <<endl; 
-  cout << "after calcForce, the COMVel of system is " << calcSysCOMVel() <<endl; 
+    //if the z-constraint molecule is still moving, just record its force
+    if(states[i] == zcsMoving){
+         fz[i] = 0;
+      Atom** movingZAtoms;
+      movingZAtoms = zconsMols[i]->getMyAtoms();
+      for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
+           movingZAtoms[j]->getFrc(force);
+           fz[i] += force[whichDirection];
+      }
+     }
+   }
+    fzOut->writeFZ(info->getTime(), zconsMols.size(), indexOfZConsMols, fz, curZPos);
+   curZconsTime += zconsTime;
+  }
+
+  zSysCOMVel = calcSysCOMVel();  
+#ifdef IS_MPI
+  if(worldRank == 0){
+#endif
+    cout << "after calcForce, the COMVel of system is " << zSysCOMVel <<endl;
+#ifdef IS_MPI
+  }
+#endif
+
 }
+
+
+/**
+ * 
+ */
  
 template<typename T> double ZConstraint<T>::calcZSys()
 {
@@ -498,7 +629,7 @@ template<typename T> double ZConstraint<T>::calcZSys()
   double totalMZ;
   double massOfCurMol;
   double COM[3];
-	
+  
   totalMass_local = 0;
   totalMZ_local = 0;
   
@@ -513,8 +644,10 @@ template<typename T> double ZConstraint<T>::calcZSys()
  
   
 #ifdef IS_MPI  
-  MPI_Allreduce(&totalMass_local, &totalMass, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
-  MPI_Allreduce(&totalMZ_local, &totalMZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);  
+  MPI_Allreduce(&totalMass_local, &totalMass, 1, 
+                      MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
+  MPI_Allreduce(&totalMZ_local, &totalMZ, 1,
+                      MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);  
  #else
   totalMass = totalMass_local;
   totalMZ = totalMZ_local;
@@ -537,8 +670,6 @@ template<typename T> void ZConstraint<T>::thermalize( 
 
 /**
  *
- *
- *
  */
 
 template<typename T> void ZConstraint<T>::zeroOutVel(){
@@ -546,33 +677,42 @@ template<typename T> void ZConstraint<T>::zeroOutVel()
   Atom** fixedZAtoms;  
   double COMvel[3];
   double vel[3];
+  double zSysCOMVel;
 
   //zero out the velocities of center of mass of fixed z-constrained molecules
   
   for(int i = 0; i < zconsMols.size(); i++){
 
-    if (states[i] == zcsFixed){	
+    if (states[i] == zcsFixed){ 
 
-	   zconsMols[i]->getCOMvel(COMvel);      
-		//cout << "before resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
+     zconsMols[i]->getCOMvel(COMvel);      
+    //cout << "before resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
 
       fixedZAtoms = zconsMols[i]->getMyAtoms(); 
-	  
+    
       for(int j =0; j < zconsMols[i]->getNAtoms(); j++){
         fixedZAtoms[j]->getVel(vel);
-	     vel[whichDirection] -= COMvel[whichDirection];
-	     fixedZAtoms[j]->setVel(vel);
+       vel[whichDirection] -= COMvel[whichDirection];
+       fixedZAtoms[j]->setVel(vel);
       }
 
-		zconsMols[i]->getCOMvel(COMvel);
-		//cout << "after resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
+    zconsMols[i]->getCOMvel(COMvel);
+    //cout << "after resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
     }
-	
+  
   }
 
-  	//cout << "before resetting the COMVel of moving molecules is " << calcMovingMolsCOMVel() <<endl;	
-  	cout << "before resetting the COMVel of sytem is " << calcSysCOMVel() << endl;	
-		   
+    //cout << "before resetting the COMVel of moving molecules is " << calcMovingMolsCOMVel() <<endl; 
+
+  zSysCOMVel = calcSysCOMVel();
+#ifdef IS_MPI
+  if(worldRank == 0){
+#endif
+    cout << "before resetting the COMVel of sytem is " << zSysCOMVel << endl;  
+#ifdef IS_MPI
+  }
+#endif
+       
   // calculate the vz of center of mass of unconstrained molecules and moving z-constrained molecules
   double MVzOfMovingMols_local;
   double MVzOfMovingMols;
@@ -591,9 +731,9 @@ template<typename T> void ZConstraint<T>::zeroOutVel()
     if (states[i] == zcsMoving){
       zconsMols[i]->getCOMvel(COMvel);
       MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];   
-      totalMassOfMovingZMols_local += massOfZConsMols[i]; 	  	
+      totalMassOfMovingZMols_local += massOfZConsMols[i];       
     }
-		
+    
   }
 
 #ifndef IS_MPI
@@ -627,20 +767,32 @@ template<typename T> void ZConstraint<T>::zeroOutVel()
     if (states[i] ==zcsMoving){
    
       movingZAtoms = zconsMols[i]->getMyAtoms();
-	   for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
+     for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
         movingZAtoms[j]->getVel(vel);
         vel[whichDirection] -= vzOfMovingMols;
-	     movingZAtoms[j]->setVel(vel);
-	  }
-	  
+       movingZAtoms[j]->setVel(vel);
+    }
+    
    }
 
  }
 
-	cout << "after resetting the COMVel of moving molecules is " << calcSysCOMVel() <<endl;
 
+  zSysCOMVel = calcSysCOMVel();
+#ifdef IS_MPI
+  if(worldRank == 0){
+#endif
+    cout << "after resetting the COMVel of moving molecules is " << zSysCOMVel << endl;  
+#ifdef IS_MPI
+  }
+#endif
+
 }
 
+/**
+ * 
+ */
+
 template<typename T> void ZConstraint<T>::doZconstraintForce(){
 
   Atom** zconsAtoms;
@@ -652,17 +804,17 @@ template<typename T> void ZConstraint<T>::doZconstrain
 
 
 
-  //constrain the molecules which do not reach the specified positions	
+  //constrain the molecules which do not reach the specified positions  
     
   //Zero Out the force of z-contrained molecules    
   totalFZ_local = 0;
 
   //calculate the total z-contrained force of fixed z-contrained molecules
-  cout << "Fixed Molecules" << endl;
+
   for(int i = 0; i < zconsMols.size(); i++){
-		
+    
     if (states[i] == zcsFixed){
-	  	
+      
       zconsMols[i]->getCOM(COM);
       zconsAtoms = zconsMols[i]->getMyAtoms();  
 
@@ -670,17 +822,25 @@ template<typename T> void ZConstraint<T>::doZconstrain
       for(int j =0; j < zconsMols[i]->getNAtoms(); j++) {
         zconsAtoms[j]->getFrc(force);
         fz[i] += force[whichDirection];      
-      }	
+      } 
       totalFZ_local += fz[i];
 
-      cout << "index: " << indexOfZConsMols[i] 
-				<<"\tcurrent zpos: " << COM[whichDirection] 
-				<< "\tcurrent fz: " <<fz[i] << endl;
+//       cout << "Fixed Molecule\tindex: " << indexOfZConsMols[i] 
+//              <<"\tcurrent zpos: " << COM[whichDirection] 
+//              << "\tcurrent fz: " <<fz[i] << endl;
 
     }
-	  
+    
   } 
-	
+
+  //calculate total z-constraint force
+#ifdef IS_MPI
+  MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
+#else
+  totalFZ = totalFZ_local;
+#endif
+
+  
   // apply negative to fixed z-constrained molecues;
   force[0]= 0;
   force[1]= 0;
@@ -689,118 +849,76 @@ template<typename T> void ZConstraint<T>::doZconstrain
   for(int i = 0; i < zconsMols.size(); i++){
 
     if (states[i] == zcsFixed){  
-	
+  
       int nAtomOfCurZConsMol = zconsMols[i]->getNAtoms();
       zconsAtoms = zconsMols[i]->getMyAtoms();  
     
       for(int j =0; j < nAtomOfCurZConsMol; j++) {
-        force[whichDirection] = -fz[i]/ nAtomOfCurZConsMol;
+      force[whichDirection] = -fz[i]/ nAtomOfCurZConsMol;
+        //force[whichDirection] = - forcePolicy->getZFOfFixedZMols(zconsMols[i], zconsAtoms[j], fz[i]);
         zconsAtoms[j]->addFrc(force);
       }
-		
+    
     }
-	
+  
   } 
 
-  cout << "after zero out z-constraint force on fixed z-constraint molecuels "
-		  << "total force is " << calcTotalForce() << endl;
+//   cout << "after zero out z-constraint force on fixed z-constraint molecuels "
+//        << "total force is " << calcTotalForce() << endl;
+
   //calculate the number of atoms of moving z-constrained molecules
   int nMovingZAtoms_local;
   int nMovingZAtoms;
-	
+  
   nMovingZAtoms_local = 0;
   for(int i = 0; i < zconsMols.size(); i++)
     if(states[i] == zcsMoving)
-	   nMovingZAtoms_local += zconsMols[i]->getNAtoms();
+     nMovingZAtoms_local += zconsMols[i]->getNAtoms();
   
 #ifdef IS_MPI
-  MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
-  MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
+  MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1, 
+                      MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
 #else
-  totalFZ = totalFZ_local;
   nMovingZAtoms = nMovingZAtoms_local;
 #endif
 
   force[0]= 0;
   force[1]= 0;
   force[2]= 0;
-  force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
 
   //modify the forces of unconstrained molecules
-  int accessCount = 0;
   for(int i = 0; i < unconsMols.size(); i++){
      
      Atom** unconsAtoms = unconsMols[i]->getMyAtoms();
      
-     for(int j = 0; j < unconsMols[i]->getNAtoms(); j++)          
+     for(int j = 0; j < unconsMols[i]->getNAtoms(); j++){          
+       force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
+       //force[whichDirection] = forcePolicy->getZFOfMovingMols(unconsAtoms[j],totalFZ);
        unconsAtoms[j]->addFrc(force);
+     }
      
   }      
 
  //modify the forces of moving z-constrained molecules
   for(int i = 0; i < zconsMols.size(); i++) {
     if (states[i] == zcsMoving){
-	 	
+    
       Atom** movingZAtoms = zconsMols[i]->getMyAtoms();     
 
-      for(int j = 0; j < zconsMols[i]->getNAtoms(); j++)
+      for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
+        force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
+        //force[whichDirection] = forcePolicy->getZFOfMovingMols(movingZAtoms[j],totalFZ);
         movingZAtoms[j]->addFrc(force);
+      }
     }
   }
-	
-  cout << "after substracting z-constraint force from moving molecuels "
-		  << "total force is " << calcTotalForce()  << endl;
 
-}
+  //cout << "after substracting z-constraint force from moving molecuels "
+  //      << "total force is " << calcTotalForce()  << endl;
 
-template<typename T> bool ZConstraint<T>::checkZConsState(){
-  double COM[3];
-  double diff;
-  
-  bool changed;
-  
-  changed = false;
-  
-  for(int i =0; i < zconsMols.size(); i++){
-
-    zconsMols[i]->getCOM(COM);
-    diff = fabs(COM[whichDirection] - zPos[i]);  
-    if (  diff <= zconsTol && states[i] == zcsMoving){
-      states[i] = zcsFixed;
-	changed = true;
-    }
-    else if ( diff > zconsTol && states[i] == zcsFixed){
-      states[i] = zcsMoving;
-	changed = true;	  
-    }
-  
-  }
-
-  return changed;
 }
 
-template<typename T> bool ZConstraint<T>::haveFixedZMols(){
-  for(int i = 0; i < zconsMols.size(); i++)
-    if (states[i] == zcsFixed)
-      return true;
-
-  return false;
-}
-
-
 /**
- *
- */
-template<typename T> bool ZConstraint<T>::haveMovingZMols(){
-  for(int i = 0; i < zconsMols.size(); i++)
-    if (states[i] == zcsMoving)
-      return true;
-
-  return false;
-  
-}
-
-/**
   * 
   *
   */
@@ -811,56 +929,157 @@ template<typename T> void ZConstraint<T>::doHarmonic()
   double harmonicF;
   double COM[3];
   double diff;
+  double totalFZ_local;
   double totalFZ;
-	
+  
   force[0] = 0;
   force[1] = 0;
   force[2] = 0;
 
-  totalFZ = 0;
+  totalFZ_local = 0;
 
-  cout << "Moving Molecules" << endl;	
   for(int i = 0; i < zconsMols.size(); i++) {
 
     if (states[i] == zcsMoving){
       zconsMols[i]->getCOM(COM);
-      cout << "index: " << indexOfZConsMols[i] <<"\tcurrent zpos: " << COM[whichDirection] << endl;
-		
-		diff = COM[whichDirection] -zPos[i];
-		
+      cout << "Moving Molecule\tindex: " << indexOfZConsMols[i] 
+	   << "\tcurrent zpos: " << COM[whichDirection] << endl;
+    
+    diff = COM[whichDirection] -zPos[i];
+    
       harmonicU = 0.5 * kz[i] * diff * diff;  
-		info->lrPot += harmonicU;
+    info->lrPot += harmonicU;
 
       harmonicF =  - kz[i] * diff;
-      totalFZ += harmonicF;
+      totalFZ_local += harmonicF;
 
-       //
-		force[whichDirection] = harmonicF / zconsMols[i]->getNAtoms();
-		
+       //adjust force
+    
       Atom** movingZAtoms = zconsMols[i]->getMyAtoms();     
 
-       for(int j = 0; j < zconsMols[i]->getNAtoms(); j++)          
+       for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){          
+        force[whichDirection] = harmonicF / zconsMols[i]->getNAtoms();
+         //force[whichDirection] = forcePolicy->getHFOfFixedZMols(zconsMols[i], movingZAtoms[j], harmonicF);
          movingZAtoms[j]->addFrc(force);
+       }
     }
 
   }
-	
+
+#ifndef IS_MPI
+  totalFZ = totalFZ_local;
+#else
+  MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);  
+#endif
+
   force[0]= 0;
   force[1]= 0;
   force[2]= 0;
-  force[whichDirection] = -totalFZ /totNumOfUnconsAtoms;
 
   //modify the forces of unconstrained molecules
   for(int i = 0; i < unconsMols.size(); i++){
      
      Atom** unconsAtoms = unconsMols[i]->getMyAtoms();
      
-     for(int j = 0; j < unconsMols[i]->getNAtoms(); j++)          
+     for(int j = 0; j < unconsMols[i]->getNAtoms(); j++){          
+       force[whichDirection] = - totalFZ /totNumOfUnconsAtoms;
+       //force[whichDirection] = - forcePolicy->getHFOfUnconsMols(unconsAtoms[j], totalFZ);
        unconsAtoms[j]->addFrc(force);     
+     }
   }   
 
 }
 
+/**
+ * 
+ */
+
+template<typename T> bool ZConstraint<T>::checkZConsState(){
+  double COM[3];
+  double diff;
+  
+  int changed_local;
+  int changed;
+  
+  changed_local = 0;
+  
+  for(int i =0; i < zconsMols.size(); i++){
+
+    zconsMols[i]->getCOM(COM);
+    diff = fabs(COM[whichDirection] - zPos[i]);  
+    if (  diff <= zconsTol && states[i] == zcsMoving){
+      states[i] = zcsFixed;
+     changed_local = 1;
+    }
+    else if ( diff > zconsTol && states[i] == zcsFixed){
+      states[i] = zcsMoving;
+     changed_local = 1;   
+    }
+  
+  }
+
+#ifndef IS_MPI
+  changed =changed_local; 
+#else
+  MPI_Allreduce(&changed_local, &changed, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
+#endif
+  
+  return (changed > 0);
+}
+
+template<typename T> bool ZConstraint<T>::haveFixedZMols(){
+
+  int havingFixed_local;
+  int havingFixed;
+
+  havingFixed_local = 0;
+
+  for(int i = 0; i < zconsMols.size(); i++)
+    if (states[i] == zcsFixed){
+      havingFixed_local = 1;
+    break;
+    }
+
+#ifndef IS_MPI
+  havingFixed = havingFixed_local;
+#else
+  MPI_Allreduce(&havingFixed_local, &havingFixed, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
+#endif
+
+  return havingFixed > 0 ? true : false;
+}
+
+
+/**
+ *
+ */
+template<typename T> bool ZConstraint<T>::haveMovingZMols(){
+
+  int havingMoving_local;
+  int havingMoving;
+
+  havingMoving_local = 0;
+
+  for(int i = 0; i < zconsMols.size(); i++)
+    if (states[i] == zcsMoving){
+      havingMoving_local = 1;
+    break;
+    }
+
+#ifndef IS_MPI
+  havingMoving = havingMoving_local;
+#else
+  MPI_Allreduce(&havingMoving_local, &havingMoving, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
+#endif
+
+  return havingMoving > 0 ? true : false;
+  
+}
+
+/**
+ *
+ */
+
 template<typename T> double ZConstraint<T>::calcMovingMolsCOMVel()
 {
   double MVzOfMovingMols_local;
@@ -882,9 +1101,9 @@ template<typename T> double ZConstraint<T>::calcMoving
     if (states[i] == zcsMoving){
       zconsMols[i]->getCOMvel(COMvel);
       MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];   
-      totalMassOfMovingZMols_local += massOfZConsMols[i]; 	  	
+      totalMassOfMovingZMols_local += massOfZConsMols[i];       
     }
-		
+    
   }
 
 #ifndef IS_MPI
@@ -901,34 +1120,46 @@ template<typename T> double ZConstraint<T>::calcMoving
   return vzOfMovingMols;
 }
 
+/**
+ *
+ */
 
 template<typename T> double ZConstraint<T>::calcSysCOMVel()
 {
   double COMvel[3];
-  double tempMVz = 0;
-		
+  double tempMVz_local;
+  double tempMVz;
+  double massOfZCons_local;
+  double massOfZCons;
+
+
+ tempMVz_local = 0;
+
   for(int i =0 ; i < nMols; i++){
     molecules[i].getCOMvel(COMvel);
-	 tempMVz += molecules[i].getTotalMass()*COMvel[whichDirection];
+   tempMVz_local += molecules[i].getTotalMass()*COMvel[whichDirection];
   }
 
-  double massOfZCons_local;
-  double massOfZCons;
-
   massOfZCons_local = 0;
-	
+  
   for(int i = 0; i < massOfZConsMols.size(); i++){
     massOfZCons_local += massOfZConsMols[i];
   }
 #ifndef IS_MPI
   massOfZCons = massOfZCons_local;
+  tempMVz = tempMVz_local;
 #else
   MPI_Allreduce(&massOfZCons_local, &massOfZCons, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
+  MPI_Allreduce(&tempMVz_local, &tempMVz, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
 #endif
 
   return tempMVz /(totalMassOfUncons + massOfZCons);
 }
 
+/**
+ *
+ */
+
 template<typename T> double ZConstraint<T>::calcTotalForce(){
 
   double force[3];  
@@ -951,3 +1182,92 @@ template<typename T> double ZConstraint<T>::calcTotalF
   return totalForce;
 
 }
+
+/**
+ *
+ */
+
+template<typename T> void ZConstraint<T>::PolicyByNumber::update(){
+  //calculate the number of atoms of moving z-constrained molecules
+  int nMovingZAtoms_local;
+  int nMovingZAtoms;
+  
+  nMovingZAtoms_local = 0;
+  for(int i = 0; i < (zconsIntegrator->zconsMols).size(); i++)
+    if((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving))
+     nMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getNAtoms();
+  
+#ifdef IS_MPI
+  MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
+#else
+  nMovingZAtoms = nMovingZAtoms_local;
+#endif
+  totNumOfMovingAtoms = nMovingZAtoms + zconsIntegrator->totNumOfUnconsAtoms;
+
+#ifdef IS_MPI
+  if(worldRank == 0){
+#endif
+    std::cerr << "\n"
+	      << "*******************************************\n"
+	      << " fiished Policy by numbr()\n"
+	      << "*******************************************\n"
+	      << "\n";
+#ifdef IS_MPI
+  }
+#endif
+}
+
+template<typename T>double ZConstraint<T>::PolicyByNumber::getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
+  return totalForce / mol->getNAtoms();
+}
+
+template<typename T> double ZConstraint<T>::PolicyByNumber::getZFOfMovingMols(Atom* atom, double totalForce){
+  return totalForce / totNumOfMovingAtoms;
+}
+
+template<typename T> double ZConstraint<T>::PolicyByNumber::getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
+    return totalForce / mol->getNAtoms();
+}
+
+template<typename T> double ZConstraint<T>::PolicyByNumber::getHFOfUnconsMols(Atom* atom, double totalForce){
+  return totalForce / zconsIntegrator->totNumOfUnconsAtoms;
+}
+
+/**
+ *
+ */
+
+template<typename T> void ZConstraint<T>::PolicyByMass::update(){
+  //calculate the number of atoms of moving z-constrained molecules
+  double massOfMovingZAtoms_local;
+  double massOfMovingZAtoms;
+  
+  massOfMovingZAtoms_local = 0;
+  for(int i = 0; i < (zconsIntegrator->zconsMols).size(); i++)
+    if((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving))
+     massOfMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getTotalMass();
+  
+#ifdef IS_MPI
+  MPI_Allreduce(&massOfMovingZAtoms_local, &massOfMovingZAtoms, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
+#else
+  massOfMovingZAtoms = massOfMovingZAtoms_local;
+#endif
+  totMassOfMovingAtoms = massOfMovingZAtoms_local + zconsIntegrator->totalMassOfUncons;
+}
+
+template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
+  return totalForce * atom->getMass() / mol->getTotalMass();
+}
+
+template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfMovingMols( Atom* atom, double totalForce){
+    return totalForce * atom->getMass() / totMassOfMovingAtoms;
+}
+
+template<typename T> double ZConstraint<T>::PolicyByMass::getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
+  return totalForce * atom->getMass() / mol->getTotalMass();
+}
+
+template<typename T> double ZConstraint<T>::PolicyByMass::getHFOfUnconsMols(Atom* atom, double totalForce){
+    return totalForce * atom->getMass() / zconsIntegrator->totalMassOfUncons;
+}
+