/* -*- 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 "mpitestconf.h"

#include <stdio.h>

#include "mpitest.h"

/*

static char MTEST_Descrip[] = "Test MPI_BAND operations on optional datatypes dupported by MPICH";

*/

/*

 * This test looks at the handling of logical and for types that are not

 * integers or are not required integers (e.g., long long).  MPICH allows

 * these as well.  A strict MPI test should not include this test.

 */

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

{

    int errs = 0;

    int rc;

    int rank, size;

    MPI_Comm comm;

    char cinbuf[3], coutbuf[3];

    signed char scinbuf[3], scoutbuf[3];

    unsigned char ucinbuf[3], ucoutbuf[3];

    short sinbuf[3], soutbuf[3];

    unsigned short usinbuf[3], usoutbuf[3];

    long linbuf[3], loutbuf[3];

    unsigned long ulinbuf[3], uloutbuf[3];

    unsigned uinbuf[3], uoutbuf[3];

    MTest_Init(&argc, &argv);

    comm = MPI_COMM_WORLD;

    /* Set errors return so that we can provide better information

     * should a routine reject one of the operand/datatype pairs */

    MPI_Errhandler_set(comm, MPI_ERRORS_RETURN);

    MPI_Comm_rank(comm, &rank;;

    MPI_Comm_size(comm, &size);

#ifndef USE_STRICT_MPI

    /* char */

    MTestPrintfMsg(10, "Reduce of MPI_CHAR\n");

    cinbuf[0] = 0xff;

    cinbuf[1] = 0;

    cinbuf[2] = (rank > 0) ? 0xff : 0xf0;

    coutbuf[0] = 0;

    coutbuf[1] = 1;

    coutbuf[2] = 1;

    rc = MPI_Reduce(cinbuf, coutbuf, 3, MPI_CHAR, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_CHAR", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (coutbuf[0] != (char) 0xff) {

                errs++;

                fprintf(stderr, "char BAND(1) test failed\n");

            }

            if (coutbuf[1]) {

                errs++;

                fprintf(stderr, "char BAND(0) test failed\n");

            }

            if (coutbuf[2] != (char) 0xf0 && size > 1) {

                errs++;

                fprintf(stderr, "char BAND(>) test failed\n");

            }

        }

    }

