Update to 2.0.0 tree from current Fremantle build

[opencv] / tests / swig_python / highgui / cvRetrieveFrame.py

diff --git a/tests/swig_python/highgui/cvRetrieveFrame.py b/tests/swig_python/highgui/cvRetrieveFrame.py
new file mode 100755 (executable)
index 0000000..2bfacc3
--- /dev/null
+++ b/tests/swig_python/highgui/cvRetrieveFrame.py
@@ -0,0 +1,53 @@
+#! /usr/bin/env python
+"""
+This script will test highgui's cvRetrieveFrame function
+"""
+
+# name of this test and it's requirements
+TESTNAME = "cvRetrieveFrame"
+REQUIRED = ["cvGrabFrame"]
+
+ 
+# needed for sys.exit(int) and .works file handling
+import os
+import sys
+import works
+
+# path to imagefiles we need
+PREFIX=os.path.join(os.environ["srcdir"],"../../opencv_extra/testdata/python/videos/")
+
+# check requirements and delete old flag file, if it exists
+if not works.check_files(REQUIRED,TESTNAME):
+       sys.exit(77)
+
+# import the necessary things for OpenCV
+from highgui import *
+from cv import *
+
+
+# create a video reader using the tiny video 'uncompressed.avi'
+video = cvCreateFileCapture(PREFIX+"uncompressed.avi")
+
+# call cvGrabFrame to grab a frame from video
+res=cvGrabFrame(video)
+
+if res==0:
+       print "(ERROR) Couldn't call cvGrabFrame()"
+       sys.exit(1)
+# call cvRetrieveFrame and check if returned image is valid
+image = cvRetrieveFrame(video)
+
+if image is None:
+       # returned image is not a correct IplImage (pointer),
+       # so return an error code
+       sys.exit(1)
+
+
+# ATTENTION: We do not release the video reader or image.
+# This is bad manners, but Python and OpenCV don't care...
+       
+# create flag file for sollowing tests
+works.set_file(TESTNAME)
+
+# return 0 ('PASS')
+sys.exit(0)