NumCpp  2.11.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
reflect2d.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:
48  template<typename dtype>
49  NdArray<dtype> reflect2d(const NdArray<dtype>& inImage, uint32 inBoundarySize)
50  {
52 
53  const Shape inShape = inImage.shape();
54  Shape outShape(inShape);
55  outShape.rows += inBoundarySize * 2;
56  outShape.cols += inBoundarySize * 2;
57 
58  NdArray<dtype> outArray(outShape);
59  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
60  Slice(inBoundarySize, inBoundarySize + inShape.cols),
61  inImage);
62 
63  for (uint32 row = 0; row < inBoundarySize; ++row)
64  {
65  // bottom
66  outArray.put(row,
67  Slice(inBoundarySize, inBoundarySize + inShape.cols),
68  inImage(inBoundarySize - row - 1, Slice(0, inShape.cols)));
69 
70  // top
71  outArray.put(row + inBoundarySize + inShape.rows,
72  Slice(inBoundarySize, inBoundarySize + inShape.cols),
73  inImage(inShape.rows - row - 1, Slice(0, inShape.cols)));
74  }
75 
76  for (uint32 col = 0; col < inBoundarySize; ++col)
77  {
78  // left
79  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
80  col,
81  inImage(Slice(0, inShape.rows), inBoundarySize - col - 1));
82 
83  // right
84  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
85  col + inBoundarySize + inShape.cols,
86  inImage(Slice(0, inShape.rows), inShape.cols - col - 1));
87  }
88 
89  // now fill in the corners
90  NdArray<dtype> lowerLeft =
91  flipud(outArray(Slice(inBoundarySize, 2 * inBoundarySize), Slice(0, inBoundarySize)));
92  NdArray<dtype> lowerRight = flipud(
93  outArray(Slice(inBoundarySize, 2 * inBoundarySize), Slice(outShape.cols - inBoundarySize, outShape.cols)));
94 
95  const uint32 upperRowStart = outShape.rows - 2 * inBoundarySize;
96  NdArray<dtype> upperLeft =
97  flipud(outArray(Slice(upperRowStart, upperRowStart + inBoundarySize), Slice(0, inBoundarySize)));
98  NdArray<dtype> upperRight = flipud(outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
99  Slice(outShape.cols - inBoundarySize, outShape.cols)));
100 
101  outArray.put(Slice(0, inBoundarySize), Slice(0, inBoundarySize), lowerLeft);
102  outArray.put(Slice(0, inBoundarySize), Slice(outShape.cols - inBoundarySize, outShape.cols), lowerRight);
103  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows), Slice(0, inBoundarySize), upperLeft);
104  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
105  Slice(outShape.cols - inBoundarySize, outShape.cols),
106  upperRight);
107 
108  return outArray;
109  }
110 } // 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:3646
const Shape & shape() const noexcept
Definition: NdArrayCore.hpp:4464
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 > reflect2d(const NdArray< dtype > &inImage, uint32 inBoundarySize)
Definition: reflect2d.hpp:49
NdArray< dtype > flipud(const NdArray< dtype > &inArray)
Definition: flipud.hpp:46
std::uint32_t uint32
Definition: Types.hpp:40