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