ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/trunk/src/applications/staticProps/RNEMDStats.cpp
Revision: 1944
Committed: Wed Nov 6 18:30:25 2013 UTC (11 years, 5 months ago) by gezelter
File size: 11203 byte(s)
Log Message:
Fixed a few bugs in the analysis code for RNEMD.

File Contents

# User Rev Content
1 gezelter 1865 /*
2     * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
3     *
4     * The University of Notre Dame grants you ("Licensee") a
5     * non-exclusive, royalty free, license to use, modify and
6     * redistribute this software in source and binary code form, provided
7     * that the following conditions are met:
8     *
9     * 1. Redistributions of source code must retain the above copyright
10     * notice, this list of conditions and the following disclaimer.
11     *
12     * 2. Redistributions in binary form must reproduce the above copyright
13     * notice, this list of conditions and the following disclaimer in the
14     * documentation and/or other materials provided with the
15     * distribution.
16     *
17     * This software is provided "AS IS," without a warranty of any
18     * kind. All express or implied conditions, representations and
19     * warranties, including any implied warranty of merchantability,
20     * fitness for a particular purpose or non-infringement, are hereby
21     * excluded. The University of Notre Dame and its licensors shall not
22     * be liable for any damages suffered by licensee as a result of
23     * using, modifying or distributing the software or its
24     * derivatives. In no event will the University of Notre Dame or its
25     * licensors be liable for any lost revenue, profit or data, or for
26     * direct, indirect, special, consequential, incidental or punitive
27     * damages, however caused and regardless of the theory of liability,
28     * arising out of the use of or inability to use software, even if the
29     * University of Notre Dame has been advised of the possibility of
30     * such damages.
31     *
32     * SUPPORT OPEN SCIENCE! If you use OpenMD or its source code in your
33     * research, please cite the appropriate papers when you publish your
34     * work. Good starting points are:
35     *
36     * [1] Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).
37     * [2] Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).
38     * [3] Sun, Lin & Gezelter, J. Chem. Phys. 128, 234107 (2008).
39     * [4] Kuang & Gezelter, J. Chem. Phys. 133, 164101 (2010).
40     */
41    
42    
43     #include <algorithm>
44     #include <fstream>
45     #include "applications/staticProps/RNEMDStats.hpp"
46 gezelter 1881 #include "primitives/Molecule.hpp"
47 gezelter 1865 #include "utils/PhysicalConstants.hpp"
48    
49     namespace OpenMD {
50    
51     RNEMDZ::RNEMDZ(SimInfo* info, const std::string& filename,
52     const std::string& sele, int nzbins)
53     : SlabStatistics(info, filename, sele, nzbins) {
54    
55     setOutputName(getPrefix(filename) + ".rnemdZ");
56    
57     temperature = new OutputData;
58     temperature->units = "K";
59     temperature->title = "Temperature";
60     temperature->dataType = odtReal;
61     temperature->dataHandling = odhAverage;
62     temperature->accumulator.reserve(nBins_);
63     for (int i = 0; i < nBins_; i++)
64     temperature->accumulator.push_back( new Accumulator() );
65     data_.push_back(temperature);
66    
67     velocity = new OutputData;
68     velocity->units = "angstroms/fs";
69     velocity->title = "Velocity";
70     velocity->dataType = odtVector3;
71     velocity->dataHandling = odhAverage;
72     velocity->accumulator.reserve(nBins_);
73     for (int i = 0; i < nBins_; i++)
74     velocity->accumulator.push_back( new VectorAccumulator() );
75     data_.push_back(velocity);
76    
77     density = new OutputData;
78     density->units = "g cm^-3";
79     density->title = "Density";
80     density->dataType = odtReal;
81     density->dataHandling = odhAverage;
82     density->accumulator.reserve(nBins_);
83     for (int i = 0; i < nBins_; i++)
84     density->accumulator.push_back( new Accumulator() );
85     data_.push_back(density);
86     }
87    
88 gezelter 1881 void RNEMDZ::processFrame(int istep) {
89 gezelter 1884 RealType z;
90    
91     hmat_ = currentSnapshot_->getHmat();
92     for (int i = 0; i < nBins_; i++) {
93     z = (((RealType)i + 0.5) / (RealType)nBins_) * hmat_(2,2);
94     dynamic_cast<Accumulator*>(z_->accumulator[i])->add(z);
95     }
96     volume_ = currentSnapshot_->getVolume();
97    
98    
99 gezelter 1881 Molecule* mol;
100     RigidBody* rb;
101     StuntDouble* sd;
102     SimInfo::MoleculeIterator mi;
103     Molecule::RigidBodyIterator rbIter;
104     int i;
105 gezelter 1865
106 gezelter 1881 vector<RealType> binMass(nBins_, 0.0);
107 gezelter 1944 vector<Vector3d> binP(nBins_, V3Zero);
108 gezelter 1881 vector<RealType> binKE(nBins_, 0.0);
109 gezelter 1883 vector<unsigned int> binDof(nBins_, 0);
110 gezelter 1881
111     for (mol = info_->beginMolecule(mi); mol != NULL;
112     mol = info_->nextMolecule(mi)) {
113    
114     // change the positions of atoms which belong to the rigidbodies
115    
116     for (rb = mol->beginRigidBody(rbIter); rb != NULL;
117     rb = mol->nextRigidBody(rbIter)) {
118 gezelter 1892 rb->updateAtomVel();
119 gezelter 1865 }
120     }
121 gezelter 1884
122 gezelter 1881 if (evaluator_.isDynamic()) {
123     seleMan_.setSelectionSet(evaluator_.evaluate());
124     }
125 gezelter 1865
126 gezelter 1881 // loop over the selected atoms:
127    
128     for (sd = seleMan_.beginSelected(i); sd != NULL;
129     sd = seleMan_.nextSelected(i)) {
130    
131     // figure out where that object is:
132     Vector3d pos = sd->getPos();
133 gezelter 1883 Vector3d vel = sd->getVel();
134     RealType m = sd->getMass();
135    
136 gezelter 1881 int bin = getBin(pos);
137 gezelter 1884
138 gezelter 1881 binMass[bin] += m;
139 gezelter 1944 binP[bin] += m * vel;
140 gezelter 1881 binKE[bin] += 0.5 * (m * vel.lengthSquare());
141     binDof[bin] += 3;
142    
143     if (sd->isDirectional()) {
144     Vector3d angMom = sd->getJ();
145     Mat3x3d I = sd->getI();
146     if (sd->isLinear()) {
147     int i = sd->linearAxis();
148     int j = (i + 1) % 3;
149     int k = (i + 2) % 3;
150     binKE[bin] += 0.5 * (angMom[j] * angMom[j] / I(j, j) +
151     angMom[k] * angMom[k] / I(k, k));
152     binDof[bin] += 2;
153     } else {
154     binKE[bin] += 0.5 * (angMom[0] * angMom[0] / I(0, 0) +
155     angMom[1] * angMom[1] / I(1, 1) +
156     angMom[2] * angMom[2] / I(2, 2));
157     binDof[bin] += 3;
158     }
159     }
160     }
161    
162 gezelter 1883 for (unsigned int i = 0; i < nBins_; i++) {
163 gezelter 1885
164 gezelter 1882 if (binDof[i] > 0) {
165     RealType temp = 2.0 * binKE[i] / (binDof[i] * PhysicalConstants::kb *
166     PhysicalConstants::energyConvert);
167     RealType den = binMass[i] * nBins_ * PhysicalConstants::densityConvert
168     / volume_;
169 gezelter 1944 Vector3d vel = binP[i] / binMass[i];
170 gezelter 1892
171 gezelter 1882 dynamic_cast<Accumulator *>(temperature->accumulator[i])->add(temp);
172     dynamic_cast<VectorAccumulator *>(velocity->accumulator[i])->add(vel);
173     dynamic_cast<Accumulator *>(density->accumulator[i])->add(den);
174     dynamic_cast<Accumulator *>(counts_->accumulator[i])->add(1);
175     }
176 gezelter 1881 }
177 gezelter 1865 }
178 gezelter 1881
179     void RNEMDZ::processStuntDouble(StuntDouble* sd, int bin) {
180     }
181 gezelter 1865
182     RNEMDR::RNEMDR(SimInfo* info, const std::string& filename,
183     const std::string& sele, int nrbins)
184     : ShellStatistics(info, filename, sele, nrbins) {
185    
186    
187     setOutputName(getPrefix(filename) + ".rnemdR");
188    
189     temperature = new OutputData;
190     temperature->units = "K";
191     temperature->title = "Temperature";
192     temperature->dataType = odtReal;
193     temperature->dataHandling = odhAverage;
194     temperature->accumulator.reserve(nBins_);
195     for (int i = 0; i < nBins_; i++)
196     temperature->accumulator.push_back( new Accumulator() );
197     data_.push_back(temperature);
198    
199     angularVelocity = new OutputData;
200     angularVelocity->units = "angstroms^2/fs";
201     angularVelocity->title = "Velocity";
202     angularVelocity->dataType = odtVector3;
203     angularVelocity->dataHandling = odhAverage;
204     angularVelocity->accumulator.reserve(nBins_);
205     for (int i = 0; i < nBins_; i++)
206     angularVelocity->accumulator.push_back( new VectorAccumulator() );
207     data_.push_back(angularVelocity);
208    
209     density = new OutputData;
210     density->units = "g cm^-3";
211     density->title = "Density";
212     density->dataType = odtReal;
213     density->dataHandling = odhAverage;
214     density->accumulator.reserve(nBins_);
215     for (int i = 0; i < nBins_; i++)
216     density->accumulator.push_back( new Accumulator() );
217     data_.push_back(density);
218     }
219    
220    
221 gezelter 1887 void RNEMDR::processFrame(int istep) {
222 gezelter 1865
223 gezelter 1887 Molecule* mol;
224     RigidBody* rb;
225     StuntDouble* sd;
226     SimInfo::MoleculeIterator mi;
227     Molecule::RigidBodyIterator rbIter;
228     int i;
229    
230     vector<RealType> binMass(nBins_, 0.0);
231 gezelter 1944 vector<Mat3x3d> binI(nBins_);
232     vector<Vector3d> binL(nBins_, V3Zero);
233 gezelter 1887 vector<RealType> binKE(nBins_, 0.0);
234 gezelter 1944 vector<int> binDof(nBins_, 0);
235 gezelter 1887
236     for (mol = info_->beginMolecule(mi); mol != NULL;
237     mol = info_->nextMolecule(mi)) {
238    
239     // change the positions of atoms which belong to the rigidbodies
240    
241     for (rb = mol->beginRigidBody(rbIter); rb != NULL;
242     rb = mol->nextRigidBody(rbIter)) {
243 gezelter 1892 rb->updateAtomVel();
244 gezelter 1865 }
245     }
246 gezelter 1887
247     if (evaluator_.isDynamic()) {
248     seleMan_.setSelectionSet(evaluator_.evaluate());
249     }
250 gezelter 1865
251 gezelter 1887 // loop over the selected atoms:
252    
253     for (sd = seleMan_.beginSelected(i); sd != NULL;
254     sd = seleMan_.nextSelected(i)) {
255 gezelter 1944
256 gezelter 1887 // figure out where that object is:
257 gezelter 1944 int bin = getBin(sd->getPos() );
258 gezelter 1865
259 gezelter 1944 if (bin >= 0 && bin < nBins_) {
260 gezelter 1887
261 gezelter 1944 Vector3d rPos = sd->getPos() - coordinateOrigin_;
262     Vector3d vel = sd->getVel();
263     RealType m = sd->getMass();
264     Vector3d L = m * cross(rPos, vel);
265     Mat3x3d I(0.0);
266     I = outProduct(rPos, rPos) * m;
267     RealType r2 = rPos.lengthSquare();
268     I(0, 0) += m * r2;
269     I(1, 1) += m * r2;
270     I(2, 2) += m * r2;
271 gezelter 1887
272 gezelter 1944 binMass[bin] += m;
273     binI[bin] += I;
274     binL[bin] += L;
275     binKE[bin] += 0.5 * (m * vel.lengthSquare());
276     binDof[bin] += 3;
277    
278     if (sd->isDirectional()) {
279     Vector3d angMom = sd->getJ();
280     Mat3x3d Ia = sd->getI();
281     if (sd->isLinear()) {
282     int i = sd->linearAxis();
283     int j = (i + 1) % 3;
284     int k = (i + 2) % 3;
285     binKE[bin] += 0.5 * (angMom[j] * angMom[j] / Ia(j, j) +
286     angMom[k] * angMom[k] / Ia(k, k));
287     binDof[bin] += 2;
288     } else {
289     binKE[bin] += 0.5 * (angMom[0] * angMom[0] / Ia(0, 0) +
290     angMom[1] * angMom[1] / Ia(1, 1) +
291     angMom[2] * angMom[2] / Ia(2, 2));
292     binDof[bin] += 3;
293     }
294 gezelter 1887 }
295     }
296     }
297 gezelter 1865
298 gezelter 1887 for (unsigned int i = 0; i < nBins_; i++) {
299     RealType rinner = (RealType)i * binWidth_;
300     RealType router = (RealType)(i+1) * binWidth_;
301     if (binDof[i] > 0) {
302     RealType temp = 2.0 * binKE[i] / (binDof[i] * PhysicalConstants::kb *
303     PhysicalConstants::energyConvert);
304     RealType den = binMass[i] * 3.0 * PhysicalConstants::densityConvert
305 gezelter 1944 / (4.0 * M_PI * (pow(router,3) - pow(rinner,3)));
306    
307     Vector3d omega = binI[i].inverse() * binL[i];
308    
309 gezelter 1887 dynamic_cast<Accumulator *>(temperature->accumulator[i])->add(temp);
310 gezelter 1944 dynamic_cast<VectorAccumulator *>(angularVelocity->accumulator[i])->add(omega);
311 gezelter 1887 dynamic_cast<Accumulator *>(density->accumulator[i])->add(den);
312     dynamic_cast<Accumulator *>(counts_->accumulator[i])->add(1);
313     }
314     }
315     }
316 gezelter 1865
317 gezelter 1887
318     void RNEMDR::processStuntDouble(StuntDouble* sd, int bin) {
319 gezelter 1865 }
320     }
321    

Properties

Name Value
svn:eol-style native