Air Pen.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np \n",
    "import cv2 \n",
    "from collections import deque \n",
    "    \n",
    "def setValues(x): \n",
    "   print(\"\") \n",
    "   \n",
    "\n",
    "cv2.namedWindow(\"Color detectors\") \n",
    "cv2.createTrackbar(\"Upper Hue\", \"Color detectors\", \n",
    "                   153, 180, setValues) \n",
    "cv2.createTrackbar(\"Upper Saturation\", \"Color detectors\", \n",
    "                   255, 255, setValues) \n",
    "cv2.createTrackbar(\"Upper Value\", \"Color detectors\",  \n",
    "                   255, 255, setValues) \n",
    "cv2.createTrackbar(\"Lower Hue\", \"Color detectors\", \n",
    "                   64, 180, setValues) \n",
    "cv2.createTrackbar(\"Lower Saturation\", \"Color detectors\",  \n",
    "                   72, 255, setValues) \n",
    "cv2.createTrackbar(\"Lower Value\", \"Color detectors\",  \n",
    "                   49, 255, setValues) \n",
    "  \n",
    "\n",
    "bpoints = [deque(maxlen = 1024)] \n",
    "gpoints = [deque(maxlen = 1024)] \n",
    "rpoints = [deque(maxlen = 1024)] \n",
    "ypoints = [deque(maxlen = 1024)] \n",
    "   \n",
    "\n",
    "blue_index = 0\n",
    "green_index = 0\n",
    "red_index = 0\n",
    "yellow_index = 0\n",
    "     \n",
    "kernel = np.ones((5, 5), np.uint8) \n",
    "   \n",
    "colors = [(255, 0, 0), (0, 255, 0),  \n",
    "          (0, 0, 255), (0, 255, 255)] \n",
    "colorIndex = 0\n",
    "   \n",
    "paintWindow = np.zeros((471, 636, 3)) + 255\n",
    "   \n",
    "cv2.namedWindow('Paint', cv2.WINDOW_AUTOSIZE) \n",
    "    \n",
    "cap = cv2.VideoCapture(0) \n",
    "   \n",
    "while True: \n",
    "       \n",
    "    ret, frame = cap.read() \n",
    "       \n",
    "    frame = cv2.flip(frame, 1) \n",
    "    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) \n",
    "   \n",
    "    u_hue = cv2.getTrackbarPos(\"Upper Hue\", \n",
    "                               \"Color detectors\") \n",
    "    u_saturation = cv2.getTrackbarPos(\"Upper Saturation\", \n",
    "                                      \"Color detectors\") \n",
    "    u_value = cv2.getTrackbarPos(\"Upper Value\", \n",
    "                                 \"Color detectors\") \n",
    "    l_hue = cv2.getTrackbarPos(\"Lower Hue\", \n",
    "                               \"Color detectors\") \n",
    "    l_saturation = cv2.getTrackbarPos(\"Lower Saturation\", \n",
    "                                      \"Color detectors\") \n",
    "    l_value = cv2.getTrackbarPos(\"Lower Value\", \n",
    "                                 \"Color detectors\") \n",
    "    Upper_hsv = np.array([u_hue, u_saturation, u_value]) \n",
    "    Lower_hsv = np.array([l_hue, l_saturation, l_value]) \n",
    "   \n",
    "   \n",
    "    frame = cv2.rectangle(frame, (40, 1), (140, 65),  \n",
    "                          (122, 122, 122), -1) \n",
    "    frame = cv2.rectangle(frame, (160, 1), (255, 65), \n",
    "                          colors[0], -1) \n",
    "    frame = cv2.rectangle(frame, (275, 1), (370, 65),  \n",
    "                          colors[1], -1) \n",
    "    frame = cv2.rectangle(frame, (390, 1), (485, 65),  \n",
    "                          colors[2], -1) \n",
    "    frame = cv2.rectangle(frame, (505, 1), (600, 65), \n",
    "                          colors[3], -1) \n",
    "      \n",
    "    cv2.putText(frame, \"CLEAR ALL\", (49, 33), \n",
    "                cv2.FONT_HERSHEY_SIMPLEX, 0.5, \n",
    "                (255, 255, 255), 2, cv2.LINE_AA) \n",
    "      \n",
    "    cv2.putText(frame, \"BLUE\", (185, 33),  \n",
    "                cv2.FONT_HERSHEY_SIMPLEX, 0.5, \n",
    "                (255, 255, 255), 2, cv2.LINE_AA) \n",
    "      \n",
    "    cv2.putText(frame, \"GREEN\", (298, 33), \n",
    "                cv2.FONT_HERSHEY_SIMPLEX, 0.5, \n",
    "                (255, 255, 255), 2, cv2.LINE_AA) \n",
    "      \n",
    "    cv2.putText(frame, \"RED\", (420, 33), \n",
    "                cv2.FONT_HERSHEY_SIMPLEX, 0.5,  \n",
    "                (255, 255, 255), 2, cv2.LINE_AA) \n",
    "      \n",
    "    cv2.putText(frame, \"YELLOW\", (520, 33), \n",
    "                cv2.FONT_HERSHEY_SIMPLEX, 0.5,  \n",
    "                (150, 150, 150), 2, cv2.LINE_AA) \n",
    "   \n",
    "\n",
    "    Mask = cv2.inRange(hsv, Lower_hsv, Upper_hsv) \n",
    "    Mask = cv2.erode(Mask, kernel, iterations = 1) \n",
    "    Mask = cv2.morphologyEx(Mask, cv2.MORPH_OPEN, kernel) \n",
    "    Mask = cv2.dilate(Mask, kernel, iterations = 1) \n",
    "   \n",
    "    cnts, _ = cv2.findContours(Mask.copy(), cv2.RETR_EXTERNAL, \n",
    "        cv2.CHAIN_APPROX_SIMPLE) \n",
    "    center = None\n",
    "   \n",
    "    if len(cnts) > 0: \n",
    "          \n",
    "        cnt = sorted(cnts, key = cv2.contourArea, reverse = True)[0] \n",
    "          \n",
    "        ((x, y), radius) = cv2.minEnclosingCircle(cnt) \n",
    "          \n",
    "        cv2.circle(frame, (int(x), int(y)), int(radius), (0, 255, 255), 2) \n",
    "           \n",
    "        M = cv2.moments(cnt) \n",
    "        center = (int(M['m10'] / M['m00']), int(M['m01'] / M['m00'])) \n",
    "    \n",
    "        if center[1] <= 65: \n",
    "              \n",
    "            # Clear Button \n",
    "            if 40 <= center[0] <= 140:  \n",
    "                bpoints = [deque(maxlen = 512)] \n",
    "                gpoints = [deque(maxlen = 512)] \n",
    "                rpoints = [deque(maxlen = 512)] \n",
    "                ypoints = [deque(maxlen = 512)] \n",
    "   \n",
    "                blue_index = 0\n",
    "                green_index = 0\n",
    "                red_index = 0\n",
    "                yellow_index = 0\n",
    "   \n",
    "                paintWindow[67:, :, :] = 255\n",
    "            elif 160 <= center[0] <= 255: \n",
    "                    colorIndex = 0 # Blue \n",
    "            elif 275 <= center[0] <= 370: \n",
    "                    colorIndex = 1 # Green \n",
    "            elif 390 <= center[0] <= 485: \n",
    "                    colorIndex = 2 # Red \n",
    "            elif 505 <= center[0] <= 600: \n",
    "                    colorIndex = 3 # Yellow \n",
    "        else : \n",
    "            if colorIndex == 0: \n",
    "                bpoints[blue_index].appendleft(center) \n",
    "            elif colorIndex == 1: \n",
    "                gpoints[green_index].appendleft(center) \n",
    "            elif colorIndex == 2: \n",
    "                rpoints[red_index].appendleft(center) \n",
    "            elif colorIndex == 3: \n",
    "                ypoints[yellow_index].appendleft(center) \n",
    "                  \n",
    "\n",
    "    else: \n",
    "        bpoints.append(deque(maxlen = 512)) \n",
    "        blue_index += 1\n",
    "        gpoints.append(deque(maxlen = 512)) \n",
    "        green_index += 1\n",
    "        rpoints.append(deque(maxlen = 512)) \n",
    "        red_index += 1\n",
    "        ypoints.append(deque(maxlen = 512)) \n",
    "        yellow_index += 1\n",
    "   \n",
    "  \n",
    "    points = [bpoints, gpoints, rpoints, ypoints] \n",
    "    for i in range(len(points)): \n",
    "          \n",
    "        for j in range(len(points[i])): \n",
    "              \n",
    "            for k in range(1, len(points[i][j])): \n",
    "                  \n",
    "                if points[i][j][k - 1] is None or points[i][j][k] is None: \n",
    "                    continue\n",
    "                      \n",
    "                cv2.line(frame, points[i][j][k - 1], points[i][j][k], colors[i], 2) \n",
    "                cv2.line(paintWindow, points[i][j][k - 1], points[i][j][k], colors[i], 2) \n",
    "   \n",
    "    cv2.imshow(\"Tracking\", frame) \n",
    "    cv2.imshow(\"Paint\", paintWindow) \n",
    "    cv2.imshow(\"mask\", Mask) \n",
    "   \n",
    "    if cv2.waitKey(1) & 0xFF == ord(\"q\"): \n",
    "        break\n",
    "        \n",
    "cap.release() \n",
    "cv2.destroyAllWindows()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}