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

/*

 *

 *  (C) 2003 by Argonne National Laboratory.

 *      See COPYRIGHT in top-level directory.

 */

#include "mpi.h"

#include <stdio.h>

#include "mpitest.h"

/*

static char MTEST_Descrip[] = "Test of broadcast with various roots and datatypes";

*/

int main(int argc, char *argv[])

{

    int errs = 0, err;

    int rank, size, root;

    int minsize = 2, count;

    MPI_Comm comm;

    MTestDatatype sendtype, recvtype;

    MTest_Init(&argc, &argv);

    /* The following illustrates the use of the routines to

     * run through a selection of communicators and datatypes.

     * Use subsets of these for tests that do not involve combinations

     * of communicators, datatypes, and counts of datatypes */

    while (MTestGetIntracommGeneral(&comm, minsize, 1)) {

        if (comm == MPI_COMM_NULL)

            continue;

#if defined BCAST_COMM_WORLD_ONLY

        if (comm != MPI_COMM_WORLD) {

            MTestFreeComm(&comm);

            continue;

        }

#endif /* BCAST_COMM_WORLD_ONLY */

        /* Determine the sender and receiver */

        MPI_Comm_rank(comm, &rank;;

        MPI_Comm_size(comm, &size);

        /* To improve reporting of problems about operations, we

         * change the error handler to errors return */

        MPI_Errhandler_set(comm, MPI_ERRORS_RETURN);

        MTEST_DATATYPE_FOR_EACH_COUNT(count) {

            /* To shorten test time, only run the default version of datatype tests

             * for comm world and run the minimum version for other communicators. */

#if defined BCAST_MIN_DATATYPES_ONLY

            MTestInitMinDatatypes();

#endif /* BCAST_MIN_DATATYPES_ONLY */

            while (MTestGetDatatypes(&sendtype, &recvtype, count)) {

                for (root = 0; root < size; root++) {

                    if (rank == root) {

                        sendtype.InitBuf(&sendtype);

                        err = MPI_Bcast(sendtype.buf, sendtype.count,

                                        sendtype.datatype, root, comm);

                        if (err) {

                            errs++;

                            MTestPrintError(err);

                        }

                    }

                    else {

                        recvtype.InitBuf(&recvtype);

                        err = MPI_Bcast(recvtype.buf, recvtype.count,

                                        recvtype.datatype, root, comm);

                        if (err) {

                            errs++;

                            fprintf(stderr, "Error with communicator %s and datatype %s\n",

                                    MTestGetIntracommName(), MTestGetDatatypeName(&recvtype));

                            MTestPrintError(err);

                        }

                        err = MTestCheckRecv(0, &recvtype);

                        if (err) {

                            errs += errs;

                        }

                    }

                }

                MTestFreeDatatype(&recvtype);

                MTestFreeDatatype(&sendtype);

            }

        }

        MTestFreeComm(&comm);

    }

    MTest_Finalize(errs);

    MPI_Finalize();

    return 0;

}