--- trunk/OOPSE-1.0/libmdtools/AtomVisitor.hpp 2004/07/23 14:53:10 1378 +++ trunk/OOPSE-1.0/libmdtools/AtomVisitor.hpp 2004/07/23 15:35:12 1379 @@ -2,6 +2,8 @@ #define _BASEATOMVISITOR_H_ #include "BaseVisitor.hpp" +#include +using namespace std; class BaseAtomVisitor : public BaseVisitor{ public: @@ -19,14 +21,22 @@ class SSDAtomVisitor : public BaseAtomVisitor{ class SSDAtomVisitor : public BaseAtomVisitor{ public: - SSDAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "SSDAtomVisitor";} + SSDAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { + visitorName = "SSDAtomVisitor"; + ssdAtomType.push_back("SSD"); + ssdAtomType.push_back("SSD_E"); + ssdAtomType.push_back("SSD_RF"); + ssdAtomType.push_back("SSD1"); + } virtual void visit(Atom* atom) {} virtual void visit(DirectionalAtom* datom); virtual void visit(RigidBody* rb) {} - virtual const string toString(); - + virtual const string toString(); + private: + inline bool isSSDAtom(const string& atomType); + vector ssdAtomType; }; class DefaultAtomVisitor : public BaseAtomVisitor{