46 |
|
#include "nonbonded/LJ.hpp" |
47 |
|
#include "utils/simError.h" |
48 |
|
|
49 |
– |
|
49 |
|
namespace OpenMD { |
50 |
|
|
51 |
< |
bool LJ::initialized_ = false; |
53 |
< |
bool LJ::shiftedPot_ = false; |
54 |
< |
bool LJ::shiftedFrc_ = false; |
55 |
< |
ForceField* LJ::forceField_ = NULL; |
56 |
< |
std::map<int, AtomType*> LJ::LJMap; |
57 |
< |
std::map<std::pair<AtomType*, AtomType*>, LJInteractionData> LJ::MixingMap; |
58 |
< |
|
59 |
< |
LJ* LJ::_instance = NULL; |
51 |
> |
LJ::LJ() : name_("LJ"), initialized_(false), forceField_(NULL) {} |
52 |
|
|
61 |
– |
LJ* LJ::Instance() { |
62 |
– |
if (!_instance) { |
63 |
– |
_instance = new LJ(); |
64 |
– |
} |
65 |
– |
return _instance; |
66 |
– |
} |
67 |
– |
|
53 |
|
LJParam LJ::getLJParam(AtomType* atomType) { |
54 |
|
|
55 |
|
// Do sanity checking on the AtomType we were passed before |
91 |
|
return ljParam.sigma; |
92 |
|
} |
93 |
|
|
109 |
– |
RealType LJ::getSigma(int atid) { |
110 |
– |
if (!initialized_) initialize(); |
111 |
– |
std::map<int, AtomType*> :: const_iterator it; |
112 |
– |
it = LJMap.find(atid); |
113 |
– |
if (it == LJMap.end()) { |
114 |
– |
sprintf( painCave.errMsg, |
115 |
– |
"LJ::getSigma could not find atid %d in LJMap\n", |
116 |
– |
(atid)); |
117 |
– |
painCave.severity = OPENMD_ERROR; |
118 |
– |
painCave.isFatal = 1; |
119 |
– |
simError(); |
120 |
– |
} |
121 |
– |
AtomType* atype = it->second; |
122 |
– |
|
123 |
– |
return getSigma(atype); |
124 |
– |
} |
125 |
– |
|
94 |
|
RealType LJ::getSigma(AtomType* atomType1, AtomType* atomType2) { |
95 |
|
RealType sigma1 = getSigma(atomType1); |
96 |
|
RealType sigma2 = getSigma(atomType2); |
97 |
|
|
98 |
|
ForceFieldOptions& fopts = forceField_->getForceFieldOptions(); |
99 |
< |
std::string DistanceMix = fopts.getDistanceMixingRule(); |
99 |
> |
string DistanceMix = fopts.getDistanceMixingRule(); |
100 |
|
toUpper(DistanceMix); |
101 |
|
|
102 |
|
if (DistanceMix == "GEOMETRIC") |
110 |
|
return ljParam.epsilon; |
111 |
|
} |
112 |
|
|
145 |
– |
RealType LJ::getEpsilon(int atid) { |
146 |
– |
if (!initialized_) initialize(); |
147 |
– |
std::map<int, AtomType*> :: const_iterator it; |
148 |
– |
it = LJMap.find(atid); |
149 |
– |
if (it == LJMap.end()) { |
150 |
– |
sprintf( painCave.errMsg, |
151 |
– |
"LJ::getEpsilon could not find atid %d in LJMap\n", |
152 |
– |
(atid)); |
153 |
– |
painCave.severity = OPENMD_ERROR; |
154 |
– |
painCave.isFatal = 1; |
155 |
– |
simError(); |
156 |
– |
} |
157 |
– |
AtomType* atype = it->second; |
158 |
– |
|
159 |
– |
return getEpsilon(atype); |
160 |
– |
} |
161 |
– |
|
113 |
|
RealType LJ::getEpsilon(AtomType* atomType1, AtomType* atomType2) { |
114 |
|
RealType epsilon1 = getEpsilon(atomType1); |
115 |
|
RealType epsilon2 = getEpsilon(atomType2); |
117 |
|
} |
118 |
|
|
119 |
|
void LJ::initialize() { |
120 |
< |
ForceField::AtomTypeContainer atomTypes = forceField_->getAtomTypes(); |
120 |
> |
ForceField::AtomTypeContainer* atomTypes = forceField_->getAtomTypes(); |
121 |
|
ForceField::AtomTypeContainer::MapTypeIterator i; |
122 |
|
AtomType* at; |
123 |
|
|
124 |
< |
for (at = atomTypes.beginType(i); at != NULL; |
125 |
< |
at = atomTypes.nextType(i)) { |
124 |
> |
for (at = atomTypes->beginType(i); at != NULL; |
125 |
> |
at = atomTypes->nextType(i)) { |
126 |
|
|
127 |
|
if (at->isLennardJones()) |
128 |
|
addType(at); |
129 |
|
} |
130 |
|
|
131 |
< |
ForceField::NonBondedInteractionTypeContainer nbiTypes = forceField_->getNonBondedInteractionTypes(); |
131 |
> |
ForceField::NonBondedInteractionTypeContainer* nbiTypes = forceField_->getNonBondedInteractionTypes(); |
132 |
|
ForceField::NonBondedInteractionTypeContainer::MapTypeIterator j; |
133 |
|
NonBondedInteractionType* nbt; |
134 |
|
|
135 |
< |
for (nbt = nbiTypes.beginType(j); nbt != NULL; |
136 |
< |
nbt = nbiTypes.nextType(j)) { |
135 |
> |
for (nbt = nbiTypes->beginType(j); nbt != NULL; |
136 |
> |
nbt = nbiTypes->nextType(j)) { |
137 |
|
|
138 |
|
if (nbt->isLennardJones()) { |
139 |
|
|
140 |
< |
std::pair<AtomType*, AtomType*> atypes = nbt->getAtomTypes(); |
140 |
> |
pair<AtomType*, AtomType*> atypes = nbt->getAtomTypes(); |
141 |
|
|
142 |
|
GenericData* data = nbt->getPropertyByName("LennardJones"); |
143 |
|
if (data == NULL) { |
178 |
|
void LJ::addType(AtomType* atomType){ |
179 |
|
RealType sigma1 = getSigma(atomType); |
180 |
|
RealType epsilon1 = getEpsilon(atomType); |
181 |
< |
|
181 |
> |
|
182 |
|
// add it to the map: |
183 |
|
AtomTypeProperties atp = atomType->getATP(); |
184 |
|
|
185 |
< |
std::pair<std::map<int,AtomType*>::iterator,bool> ret; |
186 |
< |
ret = LJMap.insert( std::pair<int, AtomType*>(atp.ident, atomType) ); |
185 |
> |
pair<map<int,AtomType*>::iterator,bool> ret; |
186 |
> |
ret = LJMap.insert( pair<int, AtomType*>(atp.ident, atomType) ); |
187 |
|
if (ret.second == false) { |
188 |
|
sprintf( painCave.errMsg, |
189 |
|
"LJ already had a previous entry with ident %d\n", |
239 |
|
} |
240 |
|
} |
241 |
|
|
242 |
< |
void LJ::calcForce(AtomType* at1, AtomType* at2, Vector3d d, |
292 |
< |
RealType rij, RealType r2, RealType rcut, RealType sw, |
293 |
< |
RealType vdwMult, RealType &vpair, RealType &pot, |
294 |
< |
Vector3d &f1) { |
295 |
< |
|
296 |
< |
if (!initialized_) initialize(); |
242 |
> |
void LJ::calcForce(InteractionData &idat) { |
243 |
|
|
244 |
< |
RealType ros; |
299 |
< |
RealType rcos; |
300 |
< |
RealType myPot = 0.0; |
301 |
< |
RealType myPotC = 0.0; |
302 |
< |
RealType myDeriv = 0.0; |
303 |
< |
RealType myDerivC = 0.0; |
244 |
> |
if (!initialized_) initialize(); |
245 |
|
|
246 |
< |
std::pair<AtomType*, AtomType*> key = std::make_pair(at1, at2); |
247 |
< |
LJInteractionData mixer = MixingMap[key]; |
307 |
< |
|
308 |
< |
RealType sigmai = mixer.sigmai; |
309 |
< |
RealType epsilon = mixer.epsilon; |
246 |
> |
map<pair<AtomType*, AtomType*>, LJInteractionData>::iterator it; |
247 |
> |
it = MixingMap.find( idat.atypes ); |
248 |
|
|
249 |
< |
|
250 |
< |
ros = rij * sigmai; |
251 |
< |
|
252 |
< |
getLJfunc(ros, myPot, myDeriv); |
249 |
> |
if (it != MixingMap.end()) { |
250 |
> |
|
251 |
> |
LJInteractionData mixer = (*it).second; |
252 |
> |
|
253 |
> |
RealType sigmai = mixer.sigmai; |
254 |
> |
RealType epsilon = mixer.epsilon; |
255 |
> |
|
256 |
> |
RealType ros; |
257 |
> |
RealType rcos; |
258 |
> |
RealType myPot = 0.0; |
259 |
> |
RealType myPotC = 0.0; |
260 |
> |
RealType myDeriv = 0.0; |
261 |
> |
RealType myDerivC = 0.0; |
262 |
> |
|
263 |
> |
ros = *(idat.rij) * sigmai; |
264 |
> |
|
265 |
> |
getLJfunc(ros, myPot, myDeriv); |
266 |
> |
|
267 |
> |
if (idat.shiftedPot) { |
268 |
> |
rcos = *(idat.rcut) * sigmai; |
269 |
> |
getLJfunc(rcos, myPotC, myDerivC); |
270 |
> |
myDerivC = 0.0; |
271 |
> |
} else if (idat.shiftedForce) { |
272 |
> |
rcos = *(idat.rcut) * sigmai; |
273 |
> |
getLJfunc(rcos, myPotC, myDerivC); |
274 |
> |
myPotC = myPotC + myDerivC * (*(idat.rij) - *(idat.rcut)) * sigmai; |
275 |
> |
} else { |
276 |
> |
myPotC = 0.0; |
277 |
> |
myDerivC = 0.0; |
278 |
> |
} |
279 |
|
|
280 |
< |
if (shiftedPot_) { |
281 |
< |
rcos = rcut * sigmai; |
282 |
< |
getLJfunc(rcos, myPotC, myDerivC); |
283 |
< |
myDerivC = 0.0; |
284 |
< |
} else if (LJ::shiftedFrc_) { |
321 |
< |
rcos = rcut * sigmai; |
322 |
< |
getLJfunc(rcos, myPotC, myDerivC); |
323 |
< |
myPotC = myPotC + myDerivC * (rij - rcut) * sigmai; |
324 |
< |
} else { |
325 |
< |
myPotC = 0.0; |
326 |
< |
myDerivC = 0.0; |
327 |
< |
} |
280 |
> |
RealType pot_temp = *(idat.vdwMult) * epsilon * (myPot - myPotC); |
281 |
> |
*(idat.vpair) += pot_temp; |
282 |
> |
|
283 |
> |
RealType dudr = *(idat.sw) * *(idat.vdwMult) * epsilon * (myDeriv - |
284 |
> |
myDerivC)*sigmai; |
285 |
|
|
286 |
< |
RealType pot_temp = vdwMult * epsilon * (myPot - myPotC); |
287 |
< |
vpair += pot_temp; |
288 |
< |
|
332 |
< |
RealType dudr = sw * vdwMult * epsilon * (myDeriv - myDerivC)*sigmai; |
333 |
< |
|
334 |
< |
pot += sw * pot_temp; |
335 |
< |
f1 = d * dudr / rij; |
336 |
< |
|
286 |
> |
(*(idat.pot))[VANDERWAALS_FAMILY] += *(idat.sw) * pot_temp; |
287 |
> |
*(idat.f1) += *(idat.d) * dudr / *(idat.rij); |
288 |
> |
} |
289 |
|
return; |
338 |
– |
|
339 |
– |
|
340 |
– |
|
290 |
|
} |
342 |
– |
|
343 |
– |
void LJ::do_lj_pair(int *atid1, int *atid2, RealType *d, RealType *rij, |
344 |
– |
RealType *r2, RealType *rcut, RealType *sw, |
345 |
– |
RealType *vdwMult, |
346 |
– |
RealType *vpair, RealType *pot, RealType *f1) { |
347 |
– |
|
348 |
– |
if (!initialized_) initialize(); |
349 |
– |
|
350 |
– |
AtomType* atype1 = LJMap[*atid1]; |
351 |
– |
AtomType* atype2 = LJMap[*atid2]; |
352 |
– |
|
353 |
– |
Vector3d disp(d[0], d[1], d[2]); |
354 |
– |
Vector3d frc(f1[0], f1[1], f1[2]); |
355 |
– |
|
356 |
– |
calcForce(atype1, atype2, disp, *rij, *r2, *rcut, *sw, *vdwMult, *vpair, |
357 |
– |
*pot, frc); |
358 |
– |
|
359 |
– |
f1[0] = frc.x(); |
360 |
– |
f1[1] = frc.y(); |
361 |
– |
f1[2] = frc.z(); |
362 |
– |
|
363 |
– |
return; |
364 |
– |
} |
291 |
|
|
292 |
|
void LJ::getLJfunc(RealType r, RealType &pot, RealType &deriv) { |
293 |
|
|
304 |
|
return; |
305 |
|
} |
306 |
|
|
307 |
< |
|
308 |
< |
void LJ::setLJDefaultCutoff(RealType *thisRcut, int *sP, int *sF) { |
309 |
< |
shiftedPot_ = (bool)(*sP); |
310 |
< |
shiftedFrc_ = (bool)(*sF); |
307 |
> |
RealType LJ::getSuggestedCutoffRadius(pair<AtomType*, AtomType*> atypes) { |
308 |
> |
if (!initialized_) initialize(); |
309 |
> |
map<pair<AtomType*, AtomType*>, LJInteractionData>::iterator it; |
310 |
> |
it = MixingMap.find(atypes); |
311 |
> |
if (it == MixingMap.end()) |
312 |
> |
return 0.0; |
313 |
> |
else { |
314 |
> |
LJInteractionData mixer = (*it).second; |
315 |
> |
return 2.5 * mixer.sigma; |
316 |
> |
} |
317 |
|
} |
386 |
– |
} |
318 |
|
|
388 |
– |
extern "C" { |
389 |
– |
|
390 |
– |
#define fortranGetSigma FC_FUNC(getsigma, GETSIGMA) |
391 |
– |
#define fortranGetEpsilon FC_FUNC(getepsilon, GETEPSILON) |
392 |
– |
#define fortranSetLJCutoff FC_FUNC(setljdefaultcutoff, SETLJDEFAULTCUTOFF) |
393 |
– |
#define fortranDoLJPair FC_FUNC(do_lj_pair, DO_LJ_PAIR) |
394 |
– |
|
395 |
– |
RealType fortranGetSigma(int* atid) { |
396 |
– |
return OpenMD::LJ::Instance()->getSigma(*atid); |
397 |
– |
} |
398 |
– |
RealType fortranGetEpsilon(int* atid) { |
399 |
– |
return OpenMD::LJ::Instance()->getEpsilon(*atid); |
400 |
– |
} |
401 |
– |
void fortranSetLJCutoff(RealType *rcut, int *shiftedPot, int *shiftedFrc) { |
402 |
– |
return OpenMD::LJ::Instance()->setLJDefaultCutoff(rcut, shiftedPot, |
403 |
– |
shiftedFrc); |
404 |
– |
} |
405 |
– |
void fortranDoLJPair(int *atid1, int *atid2, RealType *d, RealType *rij, |
406 |
– |
RealType *r2, RealType *rcut, RealType *sw, |
407 |
– |
RealType *vdwMult, RealType* vpair, RealType* pot, |
408 |
– |
RealType *f1){ |
409 |
– |
|
410 |
– |
return OpenMD::LJ::Instance()->do_lj_pair(atid1, atid2, d, rij, r2, rcut, |
411 |
– |
sw, vdwMult, vpair, pot, f1); |
412 |
– |
} |
319 |
|
} |