45 |
|
#include "primitives/Molecule.hpp" |
46 |
|
#include "brains/SimInfo.hpp" |
47 |
|
namespace oopse { |
48 |
– |
|
49 |
– |
//----------------------------------------------------------------------------// |
50 |
– |
void IgnoreVisitor::visit(Atom *atom) { |
51 |
– |
if (isIgnoreType(atom->getType())) |
52 |
– |
internalVisit(atom); |
53 |
– |
} |
54 |
– |
|
55 |
– |
void IgnoreVisitor::visit(DirectionalAtom *datom) { |
56 |
– |
if (isIgnoreType(datom->getType())) |
57 |
– |
internalVisit(datom); |
58 |
– |
} |
59 |
– |
|
60 |
– |
void IgnoreVisitor::visit(RigidBody *rb) { |
61 |
– |
std::vector<Atom *> myAtoms; |
62 |
– |
std::vector<Atom *>::iterator atomIter; |
63 |
– |
AtomInfo * atomInfo; |
64 |
– |
|
65 |
– |
if (isIgnoreType(rb->getType())) { |
66 |
– |
internalVisit(rb); |
67 |
– |
|
68 |
– |
myAtoms = rb->getAtoms(); |
69 |
– |
|
70 |
– |
for( atomIter = myAtoms.begin(); atomIter != myAtoms.end(); ++atomIter ) |
71 |
– |
internalVisit(*atomIter); |
72 |
– |
} |
73 |
– |
} |
74 |
– |
|
75 |
– |
bool IgnoreVisitor::isIgnoreType(const std::string&name) { |
76 |
– |
return itList.find(name) != itList.end() ? true : false; |
77 |
– |
} |
78 |
– |
|
79 |
– |
void IgnoreVisitor::internalVisit(StuntDouble *sd) { |
80 |
– |
info->getSelectionManager()->clearSelection(sd); |
81 |
– |
//GenericData *data; |
82 |
– |
//data = sd->getPropertyByName("IGNORE"); |
83 |
– |
// |
84 |
– |
////if this stuntdoulbe is already marked as ignore just skip it |
85 |
– |
//if (data == NULL) { |
86 |
– |
// data = new GenericData; |
87 |
– |
// data->setID("IGNORE"); |
88 |
– |
// sd->addProperty(data); |
89 |
– |
//} |
90 |
– |
} |
91 |
– |
|
92 |
– |
const std::string IgnoreVisitor::toString() { |
93 |
– |
char buffer[65535]; |
94 |
– |
std::string result; |
95 |
– |
std::set<std::string>::iterator i; |
96 |
– |
|
97 |
– |
sprintf(buffer, |
98 |
– |
"------------------------------------------------------------------\n"); |
99 |
– |
result += buffer; |
100 |
– |
|
101 |
– |
sprintf(buffer, "Visitor name: %s\n", visitorName.c_str()); |
102 |
– |
result += buffer; |
103 |
– |
|
104 |
– |
sprintf(buffer, "Visitor Description: ignore stuntdoubles\n"); |
105 |
– |
result += buffer; |
106 |
– |
|
107 |
– |
//print the ignore type list |
108 |
– |
sprintf(buffer, "Ignore type list contains below types:\n"); |
109 |
– |
result += buffer; |
110 |
– |
|
111 |
– |
for( i = itList.begin(); i != itList.end(); ++i ) { |
112 |
– |
sprintf(buffer, "%s\t", i->c_str()); |
113 |
– |
result += buffer; |
114 |
– |
} |
48 |
|
|
116 |
– |
sprintf(buffer, "\n"); |
117 |
– |
result += buffer; |
118 |
– |
|
119 |
– |
sprintf(buffer, |
120 |
– |
"------------------------------------------------------------------\n"); |
121 |
– |
result += buffer; |
122 |
– |
|
123 |
– |
return result; |
124 |
– |
} |
125 |
– |
|
126 |
– |
//----------------------------------------------------------------------------// |
127 |
– |
|
49 |
|
void WrappingVisitor::visit(Atom *atom) { |
50 |
|
internalVisit(atom); |
51 |
|
} |
217 |
|
|
218 |
|
//----------------------------------------------------------------------------// |
219 |
|
|
220 |
< |
XYZVisitor::XYZVisitor(SimInfo *info, bool printDipole) : |
221 |
< |
BaseVisitor() { |
220 |
> |
XYZVisitor::XYZVisitor(SimInfo *info) : |
221 |
> |
BaseVisitor(), seleMan(info), evaluator(info){ |
222 |
|
this->info = info; |
223 |
|
visitorName = "XYZVisitor"; |
224 |
< |
this->printDipole = printDipole; |
224 |
> |
|
225 |
> |
evaluator.loadScriptString("select all"); |
226 |
> |
|
227 |
> |
if (!evaluator.isDynamic()) { |
228 |
> |
seleMan.setSelectionSet(evaluator.evaluate()); |
229 |
> |
} |
230 |
> |
|
231 |
|
} |
232 |
+ |
|
233 |
+ |
XYZVisitor::XYZVisitor(SimInfo *info, const std::string& script) : |
234 |
+ |
BaseVisitor(), seleMan(info), evaluator(info) { |
235 |
+ |
this->info = info; |
236 |
+ |
visitorName = "XYZVisitor"; |
237 |
|
|
238 |
+ |
evaluator.loadScriptString(script); |
239 |
+ |
|
240 |
+ |
if (!evaluator.isDynamic()) { |
241 |
+ |
seleMan.setSelectionSet(evaluator.evaluate()); |
242 |
+ |
} |
243 |
+ |
|
244 |
+ |
} |
245 |
+ |
|
246 |
|
void XYZVisitor::visit(Atom *atom) { |
247 |
|
if (isSelected(atom)) |
248 |
|
internalVisit(atom); |
256 |
|
void XYZVisitor::visit(RigidBody *rb) { |
257 |
|
if (isSelected(rb)) |
258 |
|
internalVisit(rb); |
259 |
+ |
} |
260 |
+ |
|
261 |
+ |
void XYZVisitor::update() { |
262 |
+ |
//if dynamic, we need to re-evaluate the selection |
263 |
+ |
if (evaluator.isDynamic()) { |
264 |
+ |
seleMan.setSelectionSet(evaluator.evaluate()); |
265 |
+ |
} |
266 |
|
} |
267 |
|
|
268 |
|
void XYZVisitor::internalVisit(StuntDouble *sd) { |
285 |
|
|
286 |
|
for( atomInfo = atomData->beginAtomInfo(i); atomInfo; |
287 |
|
atomInfo = atomData->nextAtomInfo(i) ) { |
341 |
– |
printf("SD type is %s\n", sd->getType().c_str()); |
342 |
– |
printf("XYZVisitor thinks %s\n", atomInfo->atomTypeName.c_str()); |
343 |
– |
if (printDipole) { |
288 |
|
sprintf(buffer, |
289 |
|
"%s%15.8f%15.8f%15.8f%15.8f%15.8f%15.8f", |
290 |
|
atomInfo->atomTypeName.c_str(), |
294 |
|
atomInfo->dipole[0], |
295 |
|
atomInfo->dipole[1], |
296 |
|
atomInfo->dipole[2]); |
353 |
– |
} else { |
354 |
– |
sprintf(buffer, "%s%15.8f%15.8f%15.8f", |
355 |
– |
atomInfo->atomTypeName.c_str(), atomInfo->pos[0], |
356 |
– |
atomInfo->pos[1], atomInfo->pos[2]); |
357 |
– |
} |
358 |
– |
|
297 |
|
frame.push_back(buffer); |
298 |
|
} |
299 |
|
} |
300 |
|
|
301 |
|
bool XYZVisitor::isSelected(StuntDouble *sd) { |
302 |
< |
return info->getSelectionManager()->isSelected(sd); |
302 |
> |
return seleMan.isSelected(sd); |
303 |
|
} |
304 |
|
|
305 |
|
void XYZVisitor::writeFrame(std::ostream &outStream) { |