io.narayana.perf.Averager Maven / Gradle / Ivy

Go to download
/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2014 Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package io.narayana.perf;

import java.util.*;

public class Averager {
    public static void main(String[] args) {
        Double[][] tests = {
           { 2.0, 5.0, 6.0, 9.0, 12.0 },
           { 2.0, 5.0, 6.0, 9.0, 12.0, 26.0 },
           { 2.0 }
        };

        for (Double[] vals : tests) {
            List v1 = new ArrayList(Arrays.asList(vals));
            List v2 = new ArrayList(Arrays.asList(vals));

            double median = getMedian(v1, 0, v1.size() - 1);
            double q1 = getQ1(v1);
            double q3 = getQ3(v1);

            removeOutliers(v1);

            for (Double d : v2)
                System.out.printf("%f, ", d);

            System.out.printf("%nq1=%f q2=%f median=%f%n", q1, q3, median);
            System.out.printf("Outliers:%n");

            for (Double d : v2)
                if (!v1.contains(d))
                    System.out.printf("\t%f%n", d);
        }
    }

    static double getAverage(List values) {

        double tot = 0;

        for (Double d : values)
            tot += d;

        return tot / values.size();
    }

    static void removeOutliers(List values) {
        Collections.sort(values);

        double q1 = getQ1(values); // first interquartile
        double q3 = getQ3(values); // third interquartile
        double iqr = q3 - q1; // the interquartile range
        double outlier = 1.5 * iqr; // an outlier is any data point more than 1.5 interquartile ranges (IQRs) below the first quartile or above the third quartile.
        double lb = q1 - outlier;
        double ub = q3 + outlier;

        // remove outliers from the data
        Iterator i = values.iterator();

        while (i.hasNext()) {
            double  value = i.next();

            if (value < lb || value > ub)
                i.remove();
        }
    }

    static Double getMedian(List values, int from, int to) {
        int sz = to - from + 1;

        if (sz % 2 == 0)
             return (values.get(from + sz/2) + values.get(from + (sz/2) - 1) ) / 2;
        else
            return values.get(from + (sz / 2));
    }

    static double getQ1(List values) {
        int sz = values.size();

        if (sz == 1)
            return values.get(0);
        else if (sz % 2 == 0)
            return getMedian(values, 0, sz / 2 - 1);
        else
            return getMedian(values, 0, sz / 2 - 1);
    }

    static double getQ3(List values) {
        int sz = values.size();

        if (sz == 1)
            return values.get(0);
        else if (sz % 2 == 0)
            return getMedian(values, sz / 2, sz - 1);
        else
            return getMedian(values, sz / 2 + 1, sz - 1);
    }

}