NumCpp  2.1.0
A C++ implementation of the Python Numpy library
uniformFilter.hpp
Go to the documentation of this file.
1 #pragma once
30 
31 #include "NumCpp/Core/Shape.hpp"
32 #include "NumCpp/Core/Slice.hpp"
33 #include "NumCpp/Core/Types.hpp"
36 #include "NumCpp/NdArray.hpp"
37 
38 namespace nc
39 {
40  namespace filter
41  {
42  //============================================================================
43  // Method Description:
55  template<typename dtype>
56  NdArray<dtype> uniformFilter(const NdArray<dtype>& inImageArray, uint32 inSize,
57  Boundary inBoundaryType = Boundary::REFLECT, dtype inConstantValue = 0)
58  {
59  NdArray<dtype> arrayWithBoundary = boundary::addBoundary2d(inImageArray, inBoundaryType, inSize, inConstantValue);
60  NdArray<dtype> output(inImageArray.shape());
61 
62  const Shape inShape = inImageArray.shape();
63  const uint32 boundarySize = inSize / 2; // integer division
64  const uint32 endPointRow = boundarySize + inShape.rows;
65  const uint32 endPointCol = boundarySize + inShape.cols;
66 
67  for (uint32 row = boundarySize; row < endPointRow; ++row)
68  {
69  for (uint32 col = boundarySize; col < endPointCol; ++col)
70  {
71  NdArray<dtype> window = arrayWithBoundary(Slice(row - boundarySize, row + boundarySize + 1),
72  Slice(col - boundarySize, col + boundarySize + 1));
73 
74  output(row - boundarySize, col - boundarySize) = mean(window).item();
75  }
76  }
77 
78  return output;
79  }
80  } // namespace filter
81 } // namespace nc
nc::NdArray::item
value_type item() const
Definition: NdArrayCore.hpp:2958
nc::NdArray::shape
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4312
nc::mean
NdArray< double > mean(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: mean.hpp:54
nc::NdArray< dtype >
nc::uint32
std::uint32_t uint32
Definition: Types.hpp:41
NdArray.hpp
nc::Shape
A Shape Class for NdArrays.
Definition: Core/Shape.hpp:41
nc::Shape::cols
uint32 cols
Definition: Core/Shape.hpp:46
nc::filter::boundary::addBoundary2d
NdArray< dtype > addBoundary2d(const NdArray< dtype > &inImage, Boundary inBoundaryType, uint32 inKernalSize, dtype inConstantValue=0)
Definition: addBoundary2d.hpp:62
nc::filter::uniformFilter
NdArray< dtype > uniformFilter(const NdArray< dtype > &inImageArray, uint32 inSize, Boundary inBoundaryType=Boundary::REFLECT, dtype inConstantValue=0)
Definition: uniformFilter.hpp:56
Shape.hpp
nc::filter::Boundary::REFLECT
@ REFLECT
nc
Definition: Coordinate.hpp:45
nc::Shape::rows
uint32 rows
Definition: Core/Shape.hpp:45
nc::filter::Boundary
Boundary
Boundary condition to apply to the image filter.
Definition: Boundary.hpp:38
Types.hpp
nc::Slice
A Class for slicing into NdArrays.
Definition: Slice.hpp:44
Slice.hpp
mean.hpp
addBoundary2d.hpp