17 |
|
virtual void visit(DirectionalAtom* datom); |
18 |
|
virtual void visit(RigidBody* rb); |
19 |
|
|
20 |
– |
void addIgnoreType(const string& type) {itList.insert(type);} |
20 |
|
virtual const string toString(); |
21 |
+ |
|
22 |
+ |
void addIgnoreType(const string& type) {itList.insert(type);} |
23 |
|
|
24 |
|
protected: |
25 |
|
bool isIgnoreType(const string& name); |
37 |
|
virtual void visit(Atom* atom); |
38 |
|
virtual void visit(DirectionalAtom* datom); |
39 |
|
virtual void visit(RigidBody* rb); |
40 |
< |
|
40 |
> |
|
41 |
> |
virtual const string toString(); |
42 |
> |
|
43 |
|
protected: |
44 |
|
void internalVisit(StuntDouble* sd); |
45 |
|
SimInfo* info; |
55 |
|
vec[2] = k; |
56 |
|
} |
57 |
|
|
58 |
< |
double vec[3]; |
59 |
< |
double& operator[](int index) {return vec[index];} |
58 |
> |
int vec[3]; |
59 |
> |
int& operator[](int index) {return vec[index];} |
60 |
|
}; |
61 |
|
|
62 |
|
class ReplicateVisitor : public BaseVisitor{ |
63 |
|
public: |
64 |
< |
ReplicateVisitor(SimInfo* info, IntVec3 replicateOpt); |
64 |
> |
ReplicateVisitor(SimInfo* info, IntVec3 opt); |
65 |
|
virtual void visit(Atom* atom); |
66 |
|
virtual void visit(DirectionalAtom* datom); |
67 |
|
virtual void visit(RigidBody* rb); |
84 |
|
virtual void visit(Atom* atom); |
85 |
|
virtual void visit(DirectionalAtom* datom); |
86 |
|
virtual void visit(RigidBody* rb); |
87 |
+ |
|
88 |
+ |
virtual const string toString(); |
89 |
|
|
90 |
|
void writeFrame(ostream& outStream); |
91 |
|
void clear() {frame.clear();} |
106 |
|
PrepareVisitor() : BaseVisitor() {visitorName = "prepareVisitor";} |
107 |
|
|
108 |
|
virtual void visit(Atom* atom) {internalVisit(atom);} |
109 |
< |
virtual void visit(DirectionalAtom* datom) {internalVisit(datom);} |
109 |
> |
virtual void visit(DirectionalAtom* datom) {internalVisit((Atom*)datom);} |
110 |
|
virtual void visit(RigidBody* rb) {internalVisit(rb);} |
111 |
|
|
112 |
+ |
virtual const string toString(); |
113 |
+ |
|
114 |
|
protected: |
115 |
< |
void internalVisit(StuntDouble* sd); |
115 |
> |
void internalVisit(Atom* atom); |
116 |
> |
void internalVisit(RigidBody* rb); |
117 |
|
}; |
118 |
|
|
119 |
+ |
class WaterTypeVisitor : public BaseVisitor{ |
120 |
+ |
public: |
121 |
+ |
WaterTypeVisitor() ; |
122 |
+ |
virtual void visit(Atom* atom) {} |
123 |
+ |
virtual void visit(DirectionalAtom* datom) {} |
124 |
+ |
virtual void visit(RigidBody* rb); |
125 |
|
|
126 |
+ |
virtual const string toString(); |
127 |
+ |
|
128 |
+ |
private: |
129 |
+ |
void replaceType(string& atomType); |
130 |
+ |
|
131 |
+ |
set<string> waterTypeList; |
132 |
+ |
}; |
133 |
|
#endif //_OTHERVISITOR_H_ |