import multiprocessing
import cv2
from time import sleep
from queue import Queue
import threading
from Config import Config


class VideoReader:

    
    listOfFrames = None

    def __init__(self, config, setOfFrames = None):
        videoPath = config["inputPath"]
        if videoPath is None:
            print("Video reader needs a videoPath!")
            return None

        self.videoPath = videoPath
        self.lastFrame = 0
        #buffer = Queue([(frameNumber, frame), ])
        self.buffer = Queue(config["videoBufferLength"])
        self.vc = cv2.VideoCapture(videoPath)
        self.stopped = False
        res, image = self.vc.read()
        self.w = image.shape[1]
        self.h = image.shape[0]
        if setOfFrames is not None:
            self.listOfFrames = sorted(setOfFrames)      

    def pop(self):
        return self.buffer.get(block=True)

    def get(self):
        return self.buffer[-1]

    def fillBuffer(self):
        if self.buffer.full():
            print("VideoReader::fillBuffer was called when buffer was full.")
        self.endFrame = int(self.vc.get(cv2.CAP_PROP_FRAME_COUNT))

        #self.endFrame = 10*60*30
        if self.listOfFrames is not None:
            self.thread = threading.Thread(target=self.readFramesByList, args=())
        else:
            self.thread = threading.Thread(target=self.readFrames, args=())
        self.thread.start()

    def stop(self):
        self.thread.join()
        self.vc.release()

    def readFrames(self):
        while self.lastFrame < self.endFrame:
            res, frame = self.vc.read()
            if res:
                self.buffer.put((self.lastFrame, frame))
            self.lastFrame += 1

        self.stopped = True

    
    def readFramesByList(self):
        self.vc.set(1, self.listOfFrames[0])
        self.lastFrame = self.listOfFrames[0]
        self.endFrame = self.listOfFrames[-1]

        while self.lastFrame < self.endFrame:
            if self.lastFrame in self.listOfFrames:
                res, frame = self.vc.read()
                if res:
                    self.buffer.put((self.lastFrame, frame))
                # since the list is sorted the first element is always the lowest relevant framenumber
                # [0,1,2,3,32,33,34,35,67,68,69]
                self.listOfFrames.pop(0)
                self.lastFrame += 1
            else:
                # if current Frame number is not in list of Frames, we can skip a few frames
                self.vc.set(1, self.listOfFrames[0])
                self.lastFrame = self.listOfFrames[0]
        
        self.stopped = True
    
    def videoEnded(self):
        if self.stopped and self.buffer.empty():
            return True
        else:
            return False

    def getFPS(self):
        return self.vc.get(cv2.CAP_PROP_FPS)

    def get_file_metadata(self, path, filename, metadata):
        # Path shouldn't end with backslash, i.e. "E:\Images\Paris"
        # filename must include extension, i.e. "PID manual.pdf"
        # Returns dictionary containing all file metadata.
        sh = win32com.client.gencache.EnsureDispatch('Shell.Application', 0)
        ns = sh.NameSpace(path)

        # Enumeration is necessary because ns.GetDetailsOf only accepts an integer as 2nd argument
        file_metadata = dict()
        item = ns.ParseName(str(filename))
        for ind, attribute in enumerate(metadata):
            attr_value = ns.GetDetailsOf(item, ind)
            if attr_value:
                file_metadata[attribute] = attr_value

        return file_metadata