felix/photobooth
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Additional config variables

Browse Source
This commit is contained in:
Balthasar Reuter
2015-05-06 23:10:11 +02:00
parent 36404699e9
commit 006b4e97c4
1 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
photobooth.py
View File

@@ -30,6 +30,12 @@ image_pose = "pose.png"
# Image basename # Image basename
image_basename = "pic" image_basename = "pic"
# Waiting time in seconds for posing
pose_time = 5
# Display time for taken pictures
display_time = 10
############### ###############
### Classes ### ### Classes ###
############### ###############
@@ -61,12 +67,12 @@ class GUI_PyGame:
pygame.mouse.set_visible(False) pygame.mouse.set_visible(False)
# Store screen and size # Store screen and size
self.size = size self.size = size
# Reset and initialize screen
self.reset()
def reset(self):
self.screen = pygame.display.set_mode(self.size) self.screen = pygame.display.set_mode(self.size)
self.screen.fill((255,255,255)) # Clear screen
self.clear()
def clear(self, color=(255,255,255)):
self.screen.fill(color)
def apply(self): def apply(self):
pygame.display.update() pygame.display.update()
@@ -155,11 +161,11 @@ def handle_keypress(key):
# Take pictures # Take pictures
elif key == ord('c'): elif key == ord('c'):
# Show pose message # Show pose message
display.reset() display.clear()
display.show_picture(image_pose) display.show_picture(image_pose)
display.show_message("POSE! Taking four pictures..."); display.show_message("POSE! Taking four pictures...");
display.apply() display.apply()
time.sleep(5) time.sleep(pose_time)
# Extract display and image sizes # Extract display and image sizes
size = display.get_size() size = display.get_size()
image_size = (int(size[0]/2), int(size[1]/2)) image_size = (int(size[0]/2), int(size[1]/2))
@@ -173,15 +179,15 @@ def handle_keypress(key):
display.show_picture(filenames[2], image_size, (0,image_size[1])) display.show_picture(filenames[2], image_size, (0,image_size[1]))
display.show_picture(filenames[3], image_size, (image_size[0],image_size[1])) display.show_picture(filenames[3], image_size, (image_size[0],image_size[1]))
display.apply() display.apply()
time.sleep(10) time.sleep(display_time)
display.reset() display.clear()
def handle_exception(msg): def handle_exception(msg):
display.reset() display.clear()
display.show_message("Error: " + msg) display.show_message("Error: " + msg)
display.apply() display.apply()
time.sleep(3) time.sleep(3)
display.reset() display.clear()
def teardown(exit_code=0): def teardown(exit_code=0):
display.teardown() display.teardown()