OpenCV做个简单的画图板—变化画笔颜色，粗细

前言

最近在学习OpenCV，照着官网的教程学习，学习Gui Features in Opencv，其中主要内容参考这一章节的内容，鼠标事件部分对应前一章的学习内容，链接如下：opencv官网教程
看本博客内容基本就可以实现这个章节内容，如果还有问题可以下载链接：源程序

实现的功能

这一节主要实现的功能是，利用Trackbar相关命令，实现功能如下：

1. 实现实时画图功能；
2. 通过R,G,B三色的拖拽条，实现颜色改变，更新画笔颜色；
3. 通过拖拽条改变画笔粗细；
4. 通过拖拽条做成“开关”，实现橡皮擦功能；
5. 通过键盘按钮切换，可是实现画线或画矩形；

实现思路

1. 当按下鼠标左键开始画笔功能，移动鼠标并且左键拖拽时，画圆，从而连接成曲线；
2. 当拖拽画笔粗细拖拽条，改变画圆的半径大小；
3. 当切换开关为“橡皮擦”，改变画圆的颜色为窗口画图板的背景色，这里是白色；
4. 键盘“m”键可以切换画线模式和画矩形模式；

实现步骤

1. 建立画图板窗口。底色为白色，尺寸为350*512。#底色和尺寸可以根据自己喜好而定，但此方法的底色只有黑白灰三色。

#create a white image, a window
img = 255*np.ones((350,512,3),np.uint8)

2. 建立窗口和鼠标回调函数，用于监听鼠标事件，做出回应。

cv.namedWindow('image')
cv.setMouseCallback('image',draw_mousepush)

3. 创建各功能拖动条，用来更改画笔颜色、粗细以及橡皮擦开关功能，不设拖动条回调函数

#create trackbars for color change,switch of erase,painter thickness
cv.createTrackbar('R','image',0,255,nothing)
cv.createTrackbar('G','image',0,255,nothing)
cv.createTrackbar('B','image',0,255,nothing)

cv.createTrackbar('thickness','image',1,8,nothing)

switch = 'erase'
cv.createTrackbar(switch,'image',0,1,nothing)

4. 获取画笔信息功能函数get_painter，返回画笔颜色、画笔粗细、橡皮擦开关状态

#get info painter function
def get_painter():
    # get current positions of four trackbars
    r = cv.getTrackbarPos('R', 'image')
    g = cv.getTrackbarPos('G', 'image')
    b = cv.getTrackbarPos('B', 'image')
    thickness = cv.getTrackbarPos('thickness','image')
    s = cv.getTrackbarPos(switch,'image')
    color = (b,g,r) #color scales
    return color,thickness,s

5. 完成鼠标动作回调函数。

# mouse callback function
#initialize初始化
drawing = False #painter is false,not beginning painting
mode = False #切换画线和画矩形功能
ix, iy = 0, 0
def draw_mousepush(event,x,y,flags,param):
    global ix,iy,drawing,mode #引入全局变量
    color,thickness,s = get_painter() #从get_painter函数引入参数
# moniter mouse event.When push leftbutton down,beginning painting
    if event == cv.EVENT_LBUTTONDOWN:
        drawing = True
        ix, iy = x, y
#按住左键滑动，则开始在鼠标所在位置画圆，从而连成曲线
    elif event == cv.EVENT_MOUSEMOVE and flags == cv.EVENT_FLAG_LBUTTON:
        if drawing == True:
            if s == 1: #打开橡皮擦功能
                cv.circle(img,(x,y),10,(255,255,255),-1)
            else:
                if mode == False: #画线
                    cv.circle(img,(x,y),thickness,color,-1)
                else:
                    cv.rectangle(img,(ix,iy),(x,y),color,-1)
    elif event == cv.EVENT_LBUTTONUP:
        drawing = False

6. 建立主函数

# main function
while(1):
    color,thickness,_ = get_painter()
    cv.imshow('image',img)
    k = cv.waitKey(1) & 0xFF
    if k ==27: #ESC退出
        break
    elif k == ord('m'):
        mode = not mode

7. ESC键，清除所有窗口

cv.destroyAllWindows()

运行结果

请忽略我略丑的字体，哈哈