| 45 |  |  | 
| 46 |  | #include "brains/SimInfo.hpp" | 
| 47 |  | #include "constraints/ConstraintPair.hpp" | 
| 48 | + | #include "io/ConstraintWriter.hpp" | 
| 49 | + |  | 
| 50 |  | namespace OpenMD { | 
| 51 |  |  | 
| 52 |  | /** | 
| 56 |  | class Rattle { | 
| 57 |  | public: | 
| 58 |  | enum ConsStatus{ | 
| 59 | < | consFail = -1,  //Constraint Fail | 
| 60 | < | consSuccess = 0,  //constrain the pair by moving two elements | 
| 61 | < | consAlready = 1}; //current pair is already constrained, do not need to move the elements | 
| 59 | > | consFail = -1,   //Constraint Fail | 
| 60 | > | consSuccess = 0, //constrain the pair by moving two elements | 
| 61 | > | consAlready = 1  //current pair is already constrained, do not need to move the elements | 
| 62 | > | }; | 
| 63 |  |  | 
| 64 |  | Rattle(SimInfo* info); | 
| 65 |  | void constraintA(); | 
| 83 |  | RealType dt_; | 
| 84 |  | Snapshot* currentSnapshot_; | 
| 85 |  | bool doRattle_; | 
| 86 | + | std::string constraintOutputFile_; | 
| 87 | + | ConstraintWriter* constraintWriter_; | 
| 88 | + | RealType constraintTime_; | 
| 89 | + | RealType currConstraintTime_; | 
| 90 |  | }; | 
| 84 | – |  | 
| 91 |  | } | 
| 92 |  | #endif |