/*
 * numerical integration example, as discussed in textbook:  
 *
 * compute pi by approximating the area under the curve f(x) = 4 / (1 + x*x)
 * between 0 and 1.
 *
 * parallel version using Java and java.util.concurrent.
 *
 * number of threads is taken from environment variable NUM_THREADS.
 */
package csci3366sample.numint;
import csci3366sample.ThreadUtility;

import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.ArrayList;

public class NumIntPar2 {

    /* variables/constants to be used by all threads */

    private static final int NUM_STEPS = 400000000;
    private static final double step = 1.0/(double) NUM_STEPS; 
    private static int numThreads;

    /* main method */

    public static void main(String[] args) {

        /* get number of threads from environment variable */
        numThreads = 0;
        try {
            numThreads = ThreadUtility.getNumThreads();
        }
        catch (ThreadUtility.Exception e) {
            System.err.println(e.getMessage());
            System.exit(1);
        }

        double sum = 0.0;
 
        /* start timing */
        long startTime = System.currentTimeMillis();

        /* create executor for threads */
        ExecutorService executor = Executors.newFixedThreadPool(numThreads);

        /* create tasks (that return partial sums) and send to executor, 
           creating Future objects so we can wait for tasks to finish (and 
           get the returned values */
        ArrayList<Future<Double>> partialSums = new ArrayList<Future<Double>>();
        for (int i = 0; i < numThreads; ++i) {
            partialSums.add(executor.submit(new CodeForThread(i)));
        }

        /* shut down executor and wait for threads to finish */
        executor.shutdown();
        for (Future<Double> p : partialSums) {
            try {
                sum += p.get();
            }
            catch (ExecutionException e) {
                System.err.println("should not happen");
            }
            catch (InterruptedException e) {
                System.err.println("should not happen");
            }
        }

        /* finish computation */
        double pi = sum * step;

        /* end timing and print result */
        long endTime = System.currentTimeMillis();
        System.out.println("parallel program (v2) results with " + numThreads +
                " threads and " + NUM_STEPS + " steps:");
        System.out.println("computed pi = " + pi);
        System.out.println("difference between estimated pi and Math.PI = " +
                Math.abs(pi - Math.PI));
        System.out.println("time to compute = " + 
                (double) (endTime - startTime) / 1000);
    }

    /* static inner class to contain code to run in each thread */

    private static class CodeForThread implements Callable<Double> {

        private int myID;

        public CodeForThread(int myID) {
            this.myID = myID;
        }

        public Double call() {
            double partsum = 0.0;
            for (int i=myID; i < NUM_STEPS; i += numThreads) {
                double x = (i+0.5)*step;
                partsum += 4.0/(1.0+x*x);
            }
            return new Double(partsum);
        }
    }
}