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

/*

 *

 *  (C) 2008 by Argonne National Laboratory.

 *      See COPYRIGHT in top-level directory.

 */

#include "mpi.h"

#include "mpitestconf.h"

#ifdef HAVE_IOSTREAM

// Not all C++ compilers have iostream instead of iostream.h

#include <iostream>

#ifdef HAVE_NAMESPACE_STD

// Those that do often need the std namespace; otherwise, a bare "cout"

// is likely to fail to compile

using namespace std;

#endif

#else

#include <iostream.h>

#endif

#include <complex>

#include "mpitestcxx.h"

int main( int argc, char **argv )

{

    MPI::Intracomm comm = MPI::COMM_WORLD;

    int errs = 0;

    int size, rank;

    long sum;

    complex<float> c[2], c_out[2];

    complex<double> cd[2], cd_out[2];

#ifdef HAVE_LONG_DOUBLE

    complex<long double> cld[2], cld_out[2];

    MTEST_VG_MEM_INIT(cld, 2 * sizeof(complex<long double>));

#endif

    MTest_Init( );

    size = comm.Get_size();

    rank = comm.Get_rank();

    c[0]   = ((float)rank)*complex<float>(1,2);

    c[1]   = ((float)rank)*complex<float>(-1,4);

    cd[0]  = ((double)rank)*complex<double>(1,2);

    cd[1]  = ((double)rank)*complex<double>(-1,4);

#ifdef HAVE_LONG_DOUBLE

    cld[0] = ((long double)rank)*complex<long double>(1,2);

    cld[1] = ((long double)rank)*complex<long double>(-1,4);

#endif

    // Sums are easy - real and imaginary parts separate

    // result should be sum(0:size-1) *{ complex(1,2), complex(-1,4) }

    // The sum is (size*(size-1))/2

    comm.Allreduce( c, c_out, 2, MPI::COMPLEX, MPI::SUM );

    sum = (size * (size-1) ) / 2;

    if (c_out[0] != ((float)sum) * complex<float>(1,2)) {

	errs++;

	cout << "c_out[0] was " << c_out[0] << " expected " << 

	    ((float)sum) * complex<float>(1,2);

    }

    if (c_out[1] != ((float)sum) * complex<float>(-1,4)) {

	errs++;

	cout << "c_out[1] was " << c_out[1] << " expected " << 

	    ((float)sum) * complex<float>(-1,4);

    }

    comm.Allreduce( cd, cd_out, 2, MPI::DOUBLE_COMPLEX, MPI::SUM );

    if (cd_out[0] != ((double)sum) * complex<double>(1,2)) {

	errs++;

	cout << "cd_out[0] was " << cd_out[0] << " expected " << 

	    ((double)sum) * complex<double>(1,2);

    }

    if (cd_out[1] != ((double)sum) * complex<double>(-1,4)) {

	errs++;

	cout << "cd_out[1] was " << cd_out[1] << " expected " << 

	    ((double)sum) * complex<double>(-1,4);

    }

#ifdef HAVE_LONG_DOUBLE

    comm.Allreduce( cld, cld_out, 2, MPI::LONG_DOUBLE_COMPLEX, MPI::SUM );

    if (cld_out[0] != ((long double)sum) * complex<long double>(1,2)) {

	errs++;

	cout << "cld_out[0] was " << cld_out[0] << " expected " << 

	    ((long double)sum) * complex<long double>(1,2);

    }

    if (cld_out[1] != ((long double)sum) * complex<long double>(-1,4)) {

	errs++;

	cout << "cld_out[1] was " << cld_out[1] << " expected " << 

	    ((long double)sum) * complex<long double>(-1,4);

    }

#endif

    MTest_Finalize( errs );

    MPI::Finalize();

    return 0;

}