This CL implements Welford's algorithm for a numerically stable computation of the variance. This implementation is plugged in SamplesStatsCounter class (adapter pattern). A 'NumericalStability' unit test has been added, whose previous implementation of SamplesStatsCounter failed to pass. Follow-up CLs will factorize more occurences of duplicated and misbehaved computations. Bug: webrtc:10412 Change-Id: Id807c3d34e9c780fb1cbd769d30b655c575c88ac Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/131394 Commit-Queue: Yves Gerey <yvesg@google.com> Reviewed-by: Artem Titov <titovartem@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Cr-Commit-Position: refs/heads/master@{#27547}
88 lines
2.8 KiB
C++
88 lines
2.8 KiB
C++
/*
|
|
* Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef RTC_BASE_NUMERICS_SAMPLES_STATS_COUNTER_H_
|
|
#define RTC_BASE_NUMERICS_SAMPLES_STATS_COUNTER_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/numerics/running_statistics.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// This class extends RunningStatistics by providing GetPercentile() method,
|
|
// while slightly adapting the interface.
|
|
class SamplesStatsCounter {
|
|
public:
|
|
SamplesStatsCounter();
|
|
~SamplesStatsCounter();
|
|
SamplesStatsCounter(const SamplesStatsCounter&);
|
|
SamplesStatsCounter& operator=(const SamplesStatsCounter&);
|
|
SamplesStatsCounter(SamplesStatsCounter&&);
|
|
SamplesStatsCounter& operator=(SamplesStatsCounter&&);
|
|
|
|
// Adds sample to the stats in amortized O(1) time.
|
|
void AddSample(double value);
|
|
|
|
// Adds samples from another counter.
|
|
void AddSamples(const SamplesStatsCounter& other);
|
|
|
|
// Returns if there are any values in O(1) time.
|
|
bool IsEmpty() const { return samples_.empty(); }
|
|
|
|
// Returns min in O(1) time. This function may not be called if there are no
|
|
// samples.
|
|
double GetMin() const {
|
|
RTC_DCHECK(!IsEmpty());
|
|
return *stats_.GetMin();
|
|
}
|
|
// Returns max in O(1) time. This function may not be called if there are no
|
|
// samples.
|
|
double GetMax() const {
|
|
RTC_DCHECK(!IsEmpty());
|
|
return *stats_.GetMax();
|
|
}
|
|
// Returns average in O(1) time. This function may not be called if there are
|
|
// no samples.
|
|
double GetAverage() const {
|
|
RTC_DCHECK(!IsEmpty());
|
|
return *stats_.GetMean();
|
|
}
|
|
// Returns variance in O(1) time. This function may not be called if there are
|
|
// no samples.
|
|
double GetVariance() const {
|
|
RTC_DCHECK(!IsEmpty());
|
|
return *stats_.GetVariance();
|
|
}
|
|
// Returns standard deviation in O(1) time. This function may not be called if
|
|
// there are no samples.
|
|
double GetStandardDeviation() const {
|
|
RTC_DCHECK(!IsEmpty());
|
|
return *stats_.GetStandardDeviation();
|
|
}
|
|
// Returns percentile in O(nlogn) on first call and in O(1) after, if no
|
|
// additions were done. This function may not be called if there are no
|
|
// samples.
|
|
//
|
|
// |percentile| has to be in [0; 1]. 0 percentile is the min in the array and
|
|
// 1 percentile is the max in the array.
|
|
double GetPercentile(double percentile);
|
|
|
|
private:
|
|
RunningStatistics<double> stats_;
|
|
std::vector<double> samples_;
|
|
bool sorted_ = false;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // RTC_BASE_NUMERICS_SAMPLES_STATS_COUNTER_H_
|