--- trunk/OOPSE/libmdtools/Molecule.cpp	2004/04/12 20:32:20	1097
+++ trunk/OOPSE/libmdtools/Molecule.cpp	2004/04/15 16:18:26	1113
@@ -12,12 +12,9 @@ Molecule::Molecule( void ){
   myBonds = NULL;
   myBends = NULL;
   myTorsions = NULL;
-  myRigidBodies = NULL;
 
 }
 
-
-
 Molecule::~Molecule( void ){
   int i;
   
@@ -41,12 +38,6 @@ Molecule::~Molecule( void ){
     delete[] myTorsions;
   }
 
-  if( myRigidBodies != NULL ){
-    for(i=0; i<nRigidBodies; i++) if(myRigidBodies[i] != NULL ) 
-      delete myRigidBodies[i];
-    delete[] myRigidBodies;
-  }
-  
 }
 
 
@@ -65,6 +56,11 @@ void Molecule::initialize( molInit &theInit ){
   myBends = theInit.myBends;
   myTorsions = theInit.myTorsions;
   myRigidBodies = theInit.myRigidBodies;
+
+  myIntegrableObjects = theInit.myIntegrableObjects;
+
+  for (int i = 0; i < myRigidBodies.size(); i++) 
+      myRigidBodies[i]->calcRefCoords();
     
 }
 
@@ -72,7 +68,7 @@ void Molecule::calcForces( void ){
   
   int i;
 
-  for(i=0; i<nRigidBodies; i++) {
+  for(i=0; i<myRigidBodies.size(); i++) {
     myRigidBodies[i]->updateAtoms();
   }
 
@@ -97,6 +93,10 @@ double Molecule::getPotential( void ){
   
   int i;
   double myPot = 0.0;
+
+  for(i=0; i<myRigidBodies.size(); i++) {
+    myRigidBodies[i]->updateAtoms();
+  }
    
   for(i=0; i<nBonds; i++){
     myPot += myBonds[i]->get_potential();
@@ -135,22 +135,20 @@ void Molecule::moveCOM(double delta[3]){
   double aPos[3];
   int i, j;
 
-  for(i=0; i<nAtoms; i++) {
-    if(myAtoms[i] != NULL ) {
+  for(i=0; i<myIntegrableObjects.size(); i++) {
+    if(myIntegrableObjects[i] != NULL ) {
       
-      myAtoms[i]->getPos( aPos );
+      myIntegrableObjects[i]->getPos( aPos );
       
       for (j=0; j< 3; j++) 
         aPos[j] += delta[j];
 
-      myAtoms[i]->setPos( aPos );
+      myIntegrableObjects[i]->setPos( aPos );
     }
   }
 
-  for(i=0; i<nRigidBodies; i++) {
+  for(i=0; i<myRigidBodies.size(); i++) {
 
-    if (myRigidBodies[i] != NULL) {
-      
       myRigidBodies[i]->getPos( aPos );
 
       for (j=0; j< 3; j++) 
@@ -158,15 +156,12 @@ void Molecule::moveCOM(double delta[3]){
       
       myRigidBodies[i]->setPos( aPos );
     }
-  }  
 }
 
 void Molecule::atoms2rigidBodies( void ) {
   int i;
-  for (i = 0; i < nRigidBodies; i++) {
-    if (myRigidBodies[i] != NULL) {
-      myRigidBodies[i]->calcForcesAndTorques();   
-    }
+  for (i = 0; i < myRigidBodies.size(); i++) {
+    myRigidBodies[i]->calcForcesAndTorques();   
   }
 }
 
@@ -181,13 +176,13 @@ void Molecule::getCOM( double COM[3] ) {
 
   mtot   = 0.0;
 
-  for (i=0; i < nAtoms; i++) {
-    if (myAtoms[i] != NULL) {
+  for (i=0; i < myIntegrableObjects.size(); i++) {
+    if (myIntegrableObjects[i] != NULL) {
 
-      mass = myAtoms[i]->getMass();
+      mass = myIntegrableObjects[i]->getMass();
       mtot   += mass;
       
-      myAtoms[i]->getPos( aPos );
+      myIntegrableObjects[i]->getPos( aPos );
 
       for( j = 0; j < 3; j++) 
         COM[j] += aPos[j] * mass;
@@ -211,13 +206,13 @@ double Molecule::getCOMvel( double COMvel[3] ) {
 
   mtot   = 0.0;
 
-  for (i=0; i < nAtoms; i++) {
-    if (myAtoms[i] != NULL) {
+  for (i=0; i < myIntegrableObjects.size(); i++) {
+    if (myIntegrableObjects[i] != NULL) {
 
-      mass = myAtoms[i]->getMass();
+      mass = myIntegrableObjects[i]->getMass();
       mtot   += mass;
 
-      myAtoms[i]->getVel(aVel);
+      myIntegrableObjects[i]->getVel(aVel);
 
       for (j=0; j<3; j++) 
         COMvel[j] += aVel[j]*mass;
@@ -234,15 +229,12 @@ double Molecule::getTotalMass()
 
 double Molecule::getTotalMass()
 {
-  int natoms;
-  Atom** atoms;
+
   double totalMass;
   
-  natoms = getNAtoms();
-  atoms = getMyAtoms();
   totalMass = 0;
-  for(int i =0; i < natoms; i++){
-    totalMass += atoms[i]->getMass();
+  for(int i =0; i < myIntegrableObjects.size(); i++){
+    totalMass += myIntegrableObjects[i]->getMass();
   }
 
   return totalMass;