focus-snap.py

import piggyphoto, pygame
import os
import time
import Image, ImageFilter, ImageChops, ImageStat
from collections import deque

import focus

def quit_pressed():
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            return True
    return False

def show(file):
    picture = pygame.image.load(file)
    picture = pygame.transform.scale(picture, (1056,704))
    main_surface.blit(picture, (0, 0))
    pygame.display.flip()

C = piggyphoto.Camera()
C.leave_locked()
C.capture_preview('preview.jpg')

picture = pygame.image.load("preview.jpg")
pygame.display.set_mode(picture.get_size())
main_surface = pygame.display.get_surface()

Q = deque()
k = 1
looking_for_peak = True

while not quit_pressed():
    C.capture_preview('preview.jpg')
    show("preview.jpg")
    f = focus.estimate("preview.jpg")
    Q.append(f)
    if len(Q) > 20: 
        Q.popleft()
        
    F = max(Q)
    pygame.display.set_caption("Focus: %.4g / %.4g / %s" % (f, F, "Looking for focus peak" if looking_for_peak else "Focus peak found"))
    
    if len(Q) == 20:
        if looking_for_peak:
            if f < F * 0.8:
                looking_for_peak = False
        else:
            if f > F * 0.98:
                C.capture_image("snap%d.jpg" % k)
                show("snap%d.jpg" % k)
                k = k + 1
                Q = deque()
                time.sleep(1)
                looking_for_peak = True

C.close()