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