| 54 |
|
/** |
| 55 |
|
* |
| 56 |
|
*/ |
| 57 |
< |
void ZConsWriter::writeFZ(double time, int num, int* index, double* fz, double* curZPos){ |
| 57 |
> |
void ZConsWriter::writeFZ(double time, int num, int* index, double* fz, double* curZPos, double* zpos){ |
| 58 |
|
|
| 59 |
|
#ifndef IS_MPI |
| 60 |
|
output << time << endl; |
| 61 |
|
output << num << endl; |
| 62 |
|
|
| 63 |
|
for(int i = 0; i < num; i++) |
| 64 |
< |
output << index[i] <<"\t" << fz[i] << "\t" << curZPos[i] << endl; |
| 64 |
> |
output << index[i] <<"\t" << fz[i] << "\t" << curZPos[i] << "\t" << zpos[i] <<endl; |
| 65 |
|
|
| 66 |
|
#else |
| 67 |
|
int totalNum; |
| 76 |
|
enum CommType { RequesPosAndForce, EndOfRequest} status; |
| 77 |
|
double pos; |
| 78 |
|
double force; |
| 79 |
+ |
double zconsPos; |
| 80 |
|
int localIndex; |
| 81 |
|
MPI_Status ierr; |
| 82 |
|
int tag = 0; |
| 110 |
|
MPI_Send(&globalIndexOfCurMol, 1, MPI_INT, whichNode, tag, MPI_COMM_WORLD); |
| 111 |
|
MPI_Recv(&force, 1, MPI_DOUBLE, whichNode, tag, MPI_COMM_WORLD, &ierr); |
| 112 |
|
MPI_Recv(&pos, 1, MPI_DOUBLE, whichNode, tag, MPI_COMM_WORLD, &ierr); |
| 113 |
+ |
MPI_Recv(&zconsPos, 1, MPI_DOUBLE, whichNode, tag, MPI_COMM_WORLD, &ierr); |
| 114 |
|
} |
| 115 |
|
|
| 116 |
< |
output << globalIndexOfCurMol << "\t" << force << "\t" << pos << endl; |
| 116 |
> |
output << globalIndexOfCurMol << "\t" << force << "\t" << pos << "\t"<< zconsPos << endl; |
| 117 |
|
|
| 118 |
|
} //End of Request Loop |
| 119 |
|
|
| 145 |
|
} |
| 146 |
|
|
| 147 |
|
MPI_Send(&fz[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD); |
| 148 |
< |
MPI_Send(&curZPos[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD); |
| 148 |
> |
MPI_Send(&curZPos[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD); |
| 149 |
> |
MPI_Send(&zpos[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD); |
| 150 |
|
break; |
| 151 |
|
|
| 152 |
|
case EndOfRequest : |