Blame examples/cxx/cxxpi.cxx

Packit 0848f5
/* -*- Mode: C++; c-basic-offset:4 ; -*- */
Packit 0848f5
/*  
Packit 0848f5
 *  (C) 2004 by Argonne National Laboratory.
Packit 0848f5
 *      See COPYRIGHT in top-level directory.
Packit 0848f5
 */
Packit 0848f5
Packit 0848f5
#include "mpi.h"
Packit 0848f5
/* Support both new (e.g., has iostream) and old (requires iostream.h) 
Packit 0848f5
   C++ compilers */
Packit 0848f5
#ifdef HAVE_CXX_IOSTREAM
Packit 0848f5
#include <iostream>
Packit 0848f5
#ifdef HAVE_NAMESPACE_STD
Packit 0848f5
// Those that do often need the std namespace; otherwise, a bare "cout"
Packit 0848f5
// is likely to fail to compile
Packit 0848f5
using namespace std;
Packit 0848f5
#endif
Packit 0848f5
#else
Packit 0848f5
#include <iostream.h>
Packit 0848f5
#endif
Packit 0848f5
#ifdef HAVE_CXX_MATH
Packit 0848f5
#include <math>
Packit 0848f5
#else
Packit 0848f5
#include <math.h>
Packit 0848f5
#endif
Packit 0848f5
Packit 0848f5
double f(double);
Packit 0848f5
Packit 0848f5
double f(double a)
Packit 0848f5
{
Packit 0848f5
    return (4.0 / (1.0 + a*a));
Packit 0848f5
}
Packit 0848f5
Packit 0848f5
int main(int argc,char **argv)
Packit 0848f5
{
Packit 0848f5
    int n, myid, numprocs, i;
Packit 0848f5
    double PI25DT = 3.141592653589793238462643;
Packit 0848f5
    double mypi, pi, h, sum, x;
Packit 0848f5
    double startwtime = 0.0, endwtime;
Packit 0848f5
    int  namelen;
Packit 0848f5
    char processor_name[MPI_MAX_PROCESSOR_NAME];
Packit 0848f5
Packit 0848f5
    MPI::Init(argc,argv);
Packit 0848f5
    numprocs = MPI::COMM_WORLD.Get_size();
Packit 0848f5
    myid     = MPI::COMM_WORLD.Get_rank();
Packit 0848f5
    MPI::Get_processor_name(processor_name,namelen);
Packit 0848f5
Packit 0848f5
    cout << "Process " << myid << " of " << numprocs << " is on " <<
Packit 0848f5
	processor_name << endl;
Packit 0848f5
Packit 0848f5
    n = 10000;			/* default # of rectangles */
Packit 0848f5
    if (myid == 0)
Packit 0848f5
	startwtime = MPI::Wtime();
Packit 0848f5
Packit 0848f5
    MPI::COMM_WORLD.Bcast(&n, 1, MPI_INT, 0);
Packit 0848f5
Packit 0848f5
    h   = 1.0 / (double) n;
Packit 0848f5
    sum = 0.0;
Packit 0848f5
    /* A slightly better approach starts from large i and works back */
Packit 0848f5
    for (i = myid + 1; i <= n; i += numprocs)
Packit 0848f5
    {
Packit 0848f5
	x = h * ((double)i - 0.5);
Packit 0848f5
	sum += f(x);
Packit 0848f5
    }
Packit 0848f5
    mypi = h * sum;
Packit 0848f5
Packit 0848f5
    MPI::COMM_WORLD.Reduce(&mypi, &pi, 1, MPI_DOUBLE, MPI_SUM, 0);
Packit 0848f5
Packit 0848f5
    if (myid == 0) {
Packit 0848f5
	endwtime = MPI::Wtime();
Packit 0848f5
	cout << "pi is approximately " << pi << " Error is " <<
Packit 0848f5
	    fabs(pi - PI25DT) << endl;
Packit 0848f5
	cout << "wall clock time = " << endwtime-startwtime << endl;
Packit 0848f5
    }
Packit 0848f5
Packit 0848f5
    MPI::Finalize();
Packit 0848f5
    return 0;
Packit 0848f5
}