ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/trunk/OOPSE-2.0/src/integrators/Velocitizer.cpp
(Generate patch)

Comparing trunk/OOPSE-2.0/src/integrators/Velocitizer.cpp (file contents):
Revision 1930 by gezelter, Wed Jan 12 22:41:40 2005 UTC vs.
Revision 2077 by tim, Wed Mar 2 16:28:20 2005 UTC

# Line 1 | Line 1
1 < /*
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. Acknowledgement of the program authors must be made in any
10 < *    publication of scientific results based in part on use of the
11 < *    program.  An acceptable form of acknowledgement is citation of
12 < *    the article in which the program was described (Matthew
13 < *    A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher
14 < *    J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented
15 < *    Parallel Simulation Engine for Molecular Dynamics,"
16 < *    J. Comput. Chem. 26, pp. 252-271 (2005))
17 < *
18 < * 2. Redistributions of source code must retain the above copyright
19 < *    notice, this list of conditions and the following disclaimer.
20 < *
21 < * 3. Redistributions in binary form must reproduce the above copyright
22 < *    notice, this list of conditions and the following disclaimer in the
23 < *    documentation and/or other materials provided with the
24 < *    distribution.
25 < *
26 < * This software is provided "AS IS," without a warranty of any
27 < * kind. All express or implied conditions, representations and
28 < * warranties, including any implied warranty of merchantability,
29 < * fitness for a particular purpose or non-infringement, are hereby
30 < * excluded.  The University of Notre Dame and its licensors shall not
31 < * be liable for any damages suffered by licensee as a result of
32 < * using, modifying or distributing the software or its
33 < * derivatives. In no event will the University of Notre Dame or its
34 < * licensors be liable for any lost revenue, profit or data, or for
35 < * direct, indirect, special, consequential, incidental or punitive
36 < * damages, however caused and regardless of the theory of liability,
37 < * arising out of the use of or inability to use software, even if the
38 < * University of Notre Dame has been advised of the possibility of
39 < * such damages.
40 < */
41 <  
42 < #include "integrators/Velocitizer.hpp"
43 < #include "math/SquareMatrix3.hpp"
44 < #include "primitives/Molecule.hpp"
45 < #include "primitives/StuntDouble.hpp"
46 < #include "math/randomSPRNG.hpp"
47 <
48 < namespace oopse {
49 <
50 < void Velocitizer::velocitize(double temperature) {
51 <    Vector3d aVel;
52 <    Vector3d aJ;
53 <    Mat3x3d I;
54 <    int l;
55 <    int m;
56 <    int n;
57 <    Vector3d vdrift;
58 <    double vbar;
59 <    /**@todo refactory kb */
60 <    const double kb = 8.31451e-7; // kb in amu, angstroms, fs, etc.
61 <    double av2;
62 <    double kebar;
63 <
64 <    SimInfo::MoleculeIterator i;
65 <    Molecule::IntegrableObjectIterator j;
66 <    Molecule * mol;
67 <    StuntDouble * integrableObject;
68 <    gaussianSPRNG gaussStream(info_->getSeed());
69 <
70 <    kebar = kb * temperature * info_->getNdfRaw() / (2.0 * info_->getNdf());
71 <
72 <    for( mol = info_->beginMolecule(i); mol != NULL;
73 <        mol = info_->nextMolecule(i) ) {
74 <        for( integrableObject = mol->beginIntegrableObject(j);
75 <            integrableObject != NULL;
76 <            integrableObject = mol->nextIntegrableObject(j) ) {
77 <
78 <            // uses equipartition theory to solve for vbar in angstrom/fs
79 <
80 <            av2 = 2.0 * kebar / integrableObject->getMass();
81 <            vbar = sqrt(av2);
82 <
83 <            // picks random velocities from a gaussian distribution
84 <            // centered on vbar
85 <
86 <            for( int k = 0; k < 3; k++ ) {
87 <                aVel[k] = vbar * gaussStream.getGaussian();
88 <            }
89 <
90 <            integrableObject->setVel(aVel);
91 <
92 <            if (integrableObject->isDirectional()) {
93 <                I = integrableObject->getI();
94 <
95 <                if (integrableObject->isLinear()) {
96 <                    l = integrableObject->linearAxis();
97 <                    m = (l + 1) % 3;
98 <                    n = (l + 2) % 3;
99 <
100 <                    aJ[l] = 0.0;
101 <                    vbar = sqrt(2.0 * kebar * I(m, m));
102 <                    aJ[m] = vbar * gaussStream.getGaussian();
103 <                    vbar = sqrt(2.0 * kebar * I(n, n));
104 <                    aJ[n] = vbar * gaussStream.getGaussian();
105 <                } else {
106 <                    for( int k = 0; k < 3; k++ ) {
107 <                        vbar = sqrt(2.0 * kebar * I(k, k));
108 <                        aJ[k] = vbar * gaussStream.getGaussian();
109 <                    }
110 <                } // else isLinear
111 <
112 <                integrableObject->setJ(aJ);
113 <            }     //isDirectional
114 <        }
115 <    }             //end for (mol = beginMolecule(i); ...)
116 <
117 <
118 <
119 <    removeComDrift();
120 <
121 < }
122 <
123 <
124 <
125 < void Velocitizer::removeComDrift() {
126 <    // Get the Center of Mass drift velocity.
127 <    Vector3d vdrift = info_->getComVel();
128 <    
129 <    SimInfo::MoleculeIterator i;
130 <    Molecule::IntegrableObjectIterator j;
131 <    Molecule * mol;
132 <    StuntDouble * integrableObject;
133 <    
134 <    //  Corrects for the center of mass drift.
135 <    // sums all the momentum and divides by total mass.
136 <    for( mol = info_->beginMolecule(i); mol != NULL;
137 <        mol = info_->nextMolecule(i) ) {
138 <        for( integrableObject = mol->beginIntegrableObject(j);
139 <            integrableObject != NULL;
140 <            integrableObject = mol->nextIntegrableObject(j) ) {
141 <            integrableObject->setVel(integrableObject->getVel() - vdrift);
142 <        }
143 <    }
144 <
145 < }
146 <
147 < }
1 > /*
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. Acknowledgement of the program authors must be made in any
10 > *    publication of scientific results based in part on use of the
11 > *    program.  An acceptable form of acknowledgement is citation of
12 > *    the article in which the program was described (Matthew
13 > *    A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher
14 > *    J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented
15 > *    Parallel Simulation Engine for Molecular Dynamics,"
16 > *    J. Comput. Chem. 26, pp. 252-271 (2005))
17 > *
18 > * 2. Redistributions of source code must retain the above copyright
19 > *    notice, this list of conditions and the following disclaimer.
20 > *
21 > * 3. Redistributions in binary form must reproduce the above copyright
22 > *    notice, this list of conditions and the following disclaimer in the
23 > *    documentation and/or other materials provided with the
24 > *    distribution.
25 > *
26 > * This software is provided "AS IS," without a warranty of any
27 > * kind. All express or implied conditions, representations and
28 > * warranties, including any implied warranty of merchantability,
29 > * fitness for a particular purpose or non-infringement, are hereby
30 > * excluded.  The University of Notre Dame and its licensors shall not
31 > * be liable for any damages suffered by licensee as a result of
32 > * using, modifying or distributing the software or its
33 > * derivatives. In no event will the University of Notre Dame or its
34 > * licensors be liable for any lost revenue, profit or data, or for
35 > * direct, indirect, special, consequential, incidental or punitive
36 > * damages, however caused and regardless of the theory of liability,
37 > * arising out of the use of or inability to use software, even if the
38 > * University of Notre Dame has been advised of the possibility of
39 > * such damages.
40 > */
41 >  
42 > #include "integrators/Velocitizer.hpp"
43 > #include "math/SquareMatrix3.hpp"
44 > #include "primitives/Molecule.hpp"
45 > #include "primitives/StuntDouble.hpp"
46 >
47 > #ifndef IS_MPI
48 > #include "math/SeqRandNumGen.hpp"
49 > #else
50 > #include "math/ParallelRandNumGen.hpp"
51 > #endif
52 >
53 > namespace oopse {
54 >
55 > Velocitizer::Velocitizer(SimInfo* info) : info_(info) {
56 >
57 >    int seedValue;
58 >    Globals * simParams = info->getSimParams();
59 >
60 > #ifndef IS_MPI
61 >    if (simParams->haveSeed()) {
62 >        seedValue = simParams->getSeed();
63 >        randNumGen_ = new SeqRandNumGen(seedValue);
64 >    }else {
65 >        randNumGen_ = new SeqRandNumGen();
66 >    }    
67 > #else
68 >    if (simParams->haveSeed()) {
69 >        seedValue = simParams->getSeed();
70 >        randNumGen_ = new ParallelRandNumGen(seedValue);
71 >    }else {
72 >        randNumGen_ = new ParallelRandNumGen();
73 >    }    
74 > #endif
75 > }
76 >
77 > Velocitizer::~Velocitizer() {
78 >    delete randNumGen_;
79 > }
80 >
81 > void Velocitizer::velocitize(double temperature) {
82 >    Vector3d aVel;
83 >    Vector3d aJ;
84 >    Mat3x3d I;
85 >    int l;
86 >    int m;
87 >    int n;
88 >    Vector3d vdrift;
89 >    double vbar;
90 >    /**@todo refactory kb */
91 >    const double kb = 8.31451e-7; // kb in amu, angstroms, fs, etc.
92 >    double av2;
93 >    double kebar;
94 >
95 >    SimInfo::MoleculeIterator i;
96 >    Molecule::IntegrableObjectIterator j;
97 >    Molecule * mol;
98 >    StuntDouble * integrableObject;
99 >
100 >
101 >
102 >    kebar = kb * temperature * info_->getNdfRaw() / (2.0 * info_->getNdf());
103 >
104 >    for( mol = info_->beginMolecule(i); mol != NULL;
105 >        mol = info_->nextMolecule(i) ) {
106 >        for( integrableObject = mol->beginIntegrableObject(j);
107 >            integrableObject != NULL;
108 >            integrableObject = mol->nextIntegrableObject(j) ) {
109 >
110 >            // uses equipartition theory to solve for vbar in angstrom/fs
111 >
112 >            av2 = 2.0 * kebar / integrableObject->getMass();
113 >            vbar = sqrt(av2);
114 >
115 >            // picks random velocities from a gaussian distribution
116 >            // centered on vbar
117 >
118 >            for( int k = 0; k < 3; k++ ) {
119 >                aVel[k] = vbar * randNumGen_->randNorm(0.0, 1.0);
120 >            }
121 >
122 >            integrableObject->setVel(aVel);
123 >
124 >            if (integrableObject->isDirectional()) {
125 >                I = integrableObject->getI();
126 >
127 >                if (integrableObject->isLinear()) {
128 >                    l = integrableObject->linearAxis();
129 >                    m = (l + 1) % 3;
130 >                    n = (l + 2) % 3;
131 >
132 >                    aJ[l] = 0.0;
133 >                    vbar = sqrt(2.0 * kebar * I(m, m));
134 >                    aJ[m] = vbar * randNumGen_->randNorm(0.0, 1.0);
135 >                    vbar = sqrt(2.0 * kebar * I(n, n));
136 >                    aJ[n] = vbar * randNumGen_->randNorm(0.0, 1.0);
137 >                } else {
138 >                    for( int k = 0; k < 3; k++ ) {
139 >                        vbar = sqrt(2.0 * kebar * I(k, k));
140 >                        aJ[k] = vbar *randNumGen_->randNorm(0.0, 1.0);
141 >                    }
142 >                } // else isLinear
143 >
144 >                integrableObject->setJ(aJ);
145 >            }     //isDirectional
146 >        }
147 >    }             //end for (mol = beginMolecule(i); ...)
148 >
149 >
150 >
151 >    removeComDrift();
152 >
153 > }
154 >
155 >
156 >
157 > void Velocitizer::removeComDrift() {
158 >    // Get the Center of Mass drift velocity.
159 >    Vector3d vdrift = info_->getComVel();
160 >    
161 >    SimInfo::MoleculeIterator i;
162 >    Molecule::IntegrableObjectIterator j;
163 >    Molecule * mol;
164 >    StuntDouble * integrableObject;
165 >    
166 >    //  Corrects for the center of mass drift.
167 >    // sums all the momentum and divides by total mass.
168 >    for( mol = info_->beginMolecule(i); mol != NULL;
169 >        mol = info_->nextMolecule(i) ) {
170 >        for( integrableObject = mol->beginIntegrableObject(j);
171 >            integrableObject != NULL;
172 >            integrableObject = mol->nextIntegrableObject(j) ) {
173 >            integrableObject->setVel(integrableObject->getVel() - vdrift);
174 >        }
175 >    }
176 >
177 > }
178 >
179 > }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines