| 1 | tim | 1118 | #ifndef _RIGIDBODYVISITOR_H_ | 
| 2 |  |  | #define _RIGIDBODYVISITOR_H_ | 
| 3 |  |  | #include <iostream> | 
| 4 |  |  | #include <set> | 
| 5 |  |  | #include <string> | 
| 6 |  |  | #include "BaseVisitor.hpp" | 
| 7 |  |  | #include "GenericData.hpp" | 
| 8 |  |  |  | 
| 9 |  |  | using namespace std; | 
| 10 |  |  |  | 
| 11 |  |  | class BaseRigidBodyVisitor: public BaseVisitor{ | 
| 12 |  |  | public: | 
| 13 |  |  |  | 
| 14 |  |  | protected: | 
| 15 |  |  | BaseRigidBodyVisitor(SimInfo* info) : BaseVisitor(){ this->info = info;} | 
| 16 |  |  |  | 
| 17 |  |  | SimInfo* info; | 
| 18 |  |  | }; | 
| 19 |  |  |  | 
| 20 |  |  |  | 
| 21 |  |  | //LipidHeadVisitor  adds a pesudo atom into rigidbody which holds a dipole moment | 
| 22 |  |  | class LipidHeadVisitor : public BaseRigidBodyVisitor{ | 
| 23 |  |  | public: | 
| 24 |  |  | LipidHeadVisitor(SimInfo* info) : BaseRigidBodyVisitor(info){ visitorName = "LipidHeadVisitor";} | 
| 25 | tim | 1126 |  | 
| 26 |  |  | virtual void visit(Atom* atom) {} | 
| 27 |  |  | virtual void visit(DirectionalAtom* datom) {} | 
| 28 | tim | 1118 | virtual void visit(RigidBody* rb); | 
| 29 | tim | 1126 |  | 
| 30 | tim | 1118 | void addLipidHeadName(const string& name); | 
| 31 |  |  | virtual const string toString(); | 
| 32 |  |  |  | 
| 33 |  |  | protected: | 
| 34 |  |  | bool canVisit(const string& name); | 
| 35 |  |  |  | 
| 36 |  |  | set<string> lipidHeadName; | 
| 37 |  |  | }; | 
| 38 |  |  |  | 
| 39 | tim | 1126 | class RBCOMVisitor : public BaseRigidBodyVisitor{ | 
| 40 |  |  | public: | 
| 41 |  |  | RBCOMVisitor(SimInfo* info) : BaseRigidBodyVisitor(info){ visitorName = "RBCOMVisitor";} | 
| 42 |  |  |  | 
| 43 |  |  | virtual void visit(Atom* atom) {} | 
| 44 |  |  | virtual void visit(DirectionalAtom* datom) {} | 
| 45 |  |  | virtual void visit(RigidBody* rb); | 
| 46 |  |  |  | 
| 47 |  |  | virtual const string toString(); | 
| 48 |  |  | }; | 
| 49 | tim | 1118 | #endif |