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