#endif /* USE_STRICT_MPI */

    /* signed char */

    MTestPrintfMsg(10, "Reduce of MPI_SIGNED_CHAR\n");

    scinbuf[0] = 0xff;

    scinbuf[1] = 0;

    scinbuf[2] = (rank > 0) ? 0xff : 0xf0;

    scoutbuf[0] = 0;

    scoutbuf[1] = 1;

    scoutbuf[2] = 1;

    rc = MPI_Reduce(scinbuf, scoutbuf, 3, MPI_SIGNED_CHAR, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_SIGNED_CHAR", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (scoutbuf[0] != (signed char) 0xff) {

                errs++;

                fprintf(stderr, "signed char BAND(1) test failed\n");

            }

            if (scoutbuf[1]) {

                errs++;

                fprintf(stderr, "signed char BAND(0) test failed\n");

            }

            if (scoutbuf[2] != (signed char) 0xf0 && size > 1) {

                errs++;

                fprintf(stderr, "signed char BAND(>) test failed\n");

            }

        }

    }

    /* unsigned char */

    MTestPrintfMsg(10, "Reduce of MPI_UNSIGNED_CHAR\n");

    ucinbuf[0] = 0xff;

    ucinbuf[1] = 0;

    ucinbuf[2] = (rank > 0) ? 0xff : 0xf0;

    ucoutbuf[0] = 0;

    ucoutbuf[1] = 1;

    ucoutbuf[2] = 1;

    rc = MPI_Reduce(ucinbuf, ucoutbuf, 3, MPI_UNSIGNED_CHAR, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_UNSIGNED_CHAR", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (ucoutbuf[0] != 0xff) {

                errs++;

                fprintf(stderr, "unsigned char BAND(1) test failed\n");

            }

            if (ucoutbuf[1]) {

                errs++;

                fprintf(stderr, "unsigned char BAND(0) test failed\n");

            }

            if (ucoutbuf[2] != 0xf0 && size > 1) {

                errs++;

                fprintf(stderr, "unsigned char BAND(>) test failed\n");

            }

        }

    }

    /* bytes */

    MTestPrintfMsg(10, "Reduce of MPI_BYTE\n");

    cinbuf[0] = 0xff;

    cinbuf[1] = 0;

    cinbuf[2] = (rank > 0) ? 0xff : 0xf0;

    coutbuf[0] = 0;

    coutbuf[1] = 1;

    coutbuf[2] = 1;

    rc = MPI_Reduce(cinbuf, coutbuf, 3, MPI_BYTE, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_BYTE", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (coutbuf[0] != (char) 0xff) {

                errs++;

                fprintf(stderr, "byte BAND(1) test failed\n");

            }

            if (coutbuf[1]) {

                errs++;

                fprintf(stderr, "byte BAND(0) test failed\n");

            }

            if (coutbuf[2] != (char) 0xf0 && size > 1) {

                errs++;

                fprintf(stderr, "byte BAND(>) test failed\n");

            }

        }

    }

    /* short */

    MTestPrintfMsg(10, "Reduce of MPI_SHORT\n");

    sinbuf[0] = 0xffff;

    sinbuf[1] = 0;

    sinbuf[2] = (rank > 0) ? 0xffff : 0xf0f0;

    soutbuf[0] = 0;

    soutbuf[1] = 1;

    soutbuf[2] = 1;

    rc = MPI_Reduce(sinbuf, soutbuf, 3, MPI_SHORT, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_SHORT", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (soutbuf[0] != (short) 0xffff) {

                errs++;

                fprintf(stderr, "short BAND(1) test failed\n");

            }

            if (soutbuf[1]) {

                errs++;

                fprintf(stderr, "short BAND(0) test failed\n");

            }

            if (soutbuf[2] != (short) 0xf0f0 && size > 1) {

                errs++;

                fprintf(stderr, "short BAND(>) test failed\n");

            }

        }

    }

    MTestPrintfMsg(10, "Reduce of MPI_UNSIGNED_SHORT\n");

    /* unsigned short */

    usinbuf[0] = 0xffff;

    usinbuf[1] = 0;

    usinbuf[2] = (rank > 0) ? 0xffff : 0xf0f0;

    usoutbuf[0] = 0;

    usoutbuf[1] = 1;

    usoutbuf[2] = 1;

    rc = MPI_Reduce(usinbuf, usoutbuf, 3, MPI_UNSIGNED_SHORT, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_UNSIGNED_SHORT", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (usoutbuf[0] != 0xffff) {

                errs++;

                fprintf(stderr, "short BAND(1) test failed\n");

            }

            if (usoutbuf[1]) {

                errs++;

                fprintf(stderr, "short BAND(0) test failed\n");

            }

            if (usoutbuf[2] != 0xf0f0 && size > 1) {

                errs++;

                fprintf(stderr, "short BAND(>) test failed\n");

            }

        }

    }

    /* unsigned */

    MTestPrintfMsg(10, "Reduce of MPI_UNSIGNED\n");

    uinbuf[0] = 0xffffffff;

    uinbuf[1] = 0;

    uinbuf[2] = (rank > 0) ? 0xffffffff : 0xf0f0f0f0;

    uoutbuf[0] = 0;

    uoutbuf[1] = 1;

    uoutbuf[2] = 1;

    rc = MPI_Reduce(uinbuf, uoutbuf, 3, MPI_UNSIGNED, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_UNSIGNED", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (uoutbuf[0] != 0xffffffff) {

                errs++;

                fprintf(stderr, "unsigned BAND(1) test failed\n");

            }

            if (uoutbuf[1]) {

                errs++;

                fprintf(stderr, "unsigned BAND(0) test failed\n");

            }

            if (uoutbuf[2] != 0xf0f0f0f0 && size > 1) {

                errs++;

                fprintf(stderr, "unsigned BAND(>) test failed\n");

            }

        }

    }

    /* long */

    MTestPrintfMsg(10, "Reduce of MPI_LONG\n");

    linbuf[0] = 0xffffffff;

    linbuf[1] = 0;

    linbuf[2] = (rank > 0) ? 0xffffffff : 0xf0f0f0f0;

    loutbuf[0] = 0;

    loutbuf[1] = 1;

    loutbuf[2] = 1;

    rc = MPI_Reduce(linbuf, loutbuf, 3, MPI_LONG, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_LONG", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (loutbuf[0] != 0xffffffff) {

                errs++;

                fprintf(stderr, "long BAND(1) test failed\n");

            }

            if (loutbuf[1]) {

                errs++;

                fprintf(stderr, "long BAND(0) test failed\n");

            }

            if (loutbuf[2] != 0xf0f0f0f0 && size > 1) {

                errs++;

                fprintf(stderr, "long BAND(>) test failed\n");

            }

        }

    }

    MTestPrintfMsg(10, "Reduce of MPI_UNSIGNED_LONG\n");

    /* unsigned long */

    ulinbuf[0] = 0xffffffff;

    ulinbuf[1] = 0;

    ulinbuf[2] = (rank > 0) ? 0xffffffff : 0xf0f0f0f0;

    uloutbuf[0] = 0;

    uloutbuf[1] = 1;

    uloutbuf[2] = 1;

    rc = MPI_Reduce(ulinbuf, uloutbuf, 3, MPI_UNSIGNED_LONG, MPI_BAND, 0, comm);

    if (rc) {

        MTestPrintErrorMsg("MPI_BAND and MPI_UNSIGNED_LONG", rc);

        errs++;

    } else {

        if (rank == 0) {

            if (uloutbuf[0] != 0xffffffff) {

                errs++;

                fprintf(stderr, "unsigned long BAND(1) test failed\n");

            }

            if (uloutbuf[1]) {

                errs++;

                fprintf(stderr, "unsigned long BAND(0) test failed\n");

            }

            if (uloutbuf[2] != 0xf0f0f0f0 && size > 1) {

                errs++;

                fprintf(stderr, "unsigned long BAND(>) test failed\n");

            }

        }

    }

