| 9 |
|
#include "MoleculeStamp.hpp" |
| 10 |
|
#include "RigidBody.hpp" |
| 11 |
|
#include "CutoffGroup.hpp" |
| 12 |
+ |
#include "ConstraintPair.hpp" |
| 13 |
|
|
| 14 |
|
using namespace std; |
| 15 |
|
|
| 30 |
|
vector<RigidBody*> myRigidBodies; |
| 31 |
|
vector<StuntDouble*> myIntegrableObjects; |
| 32 |
|
vector<CutoffGroup*> myCutoffGroups; |
| 33 |
+ |
vector<ConstraintPair*> myConstraintPairs; |
| 34 |
|
} molInit; |
| 35 |
|
|
| 36 |
|
class Molecule{ |
| 64 |
|
vector<RigidBody*> getMyRigidBodies( void ) {return myRigidBodies;} |
| 65 |
|
vector<StuntDouble*>& getIntegrableObjects(void) {return myIntegrableObjects;} |
| 66 |
|
|
| 67 |
+ |
//beginCutoffGroup return the first group and initialize the iterator |
| 68 |
|
CutoffGroup* beginCutoffGroup(vector<CutoffGroup*>::iterator& i){ |
| 69 |
|
i = myCutoffGroups.begin(); |
| 70 |
|
return i != myCutoffGroups.end()? *i : NULL; |
| 71 |
|
} |
| 72 |
|
|
| 73 |
+ |
//nextCutoffGroup return next cutoff group based on the iterator |
| 74 |
|
CutoffGroup* nextCutoffGroup(vector<CutoffGroup*>::iterator& i){ |
| 75 |
|
i++; |
| 76 |
|
return i != myCutoffGroups.end()? *i : NULL; |
| 77 |
|
} |
| 78 |
|
|
| 79 |
|
int getNCutoffGroups() {return nCutoffGroups;} |
| 80 |
+ |
|
| 81 |
+ |
int getNConstrains() {return myConstraintPairs.size();} |
| 82 |
+ |
vector<ConstraintPair*>& getConstraintPairs() {return myConstraintPairs;} |
| 83 |
|
|
| 84 |
+ |
|
| 85 |
|
void setStampID( int info ) {stampID = info;} |
| 86 |
|
|
| 87 |
|
void calcForces( void ); |
| 118 |
|
vector<RigidBody*> myRigidBodies; |
| 119 |
|
vector<StuntDouble*> myIntegrableObjects; |
| 120 |
|
vector<CutoffGroup*> myCutoffGroups; |
| 121 |
+ |
vector<ConstraintPair*> myConstraintPairs; |
| 122 |
|
|
| 114 |
– |
|
| 123 |
|
}; |
| 124 |
|
|
| 125 |
|
#endif |