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 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185
|
/************************************************************************
*
* Copyright (C) 2022-2024 IRCAD France
*
* This file is part of Sight.
*
* Sight is free software: you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Sight is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with Sight. If not, see <https://www.gnu.org/licenses/>.
*
***********************************************************************/
#include "histogram.hpp"
#include <core/tools/dispatcher.hpp>
#include <data/helper/medical_image.hpp>
#include <data/thread/region_threader.hpp>
#include <numeric>
namespace sight::data::helper
{
/**
* @brief Functor use to compute the histogram of the image.
*/
struct computehistogram_functor
{
/// Parameters of the functor.
struct parameter
{
data::image::csptr image;
double bins_width {};
std::vector<double> o_histogram;
double o_min {std::numeric_limits<double>::max()};
double o_max {std::numeric_limits<double>::lowest()};
};
using vector_t = std::vector<std::vector<double> >;
//------------------------------------------------------------------------------
template<class T>
static void count_pixels(
const data::image::const_iterator<T>& _img_begin,
vector_t& _values,
T _min,
double _inv_bins_width,
std::ptrdiff_t _region_min,
std::ptrdiff_t _region_max,
std::size_t _i
)
{
const data::image::const_iterator<T> begin = _img_begin + _region_min;
const data::image::const_iterator<T> end = _img_begin + _region_max;
for(auto itr = begin ; itr != end ; ++itr)
{
const auto index = static_cast<std::size_t>(static_cast<double>(*itr - _min) * _inv_bins_width);
++_values[_i][index];
}
}
//------------------------------------------------------------------------------
template<class IMAGETYPE>
void operator()(parameter& _param)
{
data::image::csptr image = _param.image;
const auto dump_lock = image->dump_lock();
{
const auto& [min, max] = data::helper::medical_image::get_min_max<IMAGETYPE>(image);
const double inv_bins_width = 1 / _param.bins_width;
if(max > min)
{
vector_t values;
std::size_t size = static_cast<std::size_t>(static_cast<double>(max - min) * inv_bins_width) + 1;
sight::data::thread::region_threader rt;
values.resize(rt.number_of_thread());
for(auto& v : values)
{
v.resize(size, 0);
}
rt(
[begin = image->cbegin<IMAGETYPE>(), &values, min = min, inv_bins_width]
(std::ptrdiff_t _p_h1, std::ptrdiff_t _p_h2, std::size_t _p_h3, auto&& ...)
{
return computehistogram_functor::count_pixels<IMAGETYPE>(
begin,
values,
min,
inv_bins_width,
_p_h1,
_p_h2,
_p_h3
);
},
image->cend<IMAGETYPE>() - image->cbegin<IMAGETYPE>()
);
_param.o_histogram.resize(size, 0);
double num_pixels = 0;
for(std::size_t i = 0 ; i < size ; ++i)
{
for(const auto& v : values)
{
_param.o_histogram[i] += v[i];
}
num_pixels += _param.o_histogram[i];
}
SIGHT_ASSERT("The number of pixels should not be null", num_pixels != 0);
for(std::size_t i = 0 ; i < size ; ++i)
{
_param.o_histogram[i] /= num_pixels;
}
_param.o_min = static_cast<double>(min);
_param.o_max = static_cast<double>(max);
}
}
}
};
//------------------------------------------------------------------------------
void histogram::compute()
{
computehistogram_functor::parameter param;
param.image = m_image;
param.bins_width = 1.;
core::type type = m_image->type();
core::tools::dispatcher<core::tools::supported_dispatcher_types, computehistogram_functor>::invoke(type, param);
m_values = std::move(param.o_histogram);
m_max = param.o_max;
m_min = param.o_min;
}
//------------------------------------------------------------------------------
histogram::histogram_t histogram::sample(std::size_t _bin_width) const
{
const auto bin_width = static_cast<std::ptrdiff_t>(_bin_width);
const std::ptrdiff_t num_bins = static_cast<std::ptrdiff_t>(m_max - m_min) / bin_width + 1;
const auto begin = m_values.begin();
std::ptrdiff_t it = 0;
histogram_t histogram;
for(ptrdiff_t i = 0 ; i < num_bins ; ++i)
{
histogram.push_back(
std::accumulate(
begin + it,
begin + std::min(it + bin_width, static_cast<std::ptrdiff_t>(m_values.size())),
0.
)
);
it += bin_width;
}
return histogram;
}
//------------------------------------------------------------------------------
} // namespace sight::data::helper
|