| 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 |