import cv2
import os
import math
import numpy as np
import matplotlib.pyplot as plt

# Create the temp dir
if not os.path.exists('data'):
    os.makedirs('data')

# Define the color spaces
Red_lower2 = np.array([,100,100])
Red_upper2 = np.array([,255,255])
Red_lower1 = np.array([0,100,100])
Red_upper1 = np.array([6,255,255])
Orange_lower = np.array([7,100,100])
Orange_upper = np.array([22,255,255])
Yellow_lower = np.array([23,100,100])
Yellow_upper = np.array([32,255,255])
Green_lower = np.array([33,100,100])
Green_upper = np.array([75,255,255])
Cyan_lower = np.array([76,100,100])
Cyan_upper = np.array([97,255,255])
Blue_lower = np.array([93,100,100])
Blue_upper = np.array([130,255,255])
Purple_lower = np.array([130,100,100])
Purple_upper = np.array([,255,255])
Pink_lower = np.array([,100,100])
Pink_upper = np.array([,255,255])
White_lower = np.array([,100,100])
White_upper = np.array([,255,255])
Black_lower = np.array([,100,100])
Black_upper = np.array([,255,255])

# loop through video files
videopath = '/media/vm/HDD/freeloops/3D/'
imagepath = 'data/'
videos = os.listdir(videopath)
for video in videos:
    cap = cv2.VideoCapture(videopath + video)
    framerate = cap.get(cv2.CAP_PROP_FPS)
    while cap.isOpened():
        frameId = cap.get(1)
        ret, frame = cap.read()
        if not ret:
            break
        if frameId % math.floor(framerate) == 0:
            filename = imagepath + "/screenshot_" + str(int(frameId)) + ".jpg"
            cv2.imwrite(filename, frame)
    del cap
# analyze each screenshot for colors
    images = os.listdir(imagepath)
    for image in images:
        cap = cv2.imread(imagepath + image, cv2.COLOR_BGR2HSV)
        plt.imshow(cap)



# move video file according to results