#!/usr/bin/env python

import sys

import pygame
from pygame.locals import QUIT, KEYDOWN
import Image

from camera import Camera


pygame.init()
window = pygame.display.set_mode((640,480))
pygame.display.set_caption("WebCam Demo")
screen = pygame.display.get_surface()

cam = Camera()

while True:
    events = pygame.event.get()
    for event in events:
        if event.type == QUIT:
            sys.exit(0)
        elif event.type == KEYDOWN:
            im = cam.get_image()
            pg_img = pygame.image.frombuffer(im.tostring(), im.size, im.mode)
            screen.blit(pg_img, (0,0))
            pygame.display.flip()