150 |
|
void internalVisit(RigidBody* rb); |
151 |
|
}; |
152 |
|
|
153 |
< |
class WaterTypeVisitor : public BaseVisitor{ |
153 |
> |
class WaterTypeVisitor : public BaseVisitor{ |
154 |
|
public: |
155 |
|
WaterTypeVisitor() ; |
156 |
|
virtual void visit(Atom* atom) {} |
157 |
|
virtual void visit(DirectionalAtom* datom) {} |
158 |
|
virtual void visit(RigidBody* rb); |
159 |
< |
|
159 |
> |
|
160 |
|
virtual const std::string toString(); |
161 |
|
|
162 |
|
private: |
163 |
< |
void replaceType(std::string& atomType); |
164 |
< |
|
163 |
> |
std::string trimmedName(const std::string& atomType); |
164 |
> |
|
165 |
|
std::set<std::string> waterTypeList; |
166 |
|
}; |
167 |
|
|