blob: 4fddc9bc183830c0960bf3437d1725bed43e3b1f [file] [log] [blame]
mtklein90c471e2014-06-16 14:04:32 -07001#ifndef Stats_DEFINED
2#define Stats_DEFINED
3
mtklein5d9d10e2014-07-11 11:57:07 -07004#include <math.h>
5
6#include "SkString.h"
mtklein40b32be2014-07-09 08:46:49 -07007#include "SkTSort.h"
8
mtklein62386882014-07-15 10:30:31 -07009#ifdef SK_BUILD_FOR_WIN
10 static const char* kBars[] = { ".", "o", "O" };
11#else
12 static const char* kBars[] = { "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█" };
13#endif
mtklein5d9d10e2014-07-11 11:57:07 -070014
mtklein90c471e2014-06-16 14:04:32 -070015struct Stats {
16 Stats(const double samples[], int n) {
17 min = samples[0];
18 max = samples[0];
19 for (int i = 0; i < n; i++) {
20 if (samples[i] < min) { min = samples[i]; }
21 if (samples[i] > max) { max = samples[i]; }
22 }
23
24 double sum = 0.0;
25 for (int i = 0 ; i < n; i++) {
26 sum += samples[i];
27 }
28 mean = sum / n;
29
30 double err = 0.0;
31 for (int i = 0 ; i < n; i++) {
32 err += (samples[i] - mean) * (samples[i] - mean);
33 }
34 var = err / (n-1);
mtklein40b32be2014-07-09 08:46:49 -070035
36 SkAutoTMalloc<double> sorted(n);
37 memcpy(sorted.get(), samples, n * sizeof(double));
38 SkTQSort(sorted.get(), sorted.get() + n - 1);
39 median = sorted[n/2];
mtklein5d9d10e2014-07-11 11:57:07 -070040
mtkleina189ccd2014-07-14 12:28:47 -070041 // Normalize samples to [min, max] in as many quanta as we have distinct bars to print.
mtklein5d9d10e2014-07-11 11:57:07 -070042 for (int i = 0; i < n; i++) {
mtkleina189ccd2014-07-14 12:28:47 -070043 if (min == max) {
44 // All samples are the same value. Don't divide by zero.
45 plot.append(kBars[0]);
46 continue;
47 }
48
mtklein5d9d10e2014-07-11 11:57:07 -070049 double s = samples[i];
mtklein5d9d10e2014-07-11 11:57:07 -070050 s -= min;
51 s /= (max - min);
52 s *= (SK_ARRAY_COUNT(kBars) - 1);
53 const size_t bar = (size_t)round(s);
54 SK_ALWAYSBREAK(bar < SK_ARRAY_COUNT(kBars));
55 plot.append(kBars[bar]);
56 }
mtklein90c471e2014-06-16 14:04:32 -070057 }
58
59 double min;
60 double max;
mtklein5d9d10e2014-07-11 11:57:07 -070061 double mean; // Estimate of population mean.
62 double var; // Estimate of population variance.
mtklein40b32be2014-07-09 08:46:49 -070063 double median;
mtklein5d9d10e2014-07-11 11:57:07 -070064 SkString plot; // A single-line bar chart (_not_ histogram) of the samples.
mtklein90c471e2014-06-16 14:04:32 -070065};
66
67#endif//Stats_DEFINED