Compare commits

...

3 Commits
master ... dev

7 changed files with 346 additions and 301 deletions

3
.gitignore vendored
View File

@ -1,2 +1,5 @@
/.idea/
/venv/
/images/
/tmp?.jpg

27
FakeGPIO/__init__.py Normal file
View File

@ -0,0 +1,27 @@
BOARD = 1
HIGH = 1
LOW = 0
OUT = 1
IN = 0
FALLING = 0
RISING = 1
def setmode(x):
pass
def setwarnings(x):
pass
def setup(gpio, mode, initial=None):
pass
def output(gpio, state):
pass
def add_event_detect(gpio, edge, callback, bouncetime=None):
pass

BIN
FakeGPIO/__init__.pyc Normal file

Binary file not shown.

28
FakePicamera/__init__.py Normal file
View File

@ -0,0 +1,28 @@
from PIL import Image
class Overlay(object):
def __init__(self):
self.alpha = 0
class PiCamera(object):
def __init__(self):
self.resolution = None
def start_preview(self, *args, **kwargs):
pass
def add_overlay(self, *args, **kwargs):
return Overlay()
def remove_overlay(self, *args, **kwargs):
pass
def capture(self, filename, format, *args, **kwargs):
print("FAKE: capture(%s, %s)" % (filename, format))
img = Image.new("RGB", self.resolution, (255, 255, 255))
img.save(filename, format)
def close(self, *args, **kwargs):
pass

BIN
FakePicamera/__init__.pyc Normal file

Binary file not shown.

View File

@ -1,22 +1,28 @@
#!/usr/bin/python
import sys, os, inspect
import pytz, datetime
from time import sleep
from PIL import Image
from shutil import copyfile
import datetime
import imp
import inspect
import os
import sys
from math import floor
import RPi.GPIO as GPIO
from shutil import copyfile
from time import sleep
import picamera
import pygame
import random
import time
# ~ import server
from PIL import Image
try:
imp.find_module('RPi')
import RPi.GPIO as GPIO
except ImportError:
import FakeGPIO as GPIO
try:
imp.find_module('picamera')
import picamera
except ImportError:
import FakePicamera as picamera
__all__ = ["monotonic_time"]
@ -24,29 +30,29 @@ import ctypes
CLOCK_MONOTONIC_RAW = 4 # see <linux/time.h>
class timespec(ctypes.Structure):
_fields_ = [
('tv_sec', ctypes.c_long),
('tv_nsec', ctypes.c_long)
]
librt = ctypes.CDLL('librt.so.1', use_errno=True)
clock_gettime = librt.clock_gettime
clock_gettime.argtypes = [ctypes.c_int, ctypes.POINTER(timespec)]
def monotonic_time():
t = timespec()
if clock_gettime(CLOCK_MONOTONIC_RAW , ctypes.pointer(t)) != 0:
if clock_gettime(CLOCK_MONOTONIC_RAW, ctypes.pointer(t)) != 0:
errno_ = ctypes.get_errno()
raise OSError(errno_, os.strerror(errno_))
return t.tv_sec + t.tv_nsec * 1e-9
os.chdir(os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))))
# CONFIGURATION
WIDTH = 800
HEIGHT = 480
@ -75,14 +81,11 @@ THUMB_HEIGHT = int(THUMB_WIDTH / 1.333)
IMAGES_ORIGINAL = os.path.join(IMAGE_DIR, "original")
IMAGES_TODO = os.path.join(IMAGE_DIR, "todo")
# global variables
spot_mode = 0
session_start = monotonic_time()
# GPIO helpers
def led_border(state):
@ -102,14 +105,12 @@ def next_spot_mode():
set_spot_mode(spot_mode + 1)
# user interface
def button_press(gpio_id):
next_spot_mode()
# INIT GPIO
GPIO.setmode(GPIO.BOARD)
GPIO.setwarnings(False)
@ -136,18 +137,15 @@ led_border(True)
sleep(0.02)
led_border(False)
# INIT PYGAME
pygame.init()
pygame.mouse.set_visible(False)
screen = pygame.display.set_mode((WIDTH,HEIGHT))
screen = pygame.display.set_mode((WIDTH, HEIGHT))
# LOAD GRAPHICS
gfx = {}
def _load(name):
filename = "%s.png" % name
path = os.path.join(GRAPHICS_DIR, filename)
@ -157,6 +155,7 @@ def _load(name):
gfx[name] = (img, rect)
_load("begin")
_load("loading")
_load("yes")
@ -166,8 +165,6 @@ _load("end_no")
_load("saved")
_load("canceled")
# LOAD COUNTDOWN OVERLAYS
img_cdn = []
pad_cdn = []
@ -185,7 +182,6 @@ for i in range(0, COUNTDOWN):
pad_cdn.append(pad)
# FUNCTIONS ========================================================================================
def waitForEvent():
@ -201,7 +197,6 @@ def waitForEvent():
return events
def waitForTouch():
pos = (-1, -1)
while True:
@ -213,7 +208,6 @@ def waitForTouch():
return pos
def maintainance():
screen.fill(COLOR_BACK)
screen.blit(maintain, maintain_rect)
@ -223,7 +217,6 @@ def maintainance():
pos = waitForTouch()
def waitForBegin():
screen.fill(COLOR_BACK)
screen.blit(*gfx["begin"])
@ -235,17 +228,15 @@ def waitForBegin():
maintainance()
def countdown(camera, pad, img):
sleep(PIC_INTERVAL)
o = camera.add_overlay(pad.tostring(), size=img.size)
o = camera.add_overlay(pad.tobytes(), size=img.size)
o.alpha = OVERLAY_ALPHA
o.layer = 3
sleep(TEXT_INTERVAL)
camera.remove_overlay(o)
def makePhotos():
# CLEAR SCREEN
screen.fill(COLOR_BACK)
@ -284,7 +275,6 @@ def makePhotos():
camera.close()
def chooseImages():
# clear screen
screen.fill(COLOR_BACK)
@ -309,7 +299,6 @@ def chooseImages():
x = x + THUMB_WIDTH + BORDER
save = False
cancel = False
any_choice = False
@ -369,7 +358,6 @@ def chooseImages():
return (choices, save)
def saveImages(choices):
local = datetime.datetime.now()
session_age = (monotonic_time() - session_start)
@ -399,7 +387,6 @@ def saveImages(choices):
copyfile(path_original, path_todo)
try:
while 1:
waitForBegin()