statistic helper classes
test-cases modified grid importance for better trap-detection
This commit is contained in:
52
math/stats/Median.h
Normal file
52
math/stats/Median.h
Normal file
@@ -0,0 +1,52 @@
|
||||
#ifndef STATS_MEDIAN_H
|
||||
#define STATS_MEDIAN_H
|
||||
|
||||
#include <vector>
|
||||
#include <algorithm>
|
||||
|
||||
#include "../../Assertions.h"
|
||||
|
||||
namespace Stats {
|
||||
|
||||
template <typename Scalar> class Median {
|
||||
|
||||
private:
|
||||
|
||||
/** all scalar values in a sorted order (ascending) */
|
||||
std::vector<Scalar> sorted;
|
||||
|
||||
public:
|
||||
|
||||
/** add the given scalar value to the median calculation */
|
||||
void add(const Scalar value) {
|
||||
|
||||
const auto idx = std::upper_bound( sorted.begin(), sorted.end(), value );
|
||||
sorted.insert( idx, value );
|
||||
|
||||
}
|
||||
|
||||
/** get the median of all added values */
|
||||
float get() const {
|
||||
|
||||
// sanity check
|
||||
Assert::isNot0(sorted.size(), "add elements first!");
|
||||
|
||||
if (sorted.size() % 2 == 1) { // odd
|
||||
|
||||
const int idx = sorted.size()/2;
|
||||
return sorted[idx];
|
||||
|
||||
} else { // even
|
||||
|
||||
const int idx0 = sorted.size()/2 - 1;
|
||||
return (sorted[idx0] + sorted[idx0+1]) / 2;
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // STATS_MEDIAN_H
|
||||
Reference in New Issue
Block a user