| 44 |
|
|
| 45 |
|
#include <cmath> |
| 46 |
|
#include "nonbonded/GB.hpp" |
| 47 |
– |
#include "nonbonded/LJ.hpp" |
| 47 |
|
#include "utils/simError.h" |
| 48 |
|
|
| 49 |
|
using namespace std; |
| 50 |
|
namespace OpenMD { |
| 51 |
|
|
| 52 |
< |
bool GB::initialized_ = false; |
| 54 |
< |
RealType GB::mu_ = 2.0; |
| 55 |
< |
RealType GB::nu_ = 1.0; |
| 56 |
< |
ForceField* GB::forceField_ = NULL; |
| 57 |
< |
map<int, AtomType*> GB::GBMap; |
| 58 |
< |
map<pair<AtomType*, AtomType*>, GBInteractionData> GB::MixingMap; |
| 52 |
> |
GB::GB() : name_("GB"), initialized_(false), mu_(2.0), nu_(1.0), forceField_(NULL) {} |
| 53 |
|
|
| 60 |
– |
GB* GB::_instance = NULL; |
| 61 |
– |
|
| 62 |
– |
GB* GB::Instance() { |
| 63 |
– |
if (!_instance) { |
| 64 |
– |
_instance = new GB(); |
| 65 |
– |
} |
| 66 |
– |
return _instance; |
| 67 |
– |
} |
| 68 |
– |
|
| 54 |
|
GayBerneParam GB::getGayBerneParam(AtomType* atomType) { |
| 55 |
|
|
| 56 |
|
// Do sanity checking on the AtomType we were passed before |
| 90 |
|
return gbData->getData(); |
| 91 |
|
} |
| 92 |
|
|
| 93 |
+ |
LJParam GB::getLJParam(AtomType* atomType) { |
| 94 |
+ |
|
| 95 |
+ |
// Do sanity checking on the AtomType we were passed before |
| 96 |
+ |
// building any data structures: |
| 97 |
+ |
if (!atomType->isLennardJones()) { |
| 98 |
+ |
sprintf( painCave.errMsg, |
| 99 |
+ |
"GB::getLJParam was passed an atomType (%s) that does not\n" |
| 100 |
+ |
"\tappear to be a Lennard-Jones atom.\n", |
| 101 |
+ |
atomType->getName().c_str()); |
| 102 |
+ |
painCave.severity = OPENMD_ERROR; |
| 103 |
+ |
painCave.isFatal = 1; |
| 104 |
+ |
simError(); |
| 105 |
+ |
} |
| 106 |
+ |
|
| 107 |
+ |
GenericData* data = atomType->getPropertyByName("LennardJones"); |
| 108 |
+ |
if (data == NULL) { |
| 109 |
+ |
sprintf( painCave.errMsg, "GB::getLJParam could not find Lennard-Jones\n" |
| 110 |
+ |
"\tparameters for atomType %s.\n", atomType->getName().c_str()); |
| 111 |
+ |
painCave.severity = OPENMD_ERROR; |
| 112 |
+ |
painCave.isFatal = 1; |
| 113 |
+ |
simError(); |
| 114 |
+ |
} |
| 115 |
+ |
|
| 116 |
+ |
LJParamGenericData* ljData = dynamic_cast<LJParamGenericData*>(data); |
| 117 |
+ |
if (ljData == NULL) { |
| 118 |
+ |
sprintf( painCave.errMsg, |
| 119 |
+ |
"GB::getLJParam could not convert GenericData to LJParam for\n" |
| 120 |
+ |
"\tatom type %s\n", atomType->getName().c_str()); |
| 121 |
+ |
painCave.severity = OPENMD_ERROR; |
| 122 |
+ |
painCave.isFatal = 1; |
| 123 |
+ |
simError(); |
| 124 |
+ |
} |
| 125 |
+ |
|
| 126 |
+ |
return ljData->getData(); |
| 127 |
+ |
} |
| 128 |
+ |
|
| 129 |
+ |
RealType GB::getLJEpsilon(AtomType* atomType) { |
| 130 |
+ |
LJParam ljParam = getLJParam(atomType); |
| 131 |
+ |
return ljParam.epsilon; |
| 132 |
+ |
} |
| 133 |
+ |
RealType GB::getLJSigma(AtomType* atomType) { |
| 134 |
+ |
LJParam ljParam = getLJParam(atomType); |
| 135 |
+ |
return ljParam.sigma; |
| 136 |
+ |
} |
| 137 |
+ |
|
| 138 |
|
void GB::initialize() { |
| 139 |
+ |
|
| 140 |
+ |
ForceFieldOptions& fopts = forceField_->getForceFieldOptions(); |
| 141 |
+ |
mu_ = fopts.getGayBerneMu(); |
| 142 |
+ |
nu_ = fopts.getGayBerneNu(); |
| 143 |
|
ForceField::AtomTypeContainer* atomTypes = forceField_->getAtomTypes(); |
| 144 |
|
ForceField::AtomTypeContainer::MapTypeIterator i; |
| 145 |
|
AtomType* at; |
| 182 |
|
er1 = gb1.GB_eps_ratio; |
| 183 |
|
dw1 = gb1.GB_dw; |
| 184 |
|
} else if (atomType->isLennardJones()) { |
| 185 |
< |
d1 = LJ::Instance()->getSigma(atomType) / sqrt(2.0); |
| 186 |
< |
e1 = LJ::Instance()->getEpsilon(atomType); |
| 185 |
> |
d1 = getLJSigma(atomType) / sqrt(2.0); |
| 186 |
> |
e1 = getLJEpsilon(atomType); |
| 187 |
|
l1 = d1; |
| 188 |
|
er1 = 1.0; |
| 189 |
|
dw1 = 1.0; |
| 215 |
|
er2 = gb2.GB_eps_ratio; |
| 216 |
|
dw2 = gb2.GB_dw; |
| 217 |
|
} else if (atype2->isLennardJones()) { |
| 218 |
< |
d2 = LJ::Instance()->getSigma(atype2) / sqrt(2.0); |
| 219 |
< |
e2 = LJ::Instance()->getEpsilon(atype2); |
| 218 |
> |
d2 = getLJSigma(atype2) / sqrt(2.0); |
| 219 |
> |
e2 = getLJEpsilon(atype2); |
| 220 |
|
l2 = d2; |
| 221 |
|
er2 = 1.0; |
| 222 |
|
dw2 = 1.0; |
| 239 |
|
mixer.eps0 = sqrt(e1 * e2); |
| 240 |
|
|
| 241 |
|
RealType er = sqrt(er1 * er2); |
| 242 |
< |
RealType ermu = pow(er,(1.0 / mu_)); |
| 242 |
> |
RealType ermu = pow(er, (1.0 / mu_)); |
| 243 |
|
RealType xp = (1.0 - ermu) / (1.0 + ermu); |
| 244 |
|
RealType ap2 = 1.0 / (1.0 + ermu); |
| 245 |
|
|
| 247 |
|
mixer.xpap2 = xp * ap2; |
| 248 |
|
mixer.xpapi2 = xp / ap2; |
| 249 |
|
|
| 250 |
+ |
cerr << "mixer" << er1 << " " << er2 << " " << mu_ << " " << ermu << " " << xp <<" " << ap2 << "\n"; |
| 251 |
+ |
|
| 252 |
|
// only add this pairing if at least one of the atoms is a Gay-Berne atom |
| 253 |
|
|
| 254 |
|
if (atomType->isGayBerne() || atype2->isGayBerne()) { |
| 264 |
|
} |
| 265 |
|
} |
| 266 |
|
} |
| 267 |
< |
|
| 267 |
> |
|
| 268 |
> |
void GB::calcForce(InteractionData &idat) { |
| 269 |
|
|
| 233 |
– |
RealType GB::getGayBerneCut(int atid) { |
| 270 |
|
if (!initialized_) initialize(); |
| 235 |
– |
std::map<int, AtomType*> :: const_iterator it; |
| 236 |
– |
it = GBMap.find(atid); |
| 237 |
– |
if (it == GBMap.end()) { |
| 238 |
– |
sprintf( painCave.errMsg, |
| 239 |
– |
"GB::getGayBerneCut could not find atid %d in GBMap\n", |
| 240 |
– |
(atid)); |
| 241 |
– |
painCave.severity = OPENMD_ERROR; |
| 242 |
– |
painCave.isFatal = 1; |
| 243 |
– |
simError(); |
| 244 |
– |
} |
| 245 |
– |
|
| 246 |
– |
AtomType* atype = it->second; |
| 247 |
– |
|
| 248 |
– |
RealType gbCut; |
| 271 |
|
|
| 272 |
< |
if (atype->isGayBerne()) { |
| 251 |
< |
GayBerneParam gb = getGayBerneParam(atype); |
| 272 |
> |
GBInteractionData mixer = MixingMap[idat.atypes]; |
| 273 |
|
|
| 253 |
– |
// sigma is actually sqrt(2) * l for prolate ellipsoids |
| 254 |
– |
gbCut = 2.5 * sqrt(2.0) * max(gb.GB_l, gb.GB_d); |
| 255 |
– |
|
| 256 |
– |
} else if (atype->isLennardJones()) { |
| 257 |
– |
gbCut = 2.5 * LJ::Instance()->getSigma(atype); |
| 258 |
– |
} |
| 259 |
– |
|
| 260 |
– |
return gbCut; |
| 261 |
– |
} |
| 262 |
– |
|
| 263 |
– |
|
| 264 |
– |
void GB::calcForce(AtomType* at1, AtomType* at2, Vector3d d, |
| 265 |
– |
RealType r, RealType r2, RealType sw, |
| 266 |
– |
RealType vdwMult, RealType &vpair, RealType &pot, |
| 267 |
– |
RotMat3x3d A1, RotMat3x3d A2, Vector3d &f1, |
| 268 |
– |
Vector3d &t1, Vector3d &t2) { |
| 269 |
– |
|
| 270 |
– |
if (!initialized_) initialize(); |
| 271 |
– |
|
| 272 |
– |
pair<AtomType*, AtomType*> key = make_pair(at1, at2); |
| 273 |
– |
GBInteractionData mixer = MixingMap[key]; |
| 274 |
– |
|
| 274 |
|
RealType sigma0 = mixer.sigma0; |
| 275 |
|
RealType dw = mixer.dw; |
| 276 |
|
RealType eps0 = mixer.eps0; |
| 281 |
|
RealType xpap2 = mixer.xpap2; |
| 282 |
|
RealType xpapi2 = mixer.xpapi2; |
| 283 |
|
|
| 284 |
< |
Vector3d ul1 = A1.getColumn(2); |
| 285 |
< |
Vector3d ul2 = A2.getColumn(2); |
| 284 |
> |
Vector3d ul1 = idat.A1->getRow(2); |
| 285 |
> |
Vector3d ul2 = idat.A2->getRow(2); |
| 286 |
|
|
| 287 |
|
RealType a, b, g; |
| 288 |
|
|
| 289 |
< |
bool i_is_LJ = at1->isLennardJones(); |
| 290 |
< |
bool j_is_LJ = at2->isLennardJones(); |
| 291 |
< |
|
| 289 |
> |
bool i_is_LJ = idat.atypes.first->isLennardJones(); |
| 290 |
> |
bool j_is_LJ = idat.atypes.second->isLennardJones(); |
| 291 |
> |
|
| 292 |
|
if (i_is_LJ) { |
| 293 |
|
a = 0.0; |
| 294 |
|
ul1 = V3Zero; |
| 295 |
|
} else { |
| 296 |
< |
a = dot(d, ul1); |
| 296 |
> |
a = dot(*(idat.d), ul1); |
| 297 |
|
} |
| 298 |
|
|
| 299 |
|
if (j_is_LJ) { |
| 300 |
|
b = 0.0; |
| 301 |
|
ul2 = V3Zero; |
| 302 |
|
} else { |
| 303 |
< |
b = dot(d, ul2); |
| 303 |
> |
b = dot(*(idat.d), ul2); |
| 304 |
|
} |
| 305 |
|
|
| 306 |
|
if (i_is_LJ || j_is_LJ) |
| 308 |
|
else |
| 309 |
|
g = dot(ul1, ul2); |
| 310 |
|
|
| 311 |
< |
RealType au = a / r; |
| 312 |
< |
RealType bu = b / r; |
| 311 |
> |
cerr << "in GB, d = " << *(idat.d) << "\n"; |
| 312 |
> |
cerr << "abg = " << a << " " << b << " " << g <<"\n"; |
| 313 |
> |
|
| 314 |
> |
RealType au = a / *(idat.rij); |
| 315 |
> |
RealType bu = b / *(idat.rij); |
| 316 |
|
|
| 317 |
|
RealType au2 = au * au; |
| 318 |
|
RealType bu2 = bu * bu; |
| 320 |
|
|
| 321 |
|
RealType H = (xa2 * au2 + xai2 * bu2 - 2.0*x2*au*bu*g) / (1.0 - x2*g2); |
| 322 |
|
RealType Hp = (xpap2*au2 + xpapi2*bu2 - 2.0*xp2*au*bu*g) / (1.0 - xp2*g2); |
| 323 |
+ |
cerr << "xa2, xai2 " << xa2 << " " << xai2 << "\n"; |
| 324 |
+ |
cerr << "xpap2, xpapi2 " << xpap2 << " " << xpapi2 << "\n"; |
| 325 |
+ |
cerr << "H Hp = " << H << " " << Hp << "\n"; |
| 326 |
|
|
| 327 |
|
RealType sigma = sigma0 / sqrt(1.0 - H); |
| 328 |
|
RealType e1 = 1.0 / sqrt(1.0 - x2*g2); |
| 329 |
|
RealType e2 = 1.0 - Hp; |
| 330 |
|
RealType eps = eps0 * pow(e1,nu_) * pow(e2,mu_); |
| 331 |
< |
RealType BigR = dw*sigma0 / (r - sigma + dw*sigma0); |
| 331 |
> |
cerr << "eps = " << eps0 << " " << e1 << " " << nu_ << " " << e2 << " " << mu_ << "\n"; |
| 332 |
> |
RealType BigR = dw*sigma0 / (*(idat.rij) - sigma + dw*sigma0); |
| 333 |
|
|
| 334 |
|
RealType R3 = BigR*BigR*BigR; |
| 335 |
|
RealType R6 = R3*R3; |
| 337 |
|
RealType R12 = R6*R6; |
| 338 |
|
RealType R13 = R6*R7; |
| 339 |
|
|
| 340 |
< |
RealType U = vdwMult * 4.0 * eps * (R12 - R6); |
| 340 |
> |
RealType U = *(idat.vdwMult) * 4.0 * eps * (R12 - R6); |
| 341 |
|
|
| 342 |
+ |
cerr << "R12, R6, eps = " << R12 << " " << R6 << " " << eps << " " << *(idat.vdwMult) << "\n"; |
| 343 |
+ |
|
| 344 |
|
RealType s3 = sigma*sigma*sigma; |
| 345 |
|
RealType s03 = sigma0*sigma0*sigma0; |
| 346 |
|
|
| 347 |
< |
RealType pref1 = - vdwMult * 8.0 * eps * mu_ * (R12 - R6) / (e2 * r); |
| 347 |
> |
RealType pref1 = - *(idat.vdwMult) * 8.0 * eps * mu_ * (R12 - R6) / |
| 348 |
> |
(e2 * *(idat.rij)); |
| 349 |
|
|
| 350 |
< |
RealType pref2 = vdwMult * 8.0 * eps * s3 * (6.0*R13 - 3.0*R7) /(dw*r*s03); |
| 350 |
> |
RealType pref2 = *(idat.vdwMult) * 8.0 * eps * s3 * (6.0*R13 - 3.0*R7) / |
| 351 |
> |
(dw* *(idat.rij) * s03); |
| 352 |
|
|
| 353 |
< |
RealType dUdr = - (pref1 * Hp + pref2 * (sigma0*sigma0*r/s3 + H)); |
| 353 |
> |
RealType dUdr = - (pref1 * Hp + pref2 * (sigma0 * sigma0 * |
| 354 |
> |
*(idat.rij) / s3 + H)); |
| 355 |
|
|
| 356 |
|
RealType dUda = pref1 * (xpap2*au - xp2*bu*g) / (1.0 - xp2 * g2) |
| 357 |
|
+ pref2 * (xa2 * au - x2 *bu*g) / (1.0 - x2 * g2); |
| 363 |
|
+ 8.0 * eps * mu_ * (R12 - R6) * (xp2*au*bu - Hp*xp2*g) / |
| 364 |
|
(1.0 - xp2 * g2) / e2 + 8.0 * eps * s3 * (3.0 * R7 - 6.0 * R13) * |
| 365 |
|
(x2 * au * bu - H * x2 * g) / (1.0 - x2 * g2) / (dw * s03); |
| 355 |
– |
|
| 366 |
|
|
| 367 |
< |
Vector3d rhat = d / r; |
| 368 |
< |
Vector3d rxu1 = cross(d, ul1); |
| 369 |
< |
Vector3d rxu2 = cross(d, ul2); |
| 367 |
> |
cerr << pref1 << " " << pref2 << " " << dUdr <<" " << dUda << " " << dUdb << dUdg << "\n"; |
| 368 |
> |
|
| 369 |
> |
Vector3d rhat = *(idat.d) / *(idat.rij); |
| 370 |
> |
Vector3d rxu1 = cross(*(idat.d), ul1); |
| 371 |
> |
Vector3d rxu2 = cross(*(idat.d), ul2); |
| 372 |
|
Vector3d uxu = cross(ul1, ul2); |
| 373 |
+ |
|
| 374 |
+ |
cerr << "U = " << U << "\n"; |
| 375 |
+ |
cerr << "f1 = " << dUdr * rhat + dUda * ul1 + dUdb * ul2 << "\n"; |
| 376 |
+ |
cerr << "t1 = " << dUda * rxu1 - dUdg * uxu << "\n"; |
| 377 |
+ |
cerr << "t2 = " << dUdb * rxu2 - dUdg * uxu << "\n"; |
| 378 |
+ |
|
| 379 |
|
|
| 380 |
< |
pot += U*sw; |
| 381 |
< |
f1 += dUdr * rhat + dUda * ul1 + dUdb * ul2; |
| 382 |
< |
t1 += dUda * rxu1 - dUdg * uxu; |
| 383 |
< |
t2 += dUdb * rxu2 - dUdg * uxu; |
| 384 |
< |
vpair += U*sw; |
| 380 |
> |
(*(idat.pot))[VANDERWAALS_FAMILY] += U * *(idat.sw); |
| 381 |
> |
*(idat.f1) += dUdr * rhat + dUda * ul1 + dUdb * ul2; |
| 382 |
> |
*(idat.t1) += dUda * rxu1 - dUdg * uxu; |
| 383 |
> |
*(idat.t2) += dUdb * rxu2 - dUdg * uxu; |
| 384 |
> |
*(idat.vpair) += U * *(idat.sw); |
| 385 |
|
|
| 386 |
|
return; |
| 387 |
|
|
| 388 |
|
} |
| 389 |
|
|
| 390 |
< |
void GB::do_gb_pair(int *atid1, int *atid2, RealType *d, RealType *r, |
| 391 |
< |
RealType *r2, RealType *sw, RealType *vdwMult, |
| 374 |
< |
RealType *vpair, RealType *pot, RealType *A1, |
| 375 |
< |
RealType *A2, RealType *f1, RealType *t1, RealType *t2) { |
| 376 |
< |
|
| 377 |
< |
if (!initialized_) initialize(); |
| 378 |
< |
|
| 379 |
< |
AtomType* atype1 = GBMap[*atid1]; |
| 380 |
< |
AtomType* atype2 = GBMap[*atid2]; |
| 381 |
< |
|
| 382 |
< |
Vector3d disp(d); |
| 383 |
< |
Vector3d frc(f1); |
| 384 |
< |
Vector3d trq1(t1); |
| 385 |
< |
Vector3d trq2(t2); |
| 386 |
< |
RotMat3x3d Ai(A1); |
| 387 |
< |
RotMat3x3d Aj(A2); |
| 388 |
< |
|
| 389 |
< |
// Fortran has the opposite matrix ordering from c++, so we'll use |
| 390 |
< |
// transpose here. When we finish the conversion to C++, this wrapper |
| 391 |
< |
// will disappear, as will the transpose below: |
| 390 |
> |
RealType GB::getSuggestedCutoffRadius(pair<AtomType*, AtomType*> atypes) { |
| 391 |
> |
if (!initialized_) initialize(); |
| 392 |
|
|
| 393 |
< |
calcForce(atype1, atype2, disp, *r, *r2, *sw, *vdwMult, *vpair, *pot, |
| 394 |
< |
Ai.transpose(), Aj.transpose(), frc, trq1, trq1); |
| 395 |
< |
|
| 396 |
< |
f1[0] = frc.x(); |
| 397 |
< |
f1[1] = frc.y(); |
| 398 |
< |
f1[2] = frc.z(); |
| 393 |
> |
RealType cut = 0.0; |
| 394 |
|
|
| 395 |
< |
t1[0] = trq1.x(); |
| 396 |
< |
t1[1] = trq1.y(); |
| 397 |
< |
t1[2] = trq1.z(); |
| 395 |
> |
if (atypes.first->isGayBerne()) { |
| 396 |
> |
GayBerneParam gb1 = getGayBerneParam(atypes.first); |
| 397 |
> |
RealType d1 = gb1.GB_d; |
| 398 |
> |
RealType l1 = gb1.GB_l; |
| 399 |
> |
// sigma is actually sqrt(2)*l for prolate ellipsoids |
| 400 |
> |
cut = max(cut, 2.5 * sqrt(2.0) * max(d1, l1)); |
| 401 |
> |
} else if (atypes.first->isLennardJones()) { |
| 402 |
> |
cut = max(cut, 2.5 * getLJSigma(atypes.first)); |
| 403 |
> |
} |
| 404 |
|
|
| 405 |
< |
t2[0] = trq2.x(); |
| 406 |
< |
t2[1] = trq2.y(); |
| 407 |
< |
t2[2] = trq2.z(); |
| 408 |
< |
|
| 409 |
< |
return; |
| 405 |
> |
if (atypes.second->isGayBerne()) { |
| 406 |
> |
GayBerneParam gb2 = getGayBerneParam(atypes.second); |
| 407 |
> |
RealType d2 = gb2.GB_d; |
| 408 |
> |
RealType l2 = gb2.GB_l; |
| 409 |
> |
cut = max(cut, 2.5 * sqrt(2.0) * max(d2, l2)); |
| 410 |
> |
} else if (atypes.second->isLennardJones()) { |
| 411 |
> |
cut = max(cut, 2.5 * getLJSigma(atypes.second)); |
| 412 |
> |
} |
| 413 |
> |
|
| 414 |
> |
return cut; |
| 415 |
|
} |
| 416 |
|
} |
| 417 |
|
|
| 412 |
– |
extern "C" { |
| 413 |
– |
|
| 414 |
– |
#define fortranGetGayBerneCut FC_FUNC(getgaybernecut, GETGAYBERNECUT) |
| 415 |
– |
#define fortranDoGBPair FC_FUNC(do_gb_pair, DO_GB_PAIR) |
| 416 |
– |
|
| 417 |
– |
RealType fortranGetGayBerneCut(int* atid) { |
| 418 |
– |
return OpenMD::GB::Instance()->getGayBerneCut(*atid); |
| 419 |
– |
} |
| 420 |
– |
|
| 421 |
– |
void fortranDoGBPair(int *atid1, int *atid2, RealType *d, RealType *r, |
| 422 |
– |
RealType *r2, RealType *sw, RealType *vdwMult, |
| 423 |
– |
RealType *vpair, RealType *pot, RealType *A1, |
| 424 |
– |
RealType *A2, RealType *f1, RealType *t1, RealType *t2){ |
| 425 |
– |
|
| 426 |
– |
return OpenMD::GB::Instance()->do_gb_pair(atid1, atid2, d, r, r2, sw, |
| 427 |
– |
vdwMult, vpair, pot, A1, A2, f1, |
| 428 |
– |
t1, t2); |
| 429 |
– |
} |
| 430 |
– |
} |
