NumCpp  2.10.1
A Templatized Header Only C++ Implementation of the Python NumPy Library
nearest2d.hpp
Go to the documentation of this file.
1 #pragma once
29 
31 #include "NumCpp/Core/Shape.hpp"
32 #include "NumCpp/Core/Slice.hpp"
33 #include "NumCpp/Core/Types.hpp"
35 #include "NumCpp/NdArray.hpp"
36 
37 namespace nc::filter::boundary
38 {
39  //============================================================================
40  // Method Description:
47  template<typename dtype>
48  NdArray<dtype> nearest2d(const NdArray<dtype>& inImage, uint32 inBoundarySize)
49  {
51 
52  const Shape inShape = inImage.shape();
53  Shape outShape(inShape);
54  outShape.rows += inBoundarySize * 2;
55  outShape.cols += inBoundarySize * 2;
56 
57  NdArray<dtype> outArray(outShape);
58  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
59  Slice(inBoundarySize, inBoundarySize + inShape.cols),
60  inImage);
61  fillCorners(outArray, inBoundarySize);
62 
63  for (uint32 row = 0; row < inBoundarySize; ++row)
64  {
65  // bottom
66  outArray.put(row, Slice(inBoundarySize, inBoundarySize + inShape.cols), inImage(0, Slice(0, inShape.cols)));
67 
68  // top
69  outArray.put(row + inBoundarySize + inShape.rows,
70  Slice(inBoundarySize, inBoundarySize + inShape.cols),
71  inImage(inShape.rows - 1, Slice(0, inShape.cols)));
72  }
73 
74  for (uint32 col = 0; col < inBoundarySize; ++col)
75  {
76  // left
77  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows), col, inImage(Slice(0, inShape.rows), 0));
78 
79  // right
80  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
81  col + inBoundarySize + inShape.cols,
82  inImage(Slice(0, inShape.rows), inShape.cols - 1));
83  }
84 
85  return outArray;
86  }
87 } // namespace nc::filter::boundary
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:39
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:138
self_type & put(index_type inIndex, const value_type &inValue)
Definition: NdArrayCore.hpp:3616
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4402
A Shape Class for NdArrays.
Definition: Core/Shape.hpp:41
uint32 rows
Definition: Core/Shape.hpp:44
uint32 cols
Definition: Core/Shape.hpp:45
A Class for slicing into NdArrays.
Definition: Slice.hpp:45
Definition: addBoundary1d.hpp:44
NdArray< dtype > nearest2d(const NdArray< dtype > &inImage, uint32 inBoundarySize)
Definition: nearest2d.hpp:48
void fillCorners(NdArray< dtype > &inArray, uint32 inBorderWidth)
Definition: fillCorners.hpp:46
std::uint32_t uint32
Definition: Types.hpp:40