NumCpp  1.0
A C++ implementation of the Python Numpy library
wrap2d.hpp
Go to the documentation of this file.
1 #pragma once
30 
31 #include "NumCpp/NdArray.hpp"
32 #include "NumCpp/Core/Shape.hpp"
33 #include "NumCpp/Core/Slice.hpp"
34 #include "NumCpp/Core/Types.hpp"
37 
38 namespace nc
39 {
40  namespace filter
41  {
42  namespace boundary
43  {
44  //============================================================================
45  // Method Description:
53  template<typename dtype>
54  NdArray<dtype> wrap2d(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  // bottom
68  outArray.put(Slice(0, inBoundarySize),
69  Slice(inBoundarySize, inBoundarySize + inShape.cols),
70  inImage(Slice(inShape.rows - inBoundarySize, inShape.rows), Slice(0, inShape.cols)));
71 
72  // top
73  outArray.put(Slice(inShape.rows + inBoundarySize, outShape.rows),
74  Slice(inBoundarySize, inBoundarySize + inShape.cols),
75  inImage(Slice(0, inBoundarySize), Slice(0, inShape.cols)));
76 
77  // left
78  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
79  Slice(0, inBoundarySize),
80  inImage(Slice(0, inShape.rows), Slice(inShape.cols - inBoundarySize, inShape.cols)));
81 
82  // right
83  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
84  Slice(inShape.cols + inBoundarySize, outShape.cols),
85  inImage(Slice(0, inShape.rows), Slice(0, inBoundarySize)));
86 
87  // now fill in the corners
88  NdArray<dtype> lowerLeft = outArray(Slice(inBoundarySize, 2 * inBoundarySize),
89  Slice(0, inBoundarySize));
90  NdArray<dtype> lowerRight = outArray(Slice(inBoundarySize, 2 * inBoundarySize),
91  Slice(outShape.cols - inBoundarySize, outShape.cols));
92 
93  const uint32 upperRowStart = outShape.rows - 2 * inBoundarySize;
94  NdArray<dtype> upperLeft = outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
95  Slice(0, inBoundarySize));
96  NdArray<dtype> upperRight = outArray(Slice(upperRowStart, upperRowStart + inBoundarySize),
97  Slice(outShape.cols - inBoundarySize, outShape.cols));
98 
99  outArray.put(Slice(0, inBoundarySize), Slice(0, inBoundarySize), upperLeft);
100  outArray.put(Slice(0, inBoundarySize), Slice(outShape.cols - inBoundarySize, outShape.cols), upperRight);
101  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
102  Slice(0, inBoundarySize), lowerLeft);
103  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
104  Slice(outShape.cols - inBoundarySize, outShape.cols), lowerRight);
105 
106  return outArray;
107  }
108  }
109  }
110 }
StaticAsserts.hpp
nc::NdArray::shape
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4296
nc::filter::boundary::wrap2d
NdArray< dtype > wrap2d(const NdArray< dtype > &inImage, uint32 inBoundarySize)
Definition: wrap2d.hpp:54
STATIC_ASSERT_ARITHMETIC
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:38
nc::NdArray< dtype >
nc::uint32
std::uint32_t uint32
Definition: Types.hpp:41
NdArray.hpp
nc::Shape
A Shape Class for NdArrays.
Definition: Core/Shape.hpp:41
nc::NdArray::put
NdArray< dtype > & put(int32 inIndex, value_type inValue)
Definition: NdArrayCore.hpp:3667
fillCorners.hpp
nc::Shape::cols
uint32 cols
Definition: Core/Shape.hpp:46
Shape.hpp
nc
Definition: Coordinate.hpp:45
nc::Shape::rows
uint32 rows
Definition: Core/Shape.hpp:45
Types.hpp
nc::Slice
A Class for slicing into NdArrays.
Definition: Slice.hpp:44
Slice.hpp