105 |
|
Atom * atom; |
106 |
|
|
107 |
|
DataStorage& data = snapshotMan_->getCurrentSnapshot()->*storage_; |
108 |
+ |
int storageLayout = data.getStorageLayout(); |
109 |
|
|
110 |
|
totalMass = getMass(); |
111 |
|
|
118 |
|
} |
119 |
|
data.position[localIndex_] /= totalMass; |
120 |
|
} |
121 |
+ |
|
122 |
+ |
if (storageLayout & DataStorage::dslVelocity) { |
123 |
+ |
if (cutoffAtomList.size() == 1) { |
124 |
+ |
data.velocity[localIndex_] = beginAtom(i)->getVel(); |
125 |
+ |
} else { |
126 |
+ |
data.velocity[localIndex_] = V3Zero; |
127 |
+ |
for(atom = beginAtom(i); atom != NULL; atom = nextAtom(i)) { |
128 |
+ |
data.velocity[localIndex_] += atom->getMass() * atom->getVel(); |
129 |
+ |
} |
130 |
+ |
data.velocity[localIndex_] /= totalMass; |
131 |
+ |
} |
132 |
+ |
} |
133 |
|
} |
134 |
|
|
135 |
|
|
136 |
|
Vector3d getPos() { |
137 |
|
return ((snapshotMan_->getCurrentSnapshot())->*storage_).position[localIndex_]; |
138 |
|
} |
139 |
+ |
Vector3d getVel() { |
140 |
+ |
return ((snapshotMan_->getCurrentSnapshot())->*storage_).velocity[localIndex_]; |
141 |
+ |
} |
142 |
|
|
143 |
|
int getNumAtom() { |
144 |
|
return cutoffAtomList.size(); |