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