NumCpp  2.8.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
constant2d.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> constant2d(const NdArray<dtype>& inImage, uint32 inBoundarySize, dtype inConstantValue)
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),
65  inImage);
66  fillCorners(outArray, inBoundarySize, inConstantValue);
67 
68  outArray.put(Slice(0, inBoundarySize),
69  Slice(inBoundarySize, inBoundarySize + inShape.cols),
70  inConstantValue);
71  outArray.put(Slice(outShape.rows - inBoundarySize, outShape.rows),
72  Slice(inBoundarySize, inBoundarySize + inShape.cols),
73  inConstantValue);
74  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
75  Slice(0, inBoundarySize),
76  inConstantValue);
77  outArray.put(Slice(inBoundarySize, inBoundarySize + inShape.rows),
78  Slice(outShape.cols - inBoundarySize, outShape.cols),
79  inConstantValue);
80 
81  return outArray;
82  }
83  } // namespace boundary
84  } // namespace filter
85 } // 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:3663
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4276
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 > constant2d(const NdArray< dtype > &inImage, uint32 inBoundarySize, dtype inConstantValue)
Definition: constant2d.hpp:53
void fillCorners(NdArray< dtype > &inArray, uint32 inBorderWidth)
Definition: fillCorners.hpp:50
Definition: Coordinate.hpp:45
std::uint32_t uint32
Definition: Types.hpp:40