-
Notifications
You must be signed in to change notification settings - Fork 24
/
histogram_calculation.cc
68 lines (58 loc) · 2.61 KB
/
histogram_calculation.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
// ------------------------------------------------------------------------ //
// Copyright 2021 SPTK Working Group //
// //
// Licensed under the Apache License, Version 2.0 (the "License"); //
// you may not use this file except in compliance with the License. //
// You may obtain a copy of the License at //
// //
// http://www.apache.org/licenses/LICENSE-2.0 //
// //
// Unless required by applicable law or agreed to in writing, software //
// distributed under the License is distributed on an "AS IS" BASIS, //
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. //
// See the License for the specific language governing permissions and //
// limitations under the License. //
// ------------------------------------------------------------------------ //
#include "SPTK/math/histogram_calculation.h"
#include <algorithm> // std::fill
#include <cmath> // std::floor
#include <cstddef> // std::size_t
namespace sptk {
HistogramCalculation::HistogramCalculation(int num_bin, double lower_bound,
double upper_bound)
: num_bin_(num_bin),
lower_bound_(lower_bound),
upper_bound_(upper_bound),
bin_width_((upper_bound_ - lower_bound_) / num_bin_),
is_valid_(true) {
if (num_bin_ <= 0 || upper_bound_ <= lower_bound_) {
is_valid_ = false;
return;
}
}
bool HistogramCalculation::Run(const std::vector<double>& data,
std::vector<double>* histogram) const {
// Check inputs.
if (!is_valid_ || data.empty() || NULL == histogram) {
return false;
}
// Prepare memories.
if (histogram->size() != static_cast<std::size_t>(num_bin_)) {
histogram->resize(num_bin_);
}
std::fill(histogram->begin(), histogram->end(), 0.0);
const int length(static_cast<int>(data.size()));
const double* input(&(data[0]));
double* output(&((*histogram)[0]));
for (int i(0); i < length; ++i) {
if (lower_bound_ <= input[i] && input[i] < upper_bound_) {
const int bin_index(
static_cast<int>(std::floor((input[i] - lower_bound_) / bin_width_)));
++output[bin_index];
} else if (upper_bound_ == input[i]) {
++output[num_bin_ - 1];
}
}
return true;
}
} // namespace sptk