capture.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4
   5 import pygame
   6 from pygame.locals import QUIT, KEYDOWN
   7 import Image
   8
   9 from camera import Camera
  10
  11
  12 pygame.init()
  13 window = pygame.display.set_mode((640,480))
  14 pygame.display.set_caption("WebCam Demo")
  15 screen = pygame.display.get_surface()
  16
  17 cam = Camera()
  18
  19 while True:
  20     events = pygame.event.get()
  21     for event in events:
  22         if event.type == QUIT:
  23             sys.exit(0)
  24         elif event.type == KEYDOWN:
  25             im = cam.get_image()
  26             pg_img = pygame.image.frombuffer(im.tostring(), im.size, im.mode)
  27             screen.blit(pg_img, (0,0))
  28             pygame.display.flip()