| 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); |
| 143 |
|
std::set<std::string> GBLipidAtomType; |
| 144 |
|
}; |
| 145 |
|
|
| 146 |
+ |
class Ring5gbAtomVisitor : public BaseAtomVisitor{ |
| 147 |
+ |
public: |
| 148 |
+ |
Ring5gbAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
| 149 |
+ |
visitorName = "Ring5gbAtomVisitor"; |
| 150 |
+ |
Ring5gbAtomType.insert("Ring5GB"); |
| 151 |
+ |
} |
| 152 |
|
|
| 153 |
+ |
virtual void visit(Atom* atom) {} |
| 154 |
+ |
virtual void visit(DirectionalAtom* datom); |
| 155 |
+ |
virtual void visit(RigidBody* rb) {} |
| 156 |
+ |
|
| 157 |
+ |
virtual const std::string toString(); |
| 158 |
+ |
private: |
| 159 |
+ |
inline bool isRing5gbAtom(const std::string& atomType); |
| 160 |
+ |
std::set<std::string> Ring5gbAtomType; |
| 161 |
+ |
}; |
| 162 |
+ |
|
| 163 |
+ |
class HeadAtomVisitor : public BaseAtomVisitor{ |
| 164 |
+ |
public: |
| 165 |
+ |
HeadAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { |
| 166 |
+ |
visitorName = "HeadAtomVisitor"; |
| 167 |
+ |
HeadAtomType.insert("HEAD"); |
| 168 |
+ |
} |
| 169 |
+ |
|
| 170 |
+ |
virtual void visit(Atom* atom) {} |
| 171 |
+ |
virtual void visit(DirectionalAtom* datom); |
| 172 |
+ |
virtual void visit(RigidBody* rb) {} |
| 173 |
+ |
|
| 174 |
+ |
virtual const std::string toString(); |
| 175 |
+ |
private: |
| 176 |
+ |
inline bool isHeadAtom(const std::string& atomType); |
| 177 |
+ |
std::set<std::string> HeadAtomType; |
| 178 |
+ |
}; |
| 179 |
+ |
|
| 180 |
|
class DefaultAtomVisitor : public BaseAtomVisitor{ |
| 181 |
|
public: |
| 182 |
|
DefaultAtomVisitor(SimInfo* info) : BaseAtomVisitor(info) { visitorName = "DefaultAtomVisitor";} |