ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/branches/development/src/brains/ForceManager.cpp
Revision: 1535
Committed: Fri Dec 31 18:31:56 2010 UTC (14 years, 4 months ago) by gezelter
File size: 12433 byte(s)
Log Message:
Well, it compiles and builds, but still has a bus error at runtime.

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     * [4] Vardeman & Gezelter, in progress (2009).
40 gezelter 246 */
41    
42 gezelter 507 /**
43     * @file ForceManager.cpp
44     * @author tlin
45     * @date 11/09/2004
46     * @time 10:39am
47     * @version 1.0
48     */
49 gezelter 246
50     #include "brains/ForceManager.hpp"
51     #include "primitives/Molecule.hpp"
52     #include "UseTheForce/doForces_interface.h"
53 gezelter 1390 #define __OPENMD_C
54 chuckv 664 #include "UseTheForce/DarkSide/fInteractionMap.h"
55 gezelter 246 #include "utils/simError.h"
56 xsun 1215 #include "primitives/Bond.hpp"
57 tim 749 #include "primitives/Bend.hpp"
58 cli2 1275 #include "primitives/Torsion.hpp"
59     #include "primitives/Inversion.hpp"
60 gezelter 1467
61 gezelter 1390 namespace OpenMD {
62 gezelter 1469
63     ForceManager::ForceManager(SimInfo * info) : info_(info),
64     NBforcesInitialized_(false) {
65     }
66    
67 gezelter 1464 void ForceManager::calcForces() {
68 gezelter 1126
69 gezelter 1535
70 gezelter 246 if (!info_->isFortranInitialized()) {
71 gezelter 507 info_->update();
72 gezelter 1535 nbiMan_->setSimInfo(info_);
73     nbiMan_->initialize();
74     info_->setupFortran();
75 gezelter 246 }
76 gezelter 1126
77 gezelter 246 preCalculation();
78    
79     calcShortRangeInteraction();
80    
81 gezelter 1464 calcLongRangeInteraction();
82 gezelter 246
83 gezelter 1464 postCalculation();
84 tim 749
85 gezelter 507 }
86 gezelter 1126
87 gezelter 507 void ForceManager::preCalculation() {
88 gezelter 246 SimInfo::MoleculeIterator mi;
89     Molecule* mol;
90     Molecule::AtomIterator ai;
91     Atom* atom;
92     Molecule::RigidBodyIterator rbIter;
93     RigidBody* rb;
94    
95     // forces are zeroed here, before any are accumulated.
96     // NOTE: do not rezero the forces in Fortran.
97 chuckv 1245
98 gezelter 1126 for (mol = info_->beginMolecule(mi); mol != NULL;
99     mol = info_->nextMolecule(mi)) {
100 gezelter 507 for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
101     atom->zeroForcesAndTorques();
102     }
103 chuckv 1245
104 gezelter 507 //change the positions of atoms which belong to the rigidbodies
105 gezelter 1126 for (rb = mol->beginRigidBody(rbIter); rb != NULL;
106     rb = mol->nextRigidBody(rbIter)) {
107 gezelter 507 rb->zeroForcesAndTorques();
108     }
109 chuckv 1245
110 gezelter 246 }
111    
112 gezelter 1126 // Zero out the stress tensor
113     tau *= 0.0;
114    
115 gezelter 507 }
116 gezelter 1126
117 gezelter 507 void ForceManager::calcShortRangeInteraction() {
118 gezelter 246 Molecule* mol;
119     RigidBody* rb;
120     Bond* bond;
121     Bend* bend;
122     Torsion* torsion;
123 cli2 1275 Inversion* inversion;
124 gezelter 246 SimInfo::MoleculeIterator mi;
125     Molecule::RigidBodyIterator rbIter;
126     Molecule::BondIterator bondIter;;
127     Molecule::BendIterator bendIter;
128     Molecule::TorsionIterator torsionIter;
129 cli2 1275 Molecule::InversionIterator inversionIter;
130 tim 963 RealType bondPotential = 0.0;
131     RealType bendPotential = 0.0;
132     RealType torsionPotential = 0.0;
133 cli2 1275 RealType inversionPotential = 0.0;
134 gezelter 246
135     //calculate short range interactions
136 gezelter 1126 for (mol = info_->beginMolecule(mi); mol != NULL;
137     mol = info_->nextMolecule(mi)) {
138 gezelter 246
139 gezelter 507 //change the positions of atoms which belong to the rigidbodies
140 gezelter 1126 for (rb = mol->beginRigidBody(rbIter); rb != NULL;
141     rb = mol->nextRigidBody(rbIter)) {
142     rb->updateAtoms();
143 gezelter 507 }
144 gezelter 246
145 gezelter 1126 for (bond = mol->beginBond(bondIter); bond != NULL;
146     bond = mol->nextBond(bondIter)) {
147 tim 749 bond->calcForce();
148     bondPotential += bond->getPotential();
149 gezelter 507 }
150 gezelter 246
151 gezelter 1126 for (bend = mol->beginBend(bendIter); bend != NULL;
152     bend = mol->nextBend(bendIter)) {
153    
154     RealType angle;
155     bend->calcForce(angle);
156     RealType currBendPot = bend->getPotential();
157 gezelter 1448
158 gezelter 1126 bendPotential += bend->getPotential();
159     std::map<Bend*, BendDataSet>::iterator i = bendDataSets.find(bend);
160     if (i == bendDataSets.end()) {
161     BendDataSet dataSet;
162     dataSet.prev.angle = dataSet.curr.angle = angle;
163     dataSet.prev.potential = dataSet.curr.potential = currBendPot;
164     dataSet.deltaV = 0.0;
165     bendDataSets.insert(std::map<Bend*, BendDataSet>::value_type(bend, dataSet));
166     }else {
167     i->second.prev.angle = i->second.curr.angle;
168     i->second.prev.potential = i->second.curr.potential;
169     i->second.curr.angle = angle;
170     i->second.curr.potential = currBendPot;
171     i->second.deltaV = fabs(i->second.curr.potential -
172     i->second.prev.potential);
173     }
174 gezelter 507 }
175 gezelter 1126
176     for (torsion = mol->beginTorsion(torsionIter); torsion != NULL;
177     torsion = mol->nextTorsion(torsionIter)) {
178 tim 963 RealType angle;
179 gezelter 1126 torsion->calcForce(angle);
180 tim 963 RealType currTorsionPot = torsion->getPotential();
181 gezelter 1126 torsionPotential += torsion->getPotential();
182     std::map<Torsion*, TorsionDataSet>::iterator i = torsionDataSets.find(torsion);
183     if (i == torsionDataSets.end()) {
184     TorsionDataSet dataSet;
185     dataSet.prev.angle = dataSet.curr.angle = angle;
186     dataSet.prev.potential = dataSet.curr.potential = currTorsionPot;
187     dataSet.deltaV = 0.0;
188     torsionDataSets.insert(std::map<Torsion*, TorsionDataSet>::value_type(torsion, dataSet));
189     }else {
190     i->second.prev.angle = i->second.curr.angle;
191     i->second.prev.potential = i->second.curr.potential;
192     i->second.curr.angle = angle;
193     i->second.curr.potential = currTorsionPot;
194     i->second.deltaV = fabs(i->second.curr.potential -
195     i->second.prev.potential);
196     }
197     }
198 cli2 1275
199     for (inversion = mol->beginInversion(inversionIter);
200     inversion != NULL;
201     inversion = mol->nextInversion(inversionIter)) {
202     RealType angle;
203     inversion->calcForce(angle);
204     RealType currInversionPot = inversion->getPotential();
205     inversionPotential += inversion->getPotential();
206     std::map<Inversion*, InversionDataSet>::iterator i = inversionDataSets.find(inversion);
207     if (i == inversionDataSets.end()) {
208     InversionDataSet dataSet;
209     dataSet.prev.angle = dataSet.curr.angle = angle;
210     dataSet.prev.potential = dataSet.curr.potential = currInversionPot;
211     dataSet.deltaV = 0.0;
212     inversionDataSets.insert(std::map<Inversion*, InversionDataSet>::value_type(inversion, dataSet));
213     }else {
214     i->second.prev.angle = i->second.curr.angle;
215     i->second.prev.potential = i->second.curr.potential;
216     i->second.curr.angle = angle;
217     i->second.curr.potential = currInversionPot;
218     i->second.deltaV = fabs(i->second.curr.potential -
219     i->second.prev.potential);
220     }
221     }
222 gezelter 246 }
223    
224 gezelter 1126 RealType shortRangePotential = bondPotential + bendPotential +
225 cli2 1275 torsionPotential + inversionPotential;
226 gezelter 246 Snapshot* curSnapshot = info_->getSnapshotManager()->getCurrentSnapshot();
227     curSnapshot->statData[Stats::SHORT_RANGE_POTENTIAL] = shortRangePotential;
228 tim 665 curSnapshot->statData[Stats::BOND_POTENTIAL] = bondPotential;
229     curSnapshot->statData[Stats::BEND_POTENTIAL] = bendPotential;
230     curSnapshot->statData[Stats::DIHEDRAL_POTENTIAL] = torsionPotential;
231 cli2 1275 curSnapshot->statData[Stats::INVERSION_POTENTIAL] = inversionPotential;
232 tim 665
233 gezelter 507 }
234 gezelter 1126
235 gezelter 1464 void ForceManager::calcLongRangeInteraction() {
236 gezelter 246 Snapshot* curSnapshot;
237     DataStorage* config;
238 tim 963 RealType* frc;
239     RealType* pos;
240     RealType* trq;
241     RealType* A;
242     RealType* electroFrame;
243     RealType* rc;
244 chuckv 1245 RealType* particlePot;
245 gezelter 246
246     //get current snapshot from SimInfo
247     curSnapshot = info_->getSnapshotManager()->getCurrentSnapshot();
248 gezelter 1126
249 gezelter 246 //get array pointers
250     config = &(curSnapshot->atomData);
251     frc = config->getArrayPointer(DataStorage::dslForce);
252     pos = config->getArrayPointer(DataStorage::dslPosition);
253     trq = config->getArrayPointer(DataStorage::dslTorque);
254     A = config->getArrayPointer(DataStorage::dslAmat);
255     electroFrame = config->getArrayPointer(DataStorage::dslElectroFrame);
256 chuckv 1245 particlePot = config->getArrayPointer(DataStorage::dslParticlePot);
257 gezelter 246
258     //calculate the center of mass of cutoff group
259     SimInfo::MoleculeIterator mi;
260     Molecule* mol;
261     Molecule::CutoffGroupIterator ci;
262     CutoffGroup* cg;
263     Vector3d com;
264     std::vector<Vector3d> rcGroup;
265 gezelter 1126
266 gezelter 246 if(info_->getNCutoffGroups() > 0){
267 gezelter 1126
268     for (mol = info_->beginMolecule(mi); mol != NULL;
269     mol = info_->nextMolecule(mi)) {
270     for(cg = mol->beginCutoffGroup(ci); cg != NULL;
271     cg = mol->nextCutoffGroup(ci)) {
272 gezelter 507 cg->getCOM(com);
273     rcGroup.push_back(com);
274 gezelter 246 }
275 gezelter 507 }// end for (mol)
276 gezelter 246
277 gezelter 507 rc = rcGroup[0].getArrayPointer();
278 gezelter 246 } else {
279 gezelter 1126 // center of mass of the group is the same as position of the atom
280     // if cutoff group does not exist
281 gezelter 507 rc = pos;
282 gezelter 246 }
283 gezelter 1126
284 gezelter 246 //initialize data before passing to fortran
285 tim 963 RealType longRangePotential[LR_POT_TYPES];
286     RealType lrPot = 0.0;
287 gezelter 246 int isError = 0;
288    
289 chuckv 664 for (int i=0; i<LR_POT_TYPES;i++){
290     longRangePotential[i]=0.0; //Initialize array
291     }
292 gezelter 1126
293 xsun 1215 doForceLoop(pos,
294     rc,
295     A,
296     electroFrame,
297     frc,
298     trq,
299     tau.getArrayPointer(),
300     longRangePotential,
301 chuckv 1245 particlePot,
302 xsun 1215 &isError );
303    
304 gezelter 246 if( isError ){
305 gezelter 507 sprintf( painCave.errMsg,
306     "Error returned from the fortran force calculation.\n" );
307     painCave.isFatal = 1;
308     simError();
309 gezelter 246 }
310 chuckv 664 for (int i=0; i<LR_POT_TYPES;i++){
311     lrPot += longRangePotential[i]; //Quick hack
312     }
313 gezelter 1503
314 gezelter 246 //store the tau and long range potential
315 chuckv 664 curSnapshot->statData[Stats::LONG_RANGE_POTENTIAL] = lrPot;
316 chrisfen 691 curSnapshot->statData[Stats::VANDERWAALS_POTENTIAL] = longRangePotential[VDW_POT];
317 tim 681 curSnapshot->statData[Stats::ELECTROSTATIC_POTENTIAL] = longRangePotential[ELECTROSTATIC_POT];
318 gezelter 507 }
319 gezelter 246
320 gezelter 1126
321 gezelter 1464 void ForceManager::postCalculation() {
322 gezelter 246 SimInfo::MoleculeIterator mi;
323     Molecule* mol;
324     Molecule::RigidBodyIterator rbIter;
325     RigidBody* rb;
326 gezelter 1126 Snapshot* curSnapshot = info_->getSnapshotManager()->getCurrentSnapshot();
327 gezelter 246
328     // collect the atomic forces onto rigid bodies
329 gezelter 1126
330     for (mol = info_->beginMolecule(mi); mol != NULL;
331     mol = info_->nextMolecule(mi)) {
332     for (rb = mol->beginRigidBody(rbIter); rb != NULL;
333     rb = mol->nextRigidBody(rbIter)) {
334 gezelter 1464 Mat3x3d rbTau = rb->calcForcesAndTorquesAndVirial();
335     tau += rbTau;
336 gezelter 507 }
337 gezelter 1126 }
338 gezelter 1464
339 gezelter 1126 #ifdef IS_MPI
340 gezelter 1464 Mat3x3d tmpTau(tau);
341     MPI_Allreduce(tmpTau.getArrayPointer(), tau.getArrayPointer(),
342     9, MPI_REALTYPE, MPI_SUM, MPI_COMM_WORLD);
343 gezelter 1126 #endif
344 gezelter 1464 curSnapshot->statData.setTau(tau);
345 gezelter 507 }
346 gezelter 246
347 gezelter 1390 } //end namespace OpenMD

Properties

Name Value
svn:executable *
svn:keywords Author Id Revision Date