| 53 |
|
#include "primitives/Molecule.hpp" |
| 54 |
|
#include "utils/MemoryUtils.hpp" |
| 55 |
|
#include "utils/simError.h" |
| 56 |
+ |
#include "utils/StringUtils.hpp" |
| 57 |
|
|
| 58 |
|
namespace OpenMD { |
| 59 |
< |
Molecule::Molecule(int stampId, int globalIndex, const std::string& molName) |
| 60 |
< |
: stampId_(stampId), globalIndex_(globalIndex), moleculeName_(molName), |
| 61 |
< |
constrainTotalCharge_(false) { |
| 59 |
> |
Molecule::Molecule(int stampId, int globalIndex, const std::string& molName, |
| 60 |
> |
int region) : stampId_(stampId), |
| 61 |
> |
globalIndex_(globalIndex), |
| 62 |
> |
moleculeName_(molName), |
| 63 |
> |
region_(region), |
| 64 |
> |
constrainTotalCharge_(false) { |
| 65 |
|
} |
| 66 |
|
|
| 67 |
|
Molecule::~Molecule() { |
| 146 |
|
|
| 147 |
|
std::set<Atom*> rigidAtoms; |
| 148 |
|
Atom* atom; |
| 149 |
< |
AtomIterator ai; |
| 149 |
> |
Atom* atom1; |
| 150 |
> |
Atom* atom2; |
| 151 |
> |
AtomIterator ai, aj; |
| 152 |
|
RigidBody* rb; |
| 153 |
|
RigidBodyIterator rbIter; |
| 154 |
+ |
Bond* bond; |
| 155 |
+ |
BondIterator bi; |
| 156 |
|
|
| 157 |
|
// Get list of all the atoms that are part of rigid bodies |
| 158 |
|
|
| 187 |
|
fluctuatingCharges_.push_back( atom ); |
| 188 |
|
} |
| 189 |
|
|
| 182 |
– |
} |
| 190 |
|
|
| 191 |
+ |
// find the electronegative atoms and add them to the |
| 192 |
+ |
// hBondAcceptors_ vector: |
| 193 |
+ |
|
| 194 |
+ |
for (atom = beginAtom(ai); atom != NULL; atom = nextAtom(ai)) { |
| 195 |
+ |
AtomType* at = atom->getAtomType(); |
| 196 |
+ |
// get the chain of base types for this atom type: |
| 197 |
+ |
std::vector<AtomType*> ayb = at->allYourBase(); |
| 198 |
+ |
// use the last type in the chain of base types for the name: |
| 199 |
+ |
std::string bn = UpperCase(ayb[ayb.size()-1]->getName()); |
| 200 |
+ |
|
| 201 |
+ |
if (bn.compare("O")==0 || bn.compare("N")==0 |
| 202 |
+ |
|| bn.compare("F")==0) |
| 203 |
+ |
hBondAcceptors_.push_back( atom ); |
| 204 |
+ |
|
| 205 |
+ |
} |
| 206 |
+ |
|
| 207 |
+ |
// find electronegative atoms that are either bonded to |
| 208 |
+ |
// hydrogens or are present in the same rigid bodies: |
| 209 |
+ |
|
| 210 |
+ |
for (bond = beginBond(bi); bond != NULL; bond = nextBond(bi)) { |
| 211 |
+ |
Atom* atom1 = bond->getAtomA(); |
| 212 |
+ |
Atom* atom2 = bond->getAtomB(); |
| 213 |
+ |
AtomType* at1 = atom1->getAtomType(); |
| 214 |
+ |
AtomType* at2 = atom1->getAtomType(); |
| 215 |
+ |
// get the chain of base types for this atom type: |
| 216 |
+ |
std::vector<AtomType*> ayb1 = at1->allYourBase(); |
| 217 |
+ |
std::vector<AtomType*> ayb2 = at2->allYourBase(); |
| 218 |
+ |
// use the last type in the chain of base types for the name: |
| 219 |
+ |
std::string bn1 = UpperCase(ayb1[ayb1.size()-1]->getName()); |
| 220 |
+ |
std::string bn2 = UpperCase(ayb2[ayb2.size()-1]->getName()); |
| 221 |
+ |
|
| 222 |
+ |
if (bn1.compare("H")==0) { |
| 223 |
+ |
if (bn2.compare("O")==0 || bn2.compare("N")==0 |
| 224 |
+ |
|| bn2.compare("F")==0) { |
| 225 |
+ |
HBondDonor* donor = new HBondDonor(); |
| 226 |
+ |
donor->donorAtom = atom2; |
| 227 |
+ |
donor->donatedHydrogen = atom1; |
| 228 |
+ |
hBondDonors_.push_back( donor ); |
| 229 |
+ |
} |
| 230 |
+ |
} |
| 231 |
+ |
if (bn2.compare("H")==0) { |
| 232 |
+ |
if (bn1.compare("O")==0 || bn1.compare("N")==0 |
| 233 |
+ |
|| bn1.compare("F")==0) { |
| 234 |
+ |
HBondDonor* donor = new HBondDonor(); |
| 235 |
+ |
donor->donorAtom = atom1; |
| 236 |
+ |
donor->donatedHydrogen = atom2; |
| 237 |
+ |
hBondDonors_.push_back( donor ); |
| 238 |
+ |
} |
| 239 |
+ |
} |
| 240 |
+ |
} |
| 241 |
+ |
|
| 242 |
+ |
for (rb = beginRigidBody(rbIter); rb != NULL; |
| 243 |
+ |
rb = nextRigidBody(rbIter)) { |
| 244 |
+ |
for(atom1 = rb->beginAtom(ai); atom1 != NULL; |
| 245 |
+ |
atom1 = rb->nextAtom(ai)) { |
| 246 |
+ |
AtomType* at1 = atom1->getAtomType(); |
| 247 |
+ |
// get the chain of base types for this atom type: |
| 248 |
+ |
std::vector<AtomType*> ayb1 = at1->allYourBase(); |
| 249 |
+ |
// use the last type in the chain of base types for the name: |
| 250 |
+ |
std::string bn1 = UpperCase(ayb1[ayb1.size()-1]->getName()); |
| 251 |
+ |
|
| 252 |
+ |
if (bn1.compare("O")==0 || bn1.compare("N")==0 |
| 253 |
+ |
|| bn1.compare("F")==0) { |
| 254 |
+ |
for(atom2 = rb->beginAtom(aj); atom2 != NULL; |
| 255 |
+ |
atom2 = rb->nextAtom(aj)) { |
| 256 |
+ |
AtomType* at2 = atom2->getAtomType(); |
| 257 |
+ |
// get the chain of base types for this atom type: |
| 258 |
+ |
std::vector<AtomType*> ayb2 = at2->allYourBase(); |
| 259 |
+ |
// use the last type in the chain of base types for the name: |
| 260 |
+ |
std::string bn2 = UpperCase(ayb2[ayb2.size()-1]->getName()); |
| 261 |
+ |
if (bn2.compare("H")==0) { |
| 262 |
+ |
HBondDonor* donor = new HBondDonor(); |
| 263 |
+ |
donor->donorAtom = atom1; |
| 264 |
+ |
donor->donatedHydrogen = atom2; |
| 265 |
+ |
hBondDonors_.push_back( donor ); |
| 266 |
+ |
} |
| 267 |
+ |
} |
| 268 |
+ |
} |
| 269 |
+ |
} |
| 270 |
+ |
} |
| 271 |
+ |
} |
| 272 |
+ |
|
| 273 |
|
RealType Molecule::getMass() { |
| 274 |
|
StuntDouble* sd; |
| 275 |
|
std::vector<StuntDouble*>::iterator i; |
| 301 |
|
|
| 302 |
|
return com; |
| 303 |
|
} |
| 304 |
+ |
|
| 305 |
+ |
Vector3d Molecule::getCom(int snapshotNo) { |
| 306 |
+ |
StuntDouble* sd; |
| 307 |
+ |
std::vector<StuntDouble*>::iterator i; |
| 308 |
+ |
Vector3d com; |
| 309 |
+ |
RealType totalMass = 0; |
| 310 |
+ |
RealType mass; |
| 311 |
+ |
|
| 312 |
+ |
for (sd = beginIntegrableObject(i); sd != NULL; sd = |
| 313 |
+ |
nextIntegrableObject(i)){ |
| 314 |
+ |
mass = sd->getMass(); |
| 315 |
+ |
totalMass += mass; |
| 316 |
+ |
com += sd->getPos(snapshotNo) * mass; |
| 317 |
+ |
} |
| 318 |
+ |
|
| 319 |
+ |
com /= totalMass; |
| 320 |
|
|
| 321 |
+ |
return com; |
| 322 |
+ |
} |
| 323 |
+ |
|
| 324 |
|
void Molecule::moveCom(const Vector3d& delta) { |
| 325 |
|
StuntDouble* sd; |
| 326 |
|
std::vector<StuntDouble*>::iterator i; |
