[python] opencv로 태극기 그리기

import numpy as np, cv2

def draw_bar(img, pt, w, bars):
    pt = np.array(pt, np.int)
    for bar in bars:
        (x, y), h = pt, w*6
        cv2.rectangle(img, (x, y, w, h), (0,0,0), -1)
        if bar ==0:
            y = y + w*3-w//4
            h = w//2
            cv2.rectangle(img, (x, y, w, h), (255,255,255), -1)
        pt += (int(w*1.5),0)

c=200
r, sr, c2, c4 = c//2, c//4, c*2, c*4

img = np.full((c4, c4, 3), 255, np.uint8)
blue, red = (255, 0, 0), (0, 0, 255)

cv2.ellipse(img, (c2, c2), (r, r), 0, 0, 180, blue, -1)
cv2.ellipse(img, (c2, c2), (r, r), 180, 0, 180, red, -1)#원을 그리는 시작점과 끝점
cv2.ellipse(img, (c2+r-sr, c2), (sr, sr), 180, 0, 180, blue, -1)
cv2.ellipse(img, (c2-sr, c2), (sr, sr), 0, 0, 180, red, -1)

left = (c2 - c*(18+8)/24, c2 - sr)
right = (c2 + c*(18+0)/24, c2 - sr)

draw_bar(img, left, c//12, (1,1,1)) #(1,1,1)은 안 끊긴다는 것 (3개)
draw_bar(img, right, c//12, (0,0,0)) #가운데 끊김 (6개)
angle = cv2.fastAtan2(2,3)
img = cv2.warpAffine(img, cv2.getRotationMatrix2D((c2, c2), -angle*2, 1), (c4, c4))

draw_bar(img, left, c//12, (1,0,1))
draw_bar(img, right, c//12, (0,1,0))
ang = cv2.getRotationMatrix2D((c2, c2), angle, 1)
img = cv2.warpAffine(img, ang, (c4, c4))

cv2.imshow("image", img[c2-c:c2+c, c2-r*3:c2+r*3]) #슬라이스
cv2.waitKey(0)