NumCpp  2.7.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
convolve1d.hpp
Go to the documentation of this file.
1 #pragma once
29 
30 #include "NumCpp/Core/Slice.hpp"
31 #include "NumCpp/Core/Types.hpp"
34 #include "NumCpp/Functions/dot.hpp"
36 #include "NumCpp/NdArray.hpp"
37 
38 namespace nc
39 {
40  namespace filter
41  {
42  //============================================================================
43  // Method Description:
54  template<typename dtype>
55  NdArray<dtype> convolve1d(const NdArray<dtype>& inImageArray, const NdArray<dtype>& inWeights,
56  Boundary inBoundaryType = Boundary::REFLECT, dtype inConstantValue = 0)
57  {
58  const uint32 boundarySize = inWeights.size() / 2; // integer division
59  NdArray<dtype> arrayWithBoundary = boundary::addBoundary1d(inImageArray, inBoundaryType, inWeights.size(), inConstantValue);
60  NdArray<dtype> output(1, inImageArray.size());
61 
62  NdArray<dtype> weightsFlat = fliplr(inWeights.flatten());
63 
64  const uint32 endPointRow = boundarySize + inImageArray.size();
65 
66  for (uint32 i = boundarySize; i < endPointRow; ++i)
67  {
68  NdArray<dtype> window = arrayWithBoundary[Slice(i - boundarySize, i + boundarySize + 1)].flatten();
69 
70  output[i - boundarySize] = dot(window, weightsFlat).item();
71  }
72 
73  return output;
74  }
75  } // namespace filter
76 } // namespace nc
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
size_type size() const noexcept
Definition: NdArrayCore.hpp:4296
NdArray< dtype > flatten() const
Definition: NdArrayCore.hpp:2759
A Class for slicing into NdArrays.
Definition: Slice.hpp:44
NdArray< dtype > addBoundary1d(const NdArray< dtype > &inImage, Boundary inBoundaryType, uint32 inKernalSize, dtype inConstantValue=0)
Definition: addBoundary1d.hpp:60
NdArray< dtype > convolve1d(const NdArray< dtype > &inImageArray, const NdArray< dtype > &inWeights, Boundary inBoundaryType=Boundary::REFLECT, dtype inConstantValue=0)
Definition: convolve1d.hpp:55
Boundary
Boundary condition to apply to the image filter.
Definition: Boundary.hpp:37
Definition: Coordinate.hpp:45
NdArray< dtype > dot(const NdArray< dtype > &inArray1, const NdArray< dtype > &inArray2)
Definition: dot.hpp:47
NdArray< dtype > fliplr(const NdArray< dtype > &inArray)
Definition: fliplr.hpp:46
std::uint32_t uint32
Definition: Types.hpp:40