【OpenCV】示例—漫水填充

序言： 这次的示例程序类似于PS中的魔棒工具效果，可以给鼠标点击的地方改变颜色。

• 鼠标点击图中区域 -漫水填充操作
• 按键【ESC】-退出程序
• 按键【1】-进行彩色图/灰色图切换
• 按键【2】-显示/隐藏掩膜窗口
• 按键【3】-恢复原始图像
• 按键【4】-使用空范围的漫水填充
• 按键【5】-使用渐变、固定范围的漫水填充
• 按键【6】-使用渐变、浮动范围的漫水填充
• 按键【7】-操作标识符的低八位使用4位的连接模式
• 按键【8】-操作标识符的低八位使用8位的连接模式

程序代码:

#include<opencv2/imgproc/imgproc.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<iostream>
using namespace std;
using namespace cv;

//------------------------------------
//	全局变量声明
//------------------------------------
Mat g_srcImage, g_dstImage, g_grayImage, g_maskImage;//定义原始图、目标图、灰度图、掩膜图
int g_nFillMode = 1;//漫水填充的模式
int g_nLowDifference = 20, g_nUpDifference = 20;//负值最大值，正差最大值
int g_nConnectivity = 4;//表示floodFill函数标识符低八位的连通值
int g_bIsColor = true;//是否为彩色图的标识符布尔值
bool g_bUseMask = false;//是否显示掩膜窗口的布尔值
int g_nNewMaskVal = 255;//新的重新绘制的像素值

