| 45 |  |  | 
| 46 |  | #include "brains/SimInfo.hpp" | 
| 47 |  | #include "constraints/ConstraintPair.hpp" | 
| 48 | + | #include "io/ConstraintWriter.hpp" | 
| 49 | + |  | 
| 50 |  | namespace OpenMD { | 
| 51 |  |  | 
| 52 |  | class Shake { | 
| 53 |  | public: | 
| 54 |  | enum ConsStatus{ | 
| 55 | < | consFail = -1,  //Constraint Fail | 
| 56 | < | consSuccess = 0,  //constrain the pair by moving two elements | 
| 57 | < | consAlready = 1}; //current pair is already constrained, do not need to move the elements | 
| 55 | > | consFail = -1,   //Constraint Fail | 
| 56 | > | consSuccess = 0, //constrain the pair by moving two elements | 
| 57 | > | consAlready = 1  //current pair is already constrained, do not need to move the elements | 
| 58 | > | }; | 
| 59 |  |  | 
| 60 |  | Shake(SimInfo* info); | 
| 61 |  | void constraintR(); | 
| 77 |  | RealType consTolerance_; | 
| 78 |  | Snapshot* currentSnapshot_; | 
| 79 |  | bool doShake_; | 
| 80 | + | std::string constraintOutputFile_; | 
| 81 | + | ConstraintWriter* constraintWriter_; | 
| 82 | + | RealType constraintTime_; | 
| 83 | + | RealType currConstraintTime_; | 
| 84 |  | }; | 
| 85 |  | } | 
| 86 |  | #endif |