import time
import sys
from os import listdir
from os.path import isfile, join

import cv2
import numpy as np
from opticalFlow import opticalDense
from coverage import coverage

current_milli_time = lambda: int(round(time.time() * 1000))

def experiment_video(video_filename):
    cam = cv2.VideoCapture(video_filename)
    ret, frame1 = cam.read()
    prev = frame1
    prev = coverage.cloud_recognition(frame1)
    while 1:
        prev_time = current_milli_time()
        ret, frame2 = cam.read()
        next = frame2
        next = coverage.cloud_recognition(frame2)
        flow = opticalDense.calculate_opt_dense(prev, next)
#        flow = opticalDense.calculate_opt_dense(frame1, frame2)

        flow_show = cv2.resize(flow, (400, 400))
        prev_show = cv2.resize(prev, (400, 400))
        next_show = cv2.resize(next, (400, 400))
        cv2.imshow('flow?', flow_show)
        cv2.imshow('previous', prev_show)
        cv2.imshow('next', next_show)
        k = cv2.waitKey(30) & 0xff
        if (k == 27):
            break
        next_time = current_milli_time()
        print("Took ", prev_time - next_time)
        prev = next

def for_fun():
    folder_name = "test_images/"
    test_images = [f for f in listdir(folder_name) if isfile(join(folder_name, f))]
    prev = test_images[0]
    prev_img = cv2.imread(join(folder_name, prev))
    prev_img = coverage.cloud_recognition(prev_img)


    for next in test_images:
        print(next)

#for_fun()

def experiment_images(folder_name):
    test_images = [f for f in listdir(folder_name) if isfile(join(folder_name, f))]

    prev = test_images[0]
    prev_img = cv2.imread(join(folder_name, prev))
    prev_img = coverage.cloud_recognition(prev_img)

    for next in test_images[1:]:
        next_img = cv2.imread(join(folder_name, next))
        next_img = coverage.cloud_recognition(next_img)

        flow = opticalDense.calculate_opt_dense(prev_img, next_img)

        flow_show = cv2.resize(flow, (400, 400))
        prev_show = cv2.resize(prev_img, (400, 400))
        next_show = cv2.resize(next_img, (400, 400))
        cv2.imshow('flow?', flow_show)
        cv2.imshow('previous', prev_show)
        cv2.imshow('next', next_show)
        k = cv2.waitKey(30) & 0xff
        if (k == 27):
            break

        prev_img = next_img

#experiment_images("test_images/")
output = experiment_video('opticalFlow/Clouds.mp4')