42 |
|
#ifndef VISITORS_BASEATOMVISITOR_HPP |
43 |
|
#define VISITORS_BASEATOMVISITOR_HPP |
44 |
|
|
45 |
< |
#include <vector> |
45 |
> |
#include <set> |
46 |
|
|
47 |
|
#include "visitors/BaseVisitor.hpp" |
48 |
|
#include "visitors/AtomData.hpp" |
74 |
|
public: |
75 |
|
SSDAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
76 |
|
visitorName = "SSDAtomVisitor"; |
77 |
< |
ssdAtomType.push_back("SSD"); |
78 |
< |
ssdAtomType.push_back("SSD_E"); |
79 |
< |
ssdAtomType.push_back("SSD_RF"); |
80 |
< |
ssdAtomType.push_back("SSD1"); |
77 |
> |
ssdAtomType.insert("SSD"); |
78 |
> |
ssdAtomType.insert("SSD_E"); |
79 |
> |
ssdAtomType.insert("SSD_RF"); |
80 |
> |
ssdAtomType.insert("SSD1"); |
81 |
|
} |
82 |
|
|
83 |
|
virtual void visit(Atom* atom) {} |
87 |
|
virtual const std::string toString(); |
88 |
|
private: |
89 |
|
inline bool isSSDAtom(const std::string& atomType); |
90 |
< |
std::vector<std::string> ssdAtomType; |
90 |
> |
std::set<std::string> ssdAtomType; |
91 |
|
}; |
92 |
|
|
93 |
|
class LinearAtomVisitor : public BaseAtomVisitor{ |
94 |
< |
public: |
95 |
< |
LinearAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
96 |
< |
visitorName = "LinearAtomVisitor"; |
97 |
< |
linearAtomType.push_back("linear"); |
98 |
< |
} |
94 |
> |
public: |
95 |
> |
LinearAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
96 |
> |
visitorName = "LinearAtomVisitor"; |
97 |
> |
linearAtomType.insert("linear"); |
98 |
> |
} |
99 |
|
|
100 |
< |
virtual void visit(Atom* atom) {} |
101 |
< |
virtual void visit(DirectionalAtom* datom); |
102 |
< |
virtual void visit(RigidBody* rb) {} |
103 |
< |
|
104 |
< |
virtual const string toString(); |
105 |
< |
private: |
106 |
< |
inline bool isLinearAtom(const string& atomType); |
107 |
< |
vector<string> linearAtomType; |
100 |
> |
virtual void visit(Atom* atom) {} |
101 |
> |
virtual void visit(DirectionalAtom* datom); |
102 |
> |
virtual void visit(RigidBody* rb) {} |
103 |
> |
|
104 |
> |
virtual const std::string toString(); |
105 |
> |
private: |
106 |
> |
inline bool isLinearAtom(const std::string& atomType); |
107 |
> |
std::set<std::string> linearAtomType; |
108 |
|
}; |
109 |
|
|
110 |
|
|