opencv鼠标交互

#include"stdafx.h"
#include <iostream>

using namespace cv;
using namespace std;

struct Select
{
    Rect rect;//会被自动初始化为全零
    bool flag,compute;
    Select(){flag=0;compute=0;}
};
static Select select;
static Point origin;
int roadSamplePick(Mat &img,Mat &composite,Rect&roadSampleRect);

static void onMouse(int event,int x,int y,int flag,void* param)
{
    static Size size=*(Size*)param;
    switch(event)
    {
    case CV_EVENT_MOUSEMOVE:
        if(select.flag)
        {
            select.rect.x=origin.x<x?origin.x:x;
            select.rect.y=origin.y<y?origin.y:y;
            select.rect.width=abs(x-origin.x);
            select.rect.height=abs(y-origin.y);
        }
        break;
    case CV_EVENT_LBUTTONDOWN:
        origin=Point(x,y);
        select.flag=1;
        break;
    case CV_EVENT_LBUTTONUP:
        select.flag=0;//不画矩形框
        select.compute=1;//对选定区域进行计算
    }
}
int roadSamplePick(Mat &img,Mat &composite,Rect&roadSampleRect){
    Mat hsv,img_hsv[3],tempcom;
    Mat edgeh,edgev;//边缘
    Scalar mean,dev;
    cvtColor(img,hsv,CV_BGR2HSV);
    GaussianBlur(hsv,hsv,Size(3,3),1.5);
    hsv.assignTo(hsv,CV_32FC3);
    split(hsv,img_hsv);//0色调，1饱和度，2亮度
    Sobel(img_hsv[2],edgeh,-1,1,0);//垂直边缘检测(x- image derivative)
    Sobel(img_hsv[2],edgev,-1,0,1);//水平边缘检测(y- image derivative)
    tempcom=img_hsv[0]+img_hsv[1]+edgeh+edgev;//合成图像
    //取得样本区域
    
    Mat roadSample,roadSampleGray,road;//rect子图
    if(roadSampleRect.y<0)//图像过小而不能取样
    {
        std::cout<<"enable to create road sample"<<std::endl;
        return -1;
    }
    cout<<(double)img_hsv[0].data[0]<<" "<<(double)img_hsv[1].data[0]<<endl;
    roadSample=tempcom(roadSampleRect);//获取采样区域
    meanStdDev(roadSample,mean,dev);
    imshow("sample",roadSample);
    float *data=(float*)tempcom.data;
    for(int i=0;i<tempcom.total();i++)
    {
        if(data[i]<mean.val[0]-3*dev.val[0]||data[i]>mean.val[0]+3*dev.val[0])
            data[i]=0;
        else
            data[i]=255;
    }
    tempcom.assignTo(composite,CV_8UC1);
    imshow("composite",composite);
    return 0;

}


int main(int argc, unsigned char* argv[])
{
    Mat img,choose;
    Size size;
    //交互初始化
    img=imread("test2.png");
    size=img.size();
    namedWindow("image");
    setMouseCallback("image",onMouse,(void*)&size);
    imshow("image",img);

    while(1)
    {
        if(select.flag)
        {
            choose=img.clone();
            rectangle(choose,select.rect,Scalar(255,0,0),3,8,0);
            imshow("image",choose);
        }
        if(select.compute)
        {
            Mat composite;
            roadSamplePick(img,composite,select.rect);
            select.compute=0;
        }
        if(waitKey(20)=='q')//退出
            break;
    }

    return 0;
}