创建黑色画板,并支持两种画图功能

import mathimport cv2import numpy as np"""1. 创建一个黑色画板2. 输入q退出3. 输入m切换画图模式两种模式,画矩形和画圆形。用户按住鼠标左键到一个位置然后释放就可以画出对应的图像"""px, py, rx, ry = 0, 0, 0, 0# 临时保存左键按下时的x,y坐标drawing = False# 绑定窗体,窗体名为imagecv2.namedWindow("image")canvas = np.zeros((400, 400, 3))# 创建黑色画板def draw_shape(event, x, y, flags, param):global px, py, rx, ry, canvas, draw_line# 设置全局变量,这样才能够访问方法体外部的变量if not drawing:if event == cv2.EVENT_LBUTTONDOWN:# 判断当前是否为按下左键px, py = x, y# 保存按下左键的坐标if event == cv2.EVENT_LBUTTONUP:rx, ry = x, y# 保存释放左键的坐标cv2.rectangle(canvas, [px, py], [rx, ry], color=(255, 255, 255))# 根据坐标信息,画出一个黑色的矩形else:if event == cv2.EVENT_LBUTTONDOWN:# 如果用户按下了m键,则切换到画圆形模式px, py = x, yif event == cv2.EVENT_LBUTTONUP:# 计算当前释放点和上次点击的圆心距离多远dis = math.sqrt((x - px) ** 2 + (y - py) ** 2)# 两点之间距离公式cv2.circle(canvas, center=(px, py), radius=int(dis), color=(0, 0, 255))# 根据离圆心的距离画出圆cv2.setMouseCallback('image', draw_shape)# 设置opencv鼠标事件while 1:# 死循环等待用户cv2.imshow("image", canvas)# 显示当前的画板数据k = cv2.waitKey(10)# 等待10ms,判断用户的键盘数据if k == ord("q"):# 如果输入q则退出程序breakif k == ord('m'):# 如果输入m则切换模型,修改drawing变量drawing = not drawing

这段代码使用 OpenCV 创建了一个简单的画板应用,支持矩形和圆形的绘制。用户可以按下左键拖动以绘制矩形,按下右键拖动以绘制圆形。按下 ‘q’ 键退出程序,按下 ‘m’ 键切换绘制模式(矩形或圆形)。

以下是代码的主要步骤:

  1. 初始化全局变量:px, py, rx, ry 用于保存左键按下和释放时的坐标,drawing 用于标记当前绘制模式。

  2. 创建一个黑色画板:canvas = np.zeros((400, 400, 3))

  3. 定义鼠标事件处理函数 draw_shape,在函数中根据事件类型和坐标信息绘制矩形或圆形。

  4. 设置鼠标事件回调函数:cv2.setMouseCallback('image', draw_shape)

  5. 进入死循环,不断显示当前画板,并等待用户的键盘输入。

  6. 如果用户按下 ‘q’ 键,则退出程序;如果按下 ‘m’ 键,则切换绘制模式。

这个简单的画板演示了如何使用 OpenCV 处理鼠标事件来实现基本的图形绘制功能。