#!/usr/bin/env python
from sys import path
from CVtypes import cv
from time import sleep
from time import clock
from PIL import JpegImagePlugin
import wx

class ImageSource:
	def __init__(self):
		pass

	def getNextImage(self):
		pass

	def getNextSavedImage(self):
		return 'face.jpg'

class VideoCapture(ImageSource):
	def __init__():
		ImageSource.__init__()
		cap=cv.CreateCameraCapture(0)

	def getImage():
		img = cv.QueryFrame(cap)

	def getSavedImage():
		cv.SaveImage('test.jpg', img);
		cv.ShowImage('win',img)	

    def cvImageAsBitmap(img, flip=True):
        sz = cvGetSize(img)
        flags = CV_CVTIMG_SWAP_RB
        if flip:
            flags |= CV_CVTIMG_FLIP
        cvConvertImage(img, img, flags)
        bitmap = wx.BitmapFromBuffer(sz.width, sz.height, cvImageAsBuffer(img))
        return bitmap