[ViewVC] Diff of: OpenMD/branches/development/src/brains/SimInfo.cpp

Comparing branches/development/src/brains/SimInfo.cpp (file contents):
Revision 1528 by gezelter, Fri Dec 17 20:11:05 2010 UTC vs.
Revision 1769 by gezelter, Mon Jul 9 14:15:52 2012 UTC

#	Line 36 \| Line 36
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		/**
#	Line 54 \| Line 55
55		#include "math/Vector3.hpp"
56		#include "primitives/Molecule.hpp"
57		#include "primitives/StuntDouble.hpp"
57	–	#include "UseTheForce/fCutoffPolicy.h"
58	–	#include "UseTheForce/DarkSide/fSwitchingFunctionType.h"
59	–	#include "UseTheForce/doForces_interface.h"
60	–	#include "UseTheForce/DarkSide/neighborLists_interface.h"
61	–	#include "UseTheForce/DarkSide/switcheroo_interface.h"
58		#include "utils/MemoryUtils.hpp"
59		#include "utils/simError.h"
60		#include "selection/SelectionManager.hpp"
61		#include "io/ForceFieldOptions.hpp"
62	<	#include "UseTheForce/ForceField.hpp"
63	<	#include "nonbonded/InteractionManager.hpp"
68	<
69	<
62	>	#include "brains/ForceField.hpp"
63	>	#include "nonbonded/SwitchingFunction.hpp"
64		#ifdef IS_MPI
65	<	#include "UseTheForce/mpiComponentPlan.h"
66	<	#include "UseTheForce/DarkSide/simParallel_interface.h"
73	<	#endif
65	>	#include <mpi.h>
66	>	#endif
67
68		using namespace std;
69		namespace OpenMD {
#	Line 79 \| Line 72 \| 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), fortranInitialized_(false),
78	>	nConstraints_(0), nFluctuatingCharges_(0), sman_(NULL), topologyDone_(false),
79		calcBoxDipole_(false), useAtomicVirial_(true) {
80
81		MoleculeStamp* molStamp;
#	Line 136 \| Line 129 \| namespace OpenMD {
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
132	+
133		nGlobalCutoffGroups_ = nGlobalAtoms_ - nCutoffAtoms + nGroups;
134
135		//every free atom (atom does not belong to rigid bodies) is an
#	Line 231 \| Line 225 \| namespace OpenMD {
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)) {
243	–	for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
244	–	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		}
255	–
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
#	Line 279 \| Line 288 \| namespace OpenMD {
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)) {
295	–	for (integrableObject = mol->beginIntegrableObject(j); integrableObject != NULL;
296	–	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;
#	Line 358 \| Line 387 \| namespace OpenMD {
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;
365	<	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) {
#	Line 376 \| Line 404 \| namespace OpenMD {
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
#	Line 511 \| Line 539 \| namespace OpenMD {
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;
518	<	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) {
#	Line 529 \| Line 556 \| namespace OpenMD {
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
#	Line 656 \| Line 683 \| namespace OpenMD {
683		molStampIds_.insert(molStampIds_.end(), nmol, curStampId);
684		}
685
659	–	void SimInfo::update() {
686
687	<	setupSimType();
688	<	setupCutoffRadius();
689	<	setupSwitchingRadius();
690	<	setupCutoffMethod();
691	<	setupSkinThickness();
692	<	setupSwitchingFunction();
693	<	setupAccumulateBoxDipole();
694	<
695	<	#ifdef IS_MPI
670	<	setupFortranParallel();
671	<	#endif
672	<	setupFortranSim();
673	<	fortranInitialized_ = true;
674	<
687	>	/**
688	>	* update
689	>	*
690	>	* Performs the global checks and variable settings after the
691	>	* objects have been created.
692	>	*
693	>	*/
694	>	void SimInfo::update() {
695	>	setupSimVariables();
696		calcNdf();
697		calcNdfRaw();
698		calcNdfTrans();
699		}
700
701	+	/**
702	+	* getSimulatedAtomTypes
703	+	*
704	+	* Returns an STL set of AtomType* that are actually present in this
705	+	* simulation. Must query all processors to assemble this information.
706	+	*
707	+	*/
708		set<AtomType*> SimInfo::getSimulatedAtomTypes() {
709		SimInfo::MoleculeIterator mi;
710		Molecule* mol;
#	Line 685 \| Line 713 \| namespace OpenMD {
713		set<AtomType*> atomTypes;
714
715		for(mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
716	<
717	<	for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
716	>	for(atom = mol->beginAtom(ai); atom != NULL;
717	>	atom = mol->nextAtom(ai)) {
718		atomTypes.insert(atom->getAtomType());
719	<	}
720	<
693	<	}
719	>	}
720	>	}
721
722	<	return atomTypes;
696	<	}
722	>	#ifdef IS_MPI
723
724	<	/**
725	<	* setupCutoffRadius
700	<	*
701	<	* If the cutoffRadius was explicitly set, use that value.
702	<	* If the cutoffRadius was not explicitly set:
703	<	* Are there electrostatic atoms? Use 12.0 Angstroms.
704	<	* No electrostatic atoms? Poll the atom types present in the
705	<	* simulation for suggested cutoff values (e.g. 2.5 * sigma).
706	<	* Use the maximum suggested value that was found.
707	<	*/
708	<	void SimInfo::setupCutoffRadius() {
724	>	// loop over the found atom types on this processor, and add their
725	>	// numerical idents to a vector:
726
727	<	if (simParams_->haveCutoffRadius()) {
728	<	cutoffRadius_ = simParams_->getCutoffRadius();
729	<	} else {
730	<	if (usesElectrostaticAtoms_) {
714	<	sprintf(painCave.errMsg,
715	<	"SimInfo Warning: No value was set for the cutoffRadius.\n"
716	<	"\tOpenMD will use a default value of 12.0 angstroms"
717	<	"\tfor the cutoffRadius.\n");
718	<	painCave.isFatal = 0;
719	<	simError();
720	<	cutoffRadius_ = 12.0;
721	<	} else {
722	<	RealType thisCut;
723	<	set<AtomType*>::iterator i;
724	<	set<AtomType*> atomTypes;
725	<	atomTypes = getSimulatedAtomTypes();
726	<	for (i = atomTypes.begin(); i != atomTypes.end(); ++i) {
727	<	thisCut = InteractionManager::Instance()->getSuggestedCutoffRadius((*i));
728	<	cutoffRadius_ = max(thisCut, cutoffRadius_);
729	<	}
730	<	sprintf(painCave.errMsg,
731	<	"SimInfo Warning: No value was set for the cutoffRadius.\n"
732	<	"\tOpenMD will use %lf angstroms.\n",
733	<	cutoffRadius_);
734	<	painCave.isFatal = 0;
735	<	simError();
736	<	}
737	<	}
727	>	vector<int> foundTypes;
728	>	set<AtomType*>::iterator i;
729	>	for (i = atomTypes.begin(); i != atomTypes.end(); ++i)
730	>	foundTypes.push_back( (*i)->getIdent() );
731
732	<	InteractionManager::Instance()->setCutoffRadius(cutoffRadius_);
733	<	}
734	<
735	<	/**
736	<	* setupSwitchingRadius
737	<	*
738	<	* If the switchingRadius was explicitly set, use that value (but check it)
739	<	* If the switchingRadius was not explicitly set: use 0.85 * cutoffRadius_
740	<	*/
741	<	void SimInfo::setupSwitchingRadius() {
732	>	// count_local holds the number of found types on this processor
733	>	int count_local = foundTypes.size();
734	>
735	>	int nproc = MPI::COMM_WORLD.Get_size();
736	>
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	>	// 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	<	if (simParams_->haveSwitchingRadius()) {
758	<	switchingRadius_ = simParams_->getSwitchingRadius();
759	<	if (switchingRadius_ > cutoffRadius_) {
760	<	sprintf(painCave.errMsg,
754	<	"SimInfo Error: switchingRadius (%f) is larger than cutoffRadius(%f)\n",
755	<	switchingRadius_, cutoffRadius_);
756	<	painCave.isFatal = 1;
757	<	simError();
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],
760	>	MPI::INT);
761
762	<	}
760	<	} else {
761	<	switchingRadius_ = 0.85 * cutoffRadius_;
762	<	sprintf(painCave.errMsg,
763	<	"SimInfo Warning: No value was set for the switchingRadius.\n"
764	<	"\tOpenMD will use a default value of 85 percent of the cutoffRadius.\n"
765	<	"\tswitchingRadius = %f. for this simulation\n", switchingRadius_);
766	<	painCave.isFatal = 0;
767	<	simError();
768	<	}
769	<	InteractionManager::Instance()->setSwitchingRadius(switchingRadius_);
770	<	}
762	>	vector<int>::iterator j;
763
764	<	/**
765	<	* setupSkinThickness
766	<	*
767	<	* If the skinThickness was explicitly set, use that value (but check it)
768	<	* If the skinThickness was not explicitly set: use 1.0 angstroms
769	<	*/
770	<	void SimInfo::setupSkinThickness() {
771	<	if (simParams_->haveSkinThickness()) {
772	<	skinThickness_ = simParams_->getSkinThickness();
773	<	} else {
774	<	skinThickness_ = 1.0;
775	<	sprintf(painCave.errMsg,
776	<	"SimInfo Warning: No value was set for the skinThickness.\n"
777	<	"\tOpenMD will use a default value of %f Angstroms\n"
778	<	"\tfor this simulation\n", skinThickness_);
779	<	painCave.isFatal = 0;
788	<	simError();
789	<	}
764	>	// foundIdents is a stl set, so inserting an already found ident
765	>	// will have no effect.
766	>	set<int> foundIdents;
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)
775	>	atomTypes.insert( forceField_->getAtomType((*it)) );
776	>
777	>	#endif
778	>
779	>	return atomTypes;
780		}
781
782	<	void SimInfo::setupSimType() {
782	>	void SimInfo::setupSimVariables() {
783	>	useAtomicVirial_ = simParams_->getUseAtomicVirial();
784	>	// we only call setAccumulateBoxDipole if the accumulateBoxDipole parameter is true
785	>	calcBoxDipole_ = false;
786	>	if ( simParams_->haveAccumulateBoxDipole() )
787	>	if ( simParams_->getAccumulateBoxDipole() ) {
788	>	calcBoxDipole_ = true;
789	>	}
790	>
791		set<AtomType*>::iterator i;
792		set<AtomType*> atomTypes;
793	<	atomTypes = getSimulatedAtomTypes();
794	<
795	<	useAtomicVirial_ = simParams_->getUseAtomicVirial();
796	<
797	<	int usesElectrostatic = 0;
800	<	int usesMetallic = 0;
801	<	int usesDirectional = 0;
793	>	atomTypes = getSimulatedAtomTypes();
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	<	fInfo_.SIM_uses_PBC = usesPeriodicBoundaries_;
832	<	fInfo_.SIM_uses_DirectionalAtoms = usesDirectionalAtoms_;
833	<	fInfo_.SIM_uses_MetallicAtoms = usesMetallicAtoms_;
834	<	fInfo_.SIM_requires_SkipCorrection = usesElectrostaticAtoms_;
824	<	fInfo_.SIM_requires_SelfCorrection = usesElectrostaticAtoms_;
825	<	fInfo_.SIM_uses_AtomicVirial = usesAtomicVirial_;
831	>
832	>	requiresPrepair_ = usesMetallicAtoms_ ? true : false;
833	>	requiresSkipCorrection_ = usesElectrostaticAtoms_ ? true : false;
834	>	requiresSelfCorrection_ = usesElectrostaticAtoms_ ? true : false;
835		}
836
837	<	void SimInfo::setupFortranSim() {
838	<	int isError;
839	<	int nExclude, nOneTwo, nOneThree, nOneFour;
840	<	vector<int> fortranGlobalGroupMembership;
837	>
838	>	vector<int> SimInfo::getGlobalAtomIndices() {
839	>	SimInfo::MoleculeIterator mi;
840	>	Molecule* mol;
841	>	Molecule::AtomIterator ai;
842	>	Atom* atom;
843	>
844	>	vector<int> GlobalAtomIndices(getNAtoms(), 0);
845
846	<	notifyFortranSkinThickness(&skinThickness_);
846	>	for (mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
847	>
848	>	for (atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
849	>	GlobalAtomIndices[atom->getLocalIndex()] = atom->getGlobalIndex();
850	>	}
851	>	}
852	>	return GlobalAtomIndices;
853	>	}
854
835	–	int ljsp = cutoffMethod_ == SHIFTED_POTENTIAL ? 1 : 0;
836	–	int ljsf = cutoffMethod_ == SHIFTED_FORCE ? 1 : 0;
837	–	notifyFortranCutoffs(&cutoffRadius_, &switchingRadius_, &ljsp, &ljsf);
855
856	<	isError = 0;
856	>	vector<int> SimInfo::getGlobalGroupIndices() {
857	>	SimInfo::MoleculeIterator mi;
858	>	Molecule* mol;
859	>	Molecule::CutoffGroupIterator ci;
860	>	CutoffGroup* cg;
861
862	<	//globalGroupMembership_ is filled by SimCreator
863	<	for (int i = 0; i < nGlobalAtoms_; i++) {
864	<	fortranGlobalGroupMembership.push_back(globalGroupMembership_[i] + 1);
862	>	vector<int> GlobalGroupIndices;
863	>
864	>	for (mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
865	>
866	>	//local index of cutoff group is trivial, it only depends on the
867	>	//order of travesing
868	>	for (cg = mol->beginCutoffGroup(ci); cg != NULL;
869	>	cg = mol->nextCutoffGroup(ci)) {
870	>	GlobalGroupIndices.push_back(cg->getGlobalIndex());
871	>	}
872		}
873	+	return GlobalGroupIndices;
874	+	}
875
876	+
877	+	void SimInfo::prepareTopology() {
878	+	int nExclude, nOneTwo, nOneThree, nOneFour;
879	+
880		//calculate mass ratio of cutoff group
847	–	vector<RealType> mfact;
881		SimInfo::MoleculeIterator mi;
882		Molecule* mol;
883		Molecule::CutoffGroupIterator ci;
#	Line 853 \| Line 886 \| namespace OpenMD {
886		Atom* atom;
887		RealType totalMass;
888
889	<	//to avoid memory reallocation, reserve enough space for mfact
890	<	mfact.reserve(getNCutoffGroups());
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_.resize(getNAtoms(), 1.0);
899
900		for(mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
901	<	for (cg = mol->beginCutoffGroup(ci); cg != NULL; cg = mol->nextCutoffGroup(ci)) {
901	>	for (cg = mol->beginCutoffGroup(ci); cg != NULL;
902	>	cg = mol->nextCutoffGroup(ci)) {
903
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	<	mfact.push_back(atom->getMass()/totalMass);
907	>	if (totalMass != 0)
908	>	massFactors_[atom->getLocalIndex()] = atom->getMass()/totalMass;
909		else
910	<	mfact.push_back( 1.0 );
910	>	massFactors_[atom->getLocalIndex()] = 1.0;
911		}
912		}
913		}
914
915	<	//fill ident array of local atoms (it is actually ident of AtomType, it is so confusing !!!)
874	<	vector<int> identArray;
915	>	// Build the identArray_
916
917	<	//to avoid memory reallocation, reserve enough space identArray
918	<	identArray.reserve(getNAtoms());
878	<
917	>	identArray_.clear();
918	>	identArray_.reserve(getNAtoms());
919		for(mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
920		for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
921	<	identArray.push_back(atom->getIdent());
921	>	identArray_.push_back(atom->getIdent());
922		}
923		}
884	–
885	–	//fill molMembershipArray
886	–	//molMembershipArray is filled by SimCreator
887	–	vector<int> molMembershipArray(nGlobalAtoms_);
888	–	for (int i = 0; i < nGlobalAtoms_; i++) {
889	–	molMembershipArray[i] = globalMolMembership_[i] + 1;
890	–	}
924
925	<	//setup fortran simulation
925	>	//scan topology
926
927		nExclude = excludedInteractions_.getSize();
928		nOneTwo = oneTwoInteractions_.getSize();
#	Line 901 \| Line 934 \| namespace OpenMD {
934		int* oneThreeList = oneThreeInteractions_.getPairList();
935		int* oneFourList = oneFourInteractions_.getPairList();
936
937	<	setFortranSim( &fInfo_, &nGlobalAtoms_, &nAtoms_, &identArray[0],
905	<	&nExclude, excludeList,
906	<	&nOneTwo, oneTwoList,
907	<	&nOneThree, oneThreeList,
908	<	&nOneFour, oneFourList,
909	<	&molMembershipArray[0], &mfact[0], &nCutoffGroups_,
910	<	&fortranGlobalGroupMembership[0], &isError);
911	<
912	<	if( isError ){
913	<
914	<	sprintf( painCave.errMsg,
915	<	"There was an error setting the simulation information in fortran.\n" );
916	<	painCave.isFatal = 1;
917	<	painCave.severity = OPENMD_ERROR;
918	<	simError();
919	<	}
920	<
921	<
922	<	sprintf( checkPointMsg,
923	<	"succesfully sent the simulation information to fortran.\n");
924	<
925	<	errorCheckPoint();
926	<
927	<	// Setup number of neighbors in neighbor list if present
928	<	if (simParams_->haveNeighborListNeighbors()) {
929	<	int nlistNeighbors = simParams_->getNeighborListNeighbors();
930	<	setNeighbors(&nlistNeighbors);
931	<	}
932	<
933	<
937	>	topologyDone_ = true;
938		}
939
936	–
937	–	void SimInfo::setupFortranParallel() {
938	–	#ifdef IS_MPI
939	–	//SimInfo is responsible for creating localToGlobalAtomIndex and localToGlobalGroupIndex
940	–	vector<int> localToGlobalAtomIndex(getNAtoms(), 0);
941	–	vector<int> localToGlobalCutoffGroupIndex;
942	–	SimInfo::MoleculeIterator mi;
943	–	Molecule::AtomIterator ai;
944	–	Molecule::CutoffGroupIterator ci;
945	–	Molecule* mol;
946	–	Atom* atom;
947	–	CutoffGroup* cg;
948	–	mpiSimData parallelData;
949	–	int isError;
950	–
951	–	for (mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
952	–
953	–	//local index(index in DataStorge) of atom is important
954	–	for (atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
955	–	localToGlobalAtomIndex[atom->getLocalIndex()] = atom->getGlobalIndex() + 1;
956	–	}
957	–
958	–	//local index of cutoff group is trivial, it only depends on the order of travesing
959	–	for (cg = mol->beginCutoffGroup(ci); cg != NULL; cg = mol->nextCutoffGroup(ci)) {
960	–	localToGlobalCutoffGroupIndex.push_back(cg->getGlobalIndex() + 1);
961	–	}
962	–
963	–	}
964	–
965	–	//fill up mpiSimData struct
966	–	parallelData.nMolGlobal = getNGlobalMolecules();
967	–	parallelData.nMolLocal = getNMolecules();
968	–	parallelData.nAtomsGlobal = getNGlobalAtoms();
969	–	parallelData.nAtomsLocal = getNAtoms();
970	–	parallelData.nGroupsGlobal = getNGlobalCutoffGroups();
971	–	parallelData.nGroupsLocal = getNCutoffGroups();
972	–	parallelData.myNode = worldRank;
973	–	MPI_Comm_size(MPI_COMM_WORLD, &(parallelData.nProcessors));
974	–
975	–	//pass mpiSimData struct and index arrays to fortran
976	–	setFsimParallel(&parallelData, &(parallelData.nAtomsLocal),
977	–	&localToGlobalAtomIndex[0], &(parallelData.nGroupsLocal),
978	–	&localToGlobalCutoffGroupIndex[0], &isError);
979	–
980	–	if (isError) {
981	–	sprintf(painCave.errMsg,
982	–	"mpiRefresh errror: fortran didn't like something we gave it.\n");
983	–	painCave.isFatal = 1;
984	–	simError();
985	–	}
986	–
987	–	sprintf(checkPointMsg, " mpiRefresh successful.\n");
988	–	errorCheckPoint();
989	–
990	–	#endif
991	–	}
992	–
993	–
994	–	void SimInfo::setupSwitchingFunction() {
995	–	int ft = CUBIC;
996	–
997	–	if (simParams_->haveSwitchingFunctionType()) {
998	–	string funcType = simParams_->getSwitchingFunctionType();
999	–	toUpper(funcType);
1000	–	if (funcType == "CUBIC") {
1001	–	ft = CUBIC;
1002	–	} else {
1003	–	if (funcType == "FIFTH_ORDER_POLYNOMIAL") {
1004	–	ft = FIFTH_ORDER_POLY;
1005	–	} else {
1006	–	// throw error
1007	–	sprintf( painCave.errMsg,
1008	–	"SimInfo error: Unknown switchingFunctionType. (Input file specified %s .)\n\tswitchingFunctionType must be one of: \"cubic\" or \"fifth_order_polynomial\".", funcType.c_str() );
1009	–	painCave.isFatal = 1;
1010	–	simError();
1011	–	}
1012	–	}
1013	–	}
1014	–
1015	–	// send switching function notification to switcheroo
1016	–	setFunctionType(&ft);
1017	–
1018	–	}
1019	–
1020	–	void SimInfo::setupAccumulateBoxDipole() {
1021	–
1022	–	// we only call setAccumulateBoxDipole if the accumulateBoxDipole parameter is true
1023	–	if ( simParams_->haveAccumulateBoxDipole() )
1024	–	if ( simParams_->getAccumulateBoxDipole() ) {
1025	–	calcBoxDipole_ = true;
1026	–	}
1027	–
1028	–	}
1029	–
940		void SimInfo::addProperty(GenericData* genData) {
941		properties_.addProperty(genData);
942		}
#	Line 1061 \| Line 971 \| namespace OpenMD {
971		Molecule* mol;
972		RigidBody* rb;
973		Atom* atom;
974	+	CutoffGroup* cg;
975		SimInfo::MoleculeIterator mi;
976		Molecule::RigidBodyIterator rbIter;
977	<	Molecule::AtomIterator atomIter;;
977	>	Molecule::AtomIterator atomIter;
978	>	Molecule::CutoffGroupIterator cgIter;
979
980		for (mol = beginMolecule(mi); mol != NULL; mol = nextMolecule(mi)) {
981
#	Line 1074 \| Line 986 \| namespace OpenMD {
986		for (rb = mol->beginRigidBody(rbIter); rb != NULL; rb = mol->nextRigidBody(rbIter)) {
987		rb->setSnapshotManager(sman_);
988		}
989	+
990	+	for (cg = mol->beginCutoffGroup(cgIter); cg != NULL; cg = mol->nextCutoffGroup(cgIter)) {
991	+	cg->setSnapshotManager(sman_);
992	+	}
993		}
994
995		}
1080	–
1081	–	Vector3d SimInfo::getComVel(){
1082	–	SimInfo::MoleculeIterator i;
1083	–	Molecule* mol;
1084	–
1085	–	Vector3d comVel(0.0);
1086	–	RealType totalMass = 0.0;
1087	–
1088	–
1089	–	for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) {
1090	–	RealType mass = mol->getMass();
1091	–	totalMass += mass;
1092	–	comVel += mass * mol->getComVel();
1093	–	}
1094	–
1095	–	#ifdef IS_MPI
1096	–	RealType tmpMass = totalMass;
1097	–	Vector3d tmpComVel(comVel);
1098	–	MPI_Allreduce(&tmpMass,&totalMass,1,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1099	–	MPI_Allreduce(tmpComVel.getArrayPointer(), comVel.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1100	–	#endif
1101	–
1102	–	comVel /= totalMass;
1103	–
1104	–	return comVel;
1105	–	}
1106	–
1107	–	Vector3d SimInfo::getCom(){
1108	–	SimInfo::MoleculeIterator i;
1109	–	Molecule* mol;
1110	–
1111	–	Vector3d com(0.0);
1112	–	RealType totalMass = 0.0;
1113	–
1114	–	for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) {
1115	–	RealType mass = mol->getMass();
1116	–	totalMass += mass;
1117	–	com += mass * mol->getCom();
1118	–	}
996
1120	–	#ifdef IS_MPI
1121	–	RealType tmpMass = totalMass;
1122	–	Vector3d tmpCom(com);
1123	–	MPI_Allreduce(&tmpMass,&totalMass,1,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1124	–	MPI_Allreduce(tmpCom.getArrayPointer(), com.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1125	–	#endif
997
1127	–	com /= totalMass;
1128	–
1129	–	return com;
1130	–
1131	–	}
1132	–
998		ostream& operator <<(ostream& o, SimInfo& info) {
999
1000		return o;
1001		}
1002
1003	<
1139	<	/*
1140	<	Returns center of mass and center of mass velocity in one function call.
1141	<	*/
1142	<
1143	<	void SimInfo::getComAll(Vector3d &com, Vector3d &comVel){
1144	<	SimInfo::MoleculeIterator i;
1145	<	Molecule* mol;
1146	<
1147	<
1148	<	RealType totalMass = 0.0;
1149	<
1150	<
1151	<	for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) {
1152	<	RealType mass = mol->getMass();
1153	<	totalMass += mass;
1154	<	com += mass * mol->getCom();
1155	<	comVel += mass * mol->getComVel();
1156	<	}
1157	<
1158	<	#ifdef IS_MPI
1159	<	RealType tmpMass = totalMass;
1160	<	Vector3d tmpCom(com);
1161	<	Vector3d tmpComVel(comVel);
1162	<	MPI_Allreduce(&tmpMass,&totalMass,1,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1163	<	MPI_Allreduce(tmpCom.getArrayPointer(), com.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1164	<	MPI_Allreduce(tmpComVel.getArrayPointer(), comVel.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1165	<	#endif
1166	<
1167	<	com /= totalMass;
1168	<	comVel /= totalMass;
1169	<	}
1170	<
1171	<	/*
1172	<	Return intertia tensor for entire system and angular momentum Vector.
1173	<
1174	<
1175	<	[ Ixx -Ixy -Ixz ]
1176	<	J =\| -Iyx Iyy -Iyz \|
1177	<	[ -Izx -Iyz Izz ]
1178	<	*/
1179	<
1180	<	void SimInfo::getInertiaTensor(Mat3x3d &inertiaTensor, Vector3d &angularMomentum){
1181	<
1182	<
1183	<	RealType xx = 0.0;
1184	<	RealType yy = 0.0;
1185	<	RealType zz = 0.0;
1186	<	RealType xy = 0.0;
1187	<	RealType xz = 0.0;
1188	<	RealType yz = 0.0;
1189	<	Vector3d com(0.0);
1190	<	Vector3d comVel(0.0);
1191	<
1192	<	getComAll(com, comVel);
1193	<
1194	<	SimInfo::MoleculeIterator i;
1195	<	Molecule* mol;
1196	<
1197	<	Vector3d thisq(0.0);
1198	<	Vector3d thisv(0.0);
1199	<
1200	<	RealType thisMass = 0.0;
1201	<
1202	<
1203	<
1204	<
1205	<	for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) {
1206	<
1207	<	thisq = mol->getCom()-com;
1208	<	thisv = mol->getComVel()-comVel;
1209	<	thisMass = mol->getMass();
1210	<	// Compute moment of intertia coefficients.
1211	<	xx += thisq[0]thisq[0]thisMass;
1212	<	yy += thisq[1]thisq[1]thisMass;
1213	<	zz += thisq[2]thisq[2]thisMass;
1214	<
1215	<	// compute products of intertia
1216	<	xy += thisq[0]thisq[1]thisMass;
1217	<	xz += thisq[0]thisq[2]thisMass;
1218	<	yz += thisq[1]thisq[2]thisMass;
1219	<
1220	<	angularMomentum += cross( thisq, thisv ) * thisMass;
1221	<
1222	<	}
1223	<
1224	<
1225	<	inertiaTensor(0,0) = yy + zz;
1226	<	inertiaTensor(0,1) = -xy;
1227	<	inertiaTensor(0,2) = -xz;
1228	<	inertiaTensor(1,0) = -xy;
1229	<	inertiaTensor(1,1) = xx + zz;
1230	<	inertiaTensor(1,2) = -yz;
1231	<	inertiaTensor(2,0) = -xz;
1232	<	inertiaTensor(2,1) = -yz;
1233	<	inertiaTensor(2,2) = xx + yy;
1234	<
1235	<	#ifdef IS_MPI
1236	<	Mat3x3d tmpI(inertiaTensor);
1237	<	Vector3d tmpAngMom;
1238	<	MPI_Allreduce(tmpI.getArrayPointer(), inertiaTensor.getArrayPointer(),9,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1239	<	MPI_Allreduce(tmpAngMom.getArrayPointer(), angularMomentum.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1240	<	#endif
1241	<
1242	<	return;
1243	<	}
1244	<
1245	<	//Returns the angular momentum of the system
1246	<	Vector3d SimInfo::getAngularMomentum(){
1247	<
1248	<	Vector3d com(0.0);
1249	<	Vector3d comVel(0.0);
1250	<	Vector3d angularMomentum(0.0);
1251	<
1252	<	getComAll(com,comVel);
1253	<
1254	<	SimInfo::MoleculeIterator i;
1255	<	Molecule* mol;
1256	<
1257	<	Vector3d thisr(0.0);
1258	<	Vector3d thisp(0.0);
1259	<
1260	<	RealType thisMass;
1261	<
1262	<	for (mol = beginMolecule(i); mol != NULL; mol = nextMolecule(i)) {
1263	<	thisMass = mol->getMass();
1264	<	thisr = mol->getCom()-com;
1265	<	thisp = (mol->getComVel()-comVel)*thisMass;
1266	<
1267	<	angularMomentum += cross( thisr, thisp );
1268	<
1269	<	}
1270	<
1271	<	#ifdef IS_MPI
1272	<	Vector3d tmpAngMom;
1273	<	MPI_Allreduce(tmpAngMom.getArrayPointer(), angularMomentum.getArrayPointer(),3,MPI_REALTYPE,MPI_SUM, MPI_COMM_WORLD);
1274	<	#endif
1275	<
1276	<	return angularMomentum;
1277	<	}
1278	<
1003	>
1004		StuntDouble* SimInfo::getIOIndexToIntegrableObject(int index) {
1005		return IOIndexToIntegrableObject.at(index);
1006		}
#	Line 1283 \| Line 1008 \| namespace OpenMD {
1008		void SimInfo::setIOIndexToIntegrableObject(const vector<StuntDouble*>& v) {
1009		IOIndexToIntegrableObject= v;
1010		}
1286	–
1287	–	/* Returns the Volume of the simulation based on a ellipsoid with semi-axes
1288	–	based on the radius of gyration V=4/3PiR_1R_2R_3
1289	–	where R_i are related to the principle inertia moments R_i = sqrt(C*I_i/N), this reduces to
1290	–	V = 4/3Pi(C/N)^3/2*sqrt(det(I)). See S.E. Baltazar et. al. Comp. Mat. Sci. 37 (2006) 526-536.
1291	–	*/
1292	–	void SimInfo::getGyrationalVolume(RealType &volume){
1293	–	Mat3x3d intTensor;
1294	–	RealType det;
1295	–	Vector3d dummyAngMom;
1296	–	RealType sysconstants;
1297	–	RealType geomCnst;
1298	–
1299	–	geomCnst = 3.0/2.0;
1300	–	/* Get the inertial tensor and angular momentum for free*/
1301	–	getInertiaTensor(intTensor,dummyAngMom);
1302	–
1303	–	det = intTensor.determinant();
1304	–	sysconstants = geomCnst/(RealType)nGlobalIntegrableObjects_;
1305	–	volume = 4.0/3.0NumericConstant::PIpow(sysconstants,3.0/2.0)*sqrt(det);
1306	–	return;
1307	–	}
1308	–
1309	–	void SimInfo::getGyrationalVolume(RealType &volume, RealType &detI){
1310	–	Mat3x3d intTensor;
1311	–	Vector3d dummyAngMom;
1312	–	RealType sysconstants;
1313	–	RealType geomCnst;
1314	–
1315	–	geomCnst = 3.0/2.0;
1316	–	/* Get the inertial tensor and angular momentum for free*/
1317	–	getInertiaTensor(intTensor,dummyAngMom);
1318	–
1319	–	detI = intTensor.determinant();
1320	–	sysconstants = geomCnst/(RealType)nGlobalIntegrableObjects_;
1321	–	volume = 4.0/3.0NumericConstant::PIpow(sysconstants,3.0/2.0)*sqrt(detI);
1322	–	return;
1323	–	}
1011		/*
1012		void SimInfo::setStuntDoubleFromGlobalIndex(vector<StuntDouble*> v) {
1013		assert( v.size() == nAtoms_ + nRigidBodies_);

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing branches/development/src/brains/SimInfo.cpp (file contents): Revision 1528 by gezelter, Fri Dec 17 20:11:05 2010 UTC vs. Revision 1769 by gezelter, Mon Jul 9 14:15:52 2012 UTC

Diff Legend

Comparing branches/development/src/brains/SimInfo.cpp (file contents):
Revision 1528 by gezelter, Fri Dec 17 20:11:05 2010 UTC vs.
Revision 1769 by gezelter, Mon Jul 9 14:15:52 2012 UTC