2 |
|
#include "AtomVisitor.hpp" |
3 |
|
#include "DirectionalAtom.hpp" |
4 |
|
#include "MatVec3.h" |
5 |
+ |
#include "RigidBody.hpp" |
6 |
|
|
7 |
+ |
void BaseAtomVisitor::visit(RigidBody* rb){ |
8 |
+ |
//vector<Atom*> myAtoms; |
9 |
+ |
//vector<Atom*>::iterator atomIter; |
10 |
+ |
|
11 |
+ |
//myAtoms = rb->getAtoms(); |
12 |
+ |
|
13 |
+ |
//for(atomIter = myAtoms.begin(); atomIter != myAtoms.end(); ++atomIter) |
14 |
+ |
// (*atomIter)->accept(this); |
15 |
+ |
} |
16 |
+ |
|
17 |
|
void BaseAtomVisitor::setVisited(Atom* atom){ |
18 |
|
GenericData* data; |
19 |
|
data = atom->getProperty("VISITED"); |
44 |
|
double ox[3] = {0.0, 0.0, -0.0654}; |
45 |
|
double u[3] = {0, 0, 1}; |
46 |
|
double rotMatrix[3][3]; |
47 |
+ |
double rotTrans[3][3]; |
48 |
|
AtomInfo* atomInfo; |
49 |
|
double pos[3]; |
38 |
– |
double vel[3]; |
50 |
|
double newVec[3]; |
51 |
|
double q[4]; |
52 |
|
AtomData* atomData; |
54 |
|
bool haveAtomData; |
55 |
|
|
56 |
|
//if atom is not SSD atom, just skip it |
57 |
< |
if(!strcmp(datom->getType(), "SSD")) |
57 |
> |
if(strcmp(datom->getType(), "SSD")) |
58 |
|
return; |
59 |
|
|
60 |
|
data = datom->getProperty("ATOMDATA"); |
78 |
|
datom->getPos(pos); |
79 |
|
datom->getQ(q); |
80 |
|
datom->getA(rotMatrix); |
81 |
+ |
|
82 |
+ |
// We need A^T to convert from body-fixed to space-fixed: |
83 |
+ |
transposeMat3(rotMatrix, rotTrans); |
84 |
|
|
85 |
|
//center of mass of the water molecule |
86 |
< |
matVecMul3(rotMatrix, u, newVec); |
86 |
> |
matVecMul3(rotTrans, u, newVec); |
87 |
|
atomInfo = new AtomInfo; |
88 |
|
atomInfo->AtomType = "X"; |
89 |
|
atomInfo->pos[0] = pos[0]; |
96 |
|
atomData->addAtomInfo(atomInfo); |
97 |
|
|
98 |
|
//oxygen |
99 |
< |
matVecMul3(rotMatrix, ox, newVec); |
99 |
> |
matVecMul3(rotTrans, ox, newVec); |
100 |
|
atomInfo = new AtomInfo; |
101 |
|
atomInfo->AtomType = "O"; |
102 |
|
atomInfo->pos[0] = pos[0] + newVec[0]; |
109 |
|
|
110 |
|
|
111 |
|
//hydrogen1 |
112 |
< |
matVecMul3(rotMatrix, h1, newVec); |
112 |
> |
matVecMul3(rotTrans, h1, newVec); |
113 |
|
atomInfo = new AtomInfo; |
114 |
|
atomInfo->AtomType = "H"; |
115 |
|
atomInfo->pos[0] = pos[0] + newVec[0]; |
121 |
|
atomData->addAtomInfo(atomInfo); |
122 |
|
|
123 |
|
//hydrogen2 |
124 |
< |
matVecMul3(rotMatrix, h2, newVec); |
124 |
> |
matVecMul3(rotTrans, h2, newVec); |
125 |
|
atomInfo = new AtomInfo; |
126 |
|
atomInfo->AtomType = "H"; |
127 |
|
atomInfo->pos[0] = pos[0] + newVec[0]; |
143 |
|
|
144 |
|
} |
145 |
|
|
146 |
+ |
const string SSDAtomVisitor::toString(){ |
147 |
+ |
char buffer[65535]; |
148 |
+ |
string result; |
149 |
+ |
|
150 |
+ |
sprintf(buffer ,"------------------------------------------------------------------\n"); |
151 |
+ |
result += buffer; |
152 |
+ |
|
153 |
+ |
sprintf(buffer ,"Visitor name: %s\n", visitorName.c_str()); |
154 |
+ |
result += buffer; |
155 |
+ |
|
156 |
+ |
sprintf(buffer , "Visitor Description: Convert SSD into 4 different atoms\n"); |
157 |
+ |
result += buffer; |
158 |
+ |
|
159 |
+ |
sprintf(buffer ,"------------------------------------------------------------------\n"); |
160 |
+ |
result += buffer; |
161 |
+ |
|
162 |
+ |
return result; |
163 |
+ |
} |
164 |
+ |
|
165 |
+ |
//----------------------------------------------------------------------------// |
166 |
+ |
|
167 |
|
void DefaultAtomVisitor::visit(Atom* atom){ |
168 |
|
AtomData* atomData; |
169 |
|
AtomInfo* atomInfo; |
173 |
|
return; |
174 |
|
|
175 |
|
atomInfo =new AtomInfo; |
176 |
+ |
|
177 |
+ |
atomData = new AtomData; |
178 |
+ |
atomData->setID("ATOMDATA"); |
179 |
|
|
180 |
|
atom->getPos(pos); |
181 |
|
atomInfo->AtomType = atom->getType(); |
186 |
|
atomInfo->dipole[1] = 0.0; |
187 |
|
atomInfo->dipole[2] = 0.0; |
188 |
|
|
189 |
< |
atomData = new AtomData; |
190 |
< |
atomData->setID("ATOMDATA"); |
189 |
> |
|
190 |
> |
atomData->addAtomInfo(atomInfo); |
191 |
> |
|
192 |
|
atom->addProperty(atomData); |
193 |
|
|
194 |
|
setVisited(atom); |
205 |
|
datom->getPos(pos); |
206 |
|
datom->getU(u); |
207 |
|
|
208 |
+ |
atomData = new AtomData; |
209 |
+ |
atomData->setID("ATOMDATA"); |
210 |
|
atomInfo =new AtomInfo; |
211 |
|
|
212 |
|
atomInfo->AtomType = datom->getType(); |
217 |
|
atomInfo->dipole[1] = u[1]; |
218 |
|
atomInfo->dipole[2] = u[2]; |
219 |
|
|
220 |
< |
atomData = new AtomData; |
221 |
< |
atomData->setID("ATOMDATA"); |
220 |
> |
atomData->addAtomInfo(atomInfo); |
221 |
> |
|
222 |
|
datom->addProperty(atomData); |
223 |
|
|
224 |
|
setVisited(datom); |
225 |
|
} |
226 |
< |
|
226 |
> |
|
227 |
> |
|
228 |
> |
const string DefaultAtomVisitor::toString(){ |
229 |
> |
char buffer[65535]; |
230 |
> |
string result; |
231 |
> |
|
232 |
> |
sprintf(buffer ,"------------------------------------------------------------------\n"); |
233 |
> |
result += buffer; |
234 |
> |
|
235 |
> |
sprintf(buffer ,"Visitor name: %s\n", visitorName.c_str()); |
236 |
> |
result += buffer; |
237 |
> |
|
238 |
> |
sprintf(buffer , "Visitor Description: copy atom infomation into atom data\n"); |
239 |
> |
result += buffer; |
240 |
> |
|
241 |
> |
sprintf(buffer ,"------------------------------------------------------------------\n"); |
242 |
> |
result += buffer; |
243 |
> |
|
244 |
> |
return result; |
245 |
> |
} |