added context manager for video reader

Application/VideoReader.py

@@ -1,6 +1,3 @@
-from Application.Config import Config
-
-from datetime import datetime
 from queue import Queue
 
 import cv2
@@ -9,14 +6,13 @@ import os
 
 class VideoReader:
     listOfFrames = None
-    w = 0
+    w = None
-    h = 0
+    h = None
 
     def __init__(self, config, setOfFrames=None):
         videoPath = config["inputPath"]
         if videoPath is None:
             raise Exception("ERROR: Video reader needs a videoPath!")
-
         self.videoPath = videoPath
         self.lastFrame = 0
         # buffer data struct:
@@ -31,12 +27,16 @@ class VideoReader:
         if setOfFrames is not None:
             self.listOfFrames = sorted(setOfFrames)
         
-    def getWH(self):
+    def __enter__(self):
-        '''get width and height'''
+        self.fillBuffer()
-        res, image = self.vc.read()
+        return self
-        self.w = image.shape[1]
+
-        self.h = image.shape[0]
+    def __exit__(self, type, value, traceback):
-        return (self.w, self.h)
+        self.stop()
+
+    def stop(self):
+        self.thread.join()
+        self.vc.release()
 
     def pop(self):
         return self.buffer.get(block=True)
@@ -53,10 +53,6 @@ class VideoReader:
             self.thread = threading.Thread(target=self.readFrames, args=())
         self.thread.start()
 
-    def stop(self):
-        self.thread.join()
-        self.vc.release()
-
     def readFrames(self):
         '''Reads video from start to finish'''
         while self.lastFrame < self.endFrame:
@@ -122,3 +118,11 @@ class VideoReader:
 
     def getStartTime(self):
         return self.starttime
+
+    def getWH(self):
+        '''get width and height'''
+        if self.w is None or self.h is None:
+            res, image = self.vc.read()
+            self.w = image.shape[1]
+            self.h = image.shape[0]
+        return (self.w, self.h)

test.py

@@ -0,0 +1,15 @@
+from Application.VideoReader import VideoReader
+from Application.Config import Config
+import os
+
+fileName = "out.mp4"
+dirName = os.path.join(os.path.dirname(__file__), "generate test footage")
+
+config = {}
+config["inputPath"] = os.path.join(dirName, fileName)
+config["videoBufferLength"] = 100
+
+with VideoReader(config) as reader:
+    while not reader.videoEnded():
+        framenumber, frame = reader.pop()
+        print(framenumber)