NumCpp  2.4.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:
52  template<typename dtype>
53  NdArray<dtype> wrap2d(const NdArray<dtype>& inImage, uint32 inBoundarySize)
54  {
56 
57  const Shape inShape = inImage.shape();
58  Shape outShape(inShape);
59  outShape.rows += inBoundarySize * 2;
60  outShape.cols += inBoundarySize * 2;
61 
62  NdArray<dtype> outArray(outShape);
63  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
64  Slice(inBoundarySize, inBoundarySize + inShape.cols), inImage);
65 
66  // bottom
67  outArray.put(Slice(0, inBoundarySize),
68  Slice(inBoundarySize, inBoundarySize + inShape.cols),
69  inImage(Slice(inShape.rows - inBoundarySize, inShape.rows), Slice(0, inShape.cols)));
70 
71  // top
72  outArray.put(Slice(inShape.rows + inBoundarySize, outShape.rows),
73  Slice(inBoundarySize, inBoundarySize + inShape.cols),
74  inImage(Slice(0, inBoundarySize), Slice(0, inShape.cols)));
75 
76  // left
77  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
78  Slice(0, inBoundarySize),
79  inImage(Slice(0, inShape.rows), Slice(inShape.cols - inBoundarySize, inShape.cols)));
80 
81  // right
82  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
83  Slice(inShape.cols + inBoundarySize, outShape.cols),
84  inImage(Slice(0, inShape.rows), Slice(0, inBoundarySize)));
85 
86  // now fill in the corners
87  NdArray<dtype> lowerLeft = outArray(Slice(inBoundarySize, 2 * inBoundarySize),
88  Slice(0, inBoundarySize));
89  NdArray<dtype> lowerRight = outArray(Slice(inBoundarySize, 2 * inBoundarySize),
90  Slice(outShape.cols - inBoundarySize, outShape.cols));
91 
92  const uint32 upperRowStart = outShape.rows - 2 * inBoundarySize;
93  NdArray<dtype> upperLeft = outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
94  Slice(0, inBoundarySize));
95  NdArray<dtype> upperRight = outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
96  Slice(outShape.cols - inBoundarySize, outShape.cols));
97 
98  outArray.put(Slice(0, inBoundarySize), Slice(0, inBoundarySize), upperLeft);
99  outArray.put(Slice(0, inBoundarySize), Slice(outShape.cols - inBoundarySize, outShape.cols), upperRight);
100  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
101  Slice(0, inBoundarySize), lowerLeft);
102  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
103  Slice(outShape.cols - inBoundarySize, outShape.cols), lowerRight);
104 
105  return outArray;
106  }
107  } // namespace boundary
108  } // namespace filter
109 } // namespace nc
StaticAsserts.hpp
nc::NdArray::shape
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4356
nc::filter::boundary::wrap2d
NdArray< dtype > wrap2d(const NdArray< dtype > &inImage, uint32 inBoundarySize)
Definition: wrap2d.hpp:53
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
fillCorners.hpp
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
Types.hpp
nc::Slice
A Class for slicing into NdArrays.
Definition: Slice.hpp:43
Slice.hpp