Classified the programs

canvas.py

@@ -0,0 +1,19 @@
+import numpy as np
+import cv2
+
+
+class canvas():
+
+    def __init__(self, width, height):
+        self.width = width
+        self.height = height
+        self.new_canvas = np.zeros((self.height, self.width, 3), np.uint8)
+
+    def set_color(self, B, G, R):
+        self.color = (B, G, R)
+
+    def set_bgColor(self, B, G, R):
+        self.new_canvas[:, :] = self.color
+
+    def show_canvas(self):
+        cv2.imshow(self.new_canvas)

fingerTrack.py

@@ -0,0 +1,64 @@
+import numpy as np
+import cv2
+import math
+
+
+class finger_track():
+
+    def __init__(self):
+        """
+        """
+        self.frame_num = 0
+        self.cx = 0
+        self.cy = 0
+        self.path = []
+        self.red_maskL = [np.array([0, 150, 100]), np.array([178, 150, 100])]
+        self.red_maskH = [np.array([2, 255, 255]), np.array([180, 255, 255])]
+        self.refreshDelay = 3
+
+    def BGR2HSV(self, frame):
+        """This functions takes in a frame and converts it from BGR to BGR2HSV
+        """
+        return cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
+
+    def red_mask(self, frame):
+        """This function generates a red mask based on the frame being passed in
+        """
+        mask1 = cv2.inRange(frame, self.red_maskL[0], self.red_maskH[0])
+        mask2 = cv2.inRange(frame, self.red_maskL[1], self.red_maskH[1])
+        return (mask1 | mask2)
+
+    def find_center(self, mask, target):
+        im2, contours, hierarchy = cv2.findContours(mask, 1, 2)
+        try:
+            if self.frame_num > self.refreshDelay or self.frame_num == 0:
+                cnt = contours[0]
+                M = cv2.moments(cnt)
+                print(M['m10'] / M['m00'])
+                self.cx = int(M['m10'] / M['m00'])
+                self.cy = int(M['m01'] / M['m00'])
+                self.frame_num = 0
+                self.path.append((self.cx, self.cy))
+                # if diffx > 100 or diffy > 100:
+                #     print('too far')
+                # else:
+            cv2.circle(target, (self.cx, self.cy), 2, (0, 255, 0), -1)
+        except IndexError:
+            """"""
+
+    def draw(self, target):
+        for i in range(len(self.path)):
+            # TODO: Add if statements to make sure that any outliers
+            # would be removed from the list or ignored when drawing
+            # the linewidth
+
+            # diffx = math.fabs(self.cx-self.path[-1][0])
+            # print(diffx)
+            # diffy = math.fabs(cy-path[-1][1])
+            # print(diffy)
+            if len(self.path) == 1:
+                break
+            # elif math.sqrt(diffx**2+diffy**2) > 30:
+                # break
+            elif i < (len(self.path)-1):
+                cv2.line(target, self.path[i], self.path[i+1], (255, 0, 0), 3)

main.py

@@ -0,0 +1,9 @@
+from fingerTrack import *
+from canvas import *
+
+def main():
+    """
+    """
+
+if __name__ == "__main__":
+    main()