# | Line 1 | Line 1 | |
---|---|---|
1 | < | /** |
2 | < | * @file ForceDecomposition.cpp |
3 | < | * @author Charles Vardeman <cvardema.at.nd.edu> |
4 | < | * @date 08/18/2010 |
5 | < | * @time 11:56am |
6 | < | * @version 1.0 |
1 | > | /* |
2 | > | * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved. |
3 | * | |
8 | – | * @section LICENSE |
9 | – | * Copyright (c) 2010 The University of Notre Dame. All Rights Reserved. |
10 | – | * |
4 | * The University of Notre Dame grants you ("Licensee") a | |
5 | * non-exclusive, royalty free, license to use, modify and | |
6 | * redistribute this software in source and binary code form, provided | |
# | Line 45 | Line 38 | |
38 | * [3] Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008). | |
39 | * [4] Vardeman & Gezelter, in progress (2009). | |
40 | */ | |
41 | + | |
42 | + | #ifndef PARALLEL_FORCEMATRIXDECOMPOSITION_HPP |
43 | + | #define PARALLEL_FORCEMATRIXDECOMPOSITION_HPP |
44 | ||
45 | + | #include "parallel/ForceDecomposition.hpp" |
46 | + | #include "math/SquareMatrix3.hpp" |
47 | + | #include "brains/Snapshot.hpp" |
48 | ||
49 | < | /* -*- c++ -*- */ |
50 | < | #ifndef FORCEDECOMPOSITION_PARALLEL_HPP |
51 | < | #define FORCEDECOMPOSITION_PARALLEL_HPP |
49 | > | #ifdef IS_MPI |
50 | > | #include "parallel/Communicator.hpp" |
51 | > | #endif |
52 | ||
53 | + | using namespace std; |
54 | + | namespace OpenMD { |
55 | + | |
56 | + | class ForceMatrixDecomposition : public ForceDecomposition { |
57 | + | public: |
58 | + | ForceMatrixDecomposition(SimInfo* info) : ForceDecomposition(info) {}; |
59 | ||
60 | < | namespace OpenMD{ |
60 | > | void distributeInitialData(); |
61 | > | void distributeData(); |
62 | > | void collectIntermediateData(); |
63 | > | void distributeIntermediateData(); |
64 | > | void collectData(); |
65 | ||
66 | < | class ForceDecomposition |
67 | < | { |
59 | < | public: |
60 | < | ForceDecomposition(); |
61 | < | virtual ~ForceDecomposition(); |
62 | < | void gather(); |
63 | < | void scatter(); |
64 | < | |
65 | < | protected: |
66 | < | private: |
67 | < | int myRank_; |
68 | < | static bool isColumn_; |
69 | < | static int myIndex_: |
70 | < | }; |
66 | > | // neighbor list routines |
67 | > | vector<pair<int, int> > buildNeighborList(); |
68 | ||
69 | + | // group bookkeeping |
70 | + | pair<int, int> getGroupTypes(int cg1, int cg2); |
71 | ||
72 | + | // Group->atom bookkeeping |
73 | + | vector<int> getAtomsInGroupRow(int cg1); |
74 | + | vector<int> getAtomsInGroupColumn(int cg2); |
75 | + | Vector3d getAtomToGroupVectorRow(int atom1, int cg1); |
76 | + | Vector3d getAtomToGroupVectorColumn(int atom2, int cg2); |
77 | + | RealType getMassFactorRow(int atom1); |
78 | + | RealType getMassFactorColumn(int atom2); |
79 | ||
80 | < | } |
80 | > | // spatial data |
81 | > | Vector3d getIntergroupVector(int cg1, int cg2); |
82 | > | Vector3d getInteratomicVector(int atom1, int atom2); |
83 | > | |
84 | > | // atom bookkeeping |
85 | > | vector<int> getAtomList(); |
86 | > | vector<int> getSkipsForAtom(int atom1); |
87 | > | bool skipAtomPair(int atom1, int atom2); |
88 | > | void addForceToAtomRow(int atom1, Vector3d fg); |
89 | > | void addForceToAtomColumn(int atom2, Vector3d fg); |
90 | ||
91 | + | // filling interaction blocks with pointers |
92 | + | InteractionData fillInteractionData(int atom1, int atom2); |
93 | + | InteractionData fillSkipData(int atom1, int atom2); |
94 | ||
95 | + | private: |
96 | + | int nLocal_; |
97 | + | int nGroups_; |
98 | + | vector<int> AtomLocalToGlobal; |
99 | + | vector<int> cgLocalToGlobal; |
100 | + | vector<RealType> pot_local; |
101 | + | vector<RealType> massFactorsLocal; |
102 | ||
103 | + | #ifdef IS_MPI |
104 | + | DataStorage atomRowData; |
105 | + | DataStorage atomColData; |
106 | + | DataStorage cgRowData; |
107 | + | DataStorage cgColData; |
108 | ||
109 | + | int nAtomsInRow_; |
110 | + | int nAtomsInCol_; |
111 | + | int nGroupsInRow_; |
112 | + | int nGroupsInCol_; |
113 | ||
114 | + | Communicator<Row, int>* AtomCommIntRow; |
115 | + | Communicator<Row, RealType>* AtomCommRealRow; |
116 | + | Communicator<Row, Vector3d>* AtomCommVectorRow; |
117 | + | Communicator<Row, Mat3x3d>* AtomCommMatrixRow; |
118 | ||
119 | + | Communicator<Column, int>* AtomCommIntColumn; |
120 | + | Communicator<Column, RealType>* AtomCommRealColumn; |
121 | + | Communicator<Column, Vector3d>* AtomCommVectorColumn; |
122 | + | Communicator<Column, Mat3x3d>* AtomCommMatrixColumn; |
123 | ||
124 | + | Communicator<Row, int>* cgCommIntRow; |
125 | + | Communicator<Row, Vector3d>* cgCommVectorRow; |
126 | + | Communicator<Column, int>* cgCommIntColumn; |
127 | + | Communicator<Column, Vector3d>* cgCommVectorColumn; |
128 | ||
129 | + | vector<vector<RealType> > pot_row; |
130 | + | vector<vector<RealType> > pot_col; |
131 | + | |
132 | + | vector<int> identsRow; |
133 | + | vector<int> identsCol; |
134 | + | |
135 | + | vector<int> AtomRowToGlobal; |
136 | + | vector<int> AtomColToGlobal; |
137 | + | |
138 | + | vector<int> cgRowToGlobal; |
139 | + | vector<int> cgColToGlobal; |
140 | + | |
141 | + | vector<vector<int> > cellListRow_; |
142 | + | vector<vector<int> > cellListCol_; |
143 | + | |
144 | + | vector<vector<int> > groupListRow_; |
145 | + | vector<vector<int> > groupListCol_; |
146 | + | |
147 | + | vector<RealType> massFactorsRow; |
148 | + | vector<RealType> massFactorsCol; |
149 | + | |
150 | #endif | |
151 | ||
152 | + | }; |
153 | ||
154 | + | } |
155 | + | #endif |
156 | + |
– | Removed lines |
+ | Added lines |
< | Changed lines |
> | Changed lines |