python source code of videocapturethreading

import time
import unittest

import cv2

from gfd.py.video.capture import VideoCaptureTreading


class VideoCaptureTest(unittest.TestCase):
    def setUp(self):
        pass

    def _run(self, n_frames=500, width=1280, height=720, with_threading=False):
        if with_threading:
            cap = VideoCaptureTreading(0)
        else:
            cap = cv2.VideoCapture(0)
        cap.set(cv2.CAP_PROP_FRAME_WIDTH, width)
        cap.set(cv2.CAP_PROP_FRAME_HEIGHT, height)
        if with_threading:
            cap.start()
        t0 = time.time()
        i = 0
        while i < n_frames:
            _, frame = cap.read()
            cv2.imshow('Frame', frame)
            cv2.waitKey(1) & 0xFF
            i += 1
        print('[i] Frames per second: {:.2f}, with_threading={}'.format(n_frames / (time.time() - t0), with_threading))
        if with_threading:
            cap.stop()
        cv2.destroyAllWindows()

    def test_video_capture(self):
        n_frames = 500
        self._run(n_frames, 1280, 720, False)

    def test_video_capture_threading(self):
        n_frames = 500
        self._run(n_frames, 1280, 720, True)


if __name__ == '__main__':
    unittest.main()