capture.py

   1 #!/usr/bin/env python
   2
   3 """Go image capture"""
   4
   5 import os
   6 import sys
   7 import argparse
   8 import time
   9
  10 import pygame
  11 from pygame.locals import QUIT, KEYDOWN
  12 import Image
  13
  14 from camera import Camera
  15
  16 class Screen:
  17     def __init__(self, res):
  18         pygame.init()
  19         pygame.display.set_mode(res)
  20         pygame.display.set_caption("Go image capture")
  21         self._screen = pygame.display.get_surface()
  22
  23     def display_picture(self, im):
  24         pg_img = pygame.image.frombuffer(im.tostring(), im.size, im.mode)
  25         self._screen.blit(pg_img, (0,0))
  26         pygame.display.flip()
  27
  28
  29 def main():
  30     parser = argparse.ArgumentParser(description=__doc__)
  31     parser.add_argument('-d', type=int, default=0,
  32                         help="video device id")
  33     parser.add_argument('-a', type=int, default=0,
  34                         help="take picture automaticaly every A seconds")
  35     parser.add_argument('-r', type=int, nargs=2, default=[640, 480],
  36                         help="set camera resolution")
  37     args = parser.parse_args()
  38
  39     res=(args.r[0], args.r[1])
  40
  41     cam = Camera(vid=args.d, res=res)
  42     screen = Screen(res)
  43
  44     im_number = 0
  45
  46     saving_dir = "./captured/" + time.strftime("%Y-%m-%d %H:%M/")
  47
  48     if not os.path.isdir(saving_dir):
  49         os.makedirs(saving_dir)
  50
  51     done = False
  52     clock = pygame.time.Clock()
  53     while not done: #live preview
  54         for event in pygame.event.get():
  55             if event.type == pygame.QUIT:
  56                 done = True
  57                 del cam
  58                 sys.exit()
  59             if event.type == pygame.KEYDOWN:
  60                 done = True
  61
  62         im = cam.get_image()
  63         screen.display_picture(im)
  64         clock.tick(5)
  65
  66
  67     if args.a > 0:
  68         last = 0
  69         clock = pygame.time.Clock()
  70         done = False
  71
  72         while not done:
  73             for event in pygame.event.get():
  74                 if event.type == pygame.QUIT:
  75                     done = True
  76             if time.time() - last > args.a:
  77                 last = time.time()
  78                 im = cam.get_image()
  79                 screen.display_picture(im)
  80                 im.save(saving_dir + "{0:0>3}.jpg".format(im_number), 'JPEG')
  81                 im_number += 1
  82             clock.tick(15)
  83
  84     else:
  85         while True:
  86             event = pygame.event.wait()
  87             if event.type == QUIT:
  88                 break
  89             if event.type != KEYDOWN:
  90                 continue
  91
  92             im = cam.get_image()
  93             screen.display_picture(im)
  94             im.save(saving_dir + "{0:0>3}.jpg".format(im_number), 'JPEG')
  95             im_number += 1
  96
  97     del cam
  98
  99 if __name__ == '__main__':
 100     try:
 101         main()
 102     finally:
 103         pygame.quit()