Opencv实现的简易绘图工具

第一次写博，还是个菜鸟。最近开始学习Opencv，试着写了个简易的绘图工具(目前只写了画线和橡皮擦部分，画其它图形还有待往里添加)，也算是记录自己的学习之路。

#include "stdafx.h"

#include "cv.h"

#include "highgui.h"

#define SHAPE_LINE 1

#define SHAPE_ERASER 2

int g_style;

CvRect rect;

CvPoint g_StartPoint;

CvPoint g_EndPoint;

CvPoint p_Start;       //记录鼠标位置左上方点

CvPoint p_End;			//记录鼠标位置右下方点

bool drawing=false;       //是否处于画图状态

bool erasering=false;    //是否处于擦除状态

void callback(int event,int x,int y,int flags,void* param);

void DrawLine(IplImage* img);

//简易绘图工具，实现了画线和橡皮擦功能，绘图功能还有待往里添加

int main(int argc,char** argv)

{

	IplImage* img=cvCreateImage(cvSize(512,512),IPL_DEPTH_8U,3);

	cvSet(img,cvScalar(255,255,255));

	IplImage* temp=cvCloneImage(img);

	cvCopy(img,temp);

	cvNamedWindow("简易绘图工具");

	cvSetMouseCallback("简易绘图工具",callback,img);

	printf("请键入要选择执行的操作:‘l’——”画线“,‘e’——”橡皮“\n");

	char select='l';

	while(1)

	{

		cvCopyImage(img,temp);      //始终在原图image上画结果(和鼠标事件绑定)，先将image复制给temp,然后在临时图temp画出画图的过程(在main函数中画过程图形),然后用temp来显示图像

		switch(select)

		{

		case 'l':

			g_style=SHAPE_LINE;

			break;

		case 'e':

			g_style=SHAPE_ERASER;

			break;

		case 27:

			return 0;

		}

		if(g_style==SHAPE_LINE&&drawing)     //画鼠标在按住移动过程中画线

			cvDrawLine(temp,g_StartPoint,g_EndPoint,cvScalar(0,0,0));

		if(g_style==SHAPE_ERASER)

		{

			cvRectangle(temp,p_Start,p_End,cvScalar(0,0,0));  //画出橡皮矩形边框

		}

		cvShowImage("简易绘图工具",temp);

		select=cvWaitKey(30);

	}

	cvReleaseImage(&img);

	cvReleaseImage(&temp);

	cvDestroyWindow("简易绘图工具");

	return 0;

}

void callback(int event,int x,int y,int flags,void* param)

{

	IplImage* img=(IplImage*)param;

	switch(event)

	{

	case CV_EVENT_LBUTTONDOWN:

		{

			if(SHAPE_LINE==g_style)

			{

				drawing=true;

				g_StartPoint=cvPoint(x,y);

				g_EndPoint=g_StartPoint;   //此处将终点坐标设为同起始点避免记住前一条直线的终点坐标

			}

			if(SHAPE_ERASER==g_style)

			{

				erasering=true;

				//注意边界问题要适当修改ROI有效区域

				rect.x=x-20;  rect.y=y-20;		//注意坐标的计算,一般原点在窗口的左上角(这和操作系统等因素有关,IplImage结构中有个origin属性可以设置图像的原点)

				rect.width=40;  rect.height=40;

				p_Start.x=x-20; p_Start.y=y-20;

				p_End.x=x+20;   p_End.y=y+20;

				if(x>-20&&x<532&&y>-20&&y<532)

				{

					cvSetImageROI(img,rect);

					cvSet(img,cvScalar(255,255,255));

					cvResetImageROI(img);

				}

			}

		}

		break;

	case CV_EVENT_MOUSEMOVE:

		{

			p_Start.x=x-20; p_Start.y=y-20;      //随时定位鼠标位置画橡皮矩形

			p_End.x=x+20;   p_End.y=y+20;

			if(SHAPE_LINE==g_style)

			{

				if(drawing)

				{

					g_EndPoint=cvPoint(x,y);

				}

			}

			if(SHAPE_ERASER==g_style)

			{

				rect.x=x-20;  rect.y=y-20;

				rect.width=40;  rect.height=40;

				if(erasering)

				{

					if(x>-20&&x<532&&y>-20&&y<532)

					{

						cvSetImageROI(img,rect);

						cvSet(img,cvScalar(255,255,255));

						cvResetImageROI(img);

					}

				}

			}

		}

		break;

	case CV_EVENT_LBUTTONUP:

		{

			if(SHAPE_LINE==g_style)

			{

				drawing=false;

				cvDrawLine(img,g_StartPoint,g_EndPoint,cvScalar(255,0,0));  //(255,0,0)此处画出的是蓝色,即BGR

			}

			if(SHAPE_ERASER==g_style)

			{

				erasering=false;

			}

		}

	}

}

Opencv实现的简易绘图工具的相关教程结束。