pacman.game.util.Stats Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pacman-main Show documentation
The main code for Ms. Pac-Man Vs Ghosts
There is a newer version: 2.0.1.0
package pacman.game.util;

/**
 * Holds iteratively built statistical observations without storing the individual observations
 * 
 * Created by pwillic on 26/02/2016.
 */
public class Stats {

    private double average;
    private double sum;
    private double sumsq;
    private double sd;
    private int n;

    private double min = Double.POSITIVE_INFINITY;
    private double max = Double.NEGATIVE_INFINITY;

    private boolean computed;

    private String description;
    private long msTaken;

    public Stats(String description) {
        this.description = description;
    }

    public Stats(long msTaken, String description, double max, double min, int n, double sumsq, double sum) {
        this.msTaken = msTaken;
        this.description = description;
        this.max = max;
        this.min = min;
        this.n = n;
        this.sumsq = sumsq;
        this.sum = sum;
    }

    public void add(double observation) {
        n++;
        sum += observation;
        sumsq += (observation * observation);
        if (observation < min) min = observation;
        if (observation > max) max = observation;
        computed = false;
    }

    private void compute() {
        if (!computed) {
            average = sum / n;
            double num = sumsq - (n * average * average);
            if (num < 0) {
                num = 0;
            }
            sd = Math.sqrt(num / (n - 1));
            computed = true;
        }
    }

    public void add(Stats other){
        n += other.n;
        sum += other.sum;
        sumsq += other.sumsq;
        if(other.min < min) min = other.min;
        if(other.max > max) max = other.max;
        computed = false;
        this.msTaken += other.msTaken;
    }

    public double getAverage() {
        if(!computed) compute();
        return average;
    }

    public int getN() {
        return n;
    }

    public double getSum() {
        return sum;
    }

    public double getMin() {
        return min;
    }

    public double getMax() {
        return max;
    }

    public double getStandardDeviation(){
        if(!computed) compute();
        return sd;
    }

    public double getStandardError(){
        if(!computed) compute();
        return sd / Math.sqrt(n);
    }

    public long getMsTaken() {
        return msTaken;
    }

    public void setMsTaken(long msTaken) {
        this.msTaken = msTaken;
    }

    public double getSumsq() {
        return sumsq;
    }

    public double getSd() {
        if(!computed) compute();
        return sd;
    }

    public boolean isComputed() {
        return computed;
    }

    public String getDescription() {
        return description;
    }

    @Override
    public String toString() {
        if(!computed) compute();
        return "Stats{" +
                "Desc=" + description +
                ", average=" + average +
                ", sum=" + sum +
                ", sumsq=" + sumsq +
                ", sd=" + sd +
                ", n=" + n +
                ", min=" + min +
                ", max=" + max +
                ", stdErr=" + getStandardError() +
                ", ms=" + msTaken +
                '}';
    }

    public static void main(String[] args) {
        Stats stats = new Stats("");
        stats.add(1.0f);
        stats.add(2.0f);
        stats.add(2.0f);
        stats.add(2.0f);
        stats.add(2.0f);
        stats.add(2.0f);

        System.out.println(stats);

    }
}