specialdetectionfilter.cpp

visual c++数字图像与图形处理中的光盘内容

	PIXELCOLORRGB rgb = Specialize(pbyRed, pbyGreen, pbyBlue, nNum);

	delete[] pbyBlue;
	delete[] pbyGreen;
	delete[] pbyRed;
	return rgb;
}


PIXELCOLORRGB CSpecialDetectionFilter::FilterPixelOnBorder(LPBYTE lpbyBitsSrc32,  int x,  int y,  int nScanWidth,  int nScanHeight)
{
	//卷积核元素的总个数;
	int nNum = m_nRows * m_nCols;

	//行字节数
	DWORD dwWidthBytes = (DWORD)nScanWidth * 4;

	//存储一个像素邻域的RGB三分量
	BYTE* pbyRed = new BYTE[nNum];
	BYTE* pbyGreen = new BYTE[nNum];
	BYTE* pbyBlue = new BYTE[nNum];

	//邻域中的左上角点.
	int xx = x - ((m_nCols - 1) / 2);
	int yy = y - ((m_nRows - 1) / 2);

	//三个数组pnRed, pnGreen, pnBlue的索引值
	int index = 0;

	//用嵌套循环获取小邻域数据.
	
	for(int i = 0;i < m_nRows;i++)
	{
		//y坐标
		int nY = yy + i;

		//复制边界
		nY = (nY < 0) ? 0 : ((nY > (nScanHeight - 1)) ? (nScanHeight - 1) : nY);

		//指针, 指向行数据
		BYTE* pbySrc = lpbyBitsSrc32 + ((DWORD)nY) * dwWidthBytes; 

		for(int j = 0;j < m_nCols;j++)
		{
			//x坐标	
			int nX = xx + j;

			//复制边界
			nX = (nX < 0) ? 0 : ((nX > (nScanWidth - 1)) ? (nScanWidth - 1) : nX);
			
			BYTE* pbyRaw = pbySrc + 4 * nX;
			
			//记录颜色分量
			pbyBlue[index] = *pbyRaw++; 
			pbyGreen[index] = *pbyRaw++;
			pbyRed[index] = *pbyRaw++;
			index++;
		}
	}

	//RGB颜色结构, 在 IMG.H 中定义
	//特殊计算
	PIXELCOLORRGB rgb = Specialize(pbyRed, pbyGreen, pbyBlue, nNum);

	delete[] pbyBlue;
	delete[] pbyGreen;
	delete[] pbyRed;
	return rgb;

}

