import os
import cv2
from src.VHSImage import generateVHSStyle
from os.path import isfile, join
import numpy as np
import subprocess
import logzero
from logzero import logger
from logzero import setup_logger


def SaveVid(path):
    vidObj = cv2.VideoCapture(path)
    count = 0
    success = 1
    while success:
        success, image = vidObj.read()
        cv2.imwrite("frames/" + str(count) + ".jpg", image)
        # os.rename("frames/"+str(count)+".jpg", os.path.splitext("frames/"+str(count)+".jpg")[0])
        count += 1


def Style(pathToFrames):
    files = [f for f in os.listdir(pathToFrames) if isfile(join(pathToFrames, f))]
    count = 0
    for i in files:
        count += 1
        f = str(i)
        fi = pathToFrames + f
        out = fi + ".jpg"

        generateVHSStyle(fi, out, silence=True)
        os.rename(out, fi)
        print("--------")
        print("On Frame: ")
        print(count)
        print("Out of")
        print(len(files))
        print("--------")
    cwd = os.getcwd()
    os.chdir(pathToFrames)
    c = "find ./ -name \"*.jpg\" -exec sh -c 'mv $0 `basename \"$0\" .jpg`' '{}' \;    ;"
    os.system(c)
    os.chdir(cwd)


def generateVideo(outfile, path, infile):
    frame_array = []
    files = [int(f) for f in os.listdir(path) if isfile(join(path, f))]
    files.sort()

    duration = subprocess.check_output(
        [
            "ffprobe",
            "-i",
            infile,
            "-show_entries",
            "format=duration",
            "-v",
            "quiet",
            "-of",
            "csv=%s" % ("p=0"),
        ]
    )
    fps = len(files) / float(duration)
    print("FPS", fps)

    for i in range(len(files)):
        filename = path + str(files[i])
        img = cv2.imread(filename)
        height, width, layers = img.shape
        size = (width, height)
        frame_array.append(img)
    out = cv2.VideoWriter(outfile, cv2.VideoWriter_fourcc(*"MP4V"), fps, size)
    for i in range(len(frame_array)):
        out.write(frame_array[i])
    out.release()


def VHS_Vid(infile, outfile):
    path = "./frames/"
    os.system("rm frames/*")
    os.system("mkdir frames")
    logger.info("Exctracting Frames")
    try:
        SaveVid(infile)
    except:
        logger.debug("Extracted Frames")
    logger.info("Applying A E S T H E T I C S")
    Style(path)
    logger.info("Generating Vidio")
    generateVideo("temp.mp4", path, infile)
    logger.info("Extracting audio of original video")
    os.system("ffmpeg -i %(infile)s -vn -acodec copy output-audio.aac")
    logger.info("Merging audio")
    os.system("ffmpeg -i temp.mp4 -i output-audio.aac -c copy %(outfile)s")
    logger.info("Removing residual files")
    os.remove("temp.mp4")
    os.remove("output-audio.aac")


# VHS_Vid("video.mp4","video2.mp4")