DrawMouseEvent.py 拖动鼠标时绘制矩形或者是圆圈

忘是亡心i 2022-07-12 23:24 141阅读 0赞

7.2 高级一点的示例  
现在我们来创建一个更好的程序。这次我们的程序要完成的任务是根据我  
们选择的模式在拖动鼠标时绘制矩形或者是圆圈（就像画图程序中一样）。所以  
我们的回调函数包含两部分，一部分画矩形，一部分画圆圈。这是一个典型的  
例子他可以帮助我们更好理解与构建人机交互式程序，比如物体跟踪，图像分  
割等。

\# -\*- coding: utf-8 -\*-  
"""  
Created on Sun Jan 5 11:41:39 2014  
@author: duan  
"""  
import cv2  
import numpy as np

\# 当鼠标按下时变为True  
drawing=False  
\# 如果mode 为true 绘制矩形。按下'm' 变成绘制曲线。  
mode=True  
ix,iy=-1,-1

\# 创建回调函数  
def draw\_circle(event,x,y,flags,param):  
 global ix,iy,drawing,mode  
 \# 当按下左键是返回起始位置坐标  
 if event==cv2.EVENT\_LBUTTONDOWN:  
  drawing=True  
  ix,iy=x,y  
 \# 当鼠标左键按下并移动是绘制图形。event 可以查看移动，flag 查看是否按下  
 elif event==cv2.EVENT\_MOUSEMOVE and flags==cv2.EVENT\_FLAG\_LBUTTON:  
  if drawing==True:  
   if mode==True:  
    cv2.rectangle(img,(ix,iy),(x,y),(255,0,0),-1)  
   else:  
    \# 绘制圆圈，小圆点连在一起就成了线，3 代表了笔画的粗细  
    cv2.circle(img,(x,y),3,(0,0,255),-1)  
 \# 下面注释掉的代码是起始点为圆心，起点到终点为半径的  
 \# r=int(np.sqrt((x-ix)\*\*2+(y-iy)\*\*2))  
 \# cv2.circle(img,(x,y),r,(0,0,255),-1)  
 \# 当鼠标松开停止绘画。  
 elif event==cv2.EVENT\_LBUTTONUP:  
  drawing==False  
 \# if mode==True:  
 \# cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)  
 \# else:  
 \# cv2.circle(img,(x,y),5,(0,0,255),-1)

img=np.zeros((512,512,3),np.uint8)  
cv2.namedWindow('image')  
cv2.setMouseCallback('image',draw\_circle)  
while(1):  
 cv2.imshow('image',img)  
 k=cv2.waitKey(1)&0xFF  
 if k==ord('m'):  
  mode=not mode  
 elif k==ord('c'):  
  \#Clear  
  img=np.zeros((512,512,3),np.uint8)  
 elif k==27:  
  break