PIXELCOLORRGB CSpecialDetectionFilter::Sobel(BYTE *pbyRed,  BYTE *pbyGreen,  BYTE *pbyBlue,  int nNum)
{
	int r0, g0, b0, r1, g1, b1, r2, g2, b2, r3, g3, b3;

	//水平划分
	r0 = ABS((((int)pbyRed[0] + (int)pbyRed[1] + (int)pbyRed[2])
			- ((int)pbyRed[6] + (int)pbyRed[7] + (int)pbyRed[8])) / 3);
	g0 = ABS((((int)pbyGreen[0] + (int)pbyGreen[1] + (int)pbyGreen[2])
			- ((int)pbyGreen[6] + (int)pbyGreen[7] + (int)pbyGreen[8])) / 3);
	b0 = ABS((((int)pbyBlue[0] + (int)pbyBlue[1] + (int)pbyBlue[2])
			- ((int)pbyBlue[6] + (int)pbyBlue[7] + (int)pbyBlue[8])) / 3);

	//垂直划分
	r1 = ABS((((int)pbyRed[0] + (int)pbyRed[3] + (int)pbyRed[6])
			- ((int)pbyRed[2] + (int)pbyRed[5] + (int)pbyRed[8])) / 3);
	g1 = ABS((((int)pbyGreen[0] + (int)pbyGreen[3] + (int)pbyGreen[6])
			- ((int)pbyGreen[2] + (int)pbyGreen[5] + (int)pbyGreen[8])) / 3);
	b1 = ABS((((int)pbyBlue[0] + (int)pbyBlue[3] + (int)pbyBlue[6])
			- ((int)pbyBlue[2] + (int)pbyBlue[5] + (int)pbyBlue[8])) / 3);

	//右对角划分
	r2 = ABS((((int)pbyRed[0] + (int)pbyRed[1] + (int)pbyRed[3])
			- ((int)pbyRed[5] + (int)pbyRed[7] + (int)pbyRed[8])) / 3);
	g2 = ABS((((int)pbyGreen[0] + (int)pbyGreen[1] + (int)pbyGreen[3])
			- ((int)pbyGreen[5] + (int)pbyGreen[7] + (int)pbyGreen[8])) / 3);
	b2 = ABS((((int)pbyBlue[0] + (int)pbyBlue[1] + (int)pbyBlue[3])
			- ((int)pbyBlue[5] + (int)pbyBlue[7] + (int)pbyBlue[8])) / 3);

	//左对角划分
	r3 = ABS((((int)pbyRed[1] + (int)pbyRed[2] + (int)pbyRed[5])
			- ((int)pbyRed[3] + (int)pbyRed[6] + (int)pbyRed[7])) / 3);
	g3 = ABS((((int)pbyGreen[1] + (int)pbyGreen[2] + (int)pbyGreen[5])
			- ((int)pbyGreen[3] + (int)pbyGreen[6] + (int)pbyGreen[7])) / 3);
	b3 = ABS((((int)pbyBlue[1] + (int)pbyBlue[2] + (int)pbyBlue[5])
			- ((int)pbyBlue[3] + (int)pbyBlue[6] + (int)pbyBlue[7])) / 3);

	PIXELCOLORRGB rgb;

	//max, 
	rgb.red = (BYTE)max(r0, max(r1, max(r2, r3)));
	rgb.green = (BYTE)max(g0, max(g1, max(g2, g3)));
	rgb.blue = (BYTE)max(b0, max(b1, max(b2, b3)));

	return rgb;

}

