# C3B.py
import cv2
import numpy as np

def RED_bot1(img):
    img1=np.copy(img)
    img2 = cv2.resize(img1, (LX2,LY2), interpolation=cv2.INTER_AREA)
    Y=LY2-1; XM=int(LX2/2); RED=[]; ra=1.5
    for x in range(LX2):
        if(img2[Y][x][2] > 50 and img2[Y][x][2] > 1.5*img2[Y][x][1]
            and img2[Y][x][2] > 1.5*img2[Y][x][0]): RED.append(x)
    LRED=len(RED)
    if(LRED > 0):
        MIDL=int(np.median(RED))
        cv2.line(img1,(MIDL*F,0),(MIDL*F,Y*F),(0,255,0),4)
        if(30<=MIDL<=50): DR=1
        elif(50<MIDL<80): DR=7
        elif(0<=MIDL<30): DR=3
    else: MIDL=-1; DR=-2
    return DR,LRED,MIDL,img1

n=0; F=8; LX=640; LY=480; LX2=int(LX/F); LY2=int(LY/F)
cap = cv2.VideoCapture('REDL.mp4')
while(cap.isOpened()):
    ret, frame = cap.read()
    if(ret==False): break
    DR,LRED,MIDL,img1=RED_bot1(frame)
    print(n,DR,LRED,MIDL); n+=1
    cv2.imshow('img1',img1)
    cv2.waitKey(0)
cap.release()
cv2.destroyAllWindows()
print('C3B is done...')