ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/branches/development/src/integrators/NPT.cpp
Revision: 1665
Committed: Tue Nov 22 20:38:56 2011 UTC (13 years, 5 months ago) by gezelter
File size: 10866 byte(s)
Log Message:
updated copyright notices

File Contents

# User Rev Content
1 gezelter 507 /*
2 gezelter 246 * 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 gezelter 1390 * 1. Redistributions of source code must retain the above copyright
10 gezelter 246 * notice, this list of conditions and the following disclaimer.
11     *
12 gezelter 1390 * 2. Redistributions in binary form must reproduce the above copyright
13 gezelter 246 * 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 gezelter 1390 *
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, 24107 (2008).
39 gezelter 1665 * [4] Kuang & Gezelter, J. Chem. Phys. 133, 164101 (2010).
40     * [5] Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
41 gezelter 246 */
42    
43 gezelter 2 #include <math.h>
44    
45 tim 3 #include "brains/SimInfo.hpp"
46     #include "brains/Thermo.hpp"
47 gezelter 246 #include "integrators/NPT.hpp"
48     #include "math/SquareMatrix3.hpp"
49     #include "primitives/Molecule.hpp"
50 gezelter 1390 #include "utils/PhysicalConstants.hpp"
51 tim 3 #include "utils/simError.h"
52 gezelter 2
53     // Basic isotropic thermostating and barostating via the Melchionna
54     // modification of the Hoover algorithm:
55     //
56     // Melchionna, S., Ciccotti, G., and Holian, B. L., 1993,
57     // Molec. Phys., 78, 533.
58     //
59     // and
60     //
61     // Hoover, W. G., 1986, Phys. Rev. A, 34, 2499.
62    
63 gezelter 1390 namespace OpenMD {
64 gezelter 2
65 gezelter 507 NPT::NPT(SimInfo* info) :
66 gezelter 246 VelocityVerletIntegrator(info), chiTolerance(1e-6), etaTolerance(1e-6), maxIterNum_(4) {
67 gezelter 2
68 gezelter 507 Globals* simParams = info_->getSimParams();
69 gezelter 246
70 tim 665 if (!simParams->getUseIntialExtendedSystemState()) {
71 gezelter 246 Snapshot* currSnapshot = info_->getSnapshotManager()->getCurrentSnapshot();
72     currSnapshot->setChi(0.0);
73     currSnapshot->setIntegralOfChiDt(0.0);
74     currSnapshot->setEta(Mat3x3d(0.0));
75 gezelter 507 }
76 gezelter 246
77 gezelter 507 if (!simParams->haveTargetTemp()) {
78 gezelter 246 sprintf(painCave.errMsg, "You can't use the NVT integrator without a targetTemp!\n");
79     painCave.isFatal = 1;
80 gezelter 1390 painCave.severity = OPENMD_ERROR;
81 gezelter 246 simError();
82 gezelter 507 } else {
83 gezelter 246 targetTemp = simParams->getTargetTemp();
84 gezelter 507 }
85 gezelter 2
86 gezelter 507 // We must set tauThermostat
87     if (!simParams->haveTauThermostat()) {
88 gezelter 246 sprintf(painCave.errMsg, "If you use the constant temperature\n"
89 gezelter 1277 "\tintegrator, you must set tauThermostat.\n");
90 gezelter 2
91 gezelter 1390 painCave.severity = OPENMD_ERROR;
92 gezelter 246 painCave.isFatal = 1;
93     simError();
94 gezelter 507 } else {
95 gezelter 246 tauThermostat = simParams->getTauThermostat();
96 gezelter 507 }
97 gezelter 2
98 gezelter 507 if (!simParams->haveTargetPressure()) {
99 gezelter 246 sprintf(painCave.errMsg, "NPT error: You can't use the NPT integrator\n"
100 gezelter 507 " without a targetPressure!\n");
101 gezelter 2
102 gezelter 246 painCave.isFatal = 1;
103     simError();
104 gezelter 507 } else {
105 gezelter 246 targetPressure = simParams->getTargetPressure();
106 gezelter 507 }
107 gezelter 246
108 gezelter 507 if (!simParams->haveTauBarostat()) {
109 gezelter 246 sprintf(painCave.errMsg,
110     "If you use the NPT integrator, you must set tauBarostat.\n");
111 gezelter 1390 painCave.severity = OPENMD_ERROR;
112 gezelter 246 painCave.isFatal = 1;
113     simError();
114 gezelter 507 } else {
115 gezelter 246 tauBarostat = simParams->getTauBarostat();
116 gezelter 507 }
117 gezelter 246
118 gezelter 507 tt2 = tauThermostat * tauThermostat;
119     tb2 = tauBarostat * tauBarostat;
120 gezelter 2
121 gezelter 507 update();
122     }
123 gezelter 2
124 gezelter 507 NPT::~NPT() {
125     }
126 gezelter 2
127 gezelter 507 void NPT::doUpdate() {
128 gezelter 2
129 gezelter 246 oldPos.resize(info_->getNIntegrableObjects());
130     oldVel.resize(info_->getNIntegrableObjects());
131     oldJi.resize(info_->getNIntegrableObjects());
132 gezelter 2
133 gezelter 507 }
134 gezelter 2
135 gezelter 507 void NPT::moveA() {
136 gezelter 246 SimInfo::MoleculeIterator i;
137     Molecule::IntegrableObjectIterator j;
138     Molecule* mol;
139     StuntDouble* integrableObject;
140     Vector3d Tb, ji;
141 tim 963 RealType mass;
142 gezelter 246 Vector3d vel;
143     Vector3d pos;
144     Vector3d frc;
145     Vector3d sc;
146     int index;
147 gezelter 2
148 gezelter 246 chi= currentSnapshot_->getChi();
149     integralOfChidt = currentSnapshot_->getIntegralOfChiDt();
150     loadEta();
151    
152     instaTemp =thermo.getTemperature();
153     press = thermo.getPressureTensor();
154 gezelter 1390 instaPress = PhysicalConstants::pressureConvert* (press(0, 0) + press(1, 1) + press(2, 2)) / 3.0;
155 gezelter 246 instaVol =thermo.getVolume();
156 gezelter 2
157 gezelter 246 Vector3d COM = info_->getCom();
158 gezelter 2
159 gezelter 246 //evolve velocity half step
160 gezelter 2
161 gezelter 246 calcVelScale();
162 gezelter 2
163 gezelter 246 for (mol = info_->beginMolecule(i); mol != NULL; mol = info_->nextMolecule(i)) {
164 gezelter 507 for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
165     integrableObject = mol->nextIntegrableObject(j)) {
166 gezelter 246
167 gezelter 507 vel = integrableObject->getVel();
168     frc = integrableObject->getFrc();
169 gezelter 2
170 gezelter 507 mass = integrableObject->getMass();
171 gezelter 2
172 gezelter 507 getVelScaleA(sc, vel);
173 gezelter 2
174 gezelter 507 // velocity half step (use chi from previous step here):
175 gezelter 1390 //vel[j] += dt2 * ((frc[j] / mass) * PhysicalConstants::energyConvert - sc[j]);
176     vel += dt2*PhysicalConstants::energyConvert/mass* frc - dt2*sc;
177 gezelter 507 integrableObject->setVel(vel);
178 gezelter 2
179 gezelter 507 if (integrableObject->isDirectional()) {
180 gezelter 2
181 gezelter 507 // get and convert the torque to body frame
182 gezelter 2
183 gezelter 507 Tb = integrableObject->lab2Body(integrableObject->getTrq());
184 gezelter 2
185 gezelter 507 // get the angular momentum, and propagate a half step
186 gezelter 2
187 gezelter 507 ji = integrableObject->getJ();
188 gezelter 2
189 gezelter 1390 //ji[j] += dt2 * (Tb[j] * PhysicalConstants::energyConvert - ji[j]*chi);
190     ji += dt2*PhysicalConstants::energyConvert * Tb - dt2*chi* ji;
191 gezelter 246
192 gezelter 507 rotAlgo->rotate(integrableObject, ji, dt);
193 gezelter 2
194 gezelter 507 integrableObject->setJ(ji);
195     }
196 gezelter 246
197 gezelter 507 }
198 gezelter 246 }
199     // evolve chi and eta half step
200 gezelter 2
201 gezelter 246 chi += dt2 * (instaTemp / targetTemp - 1.0) / tt2;
202    
203     evolveEtaA();
204 gezelter 2
205 gezelter 246 //calculate the integral of chidt
206     integralOfChidt += dt2 * chi;
207    
208     index = 0;
209     for (mol = info_->beginMolecule(i); mol != NULL; mol = info_->nextMolecule(i)) {
210 gezelter 507 for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
211     integrableObject = mol->nextIntegrableObject(j)) {
212     oldPos[index++] = integrableObject->getPos();
213     }
214 gezelter 2 }
215 gezelter 246
216     //the first estimation of r(t+dt) is equal to r(t)
217 gezelter 2
218 gezelter 246 for(int k = 0; k < maxIterNum_; k++) {
219 gezelter 507 index = 0;
220     for (mol = info_->beginMolecule(i); mol != NULL; mol = info_->nextMolecule(i)) {
221     for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
222     integrableObject = mol->nextIntegrableObject(j)) {
223 gezelter 2
224 gezelter 507 vel = integrableObject->getVel();
225     pos = integrableObject->getPos();
226 gezelter 2
227 gezelter 507 this->getPosScale(pos, COM, index, sc);
228 gezelter 2
229 gezelter 507 pos = oldPos[index] + dt * (vel + sc);
230     integrableObject->setPos(pos);
231 gezelter 2
232 gezelter 507 ++index;
233     }
234     }
235 gezelter 2
236 gezelter 507 rattle->constraintA();
237 gezelter 246 }
238 gezelter 2
239 gezelter 246 // Scale the box after all the positions have been moved:
240 gezelter 2
241 gezelter 246 this->scaleSimBox();
242 gezelter 2
243 gezelter 246 currentSnapshot_->setChi(chi);
244     currentSnapshot_->setIntegralOfChiDt(integralOfChidt);
245 gezelter 2
246 gezelter 246 saveEta();
247 gezelter 507 }
248 gezelter 2
249 gezelter 507 void NPT::moveB(void) {
250 gezelter 246 SimInfo::MoleculeIterator i;
251     Molecule::IntegrableObjectIterator j;
252     Molecule* mol;
253     StuntDouble* integrableObject;
254     int index;
255     Vector3d Tb;
256     Vector3d ji;
257     Vector3d sc;
258     Vector3d vel;
259     Vector3d frc;
260 tim 963 RealType mass;
261 gezelter 2
262    
263 gezelter 246 chi= currentSnapshot_->getChi();
264     integralOfChidt = currentSnapshot_->getIntegralOfChiDt();
265 tim 963 RealType oldChi = chi;
266     RealType prevChi;
267 gezelter 2
268 gezelter 246 loadEta();
269    
270     //save velocity and angular momentum
271     index = 0;
272     for (mol = info_->beginMolecule(i); mol != NULL; mol = info_->nextMolecule(i)) {
273 gezelter 507 for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
274     integrableObject = mol->nextIntegrableObject(j)) {
275 gezelter 246
276 gezelter 507 oldVel[index] = integrableObject->getVel();
277     oldJi[index] = integrableObject->getJ();
278     ++index;
279     }
280 gezelter 2 }
281    
282 gezelter 246 // do the iteration:
283     instaVol =thermo.getVolume();
284 gezelter 2
285 gezelter 246 for(int k = 0; k < maxIterNum_; k++) {
286 gezelter 507 instaTemp =thermo.getTemperature();
287     instaPress =thermo.getPressure();
288 gezelter 2
289 gezelter 507 // evolve chi another half step using the temperature at t + dt/2
290     prevChi = chi;
291     chi = oldChi + dt2 * (instaTemp / targetTemp - 1.0) / tt2;
292 gezelter 2
293 gezelter 507 //evolve eta
294     this->evolveEtaB();
295     this->calcVelScale();
296 gezelter 2
297 gezelter 507 index = 0;
298     for (mol = info_->beginMolecule(i); mol != NULL; mol = info_->nextMolecule(i)) {
299     for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
300     integrableObject = mol->nextIntegrableObject(j)) {
301 gezelter 2
302 gezelter 507 frc = integrableObject->getFrc();
303     vel = integrableObject->getVel();
304 gezelter 2
305 gezelter 507 mass = integrableObject->getMass();
306 gezelter 2
307 gezelter 507 getVelScaleB(sc, index);
308 gezelter 2
309 gezelter 507 // velocity half step
310 gezelter 1390 //vel[j] = oldVel[3 * i + j] + dt2 *((frc[j] / mass) * PhysicalConstants::energyConvert - sc[j]);
311     vel = oldVel[index] + dt2*PhysicalConstants::energyConvert/mass* frc - dt2*sc;
312 gezelter 507 integrableObject->setVel(vel);
313 gezelter 2
314 gezelter 507 if (integrableObject->isDirectional()) {
315     // get and convert the torque to body frame
316     Tb = integrableObject->lab2Body(integrableObject->getTrq());
317 gezelter 2
318 gezelter 1390 //ji[j] = oldJi[3*i + j] + dt2 * (Tb[j] * PhysicalConstants::energyConvert - oldJi[3*i+j]*chi);
319     ji = oldJi[index] + dt2*PhysicalConstants::energyConvert*Tb - dt2*chi*oldJi[index];
320 gezelter 507 integrableObject->setJ(ji);
321     }
322 gezelter 2
323 gezelter 507 ++index;
324     }
325     }
326 gezelter 246
327 gezelter 507 rattle->constraintB();
328 gezelter 2
329 gezelter 507 if ((fabs(prevChi - chi) <= chiTolerance) && this->etaConverged())
330     break;
331 gezelter 2 }
332    
333 gezelter 246 //calculate integral of chidt
334     integralOfChidt += dt2 * chi;
335 gezelter 2
336 gezelter 246 currentSnapshot_->setChi(chi);
337     currentSnapshot_->setIntegralOfChiDt(integralOfChidt);
338 gezelter 2
339 gezelter 246 saveEta();
340 gezelter 507 }
341 gezelter 2
342 tim 546 void NPT::resetIntegrator(){
343     currentSnapshot_->setChi(0.0);
344     currentSnapshot_->setIntegralOfChiDt(0.0);
345     resetEta();
346     }
347    
348    
349     void NPT::resetEta() {
350     Mat3x3d etaMat(0.0);
351     currentSnapshot_->setEta(etaMat);
352     }
353    
354 gezelter 2 }

Properties

Name Value
svn:keywords Author Id Revision Date