main.py

#!/usr/bin/python3
# coding: utf-8


from Display import Display 
from Cam import Cam 
from PIL import Image
from gpiozero import Button
from threading import Timer, Thread
import datetime
import time

resolution = (1440, 1050)

# time between pressing buzzer and the picture taken
BUZZER_DELAY = 2.0 

# Time between take_pic() and the actual DSLR click
CLICK_DELAY = 0.24

BUZZER_PIN = 3
BUTTON_UP_PIN = 5
BUTTON_DOWN_PIN = 6
BUTTON_NEXT = 10
BUTTON_PREV = 11

buzzer = Button(BUZZER_PIN)
button_up = Button(BUTTON_UP_PIN, bounce_time=0.1)
button_down = Button(BUTTON_DOWN_PIN, bounce_time=0.1)
button_next = Button(BUTTON_NEXT, bounce_time=0.1)
button_prev = Button(BUTTON_PREV, bounce_time=0.1)

date_str = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
pics_dir = './pics'
output_dir = '%s/%s' % (pics_dir, date_str)

pic_shown = -1 

# exponentially moving average of cam setting deltas
cam_setting_delta_ema = 0

def cancel_timer(timer):
    if timer != None:
        try:
            timer.cancel()
        except:
            pass

def process_image(cam, filename):
    global cam_setting_delta_ema
    config_delta = cam.check_brightness(filename)
    alpha = 0.2
    cam_setting_delta_ema = alpha * config_delta + (1-alpha) * cam_setting_delta_ema
    cam.store_pic(output_dir)
    print("cam_setting_delta_ema = %f"%cam_setting_delta_ema)
    print("delta = %f"%config_delta)
    if abs(cam_setting_delta_ema) > 0.5: 
#        cam.change_setting(round(cam_setting_delta_ema))
        cam_setting_delta_ema = 0

cam = Cam()
cam.set_pic_store_dir(output_dir)
button_up.when_pressed = cam.increase_brightness 
button_down.when_pressed = cam.decrease_brightness
with Display(resolution) as display:
    display.show_video_fullscreen()
    
#    with Image.open(".capture.jpg") as img:
#        display.show_image_fullscreen(img)

    show_video_small_timer = None
    show_history_timer = None

    def clear_timers():
        global show_video_small_timer
        global show_history_timer

        cancel_timer(show_video_small_timer)
        cancel_timer(show_history_timer)
        show_video_small_timer = None
        show_history_timer = None

    def show_history():
        global pic_shown
        pic_shown = -1
        pics = cam.get_all_pics(pics_dir)
        display.show_images_fullscreen(pics, 'minimize')

    def set_timers(show_small_video=True):
        global show_video_small_timer
        global show_history_timer

        if show_small_video:
            show_video_small_timer = Timer(10, display.show_video_small, ())
            show_video_small_timer.start()

        show_history_timer = Timer(40, show_history, ())
        show_history_timer.start()

    def on_buzzer_pushed():
        print("buzzer pushed")
        display.show_video_fullscreen()
        
        clear_timers()

        time.sleep(BUZZER_DELAY or CLICK_DELAY - CLICK_DELAY)
        Timer(CLICK_DELAY, display.flash,()).start() 
        
        try:
            filename = cam.take_pic()
        except:
            cam.stop()
            return

        with Image.open(filename) as img:
            display.show_image_fullscreen(img)
            img_processor = Thread(target=process_image, args=(cam, filename))
            img_processor.start()
            set_timers()

    def on_next_pushed():
        print("next pushed")
        global pic_shown

        clear_timers()

        images = cam.get_all_pics(pics_dir)
        if pic_shown == -1 or pic_shown == len(images) - 1:
            pic_shown = -1
            display.show_video_fullscreen()
            set_timers(False)
        else:
            if pic_shown > -1:
                pic_shown = pic_shown + 1
                
        if pic_shown > -1 and len(images) > 0: 
            with Image.open(cam.get_all_pics(pics_dir)[pic_shown]) as image:
                display.show_image_fullscreen(image)
            set_timers(True)

    def on_prev_pushed():
        print("prev pushed")
        global pic_shown
        clear_timers()

        images = cam.get_all_pics(pics_dir)
        if pic_shown > 0:
            pic_shown = pic_shown - 1
        else:
            pic_shown = len(images) - 1
        if len(images) > 0:
            with Image.open(cam.get_all_pics(pics_dir)[pic_shown]) as image:
                display.show_image_fullscreen(image)
        set_timers()
 

    buzzer.when_pressed = on_buzzer_pushed
    button_next.when_pressed = on_next_pushed
    button_prev.when_pressed = on_prev_pushed

    on_next_pushed()
#    on_buzzer_pushed()

    try:
        while cam.is_active():
            time.sleep(1)
    except (KeyboardInterrupt, SystemExit) as e:
        cancel_timer(show_video_small_timer)
        cancel_timer(show_history_timer)
        print("Closing app")