meanFilter.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "NumCpp/Core/Shape.hpp"
#include "NumCpp/Core/Slice.hpp"
#include "NumCpp/Core/Types.hpp"
#include "NumCpp/Filter/Boundaries/Boundaries2d/addBoundary2d.hpp"
#include "NumCpp/Functions/mean.hpp"
#include "NumCpp/NdArray.hpp"
 
namespace nc::filter
{
    //============================================================================
    // Method Description:
    template<typename dtype>
    NdArray<double> meanFilter(const NdArray<dtype>& inImageArray,
                               uint32                inSize,
                               Boundary              inBoundaryType  = Boundary::REFLECT,
                               dtype                 inConstantValue = 0)
    {
        NdArray<dtype> arrayWithBoundary =
            boundary::addBoundary2d(inImageArray, inBoundaryType, inSize, inConstantValue);
        NdArray<double> output(inImageArray.shape());
 
        const Shape  inShape      = inImageArray.shape();
        const uint32 boundarySize = inSize / 2; // integer division
        const uint32 endPointRow  = boundarySize + inShape.rows;
        const uint32 endPointCol  = boundarySize + inShape.cols;
 
        for (uint32 row = boundarySize; row < endPointRow; ++row)
        {
            for (uint32 col = boundarySize; col < endPointCol; ++col)
            {
                NdArray<dtype> window = arrayWithBoundary(Slice(row - boundarySize, row + boundarySize + 1),
                                                          Slice(col - boundarySize, col + boundarySize + 1));
 
                output(row - boundarySize, col - boundarySize) = mean(window).item();
            }
        }
 
        return output;
    }
} // namespace nc::filter