1 files changed, 98 insertions, 0 deletions
diff --git a/src/pygame/_camera_opencv_highgui.py b/src/pygame/_camera_opencv_highgui.py
new file mode 100644
index 0000000..7709c3c
--- /dev/null
+++ b/src/pygame/_camera_opencv_highgui.py
@@ -0,0 +1,98 @@
+
+import pygame
+import numpy
+
+import opencv
+#this is important for capturing/displaying images
+from opencv import highgui
+
+
+
+def list_cameras():
+    """
+    """
+    # -1 for opencv means get any of them.
+    return [-1]
+
+def init():
+    pass
+
+def quit():
+    pass
+
+
+class Camera:
+
+    def __init__(self, device =0, size = (640,480), mode = "RGB"):
+        """
+        """
+        self.camera = highgui.cvCreateCameraCapture(device)
+        if not self.camera:
+            raise ValueError ("Could not open camera.  Sorry.")
+
+
+    def set_controls(self, **kwargs):
+        """
+        """
+
+
+    def set_resolution(self, width, height):
+        """Sets the capture resolution. (without dialog)
+        """
+        # nothing to do here.
+        pass
+    def query_image(self):
+        return True
+
+    def stop(self):
+        pass
+
+    def start(self):
+        # do nothing here... since the camera is already open.
+        pass
+
+    def get_buffer(self):
+        """Returns a string containing the raw pixel data.
+        """
+        return self.get_surface().get_buffer()
+
+    def get_image(self, dest_surf = None):
+        return self.get_surface(dest_surf)
+
+    def get_surface(self, dest_surf = None):
+        camera = self.camera
+
+        im = highgui.cvQueryFrame(camera)
+        #convert Ipl image to PIL image
+        #print type(im)
+        if im:
+            xx = opencv.adaptors.Ipl2NumPy(im)
+            #print type(xx)
+            #print xx.iscontiguous()
+            #print dir(xx)
+            #print xx.shape
+            xxx = numpy.reshape(xx, (numpy.product(xx.shape),))
+
+            if xx.shape[2] != 3:
+                raise ValueError("not sure what to do about this size")
+
+            pg_img = pygame.image.frombuffer(xxx, (xx.shape[1],xx.shape[0]), "RGB")
+
+            # if there is a destination surface given, we blit onto that.
+            if dest_surf:
+                dest_surf.blit(pg_img, (0,0))
+            return dest_surf
+            #return pg_img
+
+
+
+if __name__ == "__main__":
+
+    # try and use this camera stuff with the pygame camera example.
+    import pygame.examples.camera
+
+    pygame.camera.Camera = Camera
+    pygame.camera.list_cameras = list_cameras
+    pygame.examples.camera.main()
+
+