--- trunk/src/restraints/MolecularRestraint.cpp	2012/08/22 02:28:28	1782
+++ trunk/src/restraints/MolecularRestraint.cpp	2015/03/05 16:30:23	2069
@@ -35,7 +35,7 @@
  *                                                                      
  * [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).
  */
@@ -71,7 +71,7 @@ namespace OpenMD {
 
       pot_ += p;
 
-      restInfo_[rtDisplacement] = std::make_pair(r, p);
+      if (printRest_) restInfo_[rtDisplacement] = std::make_pair(r, p);
 
       for(it = forces_.begin(); it != forces_.end(); ++it)
         (*it) = -kDisp_ * del * scaleFactor_;
@@ -135,9 +135,6 @@ namespace OpenMD {
       RotMat3x3d Atrans = v * w_tr.transpose();
       RotMat3x3d A = Atrans.transpose();
 
-      Vector3d eularAngles = A.toEulerAngles();
-
-
       RealType twistAngle;
       Vector3d swingAxis;
 
@@ -152,37 +149,34 @@ namespace OpenMD {
 
       if (restType_ & rtTwist){
         dTwist = twistAngle - twist0_;
-        dVdtwist = kTwist_ * sin(dTwist) ;
-        p = kTwist_ * (1.0 - cos(dTwist) ) ;
+        /// dVdtwist = kTwist_ * sin(dTwist) ;
+        /// p = kTwist_ * (1.0 - cos(dTwist) ) ;
+        dVdtwist = kTwist_ * dTwist;
+        p = 0.5 * kTwist_ * dTwist * dTwist;
         pot_ += p;
         tBody -= dVdtwist * V3Z;
-        restInfo_[rtTwist] = std::make_pair(twistAngle, p);
+        if (printRest_) restInfo_[rtTwist] = std::make_pair(twistAngle, p);
       }
 
-//       if (restType_ & rtSwing){
-//         dSwing = swingAngle - swing0_;
-//         dVdswing = kSwing_ * 2.0 * sin(2.0 * dSwing);
-//         p = kSwing_ * (1.0 - cos(2.0 * dSwing));
-//         pot_ += p;
-//         tBody -= dVdswing * swingAxis;
-//         restInfo_[rtSwing] = std::make_pair(swingAngle, p);
-//       }
-
       if (restType_ & rtSwingX){
         dSwingX = swingX - swingX0_;
-        dVdswingX = kSwingX_ * 2.0 * sin(2.0 * dSwingX);
-        p = kSwingX_ * (1.0 - cos(2.0 * dSwingX));
+        /// dVdswingX = kSwingX_ * 2.0 * sin(2.0 * dSwingX);
+        /// p = kSwingX_ * (1.0 - cos(2.0 * dSwingX));
+        dVdswingX = kSwingX_ * dSwingX;        
+        p = 0.5 * kSwingX_ * dSwingX * dSwingX;
         pot_ += p;
         tBody -= dVdswingX * V3X;
-        restInfo_[rtSwingX] = std::make_pair(swingX, p);
+        if (printRest_) restInfo_[rtSwingX] = std::make_pair(swingX, p);
       }
       if (restType_ & rtSwingY){
         dSwingY = swingY - swingY0_;
-        dVdswingY = kSwingY_ * 2.0 * sin(2.0 * dSwingY);
-        p = kSwingY_ * (1.0 - cos(2.0 * dSwingY));
+        /// dVdswingY = kSwingY_ * 2.0 * sin(2.0 * dSwingY);
+        /// p = kSwingY_ * (1.0 - cos(2.0 * dSwingY));
+        dVdswingY = kSwingY_ * dSwingY;        
+        p = 0.5 * kSwingX_ * dSwingY * dSwingY;
         pot_ += p;
         tBody -= dVdswingY * V3Y;
-        restInfo_[rtSwingY] = std::make_pair(swingY, p);
+        if (printRest_) restInfo_[rtSwingY] = std::make_pair(swingY, p);
       }