75 |
|
#include "antlr/MismatchedTokenException.hpp" |
76 |
|
#include "antlr/NoViableAltForCharException.hpp" |
77 |
|
#include "antlr/NoViableAltException.hpp" |
78 |
+ |
|
79 |
+ |
#include "types/DirectionalAdapter.hpp" |
80 |
+ |
#include "types/MultipoleAdapter.hpp" |
81 |
+ |
#include "types/EAMAdapter.hpp" |
82 |
+ |
#include "types/SuttonChenAdapter.hpp" |
83 |
+ |
#include "types/PolarizableAdapter.hpp" |
84 |
+ |
#include "types/FixedChargeAdapter.hpp" |
85 |
+ |
#include "types/FluctuatingChargeAdapter.hpp" |
86 |
|
|
87 |
|
#ifdef IS_MPI |
88 |
|
#include "mpi.h" |
422 |
|
//create the molecules |
423 |
|
createMolecules(info); |
424 |
|
|
425 |
+ |
//find the storage layout |
426 |
+ |
|
427 |
+ |
int storageLayout = computeStorageLayout(info); |
428 |
+ |
|
429 |
|
//allocate memory for DataStorage(circular reference, need to |
430 |
|
//break it) |
431 |
< |
info->setSnapshotManager(new SimSnapshotManager(info)); |
431 |
> |
info->setSnapshotManager(new SimSnapshotManager(info, storageLayout)); |
432 |
|
|
433 |
|
//set the global index of atoms, rigidbodies and cutoffgroups |
434 |
|
//(only need to be set once, the global index will never change |
665 |
|
} //end for(int i=0) |
666 |
|
} |
667 |
|
|
668 |
+ |
int SimCreator::computeStorageLayout(SimInfo* info) { |
669 |
+ |
|
670 |
+ |
Globals* simParams = info->getSimParams(); |
671 |
+ |
int nRigidBodies = info->getNGlobalRigidBodies(); |
672 |
+ |
set<AtomType*> atomTypes = info->getSimulatedAtomTypes(); |
673 |
+ |
set<AtomType*>::iterator i; |
674 |
+ |
bool hasDirectionalAtoms = false; |
675 |
+ |
bool hasFixedCharge = false; |
676 |
+ |
bool hasMultipoles = false; |
677 |
+ |
bool hasPolarizable = false; |
678 |
+ |
bool hasFluctuatingCharge = false; |
679 |
+ |
bool hasMetallic = false; |
680 |
+ |
int storageLayout = 0; |
681 |
+ |
storageLayout |= DataStorage::dslPosition; |
682 |
+ |
storageLayout |= DataStorage::dslVelocity; |
683 |
+ |
storageLayout |= DataStorage::dslForce; |
684 |
+ |
|
685 |
+ |
for (i = atomTypes.begin(); i != atomTypes.end(); ++i) { |
686 |
+ |
|
687 |
+ |
DirectionalAdapter da = DirectionalAdapter( (*i) ); |
688 |
+ |
MultipoleAdapter ma = MultipoleAdapter( (*i) ); |
689 |
+ |
EAMAdapter ea = EAMAdapter( (*i) ); |
690 |
+ |
SuttonChenAdapter sca = SuttonChenAdapter( (*i) ); |
691 |
+ |
PolarizableAdapter pa = PolarizableAdapter( (*i) ); |
692 |
+ |
FixedChargeAdapter fca = FixedChargeAdapter( (*i) ); |
693 |
+ |
FluctuatingChargeAdapter fqa = FluctuatingChargeAdapter( (*i) ); |
694 |
+ |
|
695 |
+ |
if (da.isDirectional()){ |
696 |
+ |
hasDirectionalAtoms = true; |
697 |
+ |
} |
698 |
+ |
if (ma.isMultipole()){ |
699 |
+ |
hasMultipoles = true; |
700 |
+ |
} |
701 |
+ |
if (ea.isEAM() || sca.isSuttonChen()){ |
702 |
+ |
hasMetallic = true; |
703 |
+ |
} |
704 |
+ |
if ( fca.isFixedCharge() ){ |
705 |
+ |
hasFixedCharge = true; |
706 |
+ |
} |
707 |
+ |
if ( fqa.isFluctuatingCharge() ){ |
708 |
+ |
hasFluctuatingCharge = true; |
709 |
+ |
} |
710 |
+ |
if ( pa.isPolarizable() ){ |
711 |
+ |
hasPolarizable = true; |
712 |
+ |
} |
713 |
+ |
} |
714 |
+ |
|
715 |
+ |
if (nRigidBodies > 0 || hasDirectionalAtoms) { |
716 |
+ |
storageLayout |= DataStorage::dslAmat; |
717 |
+ |
if(storageLayout & DataStorage::dslVelocity) { |
718 |
+ |
storageLayout |= DataStorage::dslAngularMomentum; |
719 |
+ |
} |
720 |
+ |
if (storageLayout & DataStorage::dslForce) { |
721 |
+ |
storageLayout |= DataStorage::dslTorque; |
722 |
+ |
} |
723 |
+ |
} |
724 |
+ |
if (hasMultipoles) { |
725 |
+ |
storageLayout |= DataStorage::dslElectroFrame; |
726 |
+ |
} |
727 |
+ |
if (hasFixedCharge || hasFluctuatingCharge) { |
728 |
+ |
storageLayout |= DataStorage::dslSkippedCharge; |
729 |
+ |
} |
730 |
+ |
if (hasMetallic) { |
731 |
+ |
storageLayout |= DataStorage::dslDensity; |
732 |
+ |
storageLayout |= DataStorage::dslFunctional; |
733 |
+ |
storageLayout |= DataStorage::dslFunctionalDerivative; |
734 |
+ |
} |
735 |
+ |
if (hasPolarizable) { |
736 |
+ |
storageLayout |= DataStorage::dslElectricField; |
737 |
+ |
} |
738 |
+ |
if (hasFluctuatingCharge){ |
739 |
+ |
storageLayout |= DataStorage::dslFlucQPosition; |
740 |
+ |
if(storageLayout & DataStorage::dslVelocity) { |
741 |
+ |
storageLayout |= DataStorage::dslFlucQVelocity; |
742 |
+ |
} |
743 |
+ |
if (storageLayout & DataStorage::dslForce) { |
744 |
+ |
storageLayout |= DataStorage::dslFlucQForce; |
745 |
+ |
} |
746 |
+ |
} |
747 |
+ |
|
748 |
+ |
// if the user has asked for them, make sure we've got the memory for the |
749 |
+ |
// objects defined. |
750 |
+ |
|
751 |
+ |
if (simParams->getOutputParticlePotential()) { |
752 |
+ |
storageLayout |= DataStorage::dslParticlePot; |
753 |
+ |
} |
754 |
+ |
|
755 |
+ |
if (simParams->havePrintHeatFlux()) { |
756 |
+ |
if (simParams->getPrintHeatFlux()) { |
757 |
+ |
storageLayout |= DataStorage::dslParticlePot; |
758 |
+ |
} |
759 |
+ |
} |
760 |
+ |
|
761 |
+ |
if (simParams->getOutputElectricField()) { |
762 |
+ |
storageLayout |= DataStorage::dslElectricField; |
763 |
+ |
} |
764 |
+ |
if (simParams->getOutputFluctuatingCharges()) { |
765 |
+ |
storageLayout |= DataStorage::dslFlucQPosition; |
766 |
+ |
storageLayout |= DataStorage::dslFlucQVelocity; |
767 |
+ |
storageLayout |= DataStorage::dslFlucQForce; |
768 |
+ |
} |
769 |
+ |
|
770 |
+ |
return storageLayout; |
771 |
+ |
} |
772 |
+ |
|
773 |
|
void SimCreator::setGlobalIndex(SimInfo *info) { |
774 |
|
SimInfo::MoleculeIterator mi; |
775 |
|
Molecule::AtomIterator ai; |