src/visitors/AtomVisitor.hpp

#ifndef _BASEATOMVISITOR_H_
#define _BASEATOMVISITOR_H_

#include "visitors/BaseVisitor.hpp"
#include <vector>
using namespace std;

class BaseAtomVisitor : public BaseVisitor{
  public:
    virtual void visit(Atom* atom) {}
    virtual void visit(DirectionalAtom* datom) {}
    virtual void visit(RigidBody* rb);
    void setVisited(Atom* atom);
    bool isVisited(Atom* atom);
    
  protected:
    BaseAtomVisitor(SimInfo* info) : BaseVisitor() {}    
    SimInfo* info;
};


class SSDAtomVisitor : public BaseAtomVisitor{
  public:
    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();
  private:
    inline bool isSSDAtom(const string& atomType);
    vector<string> ssdAtomType;   
};

class DefaultAtomVisitor : public BaseAtomVisitor{
  public:
    DefaultAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "DefaultAtomVisitor";}

    virtual void visit(Atom* atom);    
    virtual void visit(DirectionalAtom* datom);    
    virtual void visit(RigidBody* rb) {}
    
    virtual const string toString();
    
};
#endif
Revision:	3
Committed:	Fri Sep 24 16:27:58 2004 UTC (20 years, 7 months ago) by tim
File size:	1421 byte(s)
Log Message:	change the #include in source files
#	User	Rev	Content
1	gezelter	2	#ifndef _BASEATOMVISITOR_H_
2			#define _BASEATOMVISITOR_H_
3
4	tim	3	#include "visitors/BaseVisitor.hpp"
5	gezelter	2	#include <vector>
6			using namespace std;
7
8			class BaseAtomVisitor : public BaseVisitor{
9			public:
10			virtual void visit(Atom* atom) {}
11			virtual void visit(DirectionalAtom* datom) {}
12			virtual void visit(RigidBody* rb);
13			void setVisited(Atom* atom);
14			bool isVisited(Atom* atom);
15
16			protected:
17			BaseAtomVisitor(SimInfo* info) : BaseVisitor() {}
18			SimInfo* info;
19			};
20
21
22			class SSDAtomVisitor : public BaseAtomVisitor{
23			public:
24			SSDAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) {
25			visitorName = "SSDAtomVisitor";
26			ssdAtomType.push_back("SSD");
27			ssdAtomType.push_back("SSD_E");
28			ssdAtomType.push_back("SSD_RF");
29			ssdAtomType.push_back("SSD1");
30			}
31
32			virtual void visit(Atom* atom) {}
33			virtual void visit(DirectionalAtom* datom);
34			virtual void visit(RigidBody* rb) {}
35
36			virtual const string toString();
37			private:
38			inline bool isSSDAtom(const string& atomType);
39			vector<string> ssdAtomType;
40			};
41
42			class DefaultAtomVisitor : public BaseAtomVisitor{
43			public:
44			DefaultAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "DefaultAtomVisitor";}
45
46			virtual void visit(Atom* atom);
47			virtual void visit(DirectionalAtom* datom);
48			virtual void visit(RigidBody* rb) {}
49
50			virtual const string toString();
51
52			};
53			#endif