| 6 |
|
#include <mpi.h> |
| 7 |
|
#include <mpi++.h> |
| 8 |
|
#include "mpiSimulation.hpp" |
| 9 |
< |
#define TAKE_THIS_TAG 1 |
| 9 |
> |
#define TAKE_THIS_TAG_CHAR 1 |
| 10 |
> |
#define TAKE_THIS_TAG_INT 2 |
| 11 |
|
#endif //is_mpi |
| 12 |
|
|
| 13 |
|
#include "ReadWrite.hpp" |
| 139 |
|
|
| 140 |
|
which_node = AtomToProcMap[i]; |
| 141 |
|
|
| 142 |
< |
if (which_node == mpiSim->getMyNode()) { |
| 142 |
< |
sprintf( tempBuffer, |
| 143 |
< |
"%s\t%lf\t%lf\t%lf\t%lf\t%lf\t%lf\t", |
| 144 |
< |
atoms[i]->getType(), |
| 145 |
< |
atoms[i]->getX(), |
| 146 |
< |
atoms[i]->getY(), |
| 147 |
< |
atoms[i]->getZ(), |
| 148 |
< |
atoms[i]->get_vx(), |
| 149 |
< |
atoms[i]->get_vy(), |
| 150 |
< |
atoms[i]->get_vz()); |
| 151 |
< |
strcpy( writeLine, tempBuffer ); |
| 142 |
> |
if (which_node == 0 ) { |
| 143 |
|
|
| 144 |
< |
if( atoms[i]->isDirectional() ){ |
| 145 |
< |
|
| 146 |
< |
dAtom = (DirectionalAtom *)atoms[i]; |
| 147 |
< |
dAtom->getQ( q ); |
| 157 |
< |
|
| 158 |
< |
sprintf( tempBuffer, |
| 159 |
< |
"%lf\t%lf\t%lf\t%lf\t%lf\t%lf\t%lf\n", |
| 160 |
< |
q[0], |
| 161 |
< |
q[1], |
| 162 |
< |
q[2], |
| 163 |
< |
q[3], |
| 164 |
< |
dAtom->getJx(), |
| 165 |
< |
dAtom->getJy(), |
| 166 |
< |
dAtom->getJz()); |
| 167 |
< |
strcat( writeLine, tempBuffer ); |
| 144 |
> |
which_atom = i; |
| 145 |
> |
local_index=-1; |
| 146 |
> |
for (j=0; (j<mpiSim->getMyNlocal()) && (local_index < 0); j++) { |
| 147 |
> |
if (atoms[j]->getGlobalIndex() == which_atom) local_index = j; |
| 148 |
|
} |
| 149 |
< |
else |
| 150 |
< |
strcat( writeLine, "0.0\t0.0\t0.0\t0.0\t0.0\t0.0\t0.0\n" ); |
| 151 |
< |
|
| 152 |
< |
} else { |
| 153 |
< |
|
| 154 |
< |
std::cerr << "node 0: sending node " << which_node << " request for atom " << i << "\n"; |
| 155 |
< |
MPI::COMM_WORLD.Send(&i, 1, MPI_INT, which_node, TAKE_THIS_TAG); |
| 156 |
< |
std::cerr << "node 0: sent!\n"; |
| 149 |
> |
if (local_index != -1) { |
| 150 |
> |
//format the line |
| 151 |
> |
sprintf( tempBuffer, |
| 152 |
> |
"%s\t%lf\t%lf\t%lf\t%lf\t%lf\t%lf\t", |
| 153 |
> |
atoms[local_index]->getType(), |
| 154 |
> |
atoms[local_index]->getX(), |
| 155 |
> |
atoms[local_index]->getY(), |
| 156 |
> |
atoms[local_index]->getZ(), |
| 157 |
> |
atoms[local_index]->get_vx(), |
| 158 |
> |
atoms[local_index]->get_vy(), |
| 159 |
> |
atoms[local_index]->get_vz()); // check here. |
| 160 |
> |
strcpy( writeLine, tempBuffer ); |
| 161 |
> |
|
| 162 |
> |
if( atoms[local_index]->isDirectional() ){ |
| 163 |
> |
|
| 164 |
> |
dAtom = (DirectionalAtom *)atoms[local_index]; |
| 165 |
> |
dAtom->getQ( q ); |
| 166 |
> |
|
| 167 |
> |
sprintf( tempBuffer, |
| 168 |
> |
"%lf\t%lf\t%lf\t%lf\t%lf\t%lf\t%lf\n", |
| 169 |
> |
q[0], |
| 170 |
> |
q[1], |
| 171 |
> |
q[2], |
| 172 |
> |
q[3], |
| 173 |
> |
dAtom->getJx(), |
| 174 |
> |
dAtom->getJy(), |
| 175 |
> |
dAtom->getJz()); |
| 176 |
> |
strcat( writeLine, tempBuffer ); |
| 177 |
> |
|
| 178 |
> |
} |
| 179 |
> |
else |
| 180 |
> |
strcat( writeLine, "0.0\t0.0\t0.0\t0.0\t0.0\t0.0\t0.0\n" ); |
| 181 |
> |
} |
| 182 |
> |
else { |
| 183 |
> |
strcpy( writeLine, "ATOM NOT FOUND ON THIS PROCESSOR"); |
| 184 |
> |
} |
| 185 |
> |
} |
| 186 |
> |
else { |
| 187 |
> |
|
| 188 |
> |
//std::cerr << "node 0: sending node " << which_node << " request for atom " << i << "\n"; |
| 189 |
> |
MPI::COMM_WORLD.Send(&i, 1, MPI_INT, which_node, TAKE_THIS_TAG_INT); |
| 190 |
> |
//std::cerr << "node 0: sent!\n"; |
| 191 |
|
MPI::COMM_WORLD.Recv(writeLine, BUFFERSIZE, MPI_CHAR, which_node, |
| 192 |
< |
TAKE_THIS_TAG, istatus); |
| 193 |
< |
std::cerr << "node 0: got this line: " << writeLine; |
| 192 |
> |
TAKE_THIS_TAG_CHAR, istatus); |
| 193 |
> |
//std::cerr << "node 0: got this line: " << writeLine; |
| 194 |
|
} |
| 195 |
|
|
| 196 |
|
outFile << writeLine; |
| 200 |
|
// kill everyone off: |
| 201 |
|
game_over = -1; |
| 202 |
|
for (j = 0; j < mpiSim->getNumberProcessors(); j++) { |
| 203 |
< |
MPI::COMM_WORLD.Send(&game_over, 1, MPI_INT, j, TAKE_THIS_TAG); |
| 203 |
> |
MPI::COMM_WORLD.Send(&game_over, 1, MPI_INT, j, TAKE_THIS_TAG_INT); |
| 204 |
|
} |
| 205 |
|
|
| 206 |
|
} else { |
| 207 |
|
|
| 208 |
|
done = 0; |
| 209 |
|
while (!done) { |
| 210 |
< |
std::cerr << "node: " << mpiSim->getMyNode() << " Waiting for receive \n"; |
| 210 |
> |
//std::cerr << "node: " << mpiSim->getMyNode() << " Waiting for receive \n"; |
| 211 |
|
MPI::COMM_WORLD.Recv(&which_atom, 1, MPI_INT, 0, |
| 212 |
< |
TAKE_THIS_TAG, istatus); |
| 213 |
< |
std::cerr << "node: " << mpiSim->getMyNode() << " got request for atom " << which_atom << "\n"; |
| 212 |
> |
TAKE_THIS_TAG_INT, istatus); |
| 213 |
> |
//std::cerr << "node: " << mpiSim->getMyNode() << " got request for atom " << which_atom << "\n"; |
| 214 |
|
if (which_atom == -1) { |
| 215 |
|
done=1; |
| 216 |
|
continue; |
| 217 |
|
} else { |
| 218 |
|
local_index=-1; |
| 219 |
< |
for (j=0; j < mpiSim->getMyNlocal(); j++) { |
| 219 |
> |
for (j=0; (j<mpiSim->getMyNlocal()) && (local_index < 0); j++) { |
| 220 |
|
if (atoms[j]->getGlobalIndex() == which_atom) local_index = j; |
| 221 |
|
} |
| 222 |
|
if (local_index != -1) { |
| 250 |
|
} |
| 251 |
|
else |
| 252 |
|
strcat( writeLine, "0.0\t0.0\t0.0\t0.0\t0.0\t0.0\t0.0\n" ); |
| 253 |
< |
std::cerr << "node: " << mpiSim->getMyNode() << " sending this line" << writeLine; |
| 253 |
> |
// std::cerr << "node: " << mpiSim->getMyNode() << " sending this line" << writeLine; |
| 254 |
|
MPI::COMM_WORLD.Send(writeLine, BUFFERSIZE, MPI_CHAR, 0, |
| 255 |
< |
TAKE_THIS_TAG); |
| 255 |
> |
TAKE_THIS_TAG_CHAR); |
| 256 |
|
} else { |
| 257 |
|
strcpy( writeLine, "ATOM NOT FOUND ON THIS PROCESSOR"); |
| 258 |
|
MPI::COMM_WORLD.Send(writeLine, BUFFERSIZE, MPI_CHAR, 0, |
| 259 |
< |
TAKE_THIS_TAG); |
| 259 |
> |
TAKE_THIS_TAG_CHAR); |
| 260 |
|
} |
| 261 |
|
} |
| 262 |
|
} |
| 408 |
|
|
| 409 |
|
} else { |
| 410 |
|
|
| 411 |
< |
MPI::COMM_WORLD.Send(&i, 1, MPI_INT, which_node, TAKE_THIS_TAG); |
| 411 |
> |
MPI::COMM_WORLD.Send(&i, 1, MPI_INT, which_node, TAKE_THIS_TAG_INT); |
| 412 |
|
MPI::COMM_WORLD.Recv(writeLine, BUFFERSIZE, MPI_CHAR, which_node, |
| 413 |
< |
TAKE_THIS_TAG, istatus); |
| 413 |
> |
TAKE_THIS_TAG_CHAR, istatus); |
| 414 |
|
} |
| 415 |
|
|
| 416 |
|
finalOut << writeLine; |
| 419 |
|
// kill everyone off: |
| 420 |
|
game_over = -1; |
| 421 |
|
for (j = 0; j < mpiSim->getNumberProcessors(); j++) { |
| 422 |
< |
MPI::COMM_WORLD.Send(&game_over, 1, MPI_INT, j, TAKE_THIS_TAG); |
| 422 |
> |
MPI::COMM_WORLD.Send(&game_over, 1, MPI_INT, j, TAKE_THIS_TAG_INT); |
| 423 |
|
} |
| 424 |
|
|
| 425 |
|
} else { |
| 427 |
|
done = 0; |
| 428 |
|
while (!done) { |
| 429 |
|
MPI::COMM_WORLD.Recv(&which_atom, 1, MPI_INT, 0, |
| 430 |
< |
TAKE_THIS_TAG, istatus); |
| 430 |
> |
TAKE_THIS_TAG_INT, istatus); |
| 431 |
|
|
| 432 |
|
if (which_atom == -1) { |
| 433 |
|
done=1; |
| 472 |
|
strcat( writeLine, "0.0\t0.0\t0.0\t0.0\t0.0\t0.0\t0.0\n" ); |
| 473 |
|
|
| 474 |
|
MPI::COMM_WORLD.Send(writeLine, BUFFERSIZE, MPI_CHAR, 0, |
| 475 |
< |
TAKE_THIS_TAG); |
| 475 |
> |
TAKE_THIS_TAG_CHAR); |
| 476 |
|
} else { |
| 477 |
|
strcpy( writeLine, "ATOM NOT FOUND ON THIS PROCESSOR"); |
| 478 |
|
MPI::COMM_WORLD.Send(writeLine, BUFFERSIZE, MPI_CHAR, 0, |
| 479 |
< |
TAKE_THIS_TAG); |
| 479 |
> |
TAKE_THIS_TAG_CHAR); |
| 480 |
|
} |
| 481 |
|
} |
| 482 |
|
} |