MFC+OPENCV实现角点检测

//  角点检测 
//  根据《基于OpenCV的计算机视觉技术实现》

#define max_corners 200;                    //  限定的最大角点数

IplImage* srcImage  = 0;                    //  待处理的源图像
IplImage* ImageShow = 0;                    //  存储显示带角点的图像
IplImage* grayImage = 0;                    //  原始图像转换成的灰阶图像
IplImage* corners1  = 0;                    //  临时图像
IplImage* corners2  = 0;                    //  临时图像

int cornerCount0 = max_corners;
int cornerCount;                            //  实际测得角点数
int qualityLevel = 0;                       //  最小质量因子
int minDistance  = 15;                      //  角点最小距离
CvScalar color = CV_RGB(255,0,0);           //  绘图颜色
CvPoint2D32f corners[200];                  //  角点坐标
CvRect ROI_rect;                            //  测试范围
char   chek_area_state = 0;                 //  鼠标状态

void re_find_corners(int)                   //  滑动条响应函数
{
	int   i,x,y,xl,yu,xr,yd,k;
	int   radius = 5;
	int   thickness = 1;
	double quality_level = (double) qualityLevel / 100 + 0.02;
	double min_distance  = (double) minDistance;

	cornerCount=cornerCount0;               //  设置最大角点数
	cvGoodFeaturesToTrack(grayImage,        //  角点检测
		corners1,corners2,corners,&cornerCount,
		quality_level,min_distance,NULL);

	if (cornerCount>0) {                    //  测到角点
		xl=ROI_rect.x;     yu=ROI_rect.y;   //  设置初始测试范围
		xr=ROI_rect.x+ROI_rect.width;
		yd=ROI_rect.y+ROI_rect.height;
		cvCopy(srcImage,ImageShow);         //  恢复源图像
		for (i=0,k=0;i<cornerCount;i++) {
			x=(int)corners[i].x;
			y=(int)corners[i].y;
			if ((xl<x)&&(x<xr)&&(yu<y)&&(y<yd)) {  //  范围检查
				corners[k].x=corners[i].x;  //  保存范围内角点
				corners[k].y=corners[i].y;
				k++;
			}
		}
		cornerCount=k;                      //  范围内角点数
		cvCopy(srcImage,ImageShow);
		for (i=0;i<cornerCount;i++) {
			x=(int)corners[i].x;
			y=(int)corners[i].y;
			cvCircle(ImageShow,cvPoint(x,y),   //  角点处画圈
				radius,color,thickness,CV_AA,0);
		}
		cvRectangle(ImageShow,cvPoint(xl,yu),cvPoint(xr,yd),
			CV_RGB(0,255,0),thickness,CV_AA,0);  //  画矩形
		cvShowImage("image", ImageShow);    //  显示画圈图像
	}
}

void on_mouse2(int event,int x,int y,int flags,void* param)
{                                           //  鼠标响应函数
	int  thickness = 1;
	CvPoint point1,point2;

	if (event == CV_EVENT_LBUTTONDOWN) {    //  鼠标左键按下
		ROI_rect.x = x;                     //  记录检测窗口一角坐标
		ROI_rect.y = y;
		chek_area_state = 1;                //  设置状态标志
	}
	else if (chek_area_state && event == CV_EVENT_MOUSEMOVE) {  //  鼠标移动
		cvCopy(srcImage,ImageShow);         //  恢复原始图像
		point1 = cvPoint(ROI_rect.x, ROI_rect.y);
		point2 = cvPoint(x,y);              //  当前坐标
		cvRectangle(ImageShow,point1,point2,CV_RGB(0,255,0),
			thickness,CV_AA,0);         //  画矩形
		cvShowImage("image", ImageShow);    //  显示检测结果
		cvWaitKey(20);                      //  延时
	}
	else if (chek_area_state && event == CV_EVENT_LBUTTONUP) {  //  鼠标左键抬起
		ROI_rect.width  = abs(x - ROI_rect.x);  //  记录检测窗口对角坐标
		ROI_rect.height = abs(y - ROI_rect.y);

		re_find_corners(0);                 //  角点检测
		chek_area_state = 0;                //  恢复状态标志
		cvWaitKey(20); 
	}
}

void CCVMFCView::OnCornersTest()            //  角点检测
{
	if (workImg->nChannels>1) {             //  原图为真彩色图像==3
		srcImage = cvCloneImage(workImg);
	}
	else {                                  //  原图为灰阶图像
		srcImage = cvCreateImage(cvGetSize(workImg),IPL_DEPTH_8U,3);
		cvCvtColor(workImg,srcImage,CV_GRAY2BGR);
	}
	cvFlip(srcImage);

	grayImage = cvCreateImage(cvGetSize(srcImage),IPL_DEPTH_8U,1);
	cvCvtColor(srcImage,grayImage,CV_BGR2GRAY);  //  转换为灰阶图像
	ImageShow = cvCloneImage(srcImage);

	ROI_rect.x =0;
	ROI_rect.y =0;
	ROI_rect.width  = grayImage->width;
	ROI_rect.height = grayImage->height;

	corners1 = cvCreateImage(cvGetSize(grayImage),IPL_DEPTH_32F,1);
	corners2 = cvCreateImage(cvGetSize(grayImage),IPL_DEPTH_32F,1);

	cvNamedWindow("image",0);               //  设置显示窗口
	cvResizeWindow("image",325,350);        //  改变窗口尺寸

	cvCreateTrackbar("角点最小距离", "image",  //  设置距离滑动条
		&minDistance, 200,re_find_corners);
	cvCreateTrackbar("最小质量因子","image",  //  设置质量滑动条
		&qualityLevel,100,re_find_corners);

	re_find_corners(0);                     //  角点检测

	cvSetMouseCallback("image",on_mouse2,0);  //  设置鼠标响应函数

	cvWaitKey(0);                           //  等待键输入

	cvDestroyWindow( "image" );             //  关闭窗口

	cvReleaseImage(&srcImage);              //  释放图像存储单元
	cvReleaseImage(&grayImage);
	cvReleaseImage(&corners1);
	cvReleaseImage(&corners2);

	cvFlip(ImageShow);
	m_dibFlag=imageReplace(ImageShow,&workImg);  //  输出检测结果

	m_ImageType=-2;
	Invalidate();
}