98 |
|
linearAtomType.insert("linear"); |
99 |
|
} |
100 |
|
|
101 |
+ |
void addGayBerneAtomType(const std::string& atomType); |
102 |
|
virtual void visit(Atom* atom) {} |
103 |
|
virtual void visit(DirectionalAtom* datom); |
104 |
|
virtual void visit(RigidBody* rb) {} |
109 |
|
std::set<std::string> linearAtomType; |
110 |
|
}; |
111 |
|
|
112 |
+ |
class GBLipidAtomVisitor : public BaseAtomVisitor{ |
113 |
+ |
public: |
114 |
+ |
GBLipidAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
115 |
+ |
visitorName = "GBLipidAtomVisitor"; |
116 |
+ |
GBLipidAtomType.insert("GBlipid"); |
117 |
+ |
} |
118 |
|
|
119 |
+ |
virtual void visit(Atom* atom) {} |
120 |
+ |
virtual void visit(DirectionalAtom* datom); |
121 |
+ |
virtual void visit(RigidBody* rb) {} |
122 |
|
|
123 |
+ |
virtual const std::string toString(); |
124 |
+ |
private: |
125 |
+ |
inline bool isGBLipidAtom(const std::string& atomType); |
126 |
+ |
std::set<std::string> GBLipidAtomType; |
127 |
+ |
}; |
128 |
|
|
129 |
+ |
|
130 |
|
class DefaultAtomVisitor : public BaseAtomVisitor{ |
131 |
|
public: |
132 |
|
DefaultAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "DefaultAtomVisitor";} |