| 78 |
|
ssdAtomType.insert("SSD_E"); |
| 79 |
|
ssdAtomType.insert("SSD_RF"); |
| 80 |
|
ssdAtomType.insert("SSD1"); |
| 81 |
+ |
ssdAtomType.insert("TAP"); |
| 82 |
|
} |
| 83 |
|
|
| 84 |
|
virtual void visit(Atom* atom) {} |
| 91 |
|
std::set<std::string> ssdAtomType; |
| 92 |
|
}; |
| 93 |
|
|
| 94 |
+ |
class TREDAtomVisitor : public BaseAtomVisitor{ |
| 95 |
+ |
public: |
| 96 |
+ |
TREDAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
| 97 |
+ |
visitorName = "TREDAtomVisitor"; |
| 98 |
+ |
tredAtomType.insert("TRED"); |
| 99 |
+ |
} |
| 100 |
+ |
|
| 101 |
+ |
virtual void visit(Atom* atom) {} |
| 102 |
+ |
virtual void visit(DirectionalAtom* datom); |
| 103 |
+ |
virtual void visit(RigidBody* rb) {} |
| 104 |
+ |
|
| 105 |
+ |
virtual const std::string toString(); |
| 106 |
+ |
private: |
| 107 |
+ |
inline bool isTREDAtom(const std::string& atomType); |
| 108 |
+ |
std::set<std::string> tredAtomType; |
| 109 |
+ |
}; |
| 110 |
+ |
|
| 111 |
|
class LinearAtomVisitor : public BaseAtomVisitor{ |
| 112 |
|
public: |
| 113 |
|
LinearAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
| 114 |
|
visitorName = "LinearAtomVisitor"; |
| 115 |
< |
linearAtomType.insert("linear"); |
| 115 |
> |
linearAtomType.insert("GBlinear"); |
| 116 |
|
} |
| 117 |
|
|
| 118 |
+ |
void addGayBerneAtomType(const std::string& atomType); |
| 119 |
|
virtual void visit(Atom* atom) {} |
| 120 |
|
virtual void visit(DirectionalAtom* datom); |
| 121 |
|
virtual void visit(RigidBody* rb) {} |
| 126 |
|
std::set<std::string> linearAtomType; |
| 127 |
|
}; |
| 128 |
|
|
| 129 |
+ |
class GBLipidAtomVisitor : public BaseAtomVisitor{ |
| 130 |
+ |
public: |
| 131 |
+ |
GBLipidAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
| 132 |
+ |
visitorName = "GBLipidAtomVisitor"; |
| 133 |
+ |
GBLipidAtomType.insert("GBlipid"); |
| 134 |
+ |
} |
| 135 |
|
|
| 136 |
+ |
virtual void visit(Atom* atom) {} |
| 137 |
+ |
virtual void visit(DirectionalAtom* datom); |
| 138 |
+ |
virtual void visit(RigidBody* rb) {} |
| 139 |
|
|
| 140 |
+ |
virtual const std::string toString(); |
| 141 |
+ |
private: |
| 142 |
+ |
inline bool isGBLipidAtom(const std::string& atomType); |
| 143 |
+ |
std::set<std::string> GBLipidAtomType; |
| 144 |
+ |
}; |
| 145 |
|
|
| 146 |
+ |
|
| 147 |
|
class DefaultAtomVisitor : public BaseAtomVisitor{ |
| 148 |
|
public: |
| 149 |
|
DefaultAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "DefaultAtomVisitor";} |