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). |
39 |
> |
* [4] Kuang & Gezelter, J. Chem. Phys. 133, 164101 (2010). |
40 |
> |
* [5] Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011). |
41 |
|
*/ |
42 |
|
|
43 |
|
/** |
59 |
|
#include "utils/simError.h" |
60 |
|
#include "selection/SelectionManager.hpp" |
61 |
|
#include "io/ForceFieldOptions.hpp" |
62 |
< |
#include "UseTheForce/ForceField.hpp" |
62 |
> |
#include "brains/ForceField.hpp" |
63 |
|
#include "nonbonded/SwitchingFunction.hpp" |
64 |
+ |
#ifdef IS_MPI |
65 |
+ |
#include <mpi.h> |
66 |
+ |
#endif |
67 |
|
|
68 |
|
using namespace std; |
69 |
|
namespace OpenMD { |
72 |
|
forceField_(ff), simParams_(simParams), |
73 |
|
ndf_(0), fdf_local(0), ndfRaw_(0), ndfTrans_(0), nZconstraint_(0), |
74 |
|
nGlobalMols_(0), nGlobalAtoms_(0), nGlobalCutoffGroups_(0), |
75 |
< |
nGlobalIntegrableObjects_(0), nGlobalRigidBodies_(0), |
75 |
> |
nGlobalIntegrableObjects_(0), nGlobalRigidBodies_(0), nGlobalFluctuatingCharges_(0), |
76 |
|
nAtoms_(0), nBonds_(0), nBends_(0), nTorsions_(0), nInversions_(0), |
77 |
|
nRigidBodies_(0), nIntegrableObjects_(0), nCutoffGroups_(0), |
78 |
< |
nConstraints_(0), sman_(NULL), topologyDone_(false), |
78 |
> |
nConstraints_(0), nFluctuatingCharges_(0), sman_(NULL), topologyDone_(false), |
79 |
|
calcBoxDipole_(false), useAtomicVirial_(true) { |
80 |
|
|
81 |
|
MoleculeStamp* molStamp; |
129 |
|
//equal to the total number of atoms minus number of atoms belong to |
130 |
|
//cutoff group defined in meta-data file plus the number of cutoff |
131 |
|
//groups defined in meta-data file |
128 |
– |
std::cerr << "nGA = " << nGlobalAtoms_ << "\n"; |
129 |
– |
std::cerr << "nCA = " << nCutoffAtoms << "\n"; |
130 |
– |
std::cerr << "nG = " << nGroups << "\n"; |
132 |
|
|
133 |
|
nGlobalCutoffGroups_ = nGlobalAtoms_ - nCutoffAtoms + nGroups; |
133 |
– |
|
134 |
– |
std::cerr << "nGCG = " << nGlobalCutoffGroups_ << "\n"; |
134 |
|
|
135 |
|
//every free atom (atom does not belong to rigid bodies) is an |
136 |
|
//integrable object therefore the total number of integrable objects |
225 |
|
|
226 |
|
|
227 |
|
void SimInfo::calcNdf() { |
228 |
< |
int ndf_local; |
228 |
> |
int ndf_local, nfq_local; |
229 |
|
MoleculeIterator i; |
230 |
|
vector<StuntDouble*>::iterator j; |
231 |
+ |
vector<Atom*>::iterator k; |
232 |
+ |
|
233 |
|
Molecule* mol; |
234 |
< |
StuntDouble* integrableObject; |
234 |
> |
StuntDouble* sd; |
235 |
> |
Atom* atom; |
236 |
|
|
237 |
|
ndf_local = 0; |
238 |
+ |
nfq_local = 0; |
239 |
|
|
240 |
|
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
238 |
– |
for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL; |
239 |
– |
integrableObject = mol->nextIntegrableObject(j)) { |
241 |
|
|
242 |
+ |
for (sd = mol->beginIntegrableObject(j); sd != NULL; |
243 |
+ |
sd = mol->nextIntegrableObject(j)) { |
244 |
+ |
|
245 |
|
ndf_local += 3; |
246 |
|
|
247 |
< |
if (integrableObject->isDirectional()) { |
248 |
< |
if (integrableObject->isLinear()) { |
247 |
> |
if (sd->isDirectional()) { |
248 |
> |
if (sd->isLinear()) { |
249 |
|
ndf_local += 2; |
250 |
|
} else { |
251 |
|
ndf_local += 3; |
252 |
|
} |
253 |
|
} |
250 |
– |
|
254 |
|
} |
255 |
+ |
|
256 |
+ |
for (atom = mol->beginFluctuatingCharge(k); atom != NULL; |
257 |
+ |
atom = mol->nextFluctuatingCharge(k)) { |
258 |
+ |
if (atom->isFluctuatingCharge()) { |
259 |
+ |
nfq_local++; |
260 |
+ |
} |
261 |
+ |
} |
262 |
|
} |
263 |
|
|
264 |
+ |
ndfLocal_ = ndf_local; |
265 |
+ |
|
266 |
|
// n_constraints is local, so subtract them on each processor |
267 |
|
ndf_local -= nConstraints_; |
268 |
|
|
269 |
|
#ifdef IS_MPI |
270 |
|
MPI_Allreduce(&ndf_local,&ndf_,1,MPI_INT,MPI_SUM, MPI_COMM_WORLD); |
271 |
+ |
MPI_Allreduce(&nfq_local,&nGlobalFluctuatingCharges_,1, MPI_INT, MPI_SUM, MPI_COMM_WORLD); |
272 |
|
#else |
273 |
|
ndf_ = ndf_local; |
274 |
+ |
nGlobalFluctuatingCharges_ = nfq_local; |
275 |
|
#endif |
276 |
|
|
277 |
|
// nZconstraints_ is global, as are the 3 COM translations for the |
288 |
|
#endif |
289 |
|
return fdf_; |
290 |
|
} |
291 |
+ |
|
292 |
+ |
unsigned int SimInfo::getNLocalCutoffGroups(){ |
293 |
+ |
int nLocalCutoffAtoms = 0; |
294 |
+ |
Molecule* mol; |
295 |
+ |
MoleculeIterator mi; |
296 |
+ |
CutoffGroup* cg; |
297 |
+ |
Molecule::CutoffGroupIterator ci; |
298 |
|
|
299 |
+ |
for (mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) { |
300 |
+ |
|
301 |
+ |
for (cg = mol->beginCutoffGroup(ci); cg != NULL; |
302 |
+ |
cg = mol->nextCutoffGroup(ci)) { |
303 |
+ |
nLocalCutoffAtoms += cg->getNumAtom(); |
304 |
+ |
|
305 |
+ |
} |
306 |
+ |
} |
307 |
+ |
|
308 |
+ |
return nAtoms_ - nLocalCutoffAtoms + nCutoffGroups_; |
309 |
+ |
} |
310 |
+ |
|
311 |
|
void SimInfo::calcNdfRaw() { |
312 |
|
int ndfRaw_local; |
313 |
|
|
314 |
|
MoleculeIterator i; |
315 |
|
vector<StuntDouble*>::iterator j; |
316 |
|
Molecule* mol; |
317 |
< |
StuntDouble* integrableObject; |
317 |
> |
StuntDouble* sd; |
318 |
|
|
319 |
|
// Raw degrees of freedom that we have to set |
320 |
|
ndfRaw_local = 0; |
321 |
|
|
322 |
|
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
290 |
– |
for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL; |
291 |
– |
integrableObject = mol->nextIntegrableObject(j)) { |
323 |
|
|
324 |
+ |
for (sd = mol->beginIntegrableObject(j); sd != NULL; |
325 |
+ |
sd = mol->nextIntegrableObject(j)) { |
326 |
+ |
|
327 |
|
ndfRaw_local += 3; |
328 |
|
|
329 |
< |
if (integrableObject->isDirectional()) { |
330 |
< |
if (integrableObject->isLinear()) { |
329 |
> |
if (sd->isDirectional()) { |
330 |
> |
if (sd->isLinear()) { |
331 |
|
ndfRaw_local += 2; |
332 |
|
} else { |
333 |
|
ndfRaw_local += 3; |
387 |
|
Molecule::RigidBodyIterator rbIter; |
388 |
|
RigidBody* rb; |
389 |
|
Molecule::IntegrableObjectIterator ii; |
390 |
< |
StuntDouble* integrableObject; |
390 |
> |
StuntDouble* sd; |
391 |
|
|
392 |
< |
for (integrableObject = mol->beginIntegrableObject(ii); |
393 |
< |
integrableObject != NULL; |
360 |
< |
integrableObject = mol->nextIntegrableObject(ii)) { |
392 |
> |
for (sd = mol->beginIntegrableObject(ii); sd != NULL; |
393 |
> |
sd = mol->nextIntegrableObject(ii)) { |
394 |
|
|
395 |
< |
if (integrableObject->isRigidBody()) { |
396 |
< |
rb = static_cast<RigidBody*>(integrableObject); |
395 |
> |
if (sd->isRigidBody()) { |
396 |
> |
rb = static_cast<RigidBody*>(sd); |
397 |
|
vector<Atom*> atoms = rb->getAtoms(); |
398 |
|
set<int> rigidAtoms; |
399 |
|
for (int i = 0; i < static_cast<int>(atoms.size()); ++i) { |
404 |
|
} |
405 |
|
} else { |
406 |
|
set<int> oneAtomSet; |
407 |
< |
oneAtomSet.insert(integrableObject->getGlobalIndex()); |
408 |
< |
atomGroups.insert(map<int, set<int> >::value_type(integrableObject->getGlobalIndex(), oneAtomSet)); |
407 |
> |
oneAtomSet.insert(sd->getGlobalIndex()); |
408 |
> |
atomGroups.insert(map<int, set<int> >::value_type(sd->getGlobalIndex(), oneAtomSet)); |
409 |
|
} |
410 |
|
} |
411 |
|
|
539 |
|
Molecule::RigidBodyIterator rbIter; |
540 |
|
RigidBody* rb; |
541 |
|
Molecule::IntegrableObjectIterator ii; |
542 |
< |
StuntDouble* integrableObject; |
542 |
> |
StuntDouble* sd; |
543 |
|
|
544 |
< |
for (integrableObject = mol->beginIntegrableObject(ii); |
545 |
< |
integrableObject != NULL; |
513 |
< |
integrableObject = mol->nextIntegrableObject(ii)) { |
544 |
> |
for (sd = mol->beginIntegrableObject(ii); sd != NULL; |
545 |
> |
sd = mol->nextIntegrableObject(ii)) { |
546 |
|
|
547 |
< |
if (integrableObject->isRigidBody()) { |
548 |
< |
rb = static_cast<RigidBody*>(integrableObject); |
547 |
> |
if (sd->isRigidBody()) { |
548 |
> |
rb = static_cast<RigidBody*>(sd); |
549 |
|
vector<Atom*> atoms = rb->getAtoms(); |
550 |
|
set<int> rigidAtoms; |
551 |
|
for (int i = 0; i < static_cast<int>(atoms.size()); ++i) { |
556 |
|
} |
557 |
|
} else { |
558 |
|
set<int> oneAtomSet; |
559 |
< |
oneAtomSet.insert(integrableObject->getGlobalIndex()); |
560 |
< |
atomGroups.insert(map<int, set<int> >::value_type(integrableObject->getGlobalIndex(), oneAtomSet)); |
559 |
> |
oneAtomSet.insert(sd->getGlobalIndex()); |
560 |
> |
atomGroups.insert(map<int, set<int> >::value_type(sd->getGlobalIndex(), oneAtomSet)); |
561 |
|
} |
562 |
|
} |
563 |
|
|
712 |
|
Atom* atom; |
713 |
|
set<AtomType*> atomTypes; |
714 |
|
|
715 |
< |
for(mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) { |
716 |
< |
for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) { |
715 |
> |
for(mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) { |
716 |
> |
for(atom = mol->beginAtom(ai); atom != NULL; |
717 |
> |
atom = mol->nextAtom(ai)) { |
718 |
|
atomTypes.insert(atom->getAtomType()); |
719 |
|
} |
720 |
|
} |
721 |
< |
|
721 |
> |
|
722 |
|
#ifdef IS_MPI |
723 |
|
|
724 |
|
// loop over the found atom types on this processor, and add their |
725 |
|
// numerical idents to a vector: |
726 |
< |
|
726 |
> |
|
727 |
|
vector<int> foundTypes; |
728 |
|
set<AtomType*>::iterator i; |
729 |
|
for (i = atomTypes.begin(); i != atomTypes.end(); ++i) |
732 |
|
// count_local holds the number of found types on this processor |
733 |
|
int count_local = foundTypes.size(); |
734 |
|
|
735 |
< |
// count holds the total number of found types on all processors |
703 |
< |
// (some will be redundant with the ones found locally): |
704 |
< |
int count; |
705 |
< |
MPI::COMM_WORLD.Allreduce(&count_local, &count, 1, MPI::INT, MPI::SUM); |
706 |
< |
|
707 |
< |
// create a vector to hold the globally found types, and resize it: |
708 |
< |
vector<int> ftGlobal; |
709 |
< |
ftGlobal.resize(count); |
710 |
< |
vector<int> counts; |
735 |
> |
int nproc = MPI::COMM_WORLD.Get_size(); |
736 |
|
|
737 |
< |
int nproc = MPI::COMM_WORLD.Get_size(); |
738 |
< |
counts.resize(nproc); |
739 |
< |
vector<int> disps; |
740 |
< |
disps.resize(nproc); |
737 |
> |
// we need arrays to hold the counts and displacement vectors for |
738 |
> |
// all processors |
739 |
> |
vector<int> counts(nproc, 0); |
740 |
> |
vector<int> disps(nproc, 0); |
741 |
|
|
742 |
< |
// now spray out the foundTypes to all the other processors: |
742 |
> |
// fill the counts array |
743 |
> |
MPI::COMM_WORLD.Allgather(&count_local, 1, MPI::INT, &counts[0], |
744 |
> |
1, MPI::INT); |
745 |
> |
|
746 |
> |
// use the processor counts to compute the displacement array |
747 |
> |
disps[0] = 0; |
748 |
> |
int totalCount = counts[0]; |
749 |
> |
for (int iproc = 1; iproc < nproc; iproc++) { |
750 |
> |
disps[iproc] = disps[iproc-1] + counts[iproc-1]; |
751 |
> |
totalCount += counts[iproc]; |
752 |
> |
} |
753 |
> |
|
754 |
> |
// we need a (possibly redundant) set of all found types: |
755 |
> |
vector<int> ftGlobal(totalCount); |
756 |
|
|
757 |
+ |
// now spray out the foundTypes to all the other processors: |
758 |
|
MPI::COMM_WORLD.Allgatherv(&foundTypes[0], count_local, MPI::INT, |
759 |
< |
&ftGlobal[0], &counts[0], &disps[0], MPI::INT); |
759 |
> |
&ftGlobal[0], &counts[0], &disps[0], |
760 |
> |
MPI::INT); |
761 |
|
|
762 |
+ |
vector<int>::iterator j; |
763 |
+ |
|
764 |
|
// foundIdents is a stl set, so inserting an already found ident |
765 |
|
// will have no effect. |
766 |
|
set<int> foundIdents; |
767 |
< |
vector<int>::iterator j; |
767 |
> |
|
768 |
|
for (j = ftGlobal.begin(); j != ftGlobal.end(); ++j) |
769 |
|
foundIdents.insert((*j)); |
770 |
|
|
771 |
|
// now iterate over the foundIdents and get the actual atom types |
772 |
|
// that correspond to these: |
773 |
|
set<int>::iterator it; |
774 |
< |
for (it = foundIdents.begin(); it != foundIdents.end(); ++it) |
774 |
> |
for (it = foundIdents.begin(); it != foundIdents.end(); ++it) |
775 |
|
atomTypes.insert( forceField_->getAtomType((*it)) ); |
776 |
|
|
777 |
|
#endif |
778 |
< |
|
778 |
> |
|
779 |
|
return atomTypes; |
780 |
|
} |
781 |
|
|
787 |
|
if ( simParams_->getAccumulateBoxDipole() ) { |
788 |
|
calcBoxDipole_ = true; |
789 |
|
} |
790 |
< |
|
790 |
> |
|
791 |
|
set<AtomType*>::iterator i; |
792 |
|
set<AtomType*> atomTypes; |
793 |
|
atomTypes = getSimulatedAtomTypes(); |
794 |
< |
int usesElectrostatic = 0; |
795 |
< |
int usesMetallic = 0; |
796 |
< |
int usesDirectional = 0; |
794 |
> |
bool usesElectrostatic = false; |
795 |
> |
bool usesMetallic = false; |
796 |
> |
bool usesDirectional = false; |
797 |
> |
bool usesFluctuatingCharges = false; |
798 |
|
//loop over all of the atom types |
799 |
|
for (i = atomTypes.begin(); i != atomTypes.end(); ++i) { |
800 |
|
usesElectrostatic |= (*i)->isElectrostatic(); |
801 |
|
usesMetallic |= (*i)->isMetal(); |
802 |
|
usesDirectional |= (*i)->isDirectional(); |
803 |
+ |
usesFluctuatingCharges |= (*i)->isFluctuatingCharge(); |
804 |
|
} |
805 |
|
|
806 |
< |
#ifdef IS_MPI |
807 |
< |
int temp; |
806 |
> |
#ifdef IS_MPI |
807 |
> |
bool temp; |
808 |
|
temp = usesDirectional; |
809 |
< |
MPI_Allreduce(&temp, &usesDirectionalAtoms_, 1, MPI_INT, MPI_LOR, MPI_COMM_WORLD); |
810 |
< |
|
809 |
> |
MPI::COMM_WORLD.Allreduce(&temp, &usesDirectionalAtoms_, 1, MPI::BOOL, |
810 |
> |
MPI::LOR); |
811 |
> |
|
812 |
|
temp = usesMetallic; |
813 |
< |
MPI_Allreduce(&temp, &usesMetallicAtoms_, 1, MPI_INT, MPI_LOR, MPI_COMM_WORLD); |
814 |
< |
|
813 |
> |
MPI::COMM_WORLD.Allreduce(&temp, &usesMetallicAtoms_, 1, MPI::BOOL, |
814 |
> |
MPI::LOR); |
815 |
> |
|
816 |
|
temp = usesElectrostatic; |
817 |
< |
MPI_Allreduce(&temp, &usesElectrostaticAtoms_, 1, MPI_INT, MPI_LOR, MPI_COMM_WORLD); |
817 |
> |
MPI::COMM_WORLD.Allreduce(&temp, &usesElectrostaticAtoms_, 1, MPI::BOOL, |
818 |
> |
MPI::LOR); |
819 |
> |
|
820 |
> |
temp = usesFluctuatingCharges; |
821 |
> |
MPI::COMM_WORLD.Allreduce(&temp, &usesFluctuatingCharges_, 1, MPI::BOOL, |
822 |
> |
MPI::LOR); |
823 |
> |
#else |
824 |
> |
|
825 |
> |
usesDirectionalAtoms_ = usesDirectional; |
826 |
> |
usesMetallicAtoms_ = usesMetallic; |
827 |
> |
usesElectrostaticAtoms_ = usesElectrostatic; |
828 |
> |
usesFluctuatingCharges_ = usesFluctuatingCharges; |
829 |
> |
|
830 |
|
#endif |
831 |
+ |
|
832 |
+ |
requiresPrepair_ = usesMetallicAtoms_ ? true : false; |
833 |
+ |
requiresSkipCorrection_ = usesElectrostaticAtoms_ ? true : false; |
834 |
+ |
requiresSelfCorrection_ = usesElectrostaticAtoms_ ? true : false; |
835 |
|
} |
836 |
|
|
837 |
|
|
886 |
|
Atom* atom; |
887 |
|
RealType totalMass; |
888 |
|
|
889 |
< |
//to avoid memory reallocation, reserve enough space for massFactors_ |
889 |
> |
/** |
890 |
> |
* The mass factor is the relative mass of an atom to the total |
891 |
> |
* mass of the cutoff group it belongs to. By default, all atoms |
892 |
> |
* are their own cutoff groups, and therefore have mass factors of |
893 |
> |
* 1. We need some special handling for massless atoms, which |
894 |
> |
* will be treated as carrying the entire mass of the cutoff |
895 |
> |
* group. |
896 |
> |
*/ |
897 |
|
massFactors_.clear(); |
898 |
< |
massFactors_.reserve(getNCutoffGroups()); |
898 |
> |
massFactors_.resize(getNAtoms(), 1.0); |
899 |
|
|
900 |
|
for(mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) { |
901 |
|
for (cg = mol->beginCutoffGroup(ci); cg != NULL; |
904 |
|
totalMass = cg->getMass(); |
905 |
|
for(atom = cg->beginAtom(ai); atom != NULL; atom = cg->nextAtom(ai)) { |
906 |
|
// Check for massless groups - set mfact to 1 if true |
907 |
< |
if (totalMass != 0) |
908 |
< |
massFactors_.push_back(atom->getMass()/totalMass); |
907 |
> |
if (totalMass != 0) |
908 |
> |
massFactors_[atom->getLocalIndex()] = atom->getMass()/totalMass; |
909 |
|
else |
910 |
< |
massFactors_.push_back( 1.0 ); |
910 |
> |
massFactors_[atom->getLocalIndex()] = 1.0; |
911 |
|
} |
912 |
|
} |
913 |
|
} |
934 |
|
int* oneThreeList = oneThreeInteractions_.getPairList(); |
935 |
|
int* oneFourList = oneFourInteractions_.getPairList(); |
936 |
|
|
868 |
– |
//setFortranSim( &fInfo_, &nGlobalAtoms_, &nAtoms_, &identArray_[0], |
869 |
– |
// &nExclude, excludeList, |
870 |
– |
// &nOneTwo, oneTwoList, |
871 |
– |
// &nOneThree, oneThreeList, |
872 |
– |
// &nOneFour, oneFourList, |
873 |
– |
// &molMembershipArray[0], &mfact[0], &nCutoffGroups_, |
874 |
– |
// &fortranGlobalGroupMembership[0], &isError); |
875 |
– |
|
937 |
|
topologyDone_ = true; |
938 |
|
} |
939 |
|
|
991 |
|
cg->setSnapshotManager(sman_); |
992 |
|
} |
993 |
|
} |
933 |
– |
|
934 |
– |
} |
935 |
– |
|
936 |
– |
Vector3d SimInfo::getComVel(){ |
937 |
– |
SimInfo::MoleculeIterator i; |
938 |
– |
Molecule* mol; |
939 |
– |
|
940 |
– |
Vector3d comVel(0.0); |
941 |
– |
RealType totalMass = 0.0; |
994 |
|
|
943 |
– |
|
944 |
– |
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
945 |
– |
RealType mass = mol->getMass(); |
946 |
– |
totalMass += mass; |
947 |
– |
comVel += mass * mol->getComVel(); |
948 |
– |
} |
949 |
– |
|
950 |
– |
#ifdef IS_MPI |
951 |
– |
RealType tmpMass = totalMass; |
952 |
– |
Vector3d tmpComVel(comVel); |
953 |
– |
MPI_Allreduce(&tmpMass,&totalMass,1,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
954 |
– |
MPI_Allreduce(tmpComVel.getArrayPointer(), comVel.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
955 |
– |
#endif |
956 |
– |
|
957 |
– |
comVel /= totalMass; |
958 |
– |
|
959 |
– |
return comVel; |
995 |
|
} |
996 |
|
|
962 |
– |
Vector3d SimInfo::getCom(){ |
963 |
– |
SimInfo::MoleculeIterator i; |
964 |
– |
Molecule* mol; |
997 |
|
|
966 |
– |
Vector3d com(0.0); |
967 |
– |
RealType totalMass = 0.0; |
968 |
– |
|
969 |
– |
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
970 |
– |
RealType mass = mol->getMass(); |
971 |
– |
totalMass += mass; |
972 |
– |
com += mass * mol->getCom(); |
973 |
– |
} |
974 |
– |
|
975 |
– |
#ifdef IS_MPI |
976 |
– |
RealType tmpMass = totalMass; |
977 |
– |
Vector3d tmpCom(com); |
978 |
– |
MPI_Allreduce(&tmpMass,&totalMass,1,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
979 |
– |
MPI_Allreduce(tmpCom.getArrayPointer(), com.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
980 |
– |
#endif |
981 |
– |
|
982 |
– |
com /= totalMass; |
983 |
– |
|
984 |
– |
return com; |
985 |
– |
|
986 |
– |
} |
987 |
– |
|
998 |
|
ostream& operator <<(ostream& o, SimInfo& info) { |
999 |
|
|
1000 |
|
return o; |
1001 |
|
} |
1002 |
|
|
1003 |
< |
|
994 |
< |
/* |
995 |
< |
Returns center of mass and center of mass velocity in one function call. |
996 |
< |
*/ |
997 |
< |
|
998 |
< |
void SimInfo::getComAll(Vector3d &com, Vector3d &comVel){ |
999 |
< |
SimInfo::MoleculeIterator i; |
1000 |
< |
Molecule* mol; |
1001 |
< |
|
1002 |
< |
|
1003 |
< |
RealType totalMass = 0.0; |
1004 |
< |
|
1005 |
< |
|
1006 |
< |
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
1007 |
< |
RealType mass = mol->getMass(); |
1008 |
< |
totalMass += mass; |
1009 |
< |
com += mass * mol->getCom(); |
1010 |
< |
comVel += mass * mol->getComVel(); |
1011 |
< |
} |
1012 |
< |
|
1013 |
< |
#ifdef IS_MPI |
1014 |
< |
RealType tmpMass = totalMass; |
1015 |
< |
Vector3d tmpCom(com); |
1016 |
< |
Vector3d tmpComVel(comVel); |
1017 |
< |
MPI_Allreduce(&tmpMass,&totalMass,1,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
1018 |
< |
MPI_Allreduce(tmpCom.getArrayPointer(), com.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
1019 |
< |
MPI_Allreduce(tmpComVel.getArrayPointer(), comVel.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
1020 |
< |
#endif |
1021 |
< |
|
1022 |
< |
com /= totalMass; |
1023 |
< |
comVel /= totalMass; |
1024 |
< |
} |
1025 |
< |
|
1026 |
< |
/* |
1027 |
< |
Return intertia tensor for entire system and angular momentum Vector. |
1028 |
< |
|
1029 |
< |
|
1030 |
< |
[ Ixx -Ixy -Ixz ] |
1031 |
< |
J =| -Iyx Iyy -Iyz | |
1032 |
< |
[ -Izx -Iyz Izz ] |
1033 |
< |
*/ |
1034 |
< |
|
1035 |
< |
void SimInfo::getInertiaTensor(Mat3x3d &inertiaTensor, Vector3d &angularMomentum){ |
1036 |
< |
|
1037 |
< |
|
1038 |
< |
RealType xx = 0.0; |
1039 |
< |
RealType yy = 0.0; |
1040 |
< |
RealType zz = 0.0; |
1041 |
< |
RealType xy = 0.0; |
1042 |
< |
RealType xz = 0.0; |
1043 |
< |
RealType yz = 0.0; |
1044 |
< |
Vector3d com(0.0); |
1045 |
< |
Vector3d comVel(0.0); |
1046 |
< |
|
1047 |
< |
getComAll(com, comVel); |
1048 |
< |
|
1049 |
< |
SimInfo::MoleculeIterator i; |
1050 |
< |
Molecule* mol; |
1051 |
< |
|
1052 |
< |
Vector3d thisq(0.0); |
1053 |
< |
Vector3d thisv(0.0); |
1054 |
< |
|
1055 |
< |
RealType thisMass = 0.0; |
1056 |
< |
|
1057 |
< |
|
1058 |
< |
|
1059 |
< |
|
1060 |
< |
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
1061 |
< |
|
1062 |
< |
thisq = mol->getCom()-com; |
1063 |
< |
thisv = mol->getComVel()-comVel; |
1064 |
< |
thisMass = mol->getMass(); |
1065 |
< |
// Compute moment of intertia coefficients. |
1066 |
< |
xx += thisq[0]*thisq[0]*thisMass; |
1067 |
< |
yy += thisq[1]*thisq[1]*thisMass; |
1068 |
< |
zz += thisq[2]*thisq[2]*thisMass; |
1069 |
< |
|
1070 |
< |
// compute products of intertia |
1071 |
< |
xy += thisq[0]*thisq[1]*thisMass; |
1072 |
< |
xz += thisq[0]*thisq[2]*thisMass; |
1073 |
< |
yz += thisq[1]*thisq[2]*thisMass; |
1074 |
< |
|
1075 |
< |
angularMomentum += cross( thisq, thisv ) * thisMass; |
1076 |
< |
|
1077 |
< |
} |
1078 |
< |
|
1079 |
< |
|
1080 |
< |
inertiaTensor(0,0) = yy + zz; |
1081 |
< |
inertiaTensor(0,1) = -xy; |
1082 |
< |
inertiaTensor(0,2) = -xz; |
1083 |
< |
inertiaTensor(1,0) = -xy; |
1084 |
< |
inertiaTensor(1,1) = xx + zz; |
1085 |
< |
inertiaTensor(1,2) = -yz; |
1086 |
< |
inertiaTensor(2,0) = -xz; |
1087 |
< |
inertiaTensor(2,1) = -yz; |
1088 |
< |
inertiaTensor(2,2) = xx + yy; |
1089 |
< |
|
1090 |
< |
#ifdef IS_MPI |
1091 |
< |
Mat3x3d tmpI(inertiaTensor); |
1092 |
< |
Vector3d tmpAngMom; |
1093 |
< |
MPI_Allreduce(tmpI.getArrayPointer(), inertiaTensor.getArrayPointer(),9,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
1094 |
< |
MPI_Allreduce(tmpAngMom.getArrayPointer(), angularMomentum.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
1095 |
< |
#endif |
1096 |
< |
|
1097 |
< |
return; |
1098 |
< |
} |
1099 |
< |
|
1100 |
< |
//Returns the angular momentum of the system |
1101 |
< |
Vector3d SimInfo::getAngularMomentum(){ |
1102 |
< |
|
1103 |
< |
Vector3d com(0.0); |
1104 |
< |
Vector3d comVel(0.0); |
1105 |
< |
Vector3d angularMomentum(0.0); |
1106 |
< |
|
1107 |
< |
getComAll(com,comVel); |
1108 |
< |
|
1109 |
< |
SimInfo::MoleculeIterator i; |
1110 |
< |
Molecule* mol; |
1111 |
< |
|
1112 |
< |
Vector3d thisr(0.0); |
1113 |
< |
Vector3d thisp(0.0); |
1114 |
< |
|
1115 |
< |
RealType thisMass; |
1116 |
< |
|
1117 |
< |
for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) { |
1118 |
< |
thisMass = mol->getMass(); |
1119 |
< |
thisr = mol->getCom()-com; |
1120 |
< |
thisp = (mol->getComVel()-comVel)*thisMass; |
1121 |
< |
|
1122 |
< |
angularMomentum += cross( thisr, thisp ); |
1123 |
< |
|
1124 |
< |
} |
1125 |
< |
|
1126 |
< |
#ifdef IS_MPI |
1127 |
< |
Vector3d tmpAngMom; |
1128 |
< |
MPI_Allreduce(tmpAngMom.getArrayPointer(), angularMomentum.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD); |
1129 |
< |
#endif |
1130 |
< |
|
1131 |
< |
return angularMomentum; |
1132 |
< |
} |
1133 |
< |
|
1003 |
> |
|
1004 |
|
StuntDouble* SimInfo::getIOIndexToIntegrableObject(int index) { |
1005 |
|
return IOIndexToIntegrableObject.at(index); |
1006 |
|
} |
1008 |
|
void SimInfo::setIOIndexToIntegrableObject(const vector<StuntDouble*>& v) { |
1009 |
|
IOIndexToIntegrableObject= v; |
1010 |
|
} |
1141 |
– |
|
1142 |
– |
/* Returns the Volume of the simulation based on a ellipsoid with semi-axes |
1143 |
– |
based on the radius of gyration V=4/3*Pi*R_1*R_2*R_3 |
1144 |
– |
where R_i are related to the principle inertia moments R_i = sqrt(C*I_i/N), this reduces to |
1145 |
– |
V = 4/3*Pi*(C/N)^3/2*sqrt(det(I)). See S.E. Baltazar et. al. Comp. Mat. Sci. 37 (2006) 526-536. |
1146 |
– |
*/ |
1147 |
– |
void SimInfo::getGyrationalVolume(RealType &volume){ |
1148 |
– |
Mat3x3d intTensor; |
1149 |
– |
RealType det; |
1150 |
– |
Vector3d dummyAngMom; |
1151 |
– |
RealType sysconstants; |
1152 |
– |
RealType geomCnst; |
1153 |
– |
|
1154 |
– |
geomCnst = 3.0/2.0; |
1155 |
– |
/* Get the inertial tensor and angular momentum for free*/ |
1156 |
– |
getInertiaTensor(intTensor,dummyAngMom); |
1157 |
– |
|
1158 |
– |
det = intTensor.determinant(); |
1159 |
– |
sysconstants = geomCnst/(RealType)nGlobalIntegrableObjects_; |
1160 |
– |
volume = 4.0/3.0*NumericConstant::PI*pow(sysconstants,3.0/2.0)*sqrt(det); |
1161 |
– |
return; |
1162 |
– |
} |
1163 |
– |
|
1164 |
– |
void SimInfo::getGyrationalVolume(RealType &volume, RealType &detI){ |
1165 |
– |
Mat3x3d intTensor; |
1166 |
– |
Vector3d dummyAngMom; |
1167 |
– |
RealType sysconstants; |
1168 |
– |
RealType geomCnst; |
1169 |
– |
|
1170 |
– |
geomCnst = 3.0/2.0; |
1171 |
– |
/* Get the inertial tensor and angular momentum for free*/ |
1172 |
– |
getInertiaTensor(intTensor,dummyAngMom); |
1173 |
– |
|
1174 |
– |
detI = intTensor.determinant(); |
1175 |
– |
sysconstants = geomCnst/(RealType)nGlobalIntegrableObjects_; |
1176 |
– |
volume = 4.0/3.0*NumericConstant::PI*pow(sysconstants,3.0/2.0)*sqrt(detI); |
1177 |
– |
return; |
1178 |
– |
} |
1011 |
|
/* |
1012 |
|
void SimInfo::setStuntDoubleFromGlobalIndex(vector<StuntDouble*> v) { |
1013 |
|
assert( v.size() == nAtoms_ + nRigidBodies_); |