--- trunk/OOPSE/libmdtools/Molecule.hpp	2003/07/31 15:35:07	658
+++ trunk/OOPSE/libmdtools/Molecule.hpp	2004/06/04 03:15:31	1234
@@ -1,11 +1,18 @@
 #ifndef _MOLECULE_H_
 #define _MOLECULE_H_
 
+#include <set>
+#include <vector>
+
 #include "Atom.hpp"
 #include "SRI.hpp"
 #include "MoleculeStamp.hpp"
-#include "Exclude.hpp"
+#include "RigidBody.hpp"
+#include "CutoffGroup.hpp"
+#include "ConstraintPair.hpp"
 
+using namespace std;
+
 typedef struct{
   
   int stampID;   // the ID in the BASS component stamp array
@@ -13,16 +20,17 @@ typedef struct{
   int nBonds;    // ... .. ..  . .bonds .. .. . . . . 
   int nBends;    // . . . . .. . .bends . . . . .. . 
   int nTorsions; // .. . . .. . . torsions . . .. . . 
+  int nRigidBodies; // .. .. .. . rigid bodies ... .. 
   int nOriented; // .. . . . .. . oriented atoms . . . 
-  int nExcludes; // . . .. . . . .exclude pairs.. . . . 
-
+  
   Atom** myAtoms;      // the array of atoms
-  Exclude** myExcludes;// the array of Excluded pairs 
   Bond** myBonds;      // arrays of all the short range interactions
   Bend** myBends;
   Torsion** myTorsions;
-
-
+  vector<RigidBody*>   myRigidBodies;
+  vector<StuntDouble*> myIntegrableObjects;
+  vector<CutoffGroup*> myCutoffGroups;
+  vector<ConstraintPair*> myConstraintPairs;
 } molInit;
 
 class Molecule{
@@ -35,32 +43,49 @@ class Molecule{ (public)
   void initialize( molInit &theInit );
 
   void setMyIndex( int theIndex ){ myIndex = theIndex;}
-  void setGlobalIndex( int theIndex ) { globalIndex = theIndex; }
-
   int getMyIndex( void ) { return myIndex; }
+
   int getGlobalIndex( void ) { return globalIndex; }
- 
-  int getNAtoms   ( void ) {return nAtoms;}
-  int getNBonds   ( void ) {return nBonds;}
-  int getNBends   ( void ) {return nBends;}
-  int getNTorsions( void ) {return nTorsions;}
-  int getNOriented( void ) {return nOriented;}
-  int getNExcludes( void ) {return nExcludes;}
-  int getNMembers ( void ) {return nMembers;}
-  int getStampID  ( void ) {return stampID;}
+  void setGlobalIndex( int theIndex ) { globalIndex = theIndex; }
 
-  Atom**     getMyAtoms   ( void ) {return myAtoms;}
-  Bond**     getMyBonds   ( void ) {return myBonds;}
-  Bend**     getMyBends   ( void ) {return myBends;}
-  Torsion**  getMyTorsions( void ) {return myTorsions;}
-  Exclude**  getMyExcludes( void ) {return myExcludes;}
-  
-  void setStampID( int info ) {stampID = info;}
-  
+  int getNAtoms   ( void )    {return nAtoms;}
+  int getNBonds   ( void )    {return nBonds;}
+  int getNBends   ( void )    {return nBends;}
+  int getNTorsions( void )    {return nTorsions;}
+  int getNRigidBodies( void ) {return myRigidBodies.size();}
+  int getNOriented( void )    {return nOriented;}
+  int getNMembers ( void )    {return nMembers;}
+  int getStampID  ( void )    {return stampID;}
 
+  Atom**      getMyAtoms   ( void )    {return myAtoms;}
+  Bond**      getMyBonds   ( void )    {return myBonds;}
+  Bend**      getMyBends   ( void )    {return myBends;}
+  Torsion**   getMyTorsions( void )    {return myTorsions;}
+  vector<RigidBody*> getMyRigidBodies( void ) {return myRigidBodies;}
+  vector<StuntDouble*>& getIntegrableObjects(void) {return myIntegrableObjects;}
 
+  //beginCutoffGroup return the first group and initialize the iterator
+  CutoffGroup* beginCutoffGroup(vector<CutoffGroup*>::iterator& i){
+    i = myCutoffGroups.begin();
+    return i != myCutoffGroups.end()? *i : NULL;
+  }
 
+  //nextCutoffGroup return next cutoff group based on the iterator
+  CutoffGroup* nextCutoffGroup(vector<CutoffGroup*>::iterator& i){
+    i++;
+    return i != myCutoffGroups.end()? *i : NULL;
+  }
+
+  int getNCutoffGroups() {return nCutoffGroups;}
+
+  int getNConstrains() {return myConstraintPairs.size();}
+  vector<ConstraintPair*>& getConstraintPairs() {return myConstraintPairs;}
+  
+  
+  void setStampID( int info ) {stampID = info;} 
+
   void calcForces( void );
+  void atoms2rigidBodies( void );
   double getPotential( void );
   
   void printMe( void );
@@ -78,10 +103,11 @@ class Molecule{ (public)
   int nBonds;    // ... .. ..  . .bonds .. .. . . . . 
   int nBends;    // . . . . .. . .bends . . . . .. . 
   int nTorsions; // .. . . .. . . torsions . . .. . . 
+  int nRigidBodies; // .. . . .. .rigid bodies . . .. . . 
   int nOriented; // .. . . . .. . oriented atoms . . . 
   int nMembers;  // .. . . . . . .atoms (legacy code) . . . 
-  int nExcludes; // . . . . .. .. excludes .. . . 
-
+  int nCutoffGroups;
+  
   int myIndex; // mostly just for debug (and for making pressure calcs work)
   int globalIndex;
 
@@ -89,8 +115,11 @@ class Molecule{ (public)
   Bond** myBonds;     // arrays of all the short range interactions
   Bend** myBends;
   Torsion** myTorsions;
-  Exclude** myExcludes; // array of the excluded pairs of long range forces
-  
+  vector<RigidBody*>   myRigidBodies;
+  vector<StuntDouble*> myIntegrableObjects;
+  vector<CutoffGroup*> myCutoffGroups;
+  vector<ConstraintPair*> myConstraintPairs;
+
 };
 
 #endif