/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */

/*

 *

 *  (C) 2011 by Argonne National Laboratory.

 *      See COPYRIGHT in top-level directory.

 */

/* A simple test of the proposed MPI_T_ interface that queries all of

 * the control variables exposed by the MPI implememtation and prints

 * them to stdout.

 *

 * Author: Dave Goodell 

 */

#include "mpi.h"

#include <stdlib.h>

#include <stdio.h>

#include <assert.h>

#include <math.h>

#include "mpitestconf.h"

#if !defined(USE_STRICT_MPI) && defined(MPICH)

int main(int argc, char **argv)

{

    int i;

    int num;

    int rank, size;

/*#define STR_SZ (15)*/

#define STR_SZ (50)

    int name_len = STR_SZ;

    char name[STR_SZ] = "";

    int desc_len = STR_SZ;

    char desc[STR_SZ] = "";

    int verb;

    MPI_Datatype dtype;

    int count;

    int bind;

    int scope;

    int provided;

    int initialize_mpi = 0;

    MPI_T_cvar_handle handle;

    MPI_T_enum enumtype;

    provided = 0xdeadbeef;

    MPI_T_init_thread(MPI_THREAD_SINGLE, &provided);

    assert(provided != 0xdeadbeef);

    if (initialize_mpi) {

        MPI_Init(&argc, &argv);

        MPI_Comm_rank(MPI_COMM_WORLD, &rank;;

        MPI_Comm_size(MPI_COMM_WORLD, &size);

    }

    num = 0xdeadbeef;

    MPI_T_cvar_get_num(&num);

    printf("get_num=%d\n", num);

    assert(num != 0xdeadbeef);

    for (i = 0; i < num; ++i) {

        name_len = desc_len = STR_SZ;

        MPI_T_cvar_get_info(i, name, &name_len, &verb, &dtype, &enumtype, desc, &desc_len, &bind,

                            &scope);

        printf("index=%d\n", i);

        printf("--> name='%s' name_len=%d desc='%s' desc_len=%d\n", name, name_len, desc, desc_len);

        printf("--> verb=%d dtype=%#x bind=%d scope=%d\n", verb, dtype, bind, scope);

        MPI_T_cvar_handle_alloc(i, NULL, &handle, &count);

        printf("--> handle allocated: handle=%p count=%d\n", handle, count);

        if (dtype == MPI_INT) {

            int val = 0xdeadbeef;

            MPI_T_cvar_read(handle, &val;;

            printf("--> val=%d\n", val);

            ++val;

            MPI_T_cvar_write(handle, &val;;

            val = 0xdeadbeef;

            MPI_T_cvar_read(handle, &val;;

            printf("--> incremented val=%d\n", val);

        }

        else if (dtype == MPI_DOUBLE) {

            double val = NAN;

            MPI_T_cvar_read(handle, &val;;

            printf("--> val=%f\n", val);

            val *= 2.0;

            MPI_T_cvar_write(handle, &val;;

            val = NAN;

            MPI_T_cvar_read(handle, &val;;

            printf("--> doubled val=%f\n", val);

        }

        else if (dtype == MPI_CHAR) {

            char *str = malloc(count + 1);

            MPI_T_cvar_read(handle, str);

            printf("--> str='%s'\n", str);

            /* just write the string back unmodified for now */

            MPI_T_cvar_write(handle, str);

            MPI_T_cvar_read(handle, str);

            printf("--> written-then-read str='%s'\n", str);

        }

        MPI_T_cvar_handle_free(&handle);

        printf("\n");

    }

    if (initialize_mpi) {

        MPI_Finalize();

    }

    MPI_T_finalize();

    return 0;

}

#else

/* Simple null program to allow building this file with non-MPICH

   implementations */

int main(int argc, char **argv)

{

    MPI_Init(&argc, &argv);

    printf(" No Errors\n");

    MPI_Finalize();

    return 0;

}

#endif