ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/branches/development/src/primitives/CutoffGroup.hpp
(Generate patch)

Comparing branches/development/src/primitives/CutoffGroup.hpp (file contents):
Revision 1722 by gezelter, Tue Nov 22 20:38:56 2011 UTC vs.
Revision 1723 by gezelter, Thu May 24 20:59:54 2012 UTC

# Line 105 | Line 105 | namespace OpenMD {
105        Atom * atom;
106  
107        DataStorage&  data = snapshotMan_->getCurrentSnapshot()->*storage_;
108 +      int storageLayout = data.getStorageLayout();
109      
110        totalMass = getMass();
111        
# Line 117 | Line 118 | namespace OpenMD {
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();

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines