--- trunk/OOPSE-2.0/src/visitors/AtomVisitor.hpp 2004/10/21 16:22:01 1625 +++ trunk/OOPSE-2.0/src/visitors/AtomVisitor.hpp 2004/11/05 21:45:14 1718 @@ -44,6 +44,26 @@ class DefaultAtomVisitor : public BaseAtomVisitor{ vector ssdAtomType; }; +class LinearAtomVisitor : public BaseAtomVisitor{ + public: + LinearAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { + visitorName = "LinearAtomVisitor"; + linearAtomType.push_back("linear"); + } + + virtual void visit(Atom* atom) {} + virtual void visit(DirectionalAtom* datom); + virtual void visit(RigidBody* rb) {} + + virtual const string toString(); + private: + inline bool isLinearAtom(const string& atomType); + vector linearAtomType; +}; + + + + class DefaultAtomVisitor : public BaseAtomVisitor{ public: DefaultAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "DefaultAtomVisitor";}