图像填充颜色算法

图像操作中经常需要填充颜色，俗称“油漆桶”。

实现算法的关键是查找连通的同色像素，这里需要用到泛洪算法。

先定义个结构体：

typedef struct ImagePoint 
{
	UINT x, y;
}IMAGEPOINT;

实现的数据结构很多，队列、链表都可以，这里采用简单的数组实现：

void FillColor(BYTE* src, BYTE* dst, UINT width, UINT height, UINT x0, UINT y0, UINT new_color,UINT old_color,UINT set_step_color)  
{
	int m_WidthBytes = width*4;
	UINT a_new[3],a_read[3],a_old[3];
	
	a_new[0]	=new_color/65536%256;	//B
	a_new[1]	=new_color/256%256;		//G
	a_new[2]	=new_color%256;			//R
	
	//修改单击点颜色
	for(UINT k=0;k<3;k++)
		dst[y0*m_WidthBytes+x0*4+k] = a_new[k];

	a_old[0]	=old_color/65536%256;	//B
	a_old[1]	=old_color/256%256;		//G
	a_old[2]	=old_color%256;			//R
	
	//像素数组
	IMAGEPOINT * Points=new IMAGEPOINT[width*height];
	memset(Points,0,width*height);
	int n_p_begin=0,n_p_end=0;
	Points[0].x=x0;Points[0].y=y0;
	int a_directionX[4]={-1,0,1,0},a_directionY[4]={0,-1,0,1};
	//泛洪算法
	while(n_p_begin<=n_p_end)
	{
		//取像素
		IMAGEPOINT p1=Points[n_p_end--];
		int x=p1.x;
		int y=p1.y;
		for(int i=0;i<4;i++)
		{
			UINT x1=x+a_directionX[i];
			UINT y1=y+a_directionY[i];
			if(x1<0 || x1>=width || y1<0 || y1>=height)	continue;
			//读取颜色
			bool bContinue=true;
			for(int k=0;k<3 && bContinue;k++)
			{
				a_read[k]=(UINT)dst[y1*m_WidthBytes+x1*4+k];
				if(set_step_color==0)	//精确比对
				{
					if(a_read[k]!=a_old[k])
						bContinue=false;
				}
				else
				{
					//色差比对
					int test_step_color =abs(a_read[k]-a_old[k]);//色差
					if(test_step_color>set_step_color) 
						bContinue=false;		
				}
			}
			//同色
			if(bContinue)
			{
				//改成新颜色
				for(int k=0;k<3;k++)
					dst[y1*m_WidthBytes+x1*4+k]=a_new[k];
				//存入数组
				Points[++n_p_end].x=x1;Points[n_p_end].y=y1;
			}
		}
	}
	free(Points);
}

效果如下图：

算法在n_p_end小于n_p_begin终结。

代码中：a_directionX、a_directionY用于计算像素的上下左右4个相邻像素坐标，用循环处理比较方便。

考虑到相近颜色的填充需求，代码中设置色差set_step_color干预，允许RGB值有一定的值的浮动。