libs/gil/test/extension/numeric/convolve_2d.cpp

   1 //
   2 // Copyright 2019 Miral Shah <miralshah2211@gmail.com>
   3 //
   4 // Use, modification and distribution are subject to the Boost Software License,
   5 // Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
   6 // http://www.boost.org/LICENSE_1_0.txt)
   7 //
   8
   9 #include <cstddef>
  10
  11 #include <boost/gil.hpp>
  12 #include <boost/gil/extension/numeric/convolve.hpp>
  13
  14 #define BOOST_TEST_MODULE test_ext_convolve_2d
  15 #include "unit_test.hpp"
  16
  17 namespace gil = boost::gil;
  18
  19 std::uint8_t img[] =
  20 {
  21     0, 0, 0, 0, 0, 0, 0, 0, 0,
  22     0, 0, 0, 0, 0, 0, 0, 0, 0,
  23     0, 0, 255, 0, 0, 0, 255, 0, 0,
  24     0, 0, 0, 255, 0, 255, 0, 0, 0,
  25     0, 0, 0, 0, 255, 0, 0, 0, 0,
  26     0, 0, 0, 255, 0, 255, 0, 0, 0,
  27     0, 0, 255, 0, 0, 0, 255, 0, 0,
  28     0, 0, 0, 0, 0, 0, 0, 0, 0,
  29     0, 0, 0, 0, 0, 0, 0, 0, 0
  30 };
  31
  32 std::uint8_t output[] =
  33 {
  34     0, 0, 0, 0, 0, 0, 0, 0, 0,
  35     0, 28, 28, 28, 0, 28, 28, 28, 0,
  36     0, 28, 56, 56, 56, 56, 56, 28, 0,
  37     0, 28, 56, 85, 85, 85, 56, 28, 0,
  38     0, 0, 56, 85, 141, 85, 56, 0, 0,
  39     0, 28, 56, 85, 85, 85, 56, 28, 0,
  40     0, 28, 56, 56, 56, 56, 56, 28, 0,
  41     0, 28, 28, 28, 0, 28, 28, 28, 0,
  42     0, 0, 0, 0, 0, 0, 0, 0, 0
  43 };
  44
  45 BOOST_AUTO_TEST_SUITE(convolve_2d)
  46
  47 BOOST_AUTO_TEST_CASE(convolve_2d_with_normalized_mean_filter)
  48 {
  49     gil::gray8c_view_t src_view =
  50         gil::interleaved_view(9, 9, reinterpret_cast<const gil::gray8_pixel_t*>(img), 9);
  51
  52     gil::image<gil::gray8_pixel_t> temp_img(src_view.width(), src_view.height());
  53     typename gil::image<gil::gray8_pixel_t>::view_t temp_view = view(temp_img);
  54     gil::gray8_view_t dst_view(temp_view);
  55
  56     std::vector<float> v(9, 1.0f / 9.0f);
  57     gil::detail::kernel_2d<float> kernel(v.begin(), v.size(), 1, 1);
  58
  59     gil::detail::convolve_2d(src_view, kernel, dst_view);
  60
  61     gil::gray8c_view_t out_view =
  62         gil::interleaved_view(9, 9, reinterpret_cast<const gil::gray8_pixel_t*>(output), 9);
  63
  64     BOOST_TEST(gil::equal_pixels(out_view, dst_view));
  65 }
  66
  67 BOOST_AUTO_TEST_SUITE_END()