//---------------------------------
//	鼠标消息onMouse回调函数 
//---------------------------------
static void onMouse(int event, int x, int y, int, void*)
{
	//若鼠标左键按下，便返回
	if (event != EVENT_LBUTTONDOWN)
		return;
	//-------1、调用floodFill函数之前的参数准备部分
	Point seed = Point(x, y);
	//空范围的漫水填充，此值设为0，否则设为全局变量g_nLowDifference
	int LowDifference = g_nFillMode == 0 ? 0 : g_nLowDifference;

	//空范围的漫水填充，此值设为0，否则设为全局变量g_nUpDifference
	int UpDifference = g_nFillMode == 0 ? 0 : g_nUpDifference;

	//标识符的0~7位为g_nConnectivity,8~15位为g_nNewMaskVal左移八位的值，
	//16~32位为CV_FLOODFILL_FIXED_RANGE或者0
	int flags = g_nConnectivity + (g_nNewMaskVal << 8) + (g_nFillMode == 1 ? FLOODFILL_FIXED_RANGE : 0);

	//随机生成bgr值
	int b = (unsigned)theRNG() & 255;//随机返回一个0~255之间的值
	int g = (unsigned)theRNG() & 255;//随机返回一个0~255之间的值
	int r = (unsigned)theRNG() & 255;//随机返回一个0~255之间的值
	Rect ccomp;//重绘制区域的最小边界矩形区域

	//在重绘区域像素的新值，若是彩色图模式，取Scalar(b,g,r);若是灰度图，取
	//Scalar(r*0.299 + g * 0.587 + b * 0.114)
	Scalar newVal = g_bIsColor ? Scalar(b, g, r) : Scalar(r*0.299 + g * 0.587 + b * 0.114);
	Mat dst = g_bIsColor ? g_dstImage : g_grayImage;//目标赋值
	int area;

	//2、正式调用floodFill函数
	if (g_bUseMask)
	{
		threshold(g_maskImage, g_maskImage, 1, 128, THRESH_BINARY);
		area = floodFill(dst, g_maskImage, seed, newVal, &ccomp, Scalar(LowDifference, 
		LowDifference, LowDifference),
			Scalar(UpDifference, UpDifference, UpDifference), flags);
		imshow("mask", g_maskImage);
	}
	else
	{
		area = floodFill(dst, seed, newVal, &ccomp, Scalar(LowDifference, LowDifference,
		 LowDifference),
			Scalar(UpDifference, UpDifference, UpDifference), flags);
	}
	imshow("效果图", dst);
	cout << area << "个像素被重绘\n";
}
int main(int argc, char** argv)
{
	system("color 2F");
	//载入原图像
	g_srcImage = imread("E:\\Pec\\蜘蛛侠.jpg", 1);
	if (!g_srcImage.data)
	{
		cout << "读取图片错误" << endl;
		return false;
	}
	//复制原图到目标图
	g_srcImage.copyTo(g_dstImage);
	//转换三通道的image0到灰度图
	cvtColor(g_srcImage, g_grayImage, COLOR_BGR2GRAY);
	//利用image0的尺寸来初始化掩膜mask
	g_maskImage.create(g_srcImage.rows + 2, g_srcImage.cols + 2, CV_8UC1);
	namedWindow("效果图", WINDOW_AUTOSIZE);

	//创建Trackbar
	createTrackbar("负差最大值", "效果图", &g_nLowDifference, 255, 0);
	createTrackbar("正差最大值", "效果图", &g_nUpDifference, 255, 0);

	//鼠标回调函数
	setMouseCallback("效果图", onMouse, 0);

	//循环轮询按键
	while (1)
	{
		//先显示效果图
		imshow("效果图", g_bIsColor ? g_dstImage : g_grayImage);
		//获取键盘按键
		int c = waitKey(0);
		//判ESC是否按下，若按下便退出
		if ((c & 255) == 27)
		{
			cout << "程序退出......" << endl;
			break;
		}
		//根据按键的不同，进行各种操作
		switch ((char)c)
		{
			//如果键盘“1”被按下，效果图在灰度图，彩色图之间互换
		case '1':
			if (g_bIsColor)//若原来为彩色，则转换为灰度图，并且将掩膜mask所有元素设置为0
			{
				cout << "当前操作【彩色模式】切换为【灰度模式" << endl;
				cvtColor(g_srcImage, g_grayImage, COLOR_BGR2GRAY);
				g_maskImage = Scalar::all(0);//将msak所有元素设置为0
				g_bIsColor = false;//将标识符设置为false，表示当前图像不为彩色，而是灰度

			}
			else//若原来为灰度图，便将原来的彩图image0再次复制给image，并且将掩膜mask所有元素设置为0
			{
				cout << "【彩色模式】切换【灰度模式】" << endl;
				g_srcImage.copyTo(g_dstImage);
				g_maskImage = Scalar::all(0);
				g_bIsColor = true;//将标识符设置为true，表示当前的图像模式为彩色

			}
			break;
		case '2'://如果按下2，显示/隐藏掩膜窗口
			if (g_bUseMask)
			{
				destroyWindow("mask");
				g_bUseMask = false;
			}
			else
			{
				namedWindow("mask", 0);
				g_maskImage = Scalar::all(0);
				imshow("mask", g_maskImage);
				g_bUseMask = true;
			}
			break;
		case '3': //恢复原始图像
			cout << "恢复原始图像" << endl;
			g_srcImage.copyTo(g_dstImage);
			cvtColor(g_dstImage, g_grayImage, COLOR_BGR2GRAY);
			g_maskImage = Scalar::all(0);
			break;
		case '4'://使用空范围的漫水填充
			cout << "使用空范围的漫水填充" << endl;
			g_nFillMode = 0;
			break;
		case '5':
			cout << "使用渐变、固定范围的漫水填充" << endl;
			g_nFillMode = 1;
			break;
		case '6':
			cout << "使用渐变、浮动范围的漫水填充" << endl;
			g_nFillMode = 2;
			break;
		case '7'://操作标识符的低八位使用4位的连接模式
			g_nConnectivity = 4;
			break;
		case '8'://操作标识符的低八位使用8位的连接模式
			g_nConnectivity = 8;
			break;
		}
	}
	return 0;
}

未操作图像：

鼠标点击漫水填充图像：

掩膜描绘窗口