#include <math.h>
#include <stdlib.h>
#include <H5hut.h>

#include "testframe.h"
#include "params.h"

/* global */
h5_size_t grid[3];
h5_size_t layout[6];
h5_size_t fields_dims[3];

/* from write.c */
void h5b_test_write1(void);
void h5b_test_write2(void);

/* from read.c */
void h5b_test_read1(void);
void h5b_test_read2(void);

#ifdef H5_HAVE_PARALLEL
static int
_nth_root_int_divisor (const int m, const int n)
{
	int i, root;
	double p;

	p = 1.0 / (double)n;
	root = (int) ceil ( pow ( (double)m, p ) );
	for (i=root; i<=m; i++)
	{
		if (m % i == 0) return i;
	}

	return i;
}
#endif

int main(int argc, char **argv)
{
	extern h5_size_t layout[6];
#ifdef H5_HAVE_PARALLEL
	MPI_Init(&argc, &argv);

	int procs, rank;
	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
	MPI_Comm_size(MPI_COMM_WORLD, &procs);

	if (procs > MAX_MPI_TASKS) {
		fprintf(stderr,
		        "ERROR: please use <= %d MPI tasks for the test.\n",
		        MAX_MPI_TASKS);
		exit(EXIT_FAILURE);
	}

	/* make up a 3D layout */
	grid[0] = _nth_root_int_divisor (procs, 3);
	grid[1] = _nth_root_int_divisor (procs / grid[0], 2);
	grid[2] = procs / grid[0] / grid[1];

	h5_size_t i,j,k;
	k = rank % grid[2];
	j = (rank / grid[2]) % grid[1];
	i = rank / (grid[2] * grid[1]);

	layout[0] = i*NBLOCKX;
	layout[1] = (i+1)*NBLOCKX - 1;
	layout[2] = j*NBLOCKY;
	layout[3] = (j+1)*NBLOCKY - 1;
	layout[4] = k*NBLOCKZ;
	layout[5] = (k+1)*NBLOCKZ - 1;
#else // H5_HAVE_PARALLEL
	grid[0] = 1;
	grid[1] = 1;
	grid[2] = 1;

	layout[0] = 0;
	layout[1] = NBLOCKX - 1;
	layout[2] = 0;
	layout[3] = NBLOCKY - 1;
	layout[4] = 0;
	layout[5] = NBLOCKZ - 1;
#endif

	/* Initialize testing framework */
	TestInit(argv[0], NULL, NULL);

	/* Tests are generally arranged from least to most complexity... */
	AddTest("write1", h5b_test_write1, NULL, "Write 64-bit data", NULL);
	AddTest("read1", h5b_test_read1, NULL, "Read 64-bit data", NULL);
#ifdef H5_HAVE_PARALLEL
	AddTest("write2", h5b_test_write2, NULL, "Write 32-bit data", NULL);
	AddTest("read2", h5b_test_read2, NULL, "Read 32-bit data", NULL);
#endif
	/* Display testing information */
	TestInfo(argv[0]);

	/* Parse command line arguments */
	TestParseCmdLine(argc, argv);

	H5SetVerbosityLevel(GetTestVerbosity());

	/* Perform requested testing */
	PerformTests();

	/* Display test summary, if requested */
	if (GetTestSummary())
		TestSummary();

	/* Clean up test files, if allowed */
	//if (GetTestCleanup() && !getenv("HDF5_NOCLEANUP"))
	//    TestCleanup();

#ifdef H5_HAVE_PARALLEL
	TestPrintf ("reached end\n");
	fflush(stdout);
	MPI_Finalize();
#endif
	return GetTestNumErrs();
}