1 |
#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 |
virtual void visit(RigidBody* rb); |
26 |
void addLipidHeadName(const string& name); |
27 |
virtual const string toString(); |
28 |
|
29 |
protected: |
30 |
bool canVisit(const string& name); |
31 |
|
32 |
set<string> lipidHeadName; |
33 |
}; |
34 |
|
35 |
#endif |