#include "ImageApplyFilter.h"

CImageApplyFilter::CImageApplyFilter(FilterMode type)
	: m_type(type)
{
	m_kSize = (m_type == FilterMode::Sharpen || m_type == FilterMode::AverBlur) ? 5 : 9;
}

CImageApplyFilter::~CImageApplyFilter()
{
}

void CImageApplyFilter::apply(cv::Mat & pDib, int side)
{
#ifdef LOG
	FileTools::write_log("imgprc.txt", "enter CImageApplySharpen apply");
#endif // LOG

	switch (m_type)
	{
	case FilterMode::Sharpen:
	case FilterMode::Sharpen_More:
		sharpen(pDib, m_kSize);
		break;
	case FilterMode::AverBlur:
	case FilterMode::AverBlur_More:
		averblur(pDib, static_cast<int>(m_kSize));
		break;
	}

#ifdef LOG
	FileTools::write_log("imgprc.txt", "exit CImageApplySharpen apply");
#endif // LOG
}

void CImageApplyFilter::apply(std::vector<cv::Mat>& mats, bool isTwoSide)
{
	if (mats.empty()) return;

	if (!mats[0].empty())
		apply(mats[0], 0);

	if (isTwoSide && mats.size() > 1) 
	{
		if (!mats[1].empty())
			apply(mats[1], 1);
	}
}

void CImageApplyFilter::averblur(cv::Mat& src, int kSize)
{
	cv::blur(src, src, cv::Size(kSize, kSize));
}

void CImageApplyFilter::sharpen(cv::Mat& src, float kSize)
{
	float other = (1.0f - kSize) / 4;
	float kernel_data[] = { 0, other, 0, other, kSize, other, 0, other, 0 };
	cv::Mat kernel(3, 3, CV_32FC1, kernel_data);

	cv::filter2D(src, src, src.depth(), kernel);
}