PIXELCOLORRGB CSpecialDetectionFilter::Kirsch(BYTE *pbyRed,  BYTE *pbyGreen,  BYTE *pbyBlue,  int nNum)
{
	int nRed[8], nGreen[8], nBlue[8];

	//也许, 这些代码的确很长, 很哆嗦, 并且完全可以用数组来实现
	//但是, 主要的计算量就在这个函数
	//虽然长, 但效果是最高的。

	//绕中心顺时针旋转：
	
	

	//原始Kirsch方法
	//	序号0

	
	nRed[0] = (5 * ((int)pbyRed[0] + (int)pbyRed[1] + (int)pbyRed[2]) - 
		  3 * ((int)pbyRed[3] + (int)pbyRed[5] + (int)pbyRed[6] +
		  (int)pbyRed[7] + (int)pbyRed[8]));
	nGreen[0] = (5 * ((int)pbyGreen[0] + (int)pbyGreen[1] + (int)pbyGreen[2]) - 
		  3 * ((int)pbyGreen[3] + (int)pbyGreen[5] + (int)pbyGreen[6] +
		  (int)pbyGreen[7] + (int)pbyGreen[8]));
	nBlue[0] = (5 * ((int)pbyBlue[0] + (int)pbyBlue[1] + (int)pbyBlue[2]) - 
		  3 * ((int)pbyBlue[3] + (int)pbyBlue[5] + (int)pbyBlue[6] +
		  (int)pbyBlue[7] + (int)pbyBlue[8]));
	
	//	序号1
	nRed[1] = (5 * ((int)pbyRed[1] + (int)pbyRed[2] + (int)pbyRed[5]) - 
		  3 * ((int)pbyRed[0] + (int)pbyRed[3] + (int)pbyRed[6] +
		  (int)pbyRed[7] + (int)pbyRed[8]));
	nGreen[1] = (5 * ((int)pbyGreen[1] + (int)pbyGreen[2] + (int)pbyGreen[5]) - 
		  3 * ((int)pbyGreen[0] + (int)pbyGreen[3] + (int)pbyGreen[6] +
		  (int)pbyGreen[7] + (int)pbyGreen[8]));
	nBlue[1] = (5 * ((int)pbyBlue[1] + (int)pbyBlue[2] + (int)pbyBlue[5]) - 
		  3 * ((int)pbyBlue[0] + (int)pbyBlue[3] + (int)pbyBlue[6] +
		  (int)pbyBlue[7] + (int)pbyBlue[8]));

	//	序号2
	nRed[2] = (5 * ((int)pbyRed[2] + (int)pbyRed[5] + (int)pbyRed[8]) - 
		  3 * ((int)pbyRed[0] + (int)pbyRed[3] + (int)pbyRed[6] +
		  (int)pbyRed[7] + (int)pbyRed[1]));
	nGreen[2] = (5 * ((int)pbyGreen[2] + (int)pbyGreen[5] + (int)pbyGreen[8]) - 
		  3 * ((int)pbyGreen[0] + (int)pbyGreen[3] + (int)pbyGreen[6] +
		  (int)pbyGreen[7] + (int)pbyGreen[1]));
	nBlue[2] = (5 * ((int)pbyBlue[2] + (int)pbyBlue[5] + (int)pbyBlue[8]) - 
		  3 * ((int)pbyBlue[0] + (int)pbyBlue[3] + (int)pbyBlue[6] +
		  (int)pbyBlue[7] + (int)pbyBlue[1]));

	//	序号5
	nRed[3] = (5 * ((int)pbyRed[7] + (int)pbyRed[5] + (int)pbyRed[8]) - 
		  3 * ((int)pbyRed[0] + (int)pbyRed[3] + (int)pbyRed[6] +
		  (int)pbyRed[2] + (int)pbyRed[1]));
	nGreen[3] = (5 * ((int)pbyGreen[7] + (int)pbyGreen[5] + (int)pbyGreen[8]) - 
		  3 * ((int)pbyGreen[0] + (int)pbyGreen[3] + (int)pbyGreen[6] +
		  (int)pbyGreen[2] + (int)pbyGreen[1]));
	nBlue[3] = (5 * ((int)pbyBlue[7] + (int)pbyBlue[5] + (int)pbyBlue[8]) - 
		  3 * ((int)pbyBlue[0] + (int)pbyBlue[3] + (int)pbyBlue[6] +
		  (int)pbyBlue[2] + (int)pbyBlue[1]));

	//	序号8
	nRed[4] = (5 * ((int)pbyRed[7] + (int)pbyRed[6] + (int)pbyRed[8]) - 
		  3 * ((int)pbyRed[0] + (int)pbyRed[3] + (int)pbyRed[5] +
		  (int)pbyRed[2] + (int)pbyRed[1]));
	nGreen[4] = (5 * ((int)pbyGreen[7] + (int)pbyGreen[6] + (int)pbyGreen[8]) - 
		  3 * ((int)pbyGreen[0] + (int)pbyGreen[3] + (int)pbyGreen[5] +
		  (int)pbyGreen[2] + (int)pbyGreen[1]));
	nBlue[4] = (5 * ((int)pbyBlue[7] + (int)pbyBlue[6] + (int)pbyBlue[8]) - 
		  3 * ((int)pbyBlue[0] + (int)pbyBlue[3] + (int)pbyBlue[5] +
		  (int)pbyBlue[2] + (int)pbyBlue[1]));
	
	//	序号7
	nRed[5] = (5 * ((int)pbyRed[7] + (int)pbyRed[6] + (int)pbyRed[3]) - 
		  3 * ((int)pbyRed[0] + (int)pbyRed[8] + (int)pbyRed[5] +
		  (int)pbyRed[2] + (int)pbyRed[1]));
	nGreen[5] = (5 * ((int)pbyGreen[7] + (int)pbyGreen[6] + (int)pbyGreen[3]) - 
		  3 * ((int)pbyGreen[0] + (int)pbyGreen[8] + (int)pbyGreen[5] +
		  (int)pbyGreen[2] + (int)pbyGreen[1]));
	nBlue[5] = (5 * ((int)pbyBlue[7] + (int)pbyBlue[6] + (int)pbyBlue[3]) - 
		  3 * ((int)pbyBlue[0] + (int)pbyBlue[8] + (int)pbyBlue[5] +
		  (int)pbyBlue[2] + (int)pbyBlue[1]));
	//	序号6
	nRed[6] = (5 * ((int)pbyRed[0] + (int)pbyRed[6] + (int)pbyRed[3]) - 
		  3 * ((int)pbyRed[7] + (int)pbyRed[8] + (int)pbyRed[5] +
		  (int)pbyRed[2] + (int)pbyRed[1]));
	nGreen[6] = (5 * ((int)pbyGreen[0] + (int)pbyGreen[6] + (int)pbyGreen[3]) - 
		  3 * ((int)pbyGreen[7] + (int)pbyGreen[8] + (int)pbyGreen[5] +
		  (int)pbyGreen[2] + (int)pbyGreen[1]));
	nBlue[6] = (5 * ((int)pbyBlue[0] + (int)pbyBlue[6] + (int)pbyBlue[3]) - 
		  3 * ((int)pbyBlue[7] + (int)pbyBlue[8] + (int)pbyBlue[5] +
		  (int)pbyBlue[2] + (int)pbyBlue[1]));

	//	序号3
	nRed[7] = (5 * ((int)pbyRed[0] + (int)pbyRed[1] + (int)pbyRed[3]) - 
		  3 * ((int)pbyRed[7] + (int)pbyRed[8] + (int)pbyRed[5] +
		  (int)pbyRed[2] + (int)pbyRed[6]));
	nGreen[7] = (5 * ((int)pbyGreen[0] + (int)pbyGreen[1] + (int)pbyGreen[3]) - 
		  3 * ((int)pbyGreen[7] + (int)pbyGreen[8] + (int)pbyGreen[5] +
		  (int)pbyGreen[2] + (int)pbyGreen[6]));
	nBlue[7] = (5 * ((int)pbyBlue[0] + (int)pbyBlue[1] + (int)pbyBlue[3]) - 
		  3 * ((int)pbyBlue[7] + (int)pbyBlue[8] + (int)pbyBlue[5] +
		  (int)pbyBlue[2] + (int)pbyBlue[6]));

	//用第一个元素记录最大者
	for(int i = 1;i < 8;i++)
	{
		if(nRed[i] > nRed[0]) nRed[0] = nRed[i];
		if(nGreen[i] > nGreen[0]) nGreen[0] = nGreen[i];
		if(nBlue[i] > nBlue[0]) nBlue[0] = nBlue[i];
	}


	PIXELCOLORRGB rgb;
	rgb.red = (BYTE)BOUND(nRed[0], 0, 255);
	rgb.green = (BYTE)BOUND(nGreen[0], 0, 255);
	rgb.blue = (BYTE)BOUND(nBlue[0], 0, 255);

	return rgb;
}

//非线性处理
PIXELCOLORRGB CSpecialDetectionFilter::Specialize(BYTE *pbyRed,  BYTE *pbyGreen,  BYTE *pbyBlue,  int nNum)
{
	PIXELCOLORRGB rgb;

	switch(m_dwOperation)
	{
		case IMAGE_SOBEL_EDGE_DETECT:
			rgb = Sobel(pbyRed, pbyGreen, pbyBlue, nNum);
			break;
		case IMAGE_KIRSCH_EDGE_DETECT:
			rgb = Kirsch(pbyRed, pbyGreen, pbyBlue, nNum);
			break;
		default : break;
	}
	return rgb;
}