Home | History | Annotate | Download | only in util
      1 /*
      2  * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.
      3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
      4  *
      5  * This code is free software; you can redistribute it and/or modify it
      6  * under the terms of the GNU General Public License version 2 only, as
      7  * published by the Free Software Foundation.  Oracle designates this
      8  * particular file as subject to the "Classpath" exception as provided
      9  * by Oracle in the LICENSE file that accompanied this code.
     10  *
     11  * This code is distributed in the hope that it will be useful, but WITHOUT
     12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
     13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
     14  * version 2 for more details (a copy is included in the LICENSE file that
     15  * accompanied this code).
     16  *
     17  * You should have received a copy of the GNU General Public License version
     18  * 2 along with this work; if not, write to the Free Software Foundation,
     19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
     20  *
     21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
     22  * or visit www.oracle.com if you need additional information or have any
     23  * questions.
     24  */
     25 package java.util;
     26 
     27 import java.util.function.DoubleConsumer;
     28 
     29 /**
     30  * A state object for collecting statistics such as count, min, max, sum, and
     31  * average.
     32  *
     33  * <p>This class is designed to work with (though does not require)
     34  * {@linkplain java.util.stream streams}. For example, you can compute
     35  * summary statistics on a stream of doubles with:
     36  * <pre> {@code
     37  * DoubleSummaryStatistics stats = doubleStream.collect(DoubleSummaryStatistics::new,
     38  *                                                      DoubleSummaryStatistics::accept,
     39  *                                                      DoubleSummaryStatistics::combine);
     40  * }</pre>
     41  *
     42  * <p>{@code DoubleSummaryStatistics} can be used as a
     43  * {@linkplain java.util.stream.Stream#collect(Collector) reduction}
     44  * target for a {@linkplain java.util.stream.Stream stream}. For example:
     45  *
     46  * <pre> {@code
     47  * DoubleSummaryStatistics stats = people.stream()
     48  *     .collect(Collectors.summarizingDouble(Person::getWeight));
     49  *}</pre>
     50  *
     51  * This computes, in a single pass, the count of people, as well as the minimum,
     52  * maximum, sum, and average of their weights.
     53  *
     54  * @implNote This implementation is not thread safe. However, it is safe to use
     55  * {@link java.util.stream.Collectors#summarizingDouble(java.util.function.ToDoubleFunction)
     56  * Collectors.toDoubleStatistics()} on a parallel stream, because the parallel
     57  * implementation of {@link java.util.stream.Stream#collect Stream.collect()}
     58  * provides the necessary partitioning, isolation, and merging of results for
     59  * safe and efficient parallel execution.
     60  * @since 1.8
     61  */
     62 public class DoubleSummaryStatistics implements DoubleConsumer {
     63     private long count;
     64     private double sum;
     65     private double sumCompensation; // Low order bits of sum
     66     private double simpleSum; // Used to compute right sum for non-finite inputs
     67     private double min = Double.POSITIVE_INFINITY;
     68     private double max = Double.NEGATIVE_INFINITY;
     69 
     70     /**
     71      * Construct an empty instance with zero count, zero sum,
     72      * {@code Double.POSITIVE_INFINITY} min, {@code Double.NEGATIVE_INFINITY}
     73      * max and zero average.
     74      */
     75     public DoubleSummaryStatistics() { }
     76 
     77     /**
     78      * Records another value into the summary information.
     79      *
     80      * @param value the input value
     81      */
     82     @Override
     83     public void accept(double value) {
     84         ++count;
     85         simpleSum += value;
     86         sumWithCompensation(value);
     87         min = Math.min(min, value);
     88         max = Math.max(max, value);
     89     }
     90 
     91     /**
     92      * Combines the state of another {@code DoubleSummaryStatistics} into this
     93      * one.
     94      *
     95      * @param other another {@code DoubleSummaryStatistics}
     96      * @throws NullPointerException if {@code other} is null
     97      */
     98     public void combine(DoubleSummaryStatistics other) {
     99         count += other.count;
    100         simpleSum += other.simpleSum;
    101         sumWithCompensation(other.sum);
    102         sumWithCompensation(other.sumCompensation);
    103         min = Math.min(min, other.min);
    104         max = Math.max(max, other.max);
    105     }
    106 
    107     /**
    108      * Incorporate a new double value using Kahan summation /
    109      * compensated summation.
    110      */
    111     private void sumWithCompensation(double value) {
    112         double tmp = value - sumCompensation;
    113         double velvel = sum + tmp; // Little wolf of rounding error
    114         sumCompensation = (velvel - sum) - tmp;
    115         sum = velvel;
    116     }
    117 
    118     /**
    119      * Return the count of values recorded.
    120      *
    121      * @return the count of values
    122      */
    123     public final long getCount() {
    124         return count;
    125     }
    126 
    127     /**
    128      * Returns the sum of values recorded, or zero if no values have been
    129      * recorded.
    130      *
    131      * If any recorded value is a NaN or the sum is at any point a NaN
    132      * then the sum will be NaN.
    133      *
    134      * <p> The value of a floating-point sum is a function both of the
    135      * input values as well as the order of addition operations. The
    136      * order of addition operations of this method is intentionally
    137      * not defined to allow for implementation flexibility to improve
    138      * the speed and accuracy of the computed result.
    139      *
    140      * In particular, this method may be implemented using compensated
    141      * summation or other technique to reduce the error bound in the
    142      * numerical sum compared to a simple summation of {@code double}
    143      * values.
    144      *
    145      * @apiNote Values sorted by increasing absolute magnitude tend to yield
    146      * more accurate results.
    147      *
    148      * @return the sum of values, or zero if none
    149      */
    150     public final double getSum() {
    151         // Better error bounds to add both terms as the final sum
    152         double tmp =  sum + sumCompensation;
    153         if (Double.isNaN(tmp) && Double.isInfinite(simpleSum))
    154             // If the compensated sum is spuriously NaN from
    155             // accumulating one or more same-signed infinite values,
    156             // return the correctly-signed infinity stored in
    157             // simpleSum.
    158             return simpleSum;
    159         else
    160             return tmp;
    161     }
    162 
    163     /**
    164      * Returns the minimum recorded value, {@code Double.NaN} if any recorded
    165      * value was NaN or {@code Double.POSITIVE_INFINITY} if no values were
    166      * recorded. Unlike the numerical comparison operators, this method
    167      * considers negative zero to be strictly smaller than positive zero.
    168      *
    169      * @return the minimum recorded value, {@code Double.NaN} if any recorded
    170      * value was NaN or {@code Double.POSITIVE_INFINITY} if no values were
    171      * recorded
    172      */
    173     public final double getMin() {
    174         return min;
    175     }
    176 
    177     /**
    178      * Returns the maximum recorded value, {@code Double.NaN} if any recorded
    179      * value was NaN or {@code Double.NEGATIVE_INFINITY} if no values were
    180      * recorded. Unlike the numerical comparison operators, this method
    181      * considers negative zero to be strictly smaller than positive zero.
    182      *
    183      * @return the maximum recorded value, {@code Double.NaN} if any recorded
    184      * value was NaN or {@code Double.NEGATIVE_INFINITY} if no values were
    185      * recorded
    186      */
    187     public final double getMax() {
    188         return max;
    189     }
    190 
    191     /**
    192      * Returns the arithmetic mean of values recorded, or zero if no
    193      * values have been recorded.
    194      *
    195      * If any recorded value is a NaN or the sum is at any point a NaN
    196      * then the average will be code NaN.
    197      *
    198      * <p>The average returned can vary depending upon the order in
    199      * which values are recorded.
    200      *
    201      * This method may be implemented using compensated summation or
    202      * other technique to reduce the error bound in the {@link #getSum
    203      * numerical sum} used to compute the average.
    204      *
    205      * @apiNote Values sorted by increasing absolute magnitude tend to yield
    206      * more accurate results.
    207      *
    208      * @return the arithmetic mean of values, or zero if none
    209      */
    210     public final double getAverage() {
    211         return getCount() > 0 ? getSum() / getCount() : 0.0d;
    212     }
    213 
    214     /**
    215      * {@inheritDoc}
    216      *
    217      * Returns a non-empty string representation of this object suitable for
    218      * debugging. The exact presentation format is unspecified and may vary
    219      * between implementations and versions.
    220      */
    221     @Override
    222     public String toString() {
    223         return String.format(
    224             "%s{count=%d, sum=%f, min=%f, average=%f, max=%f}",
    225             this.getClass().getSimpleName(),
    226             getCount(),
    227             getSum(),
    228             getMin(),
    229             getAverage(),
    230             getMax());
    231     }
    232 }
    233