#include <iostream>

#include "mpi.h"

using namespace std;

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

        int numProcs, myrank, hostNameLen, ROOT=0;
        char processor_name[MPI_MAX_PROCESSOR_NAME];

        void* iAttrPtr=NULL;
        bool returnFlag;

        MPI::Status status;
        MPI::Init(argc,argv);

        numProcs = MPI::COMM_WORLD.Get_size();
        myrank = MPI::COMM_WORLD.Get_rank();

        MPI::Get_processor_name(processor_name, hostNameLen);

        MPI::COMM_WORLD.Set_errhandler(MPI::ERRORS_THROW_EXCEPTIONS);


        long unsigned int N=5000, nIter=5000;
        int accum, nSum;

        for (long unsigned int iIter=0; iIter<nIter; iIter++) {
                returnFlag = false;
                nSum=0;
                for (long unsigned int i=myrank; i<N; i+=numProcs) {
                        nSum += i*i*i;
                }

                if (myrank != ROOT) {
                        MPI::COMM_WORLD.Isend( &nSum, 1, MPI::INT, ROOT, 1 );
                }
                else {
                        for (int j=1; j<numProcs; j++) {
                                MPI::COMM_WORLD.Recv( &accum, 1, MPI::INT, j, 1, status );
                                nSum += accum;
                        }

                        // try {
                                returnFlag = MPI::COMM_WORLD.Get_attr(MPICH_ATTR_FAILED_PROCESSES, iAttrPtr);
                        // }
                        // catch (MPI::Exception failure) { }

                        int* intPtr;
                        intPtr = (int*) iAttrPtr;

                        if ( returnFlag && *intPtr != MPI::PROC_NULL ) {
                                cout << "[ ... ] ...... ";
                                while (*intPtr != MPI::PROC_NULL) {
                                        cout << ", " << *intPtr;
                                        intPtr++;
                                }
                                cout << " ......\n" << flush;
                        }
                }

                if (myrank == ROOT) {
                        if (iIter%1000 == 0) {
                                cout << "\t iter = " << iIter << "\t sum = " << nSum << endl;
                        }
                }

        }


        MPI::Finalize();

        return 0;
}