python opencv入门 滚动条控制调色板（5）

内容来自OpenCV-Python Tutorials 自己翻译整理

目标：
使用滚动条绑定窗口
学习如下函数 cv2.getTrackbarPos(), cv2.createTrackbar()

样例代码：
这里建立了一个简单的样例来现实用户指定的颜色。用户使用三个滚动条来指定B,G,R值，同时用户选择的颜色会显示在窗口当中。用户拖动滚动条会关联到窗口中颜色的改变。初始默认的颜色会被设置为黑色。
对于cv2.getTrackbarPos()函数，第一个参数是滚动条的名称，第二个个参数是关联的窗口名称，第三个参数是缺省值，第四个参数是是最大值，第五个参数是执行所有改变值的回调函数。回调函数总是有一个缺省参数，这个参数是滚动条的位置。在本例子当中，函数该函数不做任何事情，简单略过即可。
滚动条另外一个重要的应用是用来作为按钮或者开关。默认情况下，opencv里面没有按钮功能。所以用户需要使用滚动条来实现这样的一个功能。在本例子当中，我们创建了一个开关，这个开关控制这个应用程序的颜色变化功能是否开启，如果开关功能关闭，那么屏幕一直都是黑色的。

import cv2
import numpy as np

def nothing(x):
    pass

# 创建一个黑色背景的窗口
img = np.zeros((300,512,3), np.uint8)
cv2.namedWindow('image')

# 创建改变颜色的滚动条
cv2.createTrackbar('R','image',0,255,nothing)
cv2.createTrackbar('G','image',0,255,nothing)
cv2.createTrackbar('B','image',0,255,nothing)

# 创建控制函数的开关
switch = '0 : OFF \n1 : ON'
cv2.createTrackbar(switch, 'image',0,1,nothing)

while(1):
    cv2.imshow('image',img)
    k = cv2.waitKey(1) & 0xFF
    if k == 27:
        break

    # 获取四个滚动条的位置
    r = cv2.getTrackbarPos('R','image')
    g = cv2.getTrackbarPos('G','image')
    b = cv2.getTrackbarPos('B','image')
    s = cv2.getTrackbarPos(switch,'image')

    if s == 0:
        img[:] = 0
    else:
        img[:] = [b,g,r]#把滚动条里的颜色值赋给图片

cv2.destroyAllWindows()

练习：
创建一个画图程序，这个程序可以调整颜色和画刷的半径。使用滚动条调整。

import cv2
import numpy as np

# mouse callback function
def nothing(x):
    pass




def draw_circle(event,x,y,flags,param):
    s=cv2.getTrackbarPos('size','image')
    b=cv2.getTrackbarPos('B','image')
    r=cv2.getTrackbarPos('R','image')
    g=cv2.getTrackbarPos('G','image')

    if event == cv2.EVENT_LBUTTONDBLCLK:
        cv2.circle(img,(x,y),s,(b,g,r),-1)



# Create a black image, a window and bind the function to window
img = np.zeros((512,512,3), np.uint8)
cv2.namedWindow('image')
cv2.createTrackbar('R','image',0,255,nothing)
cv2.createTrackbar('G','image',0,255,nothing)
cv2.createTrackbar('B','image',0,255,nothing)
cv2.createTrackbar('size','image',1,255,nothing)
cv2.setMouseCallback('image',draw_circle)

while(1):
    cv2.imshow('image',img)
    if cv2.waitKey(20) & 0xFF == 27:
        break
cv2.destroyAllWindows()

双击画圈