NumCpp  2.10.1
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::filter
39 {
40  //============================================================================
41  // Method Description:
53  template<typename dtype>
55  const NdArray<dtype>& inWeights,
56  Boundary inBoundaryType = Boundary::REFLECT,
57  dtype inConstantValue = 0)
58  {
59  const uint32 boundarySize = inWeights.size() / 2; // integer division
60  NdArray<dtype> arrayWithBoundary =
61  boundary::addBoundary1d(inImageArray, inBoundaryType, inWeights.size(), inConstantValue);
62  NdArray<dtype> output(1, inImageArray.size());
63 
64  NdArray<dtype> weightsFlat = fliplr(inWeights.flatten());
65 
66  const uint32 endPointRow = boundarySize + inImageArray.size();
67 
68  for (uint32 i = boundarySize; i < endPointRow; ++i)
69  {
70  NdArray<dtype> window = arrayWithBoundary[Slice(i - boundarySize, i + boundarySize + 1)].flatten();
71 
72  output[i - boundarySize] = dot(window, weightsFlat).item();
73  }
74 
75  return output;
76  }
77 } // namespace nc::filter
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:138
size_type size() const noexcept
Definition: NdArrayCore.hpp:4415
self_type flatten() const
Definition: NdArrayCore.hpp:2770
A Class for slicing into NdArrays.
Definition: Slice.hpp:45
NdArray< dtype > addBoundary1d(const NdArray< dtype > &inImage, Boundary inBoundaryType, uint32 inKernalSize, dtype inConstantValue=0)
Definition: addBoundary1d.hpp:56
Definition: addBoundary1d.hpp:44
NdArray< dtype > convolve1d(const NdArray< dtype > &inImageArray, const NdArray< dtype > &inWeights, Boundary inBoundaryType=Boundary::REFLECT, dtype inConstantValue=0)
Definition: convolve1d.hpp:54
Boundary
Boundary condition to apply to the image filter.
Definition: Boundary.hpp:36
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