/*
 * Copyright (C) by Argonne National Laboratory
 *     See COPYRIGHT in top-level directory
 */

/* This is a simple test program to see how well balanced the output from
   MPI_Dims_create is.  This test can be run with a single process
   and does not need mpiexec.

   There is no standard result.  Instead, the output should be examined to
   see if the results are acceptable.  For example, make sure that Dims_create
   of 2*2*2 in three dimensions returns 2,2,2 (and not 1,1,8).
 */

#include "mpi.h"
#include <stdio.h>

static int verbose = 1;

int main(int argc, char *argv[])
{
    int i, j, dims[3], size;

    MPI_Init(&argc, &argv);

    for (i = 2; i < 12; i++) {
        for (j = 0; j < 2; j++)
            dims[j] = 0;
        MPI_Dims_create(i * i, 2, dims);
        if (verbose)
            printf("Dims_create(%d,2) = (%d,%d)\n", i * i, dims[0], dims[1]);
        for (j = 0; j < 3; j++)
            dims[j] = 0;
        MPI_Dims_create(i * i * i, 3, dims);
        if (verbose)
            printf("Dims_create(%d,3) = (%d,%d,%d)\n", i * i * i, dims[0], dims[1], dims[2]);
    }
    size = 2 * 5 * 7 * 11;
    for (j = 0; j < 3; j++)
        dims[j] = 0;
    MPI_Dims_create(size, 2, dims);
    if (verbose)
        printf("Dims_create(%d,2) = (%d,%d)\n", size, dims[0], dims[1]);
    for (j = 0; j < 3; j++)
        dims[j] = 0;
    MPI_Dims_create(size, 3, dims);
    if (verbose)
        printf("Dims_create(%d,3) = (%d,%d,%d)\n", size, dims[0], dims[1], dims[2]);
    size = 5 * 5 * 2 * 7 * 7 * 7;
    for (j = 0; j < 3; j++)
        dims[j] = 0;
    MPI_Dims_create(size, 2, dims);
    if (verbose)
        printf("Dims_create(%d,2) = (%d,%d)\n", size, dims[0], dims[1]);
    for (j = 0; j < 3; j++)
        dims[j] = 0;
    MPI_Dims_create(size, 3, dims);
    if (verbose)
        printf("Dims_create(%d,3) = (%d,%d,%d)\n", size, dims[0], dims[1], dims[2]);

    size = 12;
    for (j = 0; j < 3; j++)
        dims[j] = 0;
    MPI_Dims_create(size, 2, dims);
    if (verbose)
        printf("Dims_create(%d,2) = (%d,%d)\n", size, dims[0], dims[1]);

    MPI_Finalize();

    return 0;
}
