47 lines
1.0 KiB
C++
47 lines
1.0 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#ifndef JUNGFRAUJOCH_HISTOGRAM_H
|
|
#define JUNGFRAUJOCH_HISTOGRAM_H
|
|
|
|
#include <cstdint>
|
|
#include <cstddef>
|
|
#include <vector>
|
|
#include <mutex>
|
|
#include "Plot.h"
|
|
|
|
template <class T>
|
|
class Histogram {
|
|
std::vector<T> sum;
|
|
std::vector<uint64_t> count;
|
|
std::mutex m;
|
|
public:
|
|
explicit Histogram(size_t bins) : sum(bins), count(bins) {}
|
|
void Add(size_t bin, T val) {
|
|
std::unique_lock<std::mutex> ul(m);
|
|
|
|
if (bin < sum.size()) {
|
|
sum[bin] += val;
|
|
count[bin] += 1;
|
|
}
|
|
}
|
|
|
|
Plot GetPlot() {
|
|
std::unique_lock<std::mutex> ul(m);
|
|
|
|
Plot ret;
|
|
ret.x.resize(sum.size());
|
|
ret.y.resize(sum.size());
|
|
for (int i = 0; i < sum.size(); i++) {
|
|
ret.x[i] = static_cast<float>(i);
|
|
if (count[i] > 0)
|
|
ret.y[i] = static_cast<float>(sum[i]) / count[i];
|
|
else
|
|
ret.y[i] = 0;
|
|
}
|
|
return ret;
|
|
}
|
|
};
|
|
|
|
|
|
#endif //JUNGFRAUJOCH_HISTOGRAM_H
|