opencv 截取任意四边形区域的图像

截取任意四边形区域的图像。

mask就是结果。

需要定义四边形区域，分别是 tl, tr, bl, br

std::map<int, std::set<int>> generateBorders(const std::vector<cv::Point> & vecPts)
{
	std::map<int, std::set<int>> borders;
	cv::Point tl(vecPts[0]), tr(vecPts[1]), bl(vecPts[2]), br(vecPts[3]);

	// tl to tr
	double K = double(tl.y-tr.y) / (tl.x - tr.x);
	for (int i = tl.x; i < tr.x; ++i)
		borders[i].insert(cvRound(-K*(tl.x - i) + tl.y));

	// tr to br
	if (tr.x != br.x)
	{
		K = double(tr.y - br.y) / (tr.x - br.x);
		for (int i = tr.x; i < br.x; ++i)
			borders[i].insert(cvRound(-K*(tr.x - i) + tr.y));
	}

	// br to bl
	K = double(br.y - bl.y) / (br.x - bl.x);
	for (int i = bl.x; i < br.x; ++i)
		borders[i].insert(cvRound(-K*(br.x - i) + br.y));

	// bl to tl
	if (bl.x != tl.x)
	{
		K = double(bl.y - tl.y) / (bl.x - tl.x);
		for (int i = tl.x; i < bl.x; ++i)
			borders[i].insert(cvRound(-K*(bl.x - i) + bl.y));
	}

	for (auto it = borders.begin(); it!=borders.end(); ++it)
	{
		if ((*it).second.size() == 2)
			continue;

		std::set<int> newone = { *(*it).second.begin(), *(--(*it).second.end())};
		(*it).second.swap(newone);
	}

#ifdef DEBUG_CODE
	cv::Mat disp = cv::imread("1111.PNG", CV_LOAD_IMAGE_COLOR);

	for (auto it = borders.begin(); it != borders.end(); ++it)
	{
		cv::circle(disp, cv::Point((*it).first, *(*it).second.begin()), 1, cv::Scalar(255, 255, 0), -1);
		cv::circle(disp, cv::Point((*it).first, *(--(*it).second.end())), 1, cv::Scalar(0, 255, 255), -1);

	}

	cv::circle(disp, tl, 2, cv::Scalar(0, 0, 255), -1);
	cv::circle(disp, tr, 2, cv::Scalar(0, 0, 255), -1);
	cv::circle(disp, bl, 2, cv::Scalar(0, 0, 255), -1);
	cv::circle(disp, br, 2, cv::Scalar(0, 0, 255), -1);
#endif // DEBUG_CODE

	return borders;
}

cv::Mat generateMask(const cv::Mat & src,
	const std::vector<cv::Point> & vecPts)
{
	cv::Mat mask = cv::Mat::zeros(src.size(), CV_8UC1);

	cv::Point tl(vecPts[0]), tr(vecPts[1]), bl(vecPts[2]), br(vecPts[3]);
	std::map<int, std::set<int>> borders = generateBorders(vecPts);

	int minX = std::min(tl.x, bl.x),
		maxX = std::max(tr.x, br.x),
		minY = std::min(tl.y, tr.y),
		maxY = std::max(bl.y, br.y);

	uchar minZ = std::min(src.at<uchar>(tl), src.at<uchar>(tr));
	minZ = std::min(minZ, src.at<uchar>(bl));
	minZ = std::min(minZ, src.at<uchar>(br));

	for (size_t j= minY; j<maxY; ++j)
	{
		const uchar* pS = src.ptr<uchar>(j);
		uchar* pM = mask.ptr<uchar>(j);

		for (size_t i = minX; i < maxX; ++i)
		{
			// in the region.
			if (*borders[i].begin() < j && j < *(++borders[i].begin()))
			{
				pM[i] = pS[i];
			}
		}
	}

	return mask;

}