--- trunk/OOPSE/libmdtools/Integrator.cpp	2003/08/27 19:23:29	733
+++ trunk/OOPSE/libmdtools/Integrator.cpp	2004/04/20 16:56:40	1127
@@ -1,12 +1,16 @@
 #include <iostream>
-#include <cstdlib>
-#include <cmath>
+#include <stdlib.h>
+#include <math.h>
 
 #ifdef IS_MPI
 #include "mpiSimulation.hpp"
 #include <unistd.h>
 #endif //is_mpi
 
+#ifdef PROFILE
+#include "mdProfile.hpp"
+#endif // profile
+
 #include "Integrator.hpp"
 #include "simError.h"
 
@@ -25,9 +29,9 @@ template<typename T> Integrator<T>::Integrator(SimInfo
   if (info->the_integrator != NULL){
     delete info->the_integrator;
   }
-  info->the_integrator = this;
 
   nAtoms = info->n_atoms;
+  integrableObjects = info->integrableObjects;
 
   // check for constraints
 
@@ -65,7 +69,8 @@ template<typename T> void Integrator<T>::checkConstrai
 
   SRI** theArray;
   for (int i = 0; i < nMols; i++){
-    theArray = (SRI * *) molecules[i].getMyBonds();
+
+	  theArray = (SRI * *) molecules[i].getMyBonds();
     for (int j = 0; j < molecules[i].getNBonds(); j++){
       constrained = theArray[j]->is_constrained();
 
@@ -111,6 +116,7 @@ template<typename T> void Integrator<T>::checkConstrai
     }
   }
 
+
   if (nConstrained > 0){
     isConstrained = 1;
 
@@ -132,7 +138,7 @@ template<typename T> void Integrator<T>::checkConstrai
     }
 
 
-    // save oldAtoms to check for lode balanceing later on.
+    // save oldAtoms to check for lode balancing later on.
 
     oldAtoms = nAtoms;
 
@@ -147,68 +153,85 @@ template<typename T> void Integrator<T>::integrate(voi
 
 
 template<typename T> void Integrator<T>::integrate(void){
-  int i, j;                         // loop counters
 
   double runTime = info->run_time;
   double sampleTime = info->sampleTime;
   double statusTime = info->statusTime;
   double thermalTime = info->thermalTime;
+  double resetTime = info->resetTime;
 
+
   double currSample;
   double currThermal;
   double currStatus;
+  double currReset;
 
   int calcPot, calcStress;
-  int isError;
 
   tStats = new Thermo(info);
   statOut = new StatWriter(info);
   dumpOut = new DumpWriter(info);
 
   atoms = info->atoms;
-  DirectionalAtom* dAtom;
 
   dt = info->dt;
   dt2 = 0.5 * dt;
 
+  readyCheck();
+
+  // remove center of mass drift velocity (in case we passed in a configuration
+  // that was drifting
+  tStats->removeCOMdrift();
+
   // initialize the forces before the first step
 
   calcForce(1, 1);
   
+  if (nConstrained){
+    preMove();
+    constrainA();
+    calcForce(1, 1);
+    constrainB();
+  }
+  
   if (info->setTemp){
     thermalize();
   }
 
-  calcPot = 0;
-  calcStress = 0;
-  currSample = sampleTime;
-  currThermal = thermalTime;
-  currStatus = statusTime;
-  
   calcPot     = 0;
   calcStress  = 0;
   currSample  = sampleTime + info->getTime();
   currThermal = thermalTime+ info->getTime();
   currStatus  = statusTime + info->getTime();
+  currReset   = resetTime  + info->getTime();
 
   dumpOut->writeDump(info->getTime());
   statOut->writeStat(info->getTime());
 
-  readyCheck();
 
 #ifdef IS_MPI
   strcpy(checkPointMsg, "The integrator is ready to go.");
   MPIcheckPoint();
 #endif // is_mpi
 
-  while (info->getTime() < runTime){
+  while (info->getTime() < runTime && !stopIntegrator()){
     if ((info->getTime() + dt) >= currStatus){
       calcPot = 1;
       calcStress = 1;
     }
 
+#ifdef PROFILE
+    startProfile( pro1 );
+#endif
+    
     integrateStep(calcPot, calcStress);
 
+#ifdef PROFILE
+    endProfile( pro1 );
+
+    startProfile( pro2 );
+#endif // profile
+
     info->incrTime(dt);
 
     if (info->setTemp){
@@ -224,20 +247,29 @@ template<typename T> void Integrator<T>::integrate(voi
     }
 
     if (info->getTime() >= currStatus){
-      statOut->writeStat(info->getTime()); 
-      calcPot = 0; 
+      statOut->writeStat(info->getTime());
+      calcPot = 0;
       calcStress = 0;
       currStatus += statusTime;
-    } 
+    }
 
+    if (info->resetIntegrator){
+      if (info->getTime() >= currReset){
+        this->resetIntegrator();
+        currReset += resetTime;
+      }
+    }
+    
+#ifdef PROFILE
+    endProfile( pro2 );
+#endif //profile
+
 #ifdef IS_MPI
     strcpy(checkPointMsg, "successfully took a time step.");
     MPIcheckPoint();
 #endif // is_mpi
   }
 
-  dumpOut->writeFinal(info->getTime());
-
   delete dumpOut;
   delete statOut;
 }
@@ -245,13 +277,26 @@ template<typename T> void Integrator<T>::integrateStep
 template<typename T> void Integrator<T>::integrateStep(int calcPot,
                                                        int calcStress){
   // Position full step, and velocity half step
+
+#ifdef PROFILE
+  startProfile(pro3);
+#endif //profile
+
   preMove();
 
+#ifdef PROFILE
+  endProfile(pro3);
+
+  startProfile(pro4);
+#endif // profile
+
   moveA();
 
-  if (nConstrained){
-    constrainA();
-  }
+#ifdef PROFILE
+  endProfile(pro4);
+  
+  startProfile(pro5);
+#endif//profile
 
 
 #ifdef IS_MPI
@@ -269,14 +314,19 @@ template<typename T> void Integrator<T>::integrateStep
   MPIcheckPoint();
 #endif // is_mpi
 
+#ifdef PROFILE
+  endProfile( pro5 );
 
+  startProfile( pro6 );
+#endif //profile
+
   // finish the velocity  half step
 
   moveB();
 
-  if (nConstrained){
-    constrainB();
-  }
+#ifdef PROFILE
+  endProfile(pro6);
+#endif // profile
 
 #ifdef IS_MPI
   strcpy(checkPointMsg, "Succesful moveB\n");
@@ -286,21 +336,19 @@ template<typename T> void Integrator<T>::moveA(void){
 
 
 template<typename T> void Integrator<T>::moveA(void){
-  int i, j;
+  size_t i, j;
   DirectionalAtom* dAtom;
   double Tb[3], ji[3];
-  double A[3][3], I[3][3];
-  double angle;
   double vel[3], pos[3], frc[3];
   double mass;
+ 
+  for (i = 0; i < integrableObjects.size() ; i++){
+    integrableObjects[i]->getVel(vel);
+    integrableObjects[i]->getPos(pos);
+    integrableObjects[i]->getFrc(frc);
+    
+    mass = integrableObjects[i]->getMass();
 
-  for (i = 0; i < nAtoms; i++){
-    atoms[i]->getVel(vel);
-    atoms[i]->getPos(pos);
-    atoms[i]->getFrc(frc);
-
-    mass = atoms[i]->getMass();
-
     for (j = 0; j < 3; j++){
       // velocity half step
       vel[j] += (dt2 * frc[j] / mass) * eConvert;
@@ -308,96 +356,75 @@ template<typename T> void Integrator<T>::moveA(void){
       pos[j] += dt * vel[j];
     }
 
-    atoms[i]->setVel(vel);
-    atoms[i]->setPos(pos);
+    integrableObjects[i]->setVel(vel);
+    integrableObjects[i]->setPos(pos);
 
-    if (atoms[i]->isDirectional()){
-      dAtom = (DirectionalAtom *) atoms[i];
+    if (integrableObjects[i]->isDirectional()){
 
       // get and convert the torque to body frame
 
-      dAtom->getTrq(Tb);
-      dAtom->lab2Body(Tb);
+      integrableObjects[i]->getTrq(Tb);
+      integrableObjects[i]->lab2Body(Tb);
 
       // get the angular momentum, and propagate a half step
 
-      dAtom->getJ(ji);
+      integrableObjects[i]->getJ(ji);
 
       for (j = 0; j < 3; j++)
         ji[j] += (dt2 * Tb[j]) * eConvert;
 
-      // use the angular velocities to propagate the rotation matrix a
-      // full time step
+      this->rotationPropagation( integrableObjects[i], ji );
 
-      dAtom->getA(A);
-      dAtom->getI(I);
-
-      // rotate about the x-axis      
-      angle = dt2 * ji[0] / I[0][0];
-      this->rotate(1, 2, angle, ji, A); 
-
-      // rotate about the y-axis
-      angle = dt2 * ji[1] / I[1][1];
-      this->rotate(2, 0, angle, ji, A);
-
-      // rotate about the z-axis
-      angle = dt * ji[2] / I[2][2];
-      this->rotate(0, 1, angle, ji, A);
-
-      // rotate about the y-axis
-      angle = dt2 * ji[1] / I[1][1];
-      this->rotate(2, 0, angle, ji, A);
-
-      // rotate about the x-axis
-      angle = dt2 * ji[0] / I[0][0];
-      this->rotate(1, 2, angle, ji, A);
-
-
-      dAtom->setJ(ji);
-      dAtom->setA(A);
+      integrableObjects[i]->setJ(ji);
     }
   }
+
+  if (nConstrained){
+    constrainA();
+  }
 }
 
 
 template<typename T> void Integrator<T>::moveB(void){
   int i, j;
-  DirectionalAtom* dAtom;
   double Tb[3], ji[3];
   double vel[3], frc[3];
   double mass;
 
-  for (i = 0; i < nAtoms; i++){
-    atoms[i]->getVel(vel);
-    atoms[i]->getFrc(frc);
+  for (i = 0; i < integrableObjects.size(); i++){
+    integrableObjects[i]->getVel(vel);
+    integrableObjects[i]->getFrc(frc);
 
-    mass = atoms[i]->getMass();
+    mass = integrableObjects[i]->getMass();
 
     // velocity half step
     for (j = 0; j < 3; j++)
       vel[j] += (dt2 * frc[j] / mass) * eConvert;
 
-    atoms[i]->setVel(vel);
+    integrableObjects[i]->setVel(vel);
 
-    if (atoms[i]->isDirectional()){
-      dAtom = (DirectionalAtom *) atoms[i];
+    if (integrableObjects[i]->isDirectional()){
 
-      // get and convert the torque to body frame      
+      // get and convert the torque to body frame
 
-      dAtom->getTrq(Tb);
-      dAtom->lab2Body(Tb);
+      integrableObjects[i]->getTrq(Tb);
+      integrableObjects[i]->lab2Body(Tb);
 
       // get the angular momentum, and propagate a half step
 
-      dAtom->getJ(ji);
+      integrableObjects[i]->getJ(ji);
 
       for (j = 0; j < 3; j++)
         ji[j] += (dt2 * Tb[j]) * eConvert;
 
 
-      dAtom->setJ(ji);
+      integrableObjects[i]->setJ(ji);
     }
   }
+
+  if (nConstrained){
+    constrainB();
+  }
 }
 
 template<typename T> void Integrator<T>::preMove(void){
@@ -416,7 +443,7 @@ template<typename T> void Integrator<T>::constrainA(){
 }
 
 template<typename T> void Integrator<T>::constrainA(){
-  int i, j, k;
+  int i, j;
   int done;
   double posA[3], posB[3];
   double velA[3], velB[3];
@@ -556,10 +583,11 @@ template<typename T> void Integrator<T>::constrainA(){
     painCave.isFatal = 1;
     simError();
   }
+
 }
 
 template<typename T> void Integrator<T>::constrainB(void){
-  int i, j, k;
+  int i, j;
   int done;
   double posA[3], posB[3];
   double velA[3], velB[3];
@@ -568,8 +596,7 @@ template<typename T> void Integrator<T>::constrainB(vo
   int a, b, ax, ay, az, bx, by, bz;
   double rma, rmb;
   double dx, dy, dz;
-  double rabsq, pabsq, rvab;
-  double diffsq;
+  double rvab;
   double gab;
   int iteration;
 
@@ -659,6 +686,58 @@ template<typename T> void Integrator<T>::rotate(int ax
   }
 }
 
+template<typename T> void Integrator<T>::rotationPropagation
+( StuntDouble* sd, double ji[3] ){
+
+  double angle;
+  double A[3][3], I[3][3];
+  int i, j, k;
+
+  // use the angular velocities to propagate the rotation matrix a
+  // full time step
+
+  sd->getA(A);
+  sd->getI(I);
+
+  if (sd->isLinear()) {
+    i = sd->linearAxis();
+    j = (i+1)%3;
+    k = (i+2)%3;
+    
+    angle = dt2 * ji[j] / I[j][j];
+    this->rotate( k, i, angle, ji, A );
+
+    angle = dt * ji[k] / I[k][k];
+    this->rotate( i, j, angle, ji, A);
+
+    angle = dt2 * ji[j] / I[j][j];
+    this->rotate( k, i, angle, ji, A );
+
+  } else {
+    // rotate about the x-axis
+    angle = dt2 * ji[0] / I[0][0];
+    this->rotate( 1, 2, angle, ji, A );
+    
+    // rotate about the y-axis
+    angle = dt2 * ji[1] / I[1][1];
+    this->rotate( 2, 0, angle, ji, A );
+    
+    // rotate about the z-axis
+    angle = dt * ji[2] / I[2][2];
+    this->rotate( 0, 1, angle, ji, A);
+    
+    // rotate about the y-axis
+    angle = dt2 * ji[1] / I[1][1];
+    this->rotate( 2, 0, angle, ji, A );
+    
+    // rotate about the x-axis
+    angle = dt2 * ji[0] / I[0][0];
+    this->rotate( 1, 2, angle, ji, A );
+    
+  }
+  sd->setA( A  );
+}
+
 template<typename T> void Integrator<T>::rotate(int axes1, int axes2,
                                                 double angle, double ji[3],
                                                 double A[3][3]){
@@ -724,7 +803,7 @@ template<typename T> void Integrator<T>::rotate(int ax
     }
   }
 
-  // rotate the Rotation matrix acording to: 
+  // rotate the Rotation matrix acording to:
   //            A[][] = A[][] * transpose(rot[][])
 
 
@@ -749,3 +828,13 @@ template<typename T> void Integrator<T>::thermalize(){
 template<typename T> void Integrator<T>::thermalize(){
   tStats->velocitize();
 }
+
+template<typename T> double Integrator<T>::getConservedQuantity(void){
+  return tStats->getTotalE();
+}
+template<typename T> string Integrator<T>::getAdditionalParameters(void){
+  //By default, return a null string
+  //The reason we use string instead of char* is that if we use char*, we will
+  //return a pointer point to local variable which might cause problem
+  return string();
+}