| 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; |