#ifdef HAVE_LONG_LONG

    {

        long long llinbuf[3], lloutbuf[3];

        /* long long */

        llinbuf[0] = 0xffffffff;

        llinbuf[1] = 0;

        llinbuf[2] = (rank > 0) ? 0xffffffff : 0xf0f0f0f0;

        lloutbuf[0] = 0;

        lloutbuf[1] = 1;

        lloutbuf[2] = 1;

        if (MPI_LONG_LONG != MPI_DATATYPE_NULL) {

            MTestPrintfMsg(10, "Reduce of MPI_LONG_LONG\n");

            rc = MPI_Reduce(llinbuf, lloutbuf, 3, MPI_LONG_LONG, MPI_BAND, 0, comm);

            if (rc) {

                MTestPrintErrorMsg("MPI_BAND and MPI_LONG_LONG", rc);

                errs++;

            } else {

                if (rank == 0) {

                    if (lloutbuf[0] != 0xffffffff) {

                        errs++;

                        fprintf(stderr, "long long BAND(1) test failed\n");

                    }

                    if (lloutbuf[1]) {

                        errs++;

                        fprintf(stderr, "long long BAND(0) test failed\n");

                    }

                    if (lloutbuf[2] != 0xf0f0f0f0 && size > 1) {

                        errs++;

                        fprintf(stderr, "long long BAND(>) test failed\n");

                    }

                }

            }

        }

    }

#endif

    MPI_Errhandler_set(comm, MPI_ERRORS_ARE_FATAL);

    MTest_Finalize(errs);

    return MTestReturnValue(errs);

}