NumCpp  2.7.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
wrap2d.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:
51  template<typename dtype>
52  NdArray<dtype> wrap2d(const NdArray<dtype>& inImage, uint32 inBoundarySize)
53  {
55 
56  const Shape inShape = inImage.shape();
57  Shape outShape(inShape);
58  outShape.rows += inBoundarySize * 2;
59  outShape.cols += inBoundarySize * 2;
60 
61  NdArray<dtype> outArray(outShape);
62  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
63  Slice(inBoundarySize, inBoundarySize + inShape.cols), inImage);
64 
65  // bottom
66  outArray.put(Slice(0, inBoundarySize),
67  Slice(inBoundarySize, inBoundarySize + inShape.cols),
68  inImage(Slice(inShape.rows - inBoundarySize, inShape.rows), Slice(0, inShape.cols)));
69 
70  // top
71  outArray.put(Slice(inShape.rows + inBoundarySize, outShape.rows),
72  Slice(inBoundarySize, inBoundarySize + inShape.cols),
73  inImage(Slice(0, inBoundarySize), Slice(0, inShape.cols)));
74 
75  // left
76  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
77  Slice(0, inBoundarySize),
78  inImage(Slice(0, inShape.rows), Slice(inShape.cols - inBoundarySize, inShape.cols)));
79 
80  // right
81  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
82  Slice(inShape.cols + inBoundarySize, outShape.cols),
83  inImage(Slice(0, inShape.rows), Slice(0, inBoundarySize)));
84 
85  // now fill in the corners
86  NdArray<dtype> lowerLeft = outArray(Slice(inBoundarySize, 2 * inBoundarySize),
87  Slice(0, inBoundarySize));
88  NdArray<dtype> lowerRight = outArray(Slice(inBoundarySize, 2 * inBoundarySize),
89  Slice(outShape.cols - inBoundarySize, outShape.cols));
90 
91  const uint32 upperRowStart = outShape.rows - 2 * inBoundarySize;
92  NdArray<dtype> upperLeft = outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
93  Slice(0, inBoundarySize));
94  NdArray<dtype> upperRight = outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
95  Slice(outShape.cols - inBoundarySize, outShape.cols));
96 
97  outArray.put(Slice(0, inBoundarySize), Slice(0, inBoundarySize), upperLeft);
98  outArray.put(Slice(0, inBoundarySize), Slice(outShape.cols - inBoundarySize, outShape.cols), upperRight);
99  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
100  Slice(0, inBoundarySize), lowerLeft);
101  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
102  Slice(outShape.cols - inBoundarySize, outShape.cols), lowerRight);
103 
104  return outArray;
105  }
106  } // namespace boundary
107  } // namespace filter
108 } // namespace nc
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:37
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
NdArray< dtype > & put(int32 inIndex, value_type inValue)
Definition: NdArrayCore.hpp:3666
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4283
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:44
NdArray< dtype > wrap2d(const NdArray< dtype > &inImage, uint32 inBoundarySize)
Definition: wrap2d.hpp:52
Definition: Coordinate.hpp:45
std::uint32_t uint32
Definition: Types.hpp:40