--- branches/development/src/parallel/ForceDecomposition.hpp 2011/01/17 21:34:36 1540 +++ branches/development/src/parallel/ForceMatrixDecomposition.hpp 2011/05/24 21:24:45 1567 @@ -39,47 +39,114 @@ * [4] Vardeman & Gezelter, in progress (2009). */ -#ifndef PARALLEL_FORCEDECOMPOSITION_HPP -#define PARALLEL_FORCEDECOMPOSITION_HPP +#ifndef PARALLEL_FORCEMATRIXDECOMPOSITION_HPP +#define PARALLEL_FORCEMATRIXDECOMPOSITION_HPP -#include "Parallel/Decomposition.hpp" -#include "Parallel/Communicator.hpp" +#include "parallel/ForceDecomposition.hpp" #include "math/SquareMatrix3.hpp" +#include "brains/Snapshot.hpp" +#ifdef IS_MPI +#include "parallel/Communicator.hpp" +#endif + +using namespace std; namespace OpenMD { - class ForceDecomposition : public Decomposition { + class ForceMatrixDecomposition : public ForceDecomposition { public: - ForceDecomposition(Snapshot* sman); + ForceMatrixDecomposition(SimInfo* info) : ForceDecomposition(info) {sman_ = info_->getSnapshotManager();} void distributeInitialData(); void distributeData(); void collectIntermediateData(); void distributeIntermediateData(); void collectData(); - unsigned int getNcutoffGroupsI(); - unsigned int getNcutoffGroupsJ(); + // neighbor list routines + bool checkNeighborList(); + vector > buildNeighborList(); - vector getAtomsInGroupI(int whichCGI); - vector getAtomsInGroupJ(int whichCGJ); + // group bookkeeping + pair getGroupTypes(int cg1, int cg2); - AtomType* getAtomTypeI(int whichAtomI); - AtomType* getAtomTypeJ(int whichAtomJ); + // Group->atom bookkeeping + vector getAtomsInGroupRow(int cg1); + vector getAtomsInGroupColumn(int cg2); + Vector3d getAtomToGroupVectorRow(int atom1, int cg1); + Vector3d getAtomToGroupVectorColumn(int atom2, int cg2); + RealType getMfactRow(int atom1); + RealType getMfactColumn(int atom2); -#ifdef IS_MPI - - Communicator* AtomCommRealI; - Communicator* AtomCommVectorI; - Communicator* AtomCommMatrixI; + // spatial data + Vector3d getIntergroupVector(int cg1, int cg2); + Vector3d getInteratomicVector(int atom1, int atom2); + + // atom bookkeeping + vector getAtomList(); + vector getSkipsForAtom(int atom1); + bool skipAtomPair(int atom1, int atom2); + void addForceToAtomRow(int atom1, Vector3d fg); + void addForceToAtomColumn(int atom2, Vector3d fg); - Communicator* AtomCommRealJ; - Communicator* AtomCommVectorJ; - Communicator* AtomCommMatrixJ; + // filling interaction blocks with pointers + InteractionData fillInteractionData(int atom1, int atom2); + InteractionData fillSkipData(int atom1, int atom2); + SelfData fillSelfData(int atom1); - Communicator* cgCommVectorI; - Communicator* cgCommVectorJ; - + private: + SnapshotManager* sman_; + Snapshot* snap_; + int storageLayout_; + vector Cells; + + int nLocal_; + int nGroups_; + +#ifdef IS_MPI + DataStorage atomRowData; + DataStorage atomColData; + DataStorage cgRowData; + DataStorage cgColData; + + int nAtomsInRow_; + int nAtomsInCol_; + int nGroupsInRow_; + int nGroupsInCol_; + + Communicator* AtomCommIntRow; + Communicator* AtomCommRealRow; + Communicator* AtomCommVectorRow; + Communicator* AtomCommMatrixRow; + + Communicator* AtomCommIntColumn; + Communicator* AtomCommRealColumn; + Communicator* AtomCommVectorColumn; + Communicator* AtomCommMatrixColumn; + + Communicator* cgCommIntRow; + Communicator* cgCommVectorRow; + Communicator* cgCommIntColumn; + Communicator* cgCommVectorColumn; + + vector > pot_row; + vector > pot_col; + vector identsRow; + vector identsCol; + + vector AtomLocalToGlobal; + vector AtomRowToGlobal; + vector AtomColToGlobal; + vector cgLocalToGlobal; + vector cgRowToGlobal; + vector cgColToGlobal; + + vector > CellListRow; + vector > CellListCol; +#else + vector > CellList; #endif + vector pot_local; + }; }