100 lines
2.0 KiB
C++
100 lines
2.0 KiB
C++
#include "StdAfx.h"
|
|
#include "ImageApplyChannel.h"
|
|
|
|
CImageApplyChannel::CImageApplyChannel()
|
|
: m_channel(Invalid)
|
|
{
|
|
}
|
|
|
|
CImageApplyChannel::CImageApplyChannel(Channel channel)
|
|
: m_channel(channel)
|
|
{
|
|
}
|
|
|
|
CImageApplyChannel::~CImageApplyChannel(void)
|
|
{
|
|
}
|
|
|
|
void CImageApplyChannel::apply(cv::Mat& pDib,int side)
|
|
{
|
|
#ifdef LOG
|
|
FileTools::write_log("imgprc.txt", "enter CImageApplyChannel apply");
|
|
#endif // LOG
|
|
if (pDib.empty())
|
|
{
|
|
#ifdef LOG
|
|
FileTools::write_log("imgprc.txt", "exit CImageApplyChannel apply");
|
|
#endif // LOG
|
|
return;
|
|
}
|
|
|
|
cv::Mat dst(pDib.rows, pDib.cols, CV_8UC1);
|
|
switch (m_channel)
|
|
{
|
|
case Red:
|
|
cv::extractChannel(pDib, dst, 2);
|
|
break;
|
|
case Green:
|
|
cv::extractChannel(pDib, dst, 1);
|
|
break;
|
|
case Blue:
|
|
cv::extractChannel(pDib, dst, 0);
|
|
break;
|
|
case Except_Red:
|
|
except_channel(pDib, dst, 2);
|
|
break;
|
|
case Except_Green:
|
|
except_channel(pDib, dst, 1);
|
|
break;
|
|
case Except_Blue:
|
|
except_channel(pDib, dst, 0);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
pDib.release();
|
|
pDib = dst;
|
|
#ifdef LOG
|
|
FileTools::write_log("imgprc.txt", "exit CImageApplyChannel apply");
|
|
#endif // LOG
|
|
}
|
|
|
|
void CImageApplyChannel::apply(std::vector<cv::Mat>& mats, bool isTwoSide)
|
|
{
|
|
if (mats.empty()) return;
|
|
|
|
apply(mats[0], 0);
|
|
|
|
if (isTwoSide && mats.size() > 1)
|
|
apply(mats[1], 1);
|
|
}
|
|
|
|
void CImageApplyChannel::except_channel(const cv::Mat & src, cv::Mat & dst, int channel)
|
|
{
|
|
int rows = src.total();
|
|
cv::Mat src_temp(rows, 3, CV_8UC1, src.data);
|
|
cv::Mat dst_temp(rows, 1, CV_8UC1, dst.data);
|
|
|
|
cv::Mat temp1, temp2;
|
|
switch (channel)
|
|
{
|
|
case 0:
|
|
temp1 = src_temp(cv::Rect(1, 0, 1, rows));
|
|
temp2 = src_temp(cv::Rect(2, 0, 1, rows));
|
|
cv::addWeighted(temp1, 0.84, temp2, 0.16, 0, dst_temp);
|
|
break;
|
|
case 1:
|
|
temp1 = src_temp(cv::Rect(0, 0, 1, rows));
|
|
temp2 = src_temp(cv::Rect(2, 0, 1, rows));
|
|
cv::addWeighted(temp1, 0.73, temp2, 0.27, 0, dst_temp);
|
|
break;
|
|
case 2:
|
|
temp1 = src_temp(cv::Rect(0, 0, 1, rows));
|
|
temp2 = src_temp(cv::Rect(1, 0, 1, rows));
|
|
cv::addWeighted(temp1, 0.33, temp2, 0.67, 0, dst_temp);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|