OpenCV 5-2. 마우스로 도형 그리기 (직사각형)

• cv2.EVENT_LBUTTONDOWN : 마우스 왼쪽 버튼 클릭 이벤트
• cv2.EVENT_MOUSEMOVE : 마우스 이동 이벤트
• cv2.EVENT_LBUTTONUP : 마우스 왼쪽 버튼 눌러 있지 않은 이벤트 
• cv2.rectangle() : 직사각형 그리기
• math.sqrt() : 거리 구하는 함수

# -*- coding: utf-8 -*-
# opencv_009.py
# https://blog.naver.com/samsjang/220501964041

import numpy as np
import cv2
from random import shuffle    # 무작위로 섞기 위한 shuffle() 함수
import math    # MLB의 누른 지점과 땐 지점간의 거리를 구하는데 필요한 sqrt() 함수

# 직사각형과 원 그리기/도형이 그려지고 안그려지고 위한 토클
mode, drawing = True, False
# MLB 누른 지점으로 활용될 변수
ix, iy = -1, -1
B = [i for i in range(256)]
G = [i for i in range(256)]
R = [i for i in range(256)]

def onMouse(event, x, y, flags, param):
    # 전역 변수 선언한 것 사용
    global ix, iy, drawing, mode, B, G, R

    # MLB 드래그(눌려진 상태) 이벤트
    if event == cv2.EVENT_LBUTTONDOWN:
        drawing = True
        ix, iy = x, y
        shuffle(B), shuffle(G), shuffle(R)
    # 마우스 이동 이벤트
    elif event == cv2.EVENT_MOUSEMOVE:
        # MLB가 눌려저 있는지(drawing)
        if drawing:
            if mode:
                # 직사각형 그리기
                # cv2.rectangle()
                cv2.rectangle(param, (ix, iy), (x, y), (B[0], G[0], R[0]), -1)
            else:
                r = (ix-x)**2 + (iy-y)**2
                r = int(math.sqrt(r))
                cv2.circle(param, (ix, iy), r, (B[0], G[0], R[0]), -1)

    # MLB 업(떠려진 상태) 이벤트
    elif event == cv2.EVENT_LBUTTONUP:
        drawing = False
        if mode:
            cv2.rectangle(param, (ix, iy), (x, y), (B[0], G[0], R[0]), -1)
        else:
            r = (ix-x)**2 + (iy-y)**2
            r = int(math.sqrt(r))
            cv2.circle(param, (ix, iy), r, (B[0], G[0], R[0]), -1)

def mouseBrush():
    global mode

    # 512x512 검정색(np.zeros()) 캔버스 생성
    img = np.zeros((512, 512, 3), np.uint8)
    cv2.namedWindow('paint')
    cv2.setMouseCallback('paint', onMouse, param=img)

    while True:
        cv2.imshow('paint', img)
        k = cv2.waitKey(1) & 0xFF

        if k == 27:    # esc키 종료
            break
        elif k == ord('m'):    # m키 직사각형, 원 그리기 토글
            mode = not mode

    cv2.destroyAllWindows()

mouseBrush()

python, opencv, cv2, random, shuffle, math