选择角点

之前用到过对一幅图提取角点，有时候角点数目很多而且随机生成，但是自己又想到的目标位置处的角点，这就要求有一种方法能够把几个角点的位置标识出来，下面程序的主要思路是：首先提取出图中所有的角点，然后手动选择需要的角点，最后得到角点的位置。

#include <iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/opencv.hpp>
#include<cv.h>
#include<highgui.h>
using namespace cv;
using namespace std;


CvPoint pt1 = Point(0, 0);
CvPoint pt2 = Point(0, 0);
bool is_selecting = false;
//这个函数用来读取鼠标画出的方框
void cvMouseCallback(int mouseEvent, int x, int y, int flags, void* param)
{
    switch (mouseEvent)
    {
        case CV_EVENT_LBUTTONDOWN:
            pt1 = Point(x, y);
            pt2 = Point(x, y);
            is_selecting = true;
            break;
        case CV_EVENT_MOUSEMOVE:
            if (is_selecting)
                pt2 = Point(x, y);
            break;
        case CV_EVENT_LBUTTONUP:
            pt2 = Point(x, y);
            is_selecting = false;
            break;
     }
}


int main()
{
IplImage* orig = cvLoadImage("1.png");


//提取所有角点并显示；
IplImage* gray = cvCreateImage(cvGetSize(orig), IPL_DEPTH_8U, 1);
cvCvtColor(orig, gray, CV_BGR2GRAY);
IplImage* img_temp = cvCreateImage(cvGetSize(gray), 32, 1);
IplImage* img_eigen = cvCreateImage(cvGetSize(gray), 32, 1);
int count = 1000;//the maximum numbers of points;
CvPoint2D32f* corners = new CvPoint2D32f[count];
cvGoodFeaturesToTrack(gray, img_eigen, img_temp, corners, &count, 0.1, 10);
cout << "总共角点："<<count << endl;
for (int i = 0; i<count; i++)//在原图中显示所有角点；
{
cout << "NO. " << i << " point :" << corners[i].x << ";" <<
corners[i].y << endl;
cvLine(orig, cvPoint(corners[i].x, corners[i].y), cvPoint(corners[i].x,
corners[i].y), CV_RGB(255, 0, 0), 5);
}
cvNamedWindow("point");
cvShowImage("point", orig);
//cvWaitKey(0);

//选择需要角点，使用鼠标在原图中画小方框，以方框右下角的点作为目标角点；
IplImage* img_show = cvCreateImage(cvGetSize(orig), IPL_DEPTH_8U, 3);
CvPoint2D32f pos[4];//存储四个目标角点坐标；
setMouseCallback("point", cvMouseCallback);
bool shift_on = false;
for (int i = 0; i < 4; i++){//这里假设需要图中四个角处的角点
while (true)
{
cvCopy(orig, img_show);
cvRectangle(img_show, pt1, pt2, Scalar(0, 255, 0));
char key = cvWaitKey(10);
cvSetImageROI(img_show, cvRect(pt1.x, pt1.y, abs(pt2.x - pt1.x), abs(pt2.y - pt1.y)));
IplImage* roi = cvCreateImage(cvSize(abs(pt2.x - pt1.x), abs(pt2.y - pt1.y)), IPL_DEPTH_8U, 3);
cvCopy(img_show, roi);
cvResetImageROI(img_show);
if (key == 27) break;//按下Esc表示一个角点选择完毕；
}
pos[i] = cvPointTo32f(pt2);
cout <<"角点坐标：" <<pos[i].x << " , " << pos[i].y << endl;
}


cvWaitKey(0);
return 0;
}