NumCpp  2.4.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
38 {
39  namespace filter
40  {
41  namespace boundary
42  {
43  //============================================================================
44  // Method Description:
53  template<typename dtype>
54  NdArray<dtype> reflect2d(const NdArray<dtype>& inImage, uint32 inBoundarySize)
55  {
57 
58  const Shape inShape = inImage.shape();
59  Shape outShape(inShape);
60  outShape.rows += inBoundarySize * 2;
61  outShape.cols += inBoundarySize * 2;
62 
63  NdArray<dtype> outArray(outShape);
64  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
65  Slice(inBoundarySize, inBoundarySize + inShape.cols), inImage);
66 
67  for (uint32 row = 0; row < inBoundarySize; ++row)
68  {
69  // bottom
70  outArray.put(row,
71  Slice(inBoundarySize, inBoundarySize + inShape.cols),
72  inImage(inBoundarySize - row - 1, Slice(0, inShape.cols)));
73 
74  // top
75  outArray.put(row + inBoundarySize + inShape.rows,
76  Slice(inBoundarySize, inBoundarySize + inShape.cols),
77  inImage(inShape.rows - row - 1, Slice(0, inShape.cols)));
78  }
79 
80  for (uint32 col = 0; col < inBoundarySize; ++col)
81  {
82  // left
83  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
84  col,
85  inImage(Slice(0, inShape.rows), inBoundarySize - col - 1));
86 
87  // right
88  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
89  col + inBoundarySize + inShape.cols,
90  inImage(Slice(0, inShape.rows), inShape.cols - col - 1));
91  }
92 
93  // now fill in the corners
94  NdArray<dtype> lowerLeft = flipud(outArray(Slice(inBoundarySize, 2 * inBoundarySize),
95  Slice(0, inBoundarySize)));
96  NdArray<dtype> lowerRight = flipud(outArray(Slice(inBoundarySize, 2 * inBoundarySize),
97  Slice(outShape.cols - inBoundarySize, outShape.cols)));
98 
99  const uint32 upperRowStart = outShape.rows - 2 * inBoundarySize;
100  NdArray<dtype> upperLeft = flipud(outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
101  Slice(0, inBoundarySize)));
102  NdArray<dtype> upperRight = flipud(outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
103  Slice(outShape.cols - inBoundarySize, outShape.cols)));
104 
105  outArray.put(Slice(0, inBoundarySize), Slice(0, inBoundarySize), lowerLeft);
106  outArray.put(Slice(0, inBoundarySize), Slice(outShape.cols - inBoundarySize, outShape.cols), lowerRight);
107  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
108  Slice(0, inBoundarySize), upperLeft);
109  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
110  Slice(outShape.cols - inBoundarySize, outShape.cols), upperRight);
111 
112  return outArray;
113  }
114  } // namespace boundary
115  } // namespace filter
116 } // namespace nc
StaticAsserts.hpp
nc::NdArray::shape
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4356
STATIC_ASSERT_ARITHMETIC
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:37
nc::NdArray< dtype >
nc::uint32
std::uint32_t uint32
Definition: Types.hpp:40
NdArray.hpp
nc::Shape
A Shape Class for NdArrays.
Definition: Core/Shape.hpp:40
nc::NdArray::put
NdArray< dtype > & put(int32 inIndex, value_type inValue)
Definition: NdArrayCore.hpp:3729
nc::Shape::cols
uint32 cols
Definition: Core/Shape.hpp:45
Shape.hpp
nc
Definition: Coordinate.hpp:44
nc::Shape::rows
uint32 rows
Definition: Core/Shape.hpp:44
nc::flipud
NdArray< dtype > flipud(const NdArray< dtype > &inArray)
Definition: flipud.hpp:48
flipud.hpp
Types.hpp
nc::Slice
A Class for slicing into NdArrays.
Definition: Slice.hpp:43
nc::filter::boundary::reflect2d
NdArray< dtype > reflect2d(const NdArray< dtype > &inImage, uint32 inBoundarySize)
Definition: reflect2d.hpp:54
Slice.hpp