int main(int argc, char *argv[])
{
int rc;
int num_tasks;
int task_id;
double *old, *current, *next, *ret;
int t_max, i_max;
double time;
rc = MPI_Init(&argc, &argv); // Initialize MPI runtime
if (rc != MPI_SUCCESS) { // Check for success
fprintf(stderr, "Unable to set up MPI\n");
MPI_Abort(MPI_COMM_WORLD, rc); // Abort MPI runtime
}
/* Parse commandline args */
if (argc < 3) {
printf("Usage: %s i_max t_max num_threads [initial_data]\n", argv[0]);
printf(" - i_max: number of discrete amplitude points, should be >2\n");
printf(" - t_max: number of discrete timesteps, should be >=1\n");
printf(" - num_threads: number of threads to use for simulation, "
"should be >=1\n");
printf(" - initial_data: select what data should be used for the first "
"two generation.\n");
printf(" Available options are:\n");
printf(" * sin: one period of the sinus function at the start.\n");
printf(" * sinfull: entire data is filled with the sinus.\n");
printf(" * gauss: a single gauss-function at the start.\n");
printf(" * file <2 filenames>: allows you to specify a file with on "
"each line a float for both generations.\n");
return EXIT_FAILURE;
}
i_max = atoi(argv[1]);
t_max = atoi(argv[2]);
if (i_max < 3) {
printf("argument error: i_max should be >2.\n");
return EXIT_FAILURE;
}
if (t_max < 1) {
printf("argument error: t_max should be >=1.\n");
return EXIT_FAILURE;
}
if (task_id == 0) {
timer_start();
}
/* Get MPI rankings */
MPI_Comm_size(MPI_COMM_WORLD, &num_tasks);
MPI_Comm_rank(MPI_COMM_WORLD, &task_id);
/* Scale i_max to the number of threads
* with halo cells in mind(left and right) */
i_max = (i_max/num_tasks) + 2;
/* Allocate and initialize buffers. */
old = malloc(i_max * sizeof(double));
current = malloc(i_max * sizeof(double));
next = malloc(i_max * sizeof(double));
if (old == NULL || current == NULL || next == NULL) {
fprintf(stderr, "Could not allocate enough memory, aborting.\n");
return EXIT_FAILURE;
}
memset(old, 0, i_max * sizeof(double));
memset(current, 0, i_max * sizeof(double));
memset(next, 0, i_max * sizeof(double));
/* How should we will our first two generations? This is determined by the
* optional further commandline arguments.
*/
if (argc > 3) {
if (strcmp(argv[3], "sin") == 0) {
fill(old, 1, i_max/4, 0, 2*3.14, num_tasks, task_id, sin);
fill(current, 2, i_max/4, 0, 2*3.14, num_tasks, task_id, sin);
} else if (strcmp(argv[3], "sinfull") == 0) {
fill(old, 1, i_max-2, 0, 10*3.14, num_tasks, task_id, sin);
fill(current, 2, i_max-3, 0, 10*3.14, num_tasks, task_id, sin);
} else if (strcmp(argv[3], "gauss") == 0) {
fill(old, 1, i_max/4, -3, 3, num_tasks, task_id, gauss);
fill(current, 2, i_max/4, -3, 3, num_tasks, task_id, gauss);
} else if (strcmp(argv[3], "file") == 0) {
if (argc < 6) {
printf("No files specified!\n");
return EXIT_FAILURE;
}
file_read_double_array(argv[4], old, i_max);
file_read_double_array(argv[5], current, i_max);
} else {
printf("Unknown initial mode: %s.\n", argv[3]);
return EXIT_FAILURE;
}
} else {
/* Default to sinus. */
fill(old, 1, i_max/4, 0, 2*3.14, num_tasks, task_id, sin);
fill(current, 2, i_max/4, 0, 2*3.14, num_tasks, task_id, sin);
}
/* Call the actual simulation that should be implemented in simulate.c. */
ret = simulate(i_max, t_max, old, current, next, num_tasks, task_id);
/* If not controlling process, send message that calculation is done */
MPI_Barrier(MPI_COMM_WORLD);
if (!task_id) {
time = timer_end();
printf("Took %g seconds\n", time);
printf("Normalized: %g seconds\n", time / (1. * i_max * t_max));
/* Write own data */
file_write_double_array("result.txt", ret+1, i_max-1, "w");
/* Receive the data from all the other users */
for (int i = 1; i < num_tasks; i++) {
MPI_Recv(ret, i_max, MPI_DOUBLE, i, MSG_WRITE, MPI_COMM_WORLD, NULL);
file_write_double_array("result.txt", ret+1, i_max-1, "a+");
}
}
/* Non-Controller */
if (task_id) {
/* Send to controller */
MPI_Send(ret, i_max, MPI_DOUBLE, 0, MSG_WRITE, MPI_COMM_WORLD);
}
MPI_Finalize();
fprintf(stderr, "Finalized\n");
free(old);
free(current);
free(next);
fprintf(stderr, "Done\n");
return EXIT_SUCCESS;
}
int main(int argc, char *argv[])
{
double *old, *current, *next;
int t_max, i_max, num_threads;
double time;
/* Parse commandline args: i_max t_max num_threads */
if (argc < 4) {
printf("Usage: %s i_max t_max num_threads [initial_data]\n", argv[0]);
printf(" - i_max: number of discrete amplitude points, should be >2\n");
printf(" - t_max: number of discrete timesteps, should be >=1\n");
printf(" - num_threads: number of threads to use for simulation, "
"should be >=1\n");
printf(" - initial_data: select what data should be used for the first "
"two generation.\n");
printf(" Available options are:\n");
printf(" * sin: one period of the sinus function at the start.\n");
printf(" * sinfull: entire data is filled with the sinus.\n");
printf(" * gauss: a single gauss-function at the start.\n");
printf(" * file <2 filenames>: allows you to specify a file with on "
"each line a float for both generations.\n");
return EXIT_FAILURE;
}
i_max = atoi(argv[1]);
t_max = atoi(argv[2]);
num_threads = atoi(argv[3]);
if (i_max < 3) {
printf("argument error: i_max should be >2.\n");
return EXIT_FAILURE;
}
if (t_max < 1) {
printf("argument error: t_max should be >=1.\n");
return EXIT_FAILURE;
}
if (num_threads < 1) {
printf("argument error: num_threads should be >=1.\n");
return EXIT_FAILURE;
}
/* Allocate and initialize buffers. */
old = malloc(i_max * sizeof(double));
current = malloc(i_max * sizeof(double));
next = malloc(i_max * sizeof(double));
if (old == NULL || current == NULL || next == NULL) {
fprintf(stderr, "Could not allocate enough memory, aborting.\n");
return EXIT_FAILURE;
}
memset(old, 0, i_max * sizeof(double));
memset(current, 0, i_max * sizeof(double));
memset(next, 0, i_max * sizeof(double));
/* How should we will our first two generations? */
if (argc > 4) {
if (strcmp(argv[4], "sin") == 0) {
fill(old, 1, i_max/4, 0, 2*3.14, sin);
fill(current, 2, i_max/4, 0, 2*3.14, sin);
} else if (strcmp(argv[4], "sinfull") == 0) {
fill(old, 1, i_max-2, 0, 10*3.14, sin);
fill(current, 2, i_max-3, 0, 10*3.14, sin);
} else if (strcmp(argv[4], "gauss") == 0) {
fill(old, 1, i_max/4, -3, 3, gauss);
fill(current, 2, i_max/4, -3, 3, gauss);
} else if (strcmp(argv[4], "file") == 0) {
if (argc < 7) {
printf("No files specified!\n");
return EXIT_FAILURE;
}
file_read_double_array(argv[5], old, i_max);
file_read_double_array(argv[6], current, i_max);
} else {
printf("Unknown initial mode: %s.\n", argv[4]);
return EXIT_FAILURE;
}
} else {
/* Default to sinus. */
fill(old, 1, i_max/4, 0, 2*3.14, sin);
fill(current, 2, i_max/4, 0, 2*3.14, sin);
}
timer_start();
/* Call the actual simulation that should be implemented in simulate.c. */
simulate(i_max, t_max, num_threads, old, current, next);
time = timer_end();
printf("Took %g seconds\n", time);
printf("Normalized: %g seconds\n", time / (i_max * t_max));
file_write_double_array("result.txt", current, i_max);
free(old);
free(current);
free(next);
return EXIT_SUCCESS;
}
