1 |
< |
/* |
1 |
> |
/* |
2 |
|
* Copyright (c) 2005 The University of Notre Dame. All Rights Reserved. |
3 |
|
* |
4 |
|
* The University of Notre Dame grants you ("Licensee") a |
48 |
|
#include "visitors/BaseVisitor.hpp" |
49 |
|
#include "primitives/StuntDouble.hpp" |
50 |
|
#include "visitors/AtomData.hpp" |
51 |
+ |
#include "selection/SelectionManager.hpp" |
52 |
+ |
#include "selection/SelectionEvaluator.hpp" |
53 |
|
|
52 |
– |
|
54 |
|
namespace oopse { |
55 |
|
|
56 |
< |
class SimInfo; |
56 |
> |
class SimInfo; |
57 |
|
|
57 |
– |
//IgnoreVisitor will turn on the ignoring flag of the stuntdouble |
58 |
– |
class IgnoreVisitor : public BaseVisitor{ |
59 |
– |
public: |
60 |
– |
IgnoreVisitor(SimInfo* info) : BaseVisitor() {this->info = info; visitorName = "IgnoreVisitor";} |
58 |
|
|
59 |
< |
virtual void visit(Atom* atom); |
63 |
< |
virtual void visit(DirectionalAtom* datom); |
64 |
< |
virtual void visit(RigidBody* rb); |
65 |
< |
|
66 |
< |
virtual const std::string toString(); |
67 |
< |
|
68 |
< |
void addIgnoreType(const std::string& type) {itList.insert(type);} |
69 |
< |
|
70 |
< |
protected: |
71 |
< |
bool isIgnoreType(const std::string& name); |
72 |
< |
void internalVisit(StuntDouble* sd); |
73 |
< |
std::set<std::string> itList; //ignore type list; |
74 |
< |
SimInfo* info; |
75 |
< |
}; |
76 |
< |
|
77 |
< |
|
78 |
< |
class WrappingVisitor : public BaseVisitor{ |
59 |
> |
class WrappingVisitor : public BaseVisitor{ |
60 |
|
public: |
61 |
|
WrappingVisitor(SimInfo* info) : BaseVisitor() { |
62 |
|
this->info = info; |
71 |
|
protected: |
72 |
|
void internalVisit(StuntDouble* sd); |
73 |
|
SimInfo* info; |
74 |
< |
}; |
74 |
> |
}; |
75 |
|
|
76 |
|
|
77 |
< |
class ReplicateVisitor : public BaseVisitor{ |
77 |
> |
class ReplicateVisitor : public BaseVisitor{ |
78 |
|
public: |
79 |
|
ReplicateVisitor(SimInfo* info, Vector3i opt); |
80 |
|
virtual void visit(Atom* atom); |
90 |
|
std::vector<Vector3i> dir; |
91 |
|
SimInfo* info; |
92 |
|
Vector3i replicateOpt; |
93 |
< |
}; |
93 |
> |
}; |
94 |
|
|
95 |
< |
class XYZVisitor : public BaseVisitor{ |
95 |
> |
class XYZVisitor : public BaseVisitor{ |
96 |
|
public: |
116 |
– |
XYZVisitor(SimInfo* info, bool printDipole = true); |
97 |
|
|
98 |
+ |
XYZVisitor(SimInfo* info); |
99 |
+ |
|
100 |
+ |
XYZVisitor(SimInfo* info, const std::string& script); |
101 |
+ |
|
102 |
|
virtual void visit(Atom* atom); |
103 |
|
virtual void visit(DirectionalAtom* datom); |
104 |
|
virtual void visit(RigidBody* rb); |
105 |
|
|
106 |
+ |
virtual void update(); |
107 |
+ |
|
108 |
|
virtual const std::string toString(); |
109 |
|
|
110 |
|
void writeFrame(std::ostream& outStream); |
116 |
|
|
117 |
|
private: |
118 |
|
SimInfo* info; |
119 |
+ |
SelectionManager seleMan; |
120 |
+ |
SelectionEvaluator evaluator; |
121 |
|
std::vector<std::string> frame; |
134 |
– |
bool printDipole; |
135 |
– |
}; |
122 |
|
|
123 |
+ |
}; |
124 |
|
|
125 |
< |
class PrepareVisitor : public BaseVisitor{ |
125 |
> |
|
126 |
> |
class PrepareVisitor : public BaseVisitor{ |
127 |
|
public: |
128 |
|
PrepareVisitor() : BaseVisitor() {visitorName = "prepareVisitor";} |
129 |
|
|
136 |
|
protected: |
137 |
|
void internalVisit(Atom* atom); |
138 |
|
void internalVisit(RigidBody* rb); |
139 |
< |
}; |
139 |
> |
}; |
140 |
|
|
141 |
< |
class WaterTypeVisitor : public BaseVisitor{ |
141 |
> |
class WaterTypeVisitor : public BaseVisitor{ |
142 |
|
public: |
143 |
|
WaterTypeVisitor() ; |
144 |
|
virtual void visit(Atom* atom) {} |
145 |
|
virtual void visit(DirectionalAtom* datom) {} |
146 |
|
virtual void visit(RigidBody* rb); |
147 |
< |
|
147 |
> |
|
148 |
|
virtual const std::string toString(); |
149 |
|
|
150 |
|
private: |
151 |
< |
void replaceType(std::string& atomType); |
152 |
< |
|
151 |
> |
std::string trimmedName(const std::string& atomType); |
152 |
> |
|
153 |
|
std::set<std::string> waterTypeList; |
154 |
< |
}; |
154 |
> |
}; |
155 |
|
|
156 |
|
|
157 |
|
}//namespace oopse |