code_device/hgdriver/ImageProcess/ImageApplyRefuseInflow.cpp

#include "ImageApplyRefuseInflow.h"

CImageApplyRefuseInflow::CImageApplyRefuseInflow(int d, int sigmaColor, int sigmaSpace)
    : m_d(d)
    , m_sigmaColor(sigmaColor)
    , m_sigmaSpace(sigmaSpace)
{
}

CImageApplyRefuseInflow::~CImageApplyRefuseInflow()
{
}

void CImageApplyRefuseInflow::apply(cv::Mat& pDib, int side)
{
    cv::Mat dst;
    cv::bilateralFilter(pDib, dst, m_d, m_sigmaColor, m_sigmaSpace);
    pDib = dst;
}

void CImageApplyRefuseInflow::apply(std::vector<cv::Mat>& mats, bool isTwoSide)
{
    (void)isTwoSide;

    int i = 0;
    for (cv::Mat& var : mats) {
        if (i != 0 && isTwoSide == false)
            break;
        if (!var.empty())
            apply(var, 0);
        i++;
    }
}