《学习OpenCV（中文版）》第4章 练习5a

/*定义各种形状*/
#define SHAPE_NONE 0
#define SHAPE_LINE 1
#define SHAPE_RECT 2
#define SHAPE_CIRCLE 3
#define SHAPE_ELLIPSE 4
#define SHAPE_PLYGON 5

#define SHAPE_COUNT 6	//总的形状数，包括清空（SHAPE_NONE）
#define _ERASER 6		//橡皮擦标识

int width = 300;
int height = 300;
int shape_status = SHAPE_NONE;		//记录各种状态
CvPoint begin_point, end_point;		//鼠标摁下及松开时的坐标
void on_mouse_callback(int event, int x, int y, int flags, void* param) {
IplImage* img1 = (IplImage*)param;
switch(event) {
case CV_EVENT_LBUTTONDOWN:			//鼠标左边点击，确定行为
{
int perWidth = width/SHAPE_COUNT;	//每个按钮的宽度
if(y>height) {					//点击自定义的按钮
if(x<perWidth) {
shape_status = SHAPE_LINE;
} else if(x<2*perWidth) {
shape_status = SHAPE_RECT;
} else if(x<3*perWidth) {
shape_status = SHAPE_CIRCLE;
} else if(x<4*perWidth) {
shape_status = SHAPE_ELLIPSE;
} else if(x<5*perWidth) {
shape_status = SHAPE_PLYGON;
} else {
shape_status = SHAPE_NONE;
cvSetImageROI(img1, cvRect(0, 0, width, height));	//清空画面
cvZero(img1);
cvResetImageROI(img1);
}
} else {
//记录起点
if(shape_status != SHAPE_NONE) {
begin_point.x = x;
begin_point.y = y;
}
}
break;
}
case CV_EVENT_RBUTTONDOWN:
{
shape_status = _ERASER;
break;
}
case CV_EVENT_LBUTTONUP:		//鼠标左键起来时，各种画图
{
if(y<height) {			//在画面区域时
cvSetImageROI(img1, cvRect(0, 0, width, height));	//限定在画面区域，避免自定义按钮被覆盖或清空
//记录终点
if(shape_status != SHAPE_NONE) {
end_point.x = x;
end_point.y = y;
}
switch(shape_status) {
case SHAPE_LINE:
{
cvLine(img1, begin_point, end_point, cvScalar(255, 0, 0));	//画直线
break;
}
case SHAPE_RECT:
{
cvRectangle(img1, begin_point, end_point, cvScalar(0, 255, 0));	//画矩形
break;
}
case SHAPE_CIRCLE:
{
cvCircle(
img1, 
begin_point,	//以起点为圆心
std::sqrt(std::pow(1.0*(end_point.x-begin_point.x), 2)+std::pow(1.0*(end_point.y-begin_point.y), 2)*1.0f),	//两点距离为半径
cvScalar(0, 0, 255));
break;
}
case SHAPE_ELLIPSE:
{
cvEllipse(
img1, 
cvPoint((begin_point.x+end_point.x)/2, (begin_point.y+end_point.y)/2),	//两点中点为中心
cvSize(std::abs(end_point.x - begin_point.x), std::abs(end_point.y - begin_point.y)),	//两点为对角所确定的矩形区域
0,
0,
360,
cvScalar(255, 255, 0)
);
break;
}
case SHAPE_PLYGON:
//do nothing，还没想好怎么画

break;
}
cvResetImageROI(img1);
}
cvShowImage("Picture", img1);
break;
}
case CV_EVENT_RBUTTONUP:
{
shape_status = SHAPE_NONE;
break;
}
case CV_EVENT_MOUSEMOVE:
{
if(shape_status == _ERASER && y<height) {
for(int i=-2; i<=2; i++) {		//橡皮擦，以(x,y)为中心的大小为5的矩形橡皮擦
for(int j=-2; j<=2; j++) {
//所经过的点复原为0
*(cvPtr2D(img1, y+i, x+j)+0) = 0;
*(cvPtr2D(img1, y+i, x+j)+1) = 0;
*(cvPtr2D(img1, y+i, x+j)+2) = 0;
}
}
}
cvShowImage("Picture", img1);
break;
}
}
}
void myGUI5a() {
cvNamedWindow("Picture");
int delta = 40;
IplImage* img = cvCreateImage(cvSize(width, height+delta), IPL_DEPTH_8U, 3);	//加20用于做按钮
cvZero(img);
//自定义按钮
//cvLine(img, cvPoint(0, height), cvPoint(width, height), cvScalar(255, 255, 255));
int perWidth = width/SHAPE_COUNT;	//每个按钮的宽度
for(int i=0; i<SHAPE_COUNT; i++) {
int next_row = height + delta;
int next_col = (i+1)*perWidth;
int color = 255 - 20*i;			//按钮不同颜色
for(int row=height;row<next_row; row++) {
for(int col=i*perWidth;col<next_col; col++) {
*(cvPtr2D(img, row, col)+0) = color;
*(cvPtr2D(img, row, col)+1) = color;
*(cvPtr2D(img, row, col)+2) = color;
}
}
}

cvSetMouseCallback("Picture", on_mouse_callback, img);
cvShowImage("Picture", img);
cvWaitKey(0);
}