code_device/hgdriver/ImageProcess/ImageApplySplit.cpp

#include "ImageApplySplit.h"
#include <vector>

#define BPP(type,index) 
CImageApplySplit::CImageApplySplit(int multitype,bool split, bool ismulti_filter_red,int colormode) :
    m_bmulti_filter_red(ismulti_filter_red)
    , m_split(split)
	, m_multitype(multitype)
	, m_colormode(colormode)
{
}

CImageApplySplit::~CImageApplySplit(void)
{
}

std::vector<MatEx> CImageApplySplit::SplitMats(std::vector<cv::Mat>& mats, bool isTwoSide)
{
	std::vector<MatEx> rets;
	for (size_t i = 0; i < mats.size(); i++)
	{
		if (mats[i].empty())
			continue;
		if (i != 0 && isTwoSide == false)
			break;

		int bpp = getBpp(i);
		if (m_split)//<2F><><EFBFBD><EFBFBD>
		{
			std::vector<cv::Mat> retmats = apply(mats[i]);
			if (bpp != -1) {
			}
			else {//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
				if (m_colormode == 0) bpp = 1;//bw
				else if (m_colormode == 1) bpp = 8;
				else bpp = 24;
			}

			for (size_t j = 0; j < retmats.size(); j++)
			{
				if (!retmats[j].empty()) {
					MatEx matex(retmats[j], bpp);
					rets.push_back(matex);
				}
			}
		}
		else {
			MatEx matex(mats[i], bpp);
			rets.push_back(matex);
		}
	}
	return rets;
}

std::vector<cv::Mat> CImageApplySplit::apply(cv::Mat& pDib)
{
	if (pDib.empty())
		return std::vector<cv::Mat>();
	std::vector<cv::Mat> retMats;
	int heigth = pDib.rows;
	int width = pDib.cols;
	if (heigth > width)
	{
		cv::Mat matF = pDib(cv::Rect(0, 0, width, (int)(0.5 * heigth)));
		cv::Mat matB = pDib(cv::Rect(0, (int)(0.5 * heigth), width, (int)(0.5 * heigth)));
		retMats.push_back(matF);
		retMats.push_back(matB);
	}
	else
	{
		cv::Mat matF = pDib(cv::Rect(0, 0, (int)(width*0.5), heigth));
		cv::Mat matB = pDib(cv::Rect((int)(width*0.5), 0, (int)(width * 0.5), heigth));
		retMats.push_back(matF);
		retMats.push_back(matB);
	}
	return retMats;
}

int CImageApplySplit::getBpp(int matIndex)
{
	int ret = -1;
	if (m_bmulti_filter_red) {
		ret = matIndex == 0 ? 24 : 8;
	}
	else
	{
		if (m_multitype == -1)
			return ret;
		switch (m_multitype)
		{
		case 0://all
			if (matIndex == 0) ret = 24;
			else if (matIndex == 1) ret = 8;
			else ret = 1;
			break;
		case 1://clolr +gray
			if (matIndex == 0) ret = 24;
			else ret = 8;
			break;
		case 2://color+bw
			if (matIndex == 0) ret = 24;
			else ret = 1;
			break;
		case 3://gray+bw
			if (matIndex == 0) ret = 8;
			else ret = 1;
			break;
		default:
			break;
		}
	}
	return ret;
}