InteractiveProgramming/paint.py at master · Tweir195/InteractiveProgramming · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

import cv2
import numpy as np
import math

"""capturing video from camera"""
cap = cv2.VideoCapture(0)
frame_num = 0
cx = 0
cy = 0
path = []
clearpath = []
dist = []
colors = []


def map(x, oldL, oldH, newL, newH):
    """This function maps a value from one range to a differnet range

    x: the value in the old range
    oldL: the lower limit of the old range of values
    oldH: the upper limit of the old range of values
    newL: the lower limit of the new range of values
    newH: the upper limit of the new range of values
    """
    return int(((x - oldL)/(oldH-oldL))*(newH-newL)+newL)

def brush_color(hue):
    """This function takes in a uint8 value for hue and generate
    a BGR color range """
    color = np.uint8([[[hue, 255, 255]]])
    return cv2.cvtColor(color, cv2.COLOR_HSV2BGR)

while(True):
    #capture frame by frame
    ret, frame = cap.read()
    frame = cv2.flip(frame,1)
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    #find specific color
    lower_white = np.array([0, 0, 230])
    upper_white = np.array([180, 25, 255])
    lower_color = np.array([0,80,50])
    upper_color = np.array([20,100,100])
    lower_red = np.array([150,150,50])
    upper_red = np.array([180,255,150])


    mask1 = cv2.inRange(hsv, np.array([0, 150, 100]), np.array([1, 255, 255]));
    mask2 = cv2.inRange(hsv, np.array([178, 150, 100]), np.array([180, 255, 255]));
    mask = mask1 | mask2

    # Attempting Green
    # mask = cv2.inRange(hsv, np.array([50,100,100]), np.array([65,255,255]))

    mask = cv2.bilateralFilter(mask, 10, 40, 40)
    mask = cv2.blur(mask,(5,5))

    res = cv2.bitwise_and(frame,frame,mask=mask)
    mask = cv2.blur(mask,(20,20))
    # Getting a contour and the center of the contour
    im2,contours,hierarchy = cv2.findContours(mask, 1, 2)
    try:
        if frame_num > 0 or frame_num == 0:
            cnt = contours[0]
            M = cv2.moments(cnt)
            # print(M['m10']/M['m00'])
            cx = int(M['m10']/M['m00'])
            # print(cx)
            cy = int(M['m01']/M['m00'])
            # print(cy)
            frame_num = 0
            path.append((cx, cy))
            # if diffx > 100 or diffy > 100:
            #     print('too far')
            # else:
        cv2.circle(res, (cx, cy), 2, (0, 255, 0), -1)
    except IndexError:
        """"""

    # TODO: Add if statements to make sure that any outliers
    # would be removed from the list or ignored when drawing
    # the linewidth
    if len(path) == 1:
        clearpath.append(path[0])
    elif len(path) > 2:
        pair = path[-2]
        # print(pair, pair[0], cx, pair[1], cy)
        diffx = abs(cx-pair[0])
        diffy = abs(cy-pair[1])
        distance = math.sqrt(diffx**2+diffy**2)
        if distance<10:
            dist2hue = map(distance, 0.0, 10.0, 0.0, 255.0)
            paintColor = brush_color(dist2hue)
            print(paintColor[0][0][0])
            colors.append((int(paintColor[0][0][0]), int(paintColor[0][0][1]), int(paintColor[0][0][2])))
            print(colors)
            clearpath.append(pair)

    for i in range(len(clearpath)):
        if len(clearpath) < 1:
            break
        elif i<(len(clearpath)-1)<21:
            cv2.line(res, clearpath[i], clearpath[i+1], colors[i], 3)
        elif 20 < i < (len(clearpath)-1):
            cv2.rectangle(res, (0,0), (600, 400), (0,0,0))
            for j in range(20):
                cv2.line(res, clearpath[-(j+1)], clearpath[-(j+2)], colors[-(j+2)], 3)


    # print(dist)
    frame_num += 1
    # cnts = cv2.findContours(res, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
    #display the resulting frame
    cv2.imshow('frame',frame)
    #cv2.imshow('mask', mask)
    cv2.imshow('res',res)
    # cv2.imshow('counto', )
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
#when finshed, release the capture
cap.release()
cv2.destroyAllWindows()
"""playing video from file
cap = cv2.VideoCapture('test.avi')

while(cap.isOpened()):
    ret, frame = cap.read()
    cv2.waitKey(25)
    cv2.imshow('frame',frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()
"""