import detect
import timeit
import cv2
calibration_width = 0.75
image = "img7.jpg"
images = ("img.jpg", "img2.jpg", "img3.jpg", "img4.jpg", "img5.jpg", "img6.jpg", "img7.jpg", "img8.jpg")
show = False
video = True
def go():
    #for file in images:
    detect.detect(calibration_width, "img7.jpg", show)

if not video:
	elapsed_time = timeit.timeit(go, number=3)/3
	print(elapsed_time)
else :
    capture = cv2.VideoCapture('tcpclientsrc host=192.168.86.108 port=5001  ! gdpdepay !  rtph264depay ! avdec_h264 ! videoconvert ! appsink', cv2.CAP_GSTREAMER)
    # server command for pi camera /opt/vc/bin/raspivid -t 0 -w 1920 -h 1080 -hf -fps 5 -o - | gst-launch-1.0 -v fdsrc ! h264parse !  rtph264pay config-interval=1 pt=96 ! gdppay ! tcpserversink host=192.168.86.108 port=5001
    ret,frame = capture.read()
    detect.detect(calibration_width, "img7.jpg", True, False)
    detect.detect(calibration_width, frame, True, True)
    while True:
        ret,frame = capture.read()

        if not ret:
            print('empty frame')
            break
        #print('frame')
        
        cv2.imshow('Item Sorter', detect.detect(calibration_width, frame, True, True))
        if cv2.waitKey(1)&0xFF == ord('q'):
            break