import cv2
import numpy as np
##################GET the color code on the bottom line########
def bottclr(img):
    img1=np.copy(img)
    img2=cv2.resize(img1,(80,60))
    hsv=cv2.cvtColor(img2,cv2.COLOR_BGR2HSV)
    Y1=59; Y2=40; Y3=20
    for x in range(80):
        print(x,img2[Y1,x,:],hsv[Y1,x,:],img2[Y2,x,:],hsv[Y2,x,:],
          img2[Y3,x,:],hsv[Y3,x,:])
    cv2.line(img2,(0,Y1-6),(159,Y1-6),(0,255,0),1)
    cv2.line(img2,(0,Y2-6),(159,Y2-6),(0,255,0),1)
    cv2.line(img2,(0,Y3-6),(159,Y3-6),(0,255,0),1)
    img2=cv2.resize(img2,(320,240))
    cv2.imshow('fcolor',img2)
    cv2.waitKey(0)

if __name__ == '__main__':
    from time import sleep
    print('fcolor.py as a main')
    low=np.array([105,120 ,120 ])
    upp=np.array([115,255 ,255 ])
    cap = cv2.VideoCapture(0)
    cap.set(cv2.CAP_PROP_FRAME_WIDTH,640)
    cap.set(cv2.CAP_PROP_FRAME_HEIGHT,480)
    sleep(2)
    #ret,img=cap.read()
    #bottclr(img)
    for j in range(2680):
        ret,imgi=cap.read()
        img=cv2.resize(imgi,(320,240))
        hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
        mask1=cv2.inRange(hsv,low,upp)
        imgmask=cv2.bitwise_and(img,img,mask=mask1)
        img3=cv2.resize(imgmask,(640,480))
        cv2.imshow('mask1',mask1)
        cv2.imshow('img3',img3)
        if cv2.waitKey(1) & 0xFF == ord('q'): break
        print('j=',j)

    #main()

