def __init__(self, photobooth):
self.photobooth = photobooth
self.screen = photobooth.get_pygame_screen()
self.buttonhandler = photobooth.get_button_handler()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
screen_colour_fill(self.screen, config.white_colour)
def __init__(self, photobooth):
self.photobooth = photobooth
self.screen = photobooth.get_pygame_screen()
self.buttonhandler = photobooth.get_button_handler()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
screen_colour_fill(self.screen, config.white_colour)
def __init__(self, photobooth):
self.menu_text = "Create photo animation"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.gif_delay = 100
self.photo_width = 500
self.capture_delay = 2
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
def __init__(self, photobooth):
self.menu_text = "Take official profile photo (for Learn etc.)"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
# Set image definitions - width, height, dpi
self.image_defs = [['learn', 150, 150, 300], ['pure', 160, 185, 300],
['eevec', 100, 150, 300],
['office365', 300, 300, 300]]
def __init__(self, photobooth):
self.menu_text = "Take accompanied photo"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
self.chosen_accompaniment = 0
self.accompaniment_dir = self.filehandler.get_full_path(
config.images_dir, 'accompany')
self.accompany_button_overlay_image = self.filehandler.get_full_path(
config.images_dir, 'accompany_button_overlay.png')
def __init__(self, photobooth):
self.menu_text = "Take continuous photos"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
self.total_pics = 17280 # 24 hours of a photo every 5 seconds
self.capture_delay = 5
self.difference_threshold = 40
def __init__(self, photobooth):
self.menu_text = "Create photo animation"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.gif_delay = 100
self.photo_width = 500
self.capture_delay = 2
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
def __init__(self, photobooth):
self.menu_text = "Take accompanied photo"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
self.chosen_accompaniment = 0
self.accompaniment_dir = self.filehandler.get_full_path(config.images_dir, 'accompany')
self.accompany_button_overlay_image = self.filehandler.get_full_path(config.images_dir,
'accompany_button_overlay.png')
def __init__(self, photobooth):
self.menu_text = "Take official profile photo (for Learn etc.)"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
# Set image definitions - width, height, dpi
self.image_defs = [
['learn', 150, 150, 300],
['pure', 160, 185, 300],
['eevec', 100, 150, 300],
['office365', 300, 300, 300]
]
class Menus(object):
'A class to handle the Main Menu'
photobooth = None
buttonhandler = None
# Set the text attributes for the main menu heading
heading_font_colour = config.blue_colour
# Set the text attributes for the menu item display
menu_font_size = 64
menu_font_colour = config.black_colour
menu_item_alignment = "lm"
menu_item_position = 20
menu_cursor_font_size = 64
menu_cursor_font_colour = config.blue_colour
menu_item_line_spacing = 20
menu_option_rects = None
# menu_objects holds instances of each Photo Booth function class that
# is added to the Main Menu
menu_objects = []
def __init__(self, photobooth):
self.photobooth = photobooth
self.screen = photobooth.get_pygame_screen()
self.buttonhandler = photobooth.get_button_handler()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
screen_colour_fill(self.screen, config.white_colour)
def __del__(self):
print "Destructing Menus instance"
self.photobooth = None
def add_main_menu_item(self, item_class):
self.menu_objects.append(item_class)
def display_main_menu(self):
self.text_defs = []
self.image_defs = []
# Print the heading on the screen
self.textprinter.print_text([[
"Welcome to the Photo Booth", 120, self.heading_font_colour, "ct",
5
]], 0, True)
# Print the main menu items
for item_class in self.menu_objects:
self.text_defs.append([
item_class.get_menu_text(), self.menu_font_size,
self.menu_font_colour, self.menu_item_alignment,
self.menu_item_position
])
self.menu_option_rects = self.textprinter.print_text(
self.text_defs, self.menu_item_line_spacing, False)
# Print the image overlays onto the screen
self.image_defs = [[config.go_up_overlay_image, 'lb', 0, 0],
[config.go_down_overlay_image, 'rb', 0, 0],
[config.select_overlay_image, 'cb', 0, 0]]
self.imageprinter.print_images(self.image_defs, False)
def get_main_menu_selection(self):
self.cursorprinter = CursorPrinter(self.screen,
self.menu_cursor_font_size,
self.menu_cursor_font_colour)
self.menu_choice = 0
# Print the initial cursor at the first menu option
self.cursorprinter.print_cursor(self.menu_option_rects,
self.menu_choice)
while True:
self.button = self.buttonhandler.wait_for_buttons('lsr', False)
if self.button == 'l':
if self.menu_choice > 0:
self.menu_choice -= 1
self.cursorprinter.print_cursor(self.menu_option_rects,
self.menu_choice)
if self.button == 'r':
if self.menu_choice < len(self.menu_option_rects) - 1:
self.menu_choice += 1
self.cursorprinter.print_cursor(self.menu_option_rects,
self.menu_choice)
if self.button == 's':
self.buttonhandler.light_button_leds('lsr', False)
break
if self.button == 'exit':
# The user pressed the exit button - how long did they keep it pressed for?
self.start_time = time.time()
time.sleep(0.2)
self.menu_choice = -1 # -1 indicates a short exit button
while self.buttonhandler.button_is_down(
config.button_pin_exit):
time.sleep(0.2)
# If the exit button is held down for longer than 3 seconds
# then record a 'long exit button press'
if time.time() - self.start_time > 3:
self.menu_choice = -2 # -2 indicates a long exit button press
break
break
# If we have been sitting at the Main Menu for longer than screen_saver_seconds secs
# then go into screen_saver mode.
if self.button == 'screensaver':
print "Monitor going into screen saver mode."
self.photobooth.screen_saver() # HACK
pass
return self.menu_choice
def get_menu_object_at_index(self, object_index):
return self.menu_objects[object_index]
class Menus(object):
'A class to handle the Main Menu'
photobooth = None
buttonhandler = None
# Set the text attributes for the main menu heading
heading_font_colour = config.blue_colour
# Set the text attributes for the menu item display
menu_font_size = 64
menu_font_colour = config.black_colour
menu_item_alignment = "lm"
menu_item_position = 20
menu_cursor_font_size = 64
menu_cursor_font_colour = config.blue_colour
menu_item_line_spacing = 20
menu_option_rects = None
# menu_objects holds instances of each Photo Booth function class that
# is added to the Main Menu
menu_objects = []
def __init__(self, photobooth):
self.photobooth = photobooth
self.screen = photobooth.get_pygame_screen()
self.buttonhandler = photobooth.get_button_handler()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
screen_colour_fill(self.screen, config.white_colour)
def __del__(self):
print "Destructing Menus instance"
self.photobooth = None
def add_main_menu_item(self, item_class):
self.menu_objects.append(item_class)
def display_main_menu(self):
self.text_defs = []
self.image_defs = []
# Print the heading on the screen
self.textprinter.print_text( [["Welcome to the Photo Booth",
120,
self.heading_font_colour,
"ct",
5]],
0,
True )
# Print the main menu items
for item_class in self.menu_objects:
self.text_defs.append( [item_class.get_menu_text(),
self.menu_font_size,
self.menu_font_colour,
self.menu_item_alignment,
self.menu_item_position] )
self.menu_option_rects = self.textprinter.print_text( self.text_defs,
self.menu_item_line_spacing,
False )
# Print the image overlays onto the screen
self.image_defs = [
[config.go_up_overlay_image, 'lb', 0, 0],
[config.go_down_overlay_image, 'rb', 0, 0],
[config.select_overlay_image, 'cb', 0, 0]
]
self.imageprinter.print_images(self.image_defs, False)
def get_main_menu_selection(self):
self.cursorprinter = CursorPrinter(self.screen, self.menu_cursor_font_size,
self.menu_cursor_font_colour)
self.menu_choice = 0
# Print the initial cursor at the first menu option
self.cursorprinter.print_cursor(self.menu_option_rects, self.menu_choice)
while True:
self.button = self.buttonhandler.wait_for_buttons('lsr', False)
if self.button == 'l':
if self.menu_choice > 0:
self.menu_choice -= 1
self.cursorprinter.print_cursor( self.menu_option_rects, self.menu_choice)
if self.button == 'r':
if self.menu_choice < len(self.menu_option_rects) - 1:
self.menu_choice += 1
self.cursorprinter.print_cursor( self.menu_option_rects, self.menu_choice)
if self.button == 's':
self.buttonhandler.light_button_leds('lsr', False)
break
if self.button == 'exit':
# The user pressed the exit button - how long did they keep it pressed for?
self.start_time = time.time()
time.sleep(0.2)
self.menu_choice = -1 # -1 indicates a short exit button
while self.buttonhandler.button_is_down(config.button_pin_exit):
time.sleep(0.2)
# If the exit button is held down for longer than 3 seconds
# then record a 'long exit button press'
if time.time() - self.start_time > 3:
self.menu_choice = -2 # -2 indicates a long exit button press
break
break
# If we have been sitting at the Main Menu for longer than screen_saver_seconds secs
# then go into screen_saver mode.
if self.button == 'screensaver':
print "Monitor going into screen saver mode."
self.photobooth.screen_saver() # HACK
pass
return self.menu_choice
def get_menu_object_at_index(self, object_index):
return self.menu_objects[object_index]
class AnimatedPhoto(PhotoBoothFunction):
'Class to take a series of photographs and combine them into an animated gif'
def __init__(self, photobooth):
self.menu_text = "Create photo animation"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.gif_delay = 100
self.photo_width = 500
self.capture_delay = 2
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
def start(self, total_pics=PhotoBoothFunction.total_pics):
# Take and display photos
self.total_pics = total_pics
# Display the instructions for this photobooth function
total_pics_msg = str(self.total_pics) + " photo"
if self.total_pics > 1:
total_pics_msg += "s"
self.instructions = [
total_pics_msg + " will be taken",
"(red light will appear before each photo)",
"The photos are then animated together",
"Press the Start button to begin"
]
# Outer loop: Instructions screen
while True:
choice = self.display_instructions()
# If the user selected Exit, bail out
if choice == "l":
break
while True:
self.take_photos()
# NOTE: If you change any part of this text display, mirror changes in erase print below
text_rect_list = self.textprinter.print_text([["Processing. Please wait ...", 84,
config.white_colour, "cb", 10]],
0, False)
# Convert the images into an animated GIF (in a separate thread)
gif_thread = threading.Thread(target=self.process_photos)
gif_thread.start()
# Now display the original images one after the other in separate thread,
# to give the impression they are animated
# (easier then trying to display the animated GIF in pygame?)
display_thread_stop = threading.Event()
display_thread = threading.Thread(target=self.photohandler.show_photos_animated,
args=(self.image_extension, display_thread_stop))
display_thread.start()
# Wait until the thread that's creating and uploading the GIF finishes
gif_thread.join()
# Overwrite the 'Processing' message in black to erase it
screen_colour_fill(self.screen, config.black_colour, text_rect_list[0])
#self.textprinter.print_text([["Processing. Please wait ...", 84, config.black_colour, "cb", 10]],
# 0, False)
# Print the button label overlays
self.imageprinter.print_images([[config.start_overlay_image_bk_black, 'cb', 0, 0]], False)
self.imageprinter.print_images([[config.exit_side_overlay_image_bk_black, 'lb', 0, 0]],
False)
# Wait for the user to click Exit or Start
choice = ""
while True:
choice = self.buttonhandler.wait_for_buttons('ls', True)
if choice != 'screensaver':
break
# Now we can stop the thread that's animating the images on the screen
display_thread_stop.set()
display_thread.join()
# If user pressed Exit button, return to Photo Animation Instruction screen
if choice == 'l':
break
def take_photos(self):
################################# Step 1 - Initial Preparation ##########################
super(AnimatedPhoto, self).take_photos()
################################# Step 2 - Setup camera #################################
pixel_width = self.photo_width
pixel_height = self.photohandler.get_aspect_ratio_height(pixel_width)
self.camera.resolution = (pixel_width, pixel_height)
################################# Step 3 - Start camera preview ########################
screen_colour_fill(self.screen, config.black_colour)
self.camera.start_preview()
################################# Step 4 - Prepare the user ########################
time.sleep(self.prep_delay_short)
################################# Step 5 - Take Photos ################################
self.take_photos_and_close_camera(self.capture_delay)
def process_photos(self):
self.combine_into_gif()
self.create_thumbnail()
self.upload_photos()
# Create a thumbnail of the first image in the series
def create_thumbnail(self):
thumb_width = 200
file_list = self.filehandler.get_sorted_file_list(os.path.join(self.local_file_dir, '*' +
self.image_extension))
img = self.photohandler.resize_image(file_list[0], thumb_width, thumb_width)
img.save(os.path.join(self.local_file_dir, 'photobooth_thumb' + self.image_extension))
def upload_photos(self):
remote_upload_dir = time.strftime("%d-%b-%Y_A")
if self.booth_id is not "":
remote_upload_dir = self.booth_id + "_" + remote_upload_dir
file_suffix = time.strftime("%H-%M-%S")
file_defs = [
# Upload the animated GIF
[os.path.join(self.local_upload_file_dir, '*' + self.animated_image_extension),
'photobooth_photo_' + file_suffix,
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Upload just the first of the photo files
[os.path.join(self.local_file_dir, 'photobooth_thumb' + self.image_extension),
'photobooth_photo_' + file_suffix,
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Upload the HTML files to handle the animated photos
[os.path.join('html', 'individual', 'index-animated.php'), 'index',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
[os.path.join('html', 'individual', 'img-animated.php'), '',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Make sure the base .htaccess and index files are in place
[os.path.join('html', 'index.php'), '',
self.remote_file_dir, 1, True],
[os.path.join('html', 'redirect.html'), '',
self.remote_file_dir, 1, True],
[os.path.join('html', '.htaccess'), '',
self.remote_file_dir, 1, True],
# Make sure that all common files are in place
[os.path.join('html', 'common', '*'), '',
os.path.join(self.remote_file_dir, 'common'), 0, True],
]
success = self.upload_photos_using_defs(file_defs)
if success:
return remote_upload_dir
else:
return None
def combine_into_gif(self):
local_file_dir = self.filehandler.get_local_file_dir()
print "Creating animated GIF ..."
# Thanks to drumminhands
graphicsmagick = ("gm convert -delay " + str(self.gif_delay) + " " +
local_file_dir + "/*.jpg " +
self.local_upload_file_dir + "/photobooth.gif")
os.system(graphicsmagick) #make the .gif
class AccompaniedPhoto(PhotoBoothFunction):
'Class to take a photograph with a companion'
def __init__(self, photobooth):
self.menu_text = "Take accompanied photo"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
self.chosen_accompaniment = 0
self.accompaniment_dir = self.filehandler.get_full_path(config.images_dir, 'accompany')
self.accompany_button_overlay_image = self.filehandler.get_full_path(config.images_dir,
'accompany_button_overlay.png')
def start(self, total_pics=PhotoBoothFunction.total_pics):
# Take and display photos
self.total_pics = total_pics
# Display the instructions for this photobooth function
total_pics_msg = str(self.total_pics) + " photo"
if self.total_pics > 1:
total_pics_msg += "s"
self.instructions = [
"Press Left & Right to change companion",
"Press Select button to choose companion",
total_pics_msg + " will be taken",
"(red light will appear before each photo)",
"Press the Start button to begin"
]
choice = self.display_instructions()
# If the user selected Exit, bail out
if choice == "l":
return
self.take_photos()
self.photohandler.show_photos_tiled(self.image_extension)
time.sleep(2)
choice = self.user_accept_photos()
# See if user wants to accept photos
if (choice == 'r'):
self.process_photos()
remote_upload_dir = self.upload_photos()
if remote_upload_dir is None:
self.display_upload_failed_message()
else:
remote_url_prefix = self.filehandler.get_remote_url_prefix()
self.display_download_url(remote_url_prefix, remote_upload_dir)
else:
self.display_rejected_message()
def take_photos(self):
################################# Step 1 - Initial Preparation ##########################
super(AccompaniedPhoto, self).take_photos()
################################# Step 2 - Setup camera #################################
# Collect together the PNG versions of all available accompaniment files
file_pattern = self.filehandler.get_full_path(self.accompaniment_dir, "*.png")
self.accompaniment_files = self.filehandler.get_sorted_file_list(file_pattern)
################################# Step 2 - Setup camera #################################
pixel_width = self.photo_width
pixel_height = self.photohandler.get_aspect_ratio_height(pixel_width)
self.camera.resolution = (pixel_width, pixel_height)
################################# Step 3 - Start camera preview ########################
screen_colour_fill(self.screen, config.black_colour)
self.camera.start_preview()
################################# Step 4 - User make selection ########################
self.choose_accompaniment()
time.sleep(self.prep_delay_long)
################################# Step 5 - Take Photos ################################
self.take_photos_and_close_camera(self.capture_delay)
def manipulate_photo(self, filepath):
# Superimpose the accompanying image onto the captured image
# http://effbot.org/imagingbook/image.htm
# super_image is RGBA, so use it both for image and mask
if self.chosen_accompaniment > 1 and self.chosen_accompaniment < (len(self.accompaniment_files) + 2):
curr_accompaniment_file = self.accompaniment_files[self.chosen_accompaniment - 2]
curr_img = Image.open(filepath)
super_img = Image.open(curr_accompaniment_file)
curr_img.paste(super_img, None, super_img)
curr_img.save(filepath)
def upload_photos(self):
self.textprinter.print_text([["Uploading photos ...", 124, config.black_colour, "cm", 0]],
0, True)
remote_upload_dir = StringOperations().get_random_string(10)
file_defs = [
# Upload the ZIP archive of photos
[os.path.join(self.local_upload_file_dir, '*.zip'), '',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Upload just the first of the photo files
[os.path.join(self.local_file_dir, '*' + self.image_extension), 'photobooth_photo',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Upload the HTML file for this particular set of photos
[os.path.join('html', 'individual', 'index-single.html'), 'index',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Make sure the base .htaccess and index files are in place
[os.path.join('html', 'index.php'), '',
self.remote_file_dir, 1, True],
[os.path.join('html', 'redirect.html'), '',
self.remote_file_dir, 1, True],
[os.path.join('html', '.htaccess'), '',
self.remote_file_dir, 1, True],
# Make sure that all common files are in place
[os.path.join('html', 'common', '*.css'), '',
os.path.join(self.remote_file_dir, 'common'), 0, True],
]
success = self.upload_photos_using_defs(file_defs)
if success:
return remote_upload_dir
else:
return None
# Let the user select which image they want to be photographed with
def choose_accompaniment(self):
# Start with the first image (if there are any)
self.chosen_accompaniment = 2
# Use the JPG versions of accompaniment files, which have black background
file_pattern = self.filehandler.get_full_path(self.accompaniment_dir, "*.jpg")
files = self.filehandler.get_sorted_file_list(file_pattern)
# If there are no images, then chosen_accompaniment will be the blank screen
if len(files) < 1:
return 1
button_overlay = OverlayOnCamera(self.camera)
button_overlay.camera_overlay(self.accompany_button_overlay_image)
self.overlay_on_camera = OverlayOnCamera(self.camera)
self.change_accompaniment(files)
while True:
choice = self.buttonhandler.wait_for_buttons('lsr', False)
if choice == 'l':
if self.chosen_accompaniment > 1:
self.chosen_accompaniment -= 1
else:
self.chosen_accompaniment = len(files) + 1
self.change_accompaniment(files)
if choice == 'r':
if self.chosen_accompaniment < (len(files) + 1):
self.chosen_accompaniment += 1
else:
self.chosen_accompaniment = 1
self.change_accompaniment(files)
if choice == 's':
self.buttonhandler.light_button_leds('lsr', False)
break
button_overlay.remove_camera_overlay()
def change_accompaniment(self, files):
self.camera.saturation = 0
# We leave a blank space for chosen_accompaniment == 1
accompanying_file_num = self.chosen_accompaniment - 2
# If chosen_accompaniment == 1, then we don't want any images overlaid
if accompanying_file_num < 0 or accompanying_file_num > (len(files) - 1):
self.overlay_on_camera.remove_camera_overlay()
return
curr_accompaniment_file = files[accompanying_file_num]
self.overlay_on_camera.camera_overlay(curr_accompaniment_file)
# See if an opacity value in the filename [within square brackets]
filename = os.path.basename(curr_accompaniment_file)
opacity = filename[filename.find("[")+1:filename.find("]")]
if len(opacity) > 0:
opacity = int(opacity)
if opacity < -100:
opacity = -100
if opacity > 100:
opacity = 100
self.camera.saturation = opacity
class OfficialPhoto(PhotoBoothFunction):
'Class to take official portrait photographs'
def __init__(self, photobooth):
self.menu_text = "Take official profile photo (for Learn etc.)"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
# Set image definitions - width, height, dpi
self.image_defs = [
['learn', 150, 150, 300],
['pure', 160, 185, 300],
['eevec', 100, 150, 300],
['office365', 300, 300, 300]
]
def start(self, total_pics=PhotoBoothFunction.total_pics):
# Take and display photos
self.total_pics = total_pics
# Display the instructions for this photobooth function
total_pics_msg = str(self.total_pics) + " photo"
if self.total_pics > 1:
total_pics_msg += "s"
self.instructions = [
"A template will appear on screen",
"Following that, " + total_pics_msg + " will be taken",
"(red light will appear before each photo)",
"Press the Start button to begin"
]
choice = self.display_instructions()
# If the user selected Exit, bail out
if choice == "l":
return
self.take_photos()
self.photohandler.show_photos_tiled(self.image_extension)
time.sleep(2)
choice = self.user_accept_photos()
# See if user wants to accept photos
if (choice == 'r'):
self.process_photos()
remote_upload_dir = self.upload_photos()
if remote_upload_dir is None:
self.display_upload_failed_message()
else:
remote_url_prefix = self.filehandler.get_remote_url_prefix()
self.display_download_url(remote_url_prefix, remote_upload_dir)
else:
self.display_rejected_message()
def take_photos(self):
################################# Step 1 - Initial Preparation ##########################
super(OfficialPhoto, self).take_photos()
################################# Step 2 - Setup camera #################################
# Make the image square, using the photo_width
pixel_width = self.photo_width
pixel_height = self.photo_width
self.camera.resolution = (pixel_width, pixel_height)
################################# Step 3 - Start camera preview ########################
screen_colour_fill(self.screen, config.black_colour)
self.camera.start_preview()
################################# Step 4 - Prepare user ################################
self.overlay_on_camera = OverlayOnCamera(self.camera)
# Apply overlay images & messages to prepare the user
self.overlay_on_camera.camera_overlay(config.face_target_overlay_image)
time.sleep(self.prep_delay_short)
self.overlay_on_camera.camera_overlay(config.face_target_fit_face_overlay_image)
time.sleep(self.prep_delay_long)
self.overlay_on_camera.camera_overlay(config.face_target_overlay_image)
time.sleep(self.prep_delay_short)
self.overlay_on_camera.camera_overlay(config.face_target_smile_overlay_image)
time.sleep(self.prep_delay_long)
self.overlay_on_camera.camera_overlay(config.face_target_overlay_image)
time.sleep(self.prep_delay_short)
################################# Step 5 - Take Photos ################################
self.take_photos_and_close_camera(self.capture_delay)
def upload_photos(self):
self.textprinter.print_text([["Uploading photos ...", 124, config.black_colour, "cm", 0]],
0, True)
remote_upload_dir = StringOperations().get_random_string(10)
file_defs = [
# Upload the ZIP archive of photos
[os.path.join(self.local_upload_file_dir, '*.zip'), '',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Upload just the first of the photo files
[os.path.join(self.local_file_dir, '*' + self.image_extension), 'photobooth_photo',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Upload the HTML file for this particular set of photos
[os.path.join('html', 'individual', 'index-single.html'), 'index',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True],
# Make sure the base .htaccess and index files are in place
[os.path.join('html', 'index.php'), '',
self.remote_file_dir, 1, True],
[os.path.join('html', 'redirect.html'), '',
self.remote_file_dir, 1, True],
[os.path.join('html', '.htaccess'), '',
self.remote_file_dir, 1, True],
# Make sure that all common files are in place
[os.path.join('html', 'common', '*.css'), '',
os.path.join(self.remote_file_dir, 'common'), 0, True],
]
success = self.upload_photos_using_defs(file_defs)
if success:
return remote_upload_dir
else:
return None
class OfficialPhoto(PhotoBoothFunction):
'Class to take official portrait photographs'
def __init__(self, photobooth):
self.menu_text = "Take official profile photo (for Learn etc.)"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
# Set image definitions - width, height, dpi
self.image_defs = [['learn', 150, 150, 300], ['pure', 160, 185, 300],
['eevec', 100, 150, 300],
['office365', 300, 300, 300]]
def start(self, total_pics=PhotoBoothFunction.total_pics):
# Take and display photos
self.total_pics = total_pics
# Display the instructions for this photobooth function
total_pics_msg = str(self.total_pics) + " photo"
if self.total_pics > 1:
total_pics_msg += "s"
self.instructions = [
"A template will appear on screen",
"Following that, " + total_pics_msg + " will be taken",
"(red light will appear before each photo)",
"Press the Start button to begin"
]
choice = self.display_instructions()
# If the user selected Exit, bail out
if choice == "l":
return
self.take_photos()
self.photohandler.show_photos_tiled(self.image_extension)
time.sleep(2)
choice = self.user_accept_photos()
# See if user wants to accept photos
if (choice == 'r'):
self.process_photos()
remote_upload_dir = self.upload_photos()
if remote_upload_dir is None:
self.display_upload_failed_message()
else:
remote_url_prefix = self.filehandler.get_remote_url_prefix()
self.display_download_url(remote_url_prefix, remote_upload_dir)
else:
self.display_rejected_message()
def take_photos(self):
################################# Step 1 - Initial Preparation ##########################
super(OfficialPhoto, self).take_photos()
################################# Step 2 - Setup camera #################################
# Make the image square, using the photo_width
pixel_width = self.photo_width
pixel_height = self.photo_width
self.camera.resolution = (pixel_width, pixel_height)
################################# Step 3 - Start camera preview ########################
screen_colour_fill(self.screen, config.black_colour)
self.camera.start_preview()
################################# Step 4 - Prepare user ################################
self.overlay_on_camera = OverlayOnCamera(self.camera)
# Apply overlay images & messages to prepare the user
self.overlay_on_camera.camera_overlay(config.face_target_overlay_image)
time.sleep(self.prep_delay_short)
self.overlay_on_camera.camera_overlay(
config.face_target_fit_face_overlay_image)
time.sleep(self.prep_delay_long)
self.overlay_on_camera.camera_overlay(config.face_target_overlay_image)
time.sleep(self.prep_delay_short)
self.overlay_on_camera.camera_overlay(
config.face_target_smile_overlay_image)
time.sleep(self.prep_delay_long)
self.overlay_on_camera.camera_overlay(config.face_target_overlay_image)
time.sleep(self.prep_delay_short)
################################# Step 5 - Take Photos ################################
self.take_photos_and_close_camera(self.capture_delay)
def upload_photos(self):
self.textprinter.print_text(
[["Uploading photos ...", 124, config.black_colour, "cm", 0]], 0,
True)
remote_upload_dir = StringOperations().get_random_string(10)
file_defs = [
# Upload the ZIP archive of photos
[
os.path.join(self.local_upload_file_dir, '*.zip'), '',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Upload just the first of the photo files
[
os.path.join(self.local_file_dir, '*' + self.image_extension),
'photobooth_photo',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Upload the HTML file for this particular set of photos
[
os.path.join('html', 'individual', 'index-single.html'),
'index',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Make sure the base .htaccess and index files are in place
[
os.path.join('html', 'index.php'), '', self.remote_file_dir, 1,
True
],
[
os.path.join('html', 'redirect.html'), '',
self.remote_file_dir, 1, True
],
[
os.path.join('html', '.htaccess'), '', self.remote_file_dir, 1,
True
],
# Make sure that all common files are in place
[
os.path.join('html', 'common', '*.css'), '',
os.path.join(self.remote_file_dir, 'common'), 0, True
],
]
success = self.upload_photos_using_defs(file_defs)
if success:
return remote_upload_dir
else:
return None
class AccompaniedPhoto(PhotoBoothFunction):
'Class to take a photograph with a companion'
def __init__(self, photobooth):
self.menu_text = "Take accompanied photo"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
self.chosen_accompaniment = 0
self.accompaniment_dir = self.filehandler.get_full_path(
config.images_dir, 'accompany')
self.accompany_button_overlay_image = self.filehandler.get_full_path(
config.images_dir, 'accompany_button_overlay.png')
def start(self, total_pics=PhotoBoothFunction.total_pics):
# Take and display photos
self.total_pics = total_pics
# Display the instructions for this photobooth function
total_pics_msg = str(self.total_pics) + " photo"
if self.total_pics > 1:
total_pics_msg += "s"
self.instructions = [
"Press Left & Right to change companion",
"Press Select button to choose companion",
total_pics_msg + " will be taken",
"(red light will appear before each photo)",
"Press the Start button to begin"
]
choice = self.display_instructions()
# If the user selected Exit, bail out
if choice == "l":
return
self.take_photos()
self.photohandler.show_photos_tiled(self.image_extension)
time.sleep(2)
choice = self.user_accept_photos()
# See if user wants to accept photos
if (choice == 'r'):
self.process_photos()
remote_upload_dir = self.upload_photos()
if remote_upload_dir is None:
self.display_upload_failed_message()
else:
remote_url_prefix = self.filehandler.get_remote_url_prefix()
self.display_download_url(remote_url_prefix, remote_upload_dir)
else:
self.display_rejected_message()
def take_photos(self):
################################# Step 1 - Initial Preparation ##########################
super(AccompaniedPhoto, self).take_photos()
################################# Step 2 - Setup camera #################################
# Collect together the PNG versions of all available accompaniment files
file_pattern = self.filehandler.get_full_path(self.accompaniment_dir,
"*.png")
self.accompaniment_files = self.filehandler.get_sorted_file_list(
file_pattern)
################################# Step 2 - Setup camera #################################
pixel_width = self.photo_width
pixel_height = self.photohandler.get_aspect_ratio_height(pixel_width)
self.camera.resolution = (pixel_width, pixel_height)
################################# Step 3 - Start camera preview ########################
screen_colour_fill(self.screen, config.black_colour)
self.camera.start_preview()
################################# Step 4 - User make selection ########################
self.choose_accompaniment()
time.sleep(self.prep_delay_long)
################################# Step 5 - Take Photos ################################
self.take_photos_and_close_camera(self.capture_delay)
def manipulate_photo(self, filepath):
# Superimpose the accompanying image onto the captured image
# http://effbot.org/imagingbook/image.htm
# super_image is RGBA, so use it both for image and mask
if self.chosen_accompaniment > 1 and self.chosen_accompaniment < (
len(self.accompaniment_files) + 2):
curr_accompaniment_file = self.accompaniment_files[
self.chosen_accompaniment - 2]
curr_img = Image.open(filepath)
super_img = Image.open(curr_accompaniment_file)
curr_img.paste(super_img, None, super_img)
curr_img.save(filepath)
def upload_photos(self):
self.textprinter.print_text(
[["Uploading photos ...", 124, config.black_colour, "cm", 0]], 0,
True)
remote_upload_dir = StringOperations().get_random_string(10)
file_defs = [
# Upload the ZIP archive of photos
[
os.path.join(self.local_upload_file_dir, '*.zip'), '',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Upload just the first of the photo files
[
os.path.join(self.local_file_dir, '*' + self.image_extension),
'photobooth_photo',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Upload the HTML file for this particular set of photos
[
os.path.join('html', 'individual', 'index-single.html'),
'index',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Make sure the base .htaccess and index files are in place
[
os.path.join('html', 'index.php'), '', self.remote_file_dir, 1,
True
],
[
os.path.join('html', 'redirect.html'), '',
self.remote_file_dir, 1, True
],
[
os.path.join('html', '.htaccess'), '', self.remote_file_dir, 1,
True
],
# Make sure that all common files are in place
[
os.path.join('html', 'common', '*.css'), '',
os.path.join(self.remote_file_dir, 'common'), 0, True
],
]
success = self.upload_photos_using_defs(file_defs)
if success:
return remote_upload_dir
else:
return None
# Let the user select which image they want to be photographed with
def choose_accompaniment(self):
# Start with the first image (if there are any)
self.chosen_accompaniment = 2
# Use the JPG versions of accompaniment files, which have black background
file_pattern = self.filehandler.get_full_path(self.accompaniment_dir,
"*.jpg")
files = self.filehandler.get_sorted_file_list(file_pattern)
# If there are no images, then chosen_accompaniment will be the blank screen
if len(files) < 1:
return 1
button_overlay = OverlayOnCamera(self.camera)
button_overlay.camera_overlay(self.accompany_button_overlay_image)
self.overlay_on_camera = OverlayOnCamera(self.camera)
self.change_accompaniment(files)
while True:
choice = self.buttonhandler.wait_for_buttons('lsr', False)
if choice == 'l':
if self.chosen_accompaniment > 1:
self.chosen_accompaniment -= 1
else:
self.chosen_accompaniment = len(files) + 1
self.change_accompaniment(files)
if choice == 'r':
if self.chosen_accompaniment < (len(files) + 1):
self.chosen_accompaniment += 1
else:
self.chosen_accompaniment = 1
self.change_accompaniment(files)
if choice == 's':
self.buttonhandler.light_button_leds('lsr', False)
break
button_overlay.remove_camera_overlay()
def change_accompaniment(self, files):
self.camera.saturation = 0
# We leave a blank space for chosen_accompaniment == 1
accompanying_file_num = self.chosen_accompaniment - 2
# If chosen_accompaniment == 1, then we don't want any images overlaid
if accompanying_file_num < 0 or accompanying_file_num > (len(files) -
1):
self.overlay_on_camera.remove_camera_overlay()
return
curr_accompaniment_file = files[accompanying_file_num]
self.overlay_on_camera.camera_overlay(curr_accompaniment_file)
# See if an opacity value in the filename [within square brackets]
filename = os.path.basename(curr_accompaniment_file)
opacity = filename[filename.find("[") + 1:filename.find("]")]
if len(opacity) > 0:
opacity = int(opacity)
if opacity < -100:
opacity = -100
if opacity > 100:
opacity = 100
self.camera.saturation = opacity
class AnimatedPhoto(PhotoBoothFunction):
'Class to take a series of photographs and combine them into an animated gif'
def __init__(self, photobooth):
self.menu_text = "Create photo animation"
self.booth_id = photobooth.get_booth_id()
self.screen = photobooth.get_pygame_screen()
self.filehandler = photobooth.get_file_handler()
self.buttonhandler = photobooth.get_button_handler()
self.local_file_dir = self.filehandler.get_local_file_dir()
self.local_upload_file_dir = self.filehandler.get_upload_file_dir()
self.remote_file_dir = self.filehandler.get_remote_file_dir()
self.gif_delay = 100
self.photo_width = 500
self.capture_delay = 2
self.textprinter = TextPrinter(self.screen)
self.imageprinter = ImagePrinter(self.screen)
self.photohandler = PhotoHandler(self.screen, self.filehandler)
def start(self, total_pics=PhotoBoothFunction.total_pics):
# Take and display photos
self.total_pics = total_pics
# Display the instructions for this photobooth function
total_pics_msg = str(self.total_pics) + " photo"
if self.total_pics > 1:
total_pics_msg += "s"
self.instructions = [
total_pics_msg + " will be taken",
"(red light will appear before each photo)",
"The photos are then animated together",
"Press the Start button to begin"
]
# Outer loop: Instructions screen
while True:
choice = self.display_instructions()
# If the user selected Exit, bail out
if choice == "l":
break
while True:
self.take_photos()
# NOTE: If you change any part of this text display, mirror changes in erase print below
text_rect_list = self.textprinter.print_text([[
"Processing. Please wait ...", 84, config.white_colour,
"cb", 10
]], 0, False)
# Convert the images into an animated GIF (in a separate thread)
gif_thread = threading.Thread(target=self.process_photos)
gif_thread.start()
# Now display the original images one after the other in separate thread,
# to give the impression they are animated
# (easier then trying to display the animated GIF in pygame?)
display_thread_stop = threading.Event()
display_thread = threading.Thread(
target=self.photohandler.show_photos_animated,
args=(self.image_extension, display_thread_stop))
display_thread.start()
# Wait until the thread that's creating and uploading the GIF finishes
gif_thread.join()
# Overwrite the 'Processing' message in black to erase it
screen_colour_fill(self.screen, config.black_colour,
text_rect_list[0])
#self.textprinter.print_text([["Processing. Please wait ...", 84, config.black_colour, "cb", 10]],
# 0, False)
# Print the button label overlays
self.imageprinter.print_images(
[[config.start_overlay_image_bk_black, 'cb', 0, 0]], False)
self.imageprinter.print_images(
[[config.exit_side_overlay_image_bk_black, 'lb', 0, 0]],
False)
# Wait for the user to click Exit or Start
choice = ""
while True:
choice = self.buttonhandler.wait_for_buttons('ls', True)
if choice != 'screensaver':
break
# Now we can stop the thread that's animating the images on the screen
display_thread_stop.set()
display_thread.join()
# If user pressed Exit button, return to Photo Animation Instruction screen
if choice == 'l':
break
def take_photos(self):
################################# Step 1 - Initial Preparation ##########################
super(AnimatedPhoto, self).take_photos()
################################# Step 2 - Setup camera #################################
pixel_width = self.photo_width
pixel_height = self.photohandler.get_aspect_ratio_height(pixel_width)
self.camera.resolution = (pixel_width, pixel_height)
################################# Step 3 - Start camera preview ########################
screen_colour_fill(self.screen, config.black_colour)
self.camera.start_preview()
################################# Step 4 - Prepare the user ########################
time.sleep(self.prep_delay_short)
################################# Step 5 - Take Photos ################################
self.take_photos_and_close_camera(self.capture_delay)
def process_photos(self):
self.combine_into_gif()
self.create_thumbnail()
self.upload_photos()
# Create a thumbnail of the first image in the series
def create_thumbnail(self):
thumb_width = 200
file_list = self.filehandler.get_sorted_file_list(
os.path.join(self.local_file_dir, '*' + self.image_extension))
img = self.photohandler.resize_image(file_list[0], thumb_width,
thumb_width)
img.save(
os.path.join(self.local_file_dir,
'photobooth_thumb' + self.image_extension))
def upload_photos(self):
remote_upload_dir = time.strftime("%d-%b-%Y_A")
if self.booth_id is not "":
remote_upload_dir = self.booth_id + "_" + remote_upload_dir
file_suffix = time.strftime("%H-%M-%S")
file_defs = [
# Upload the animated GIF
[
os.path.join(self.local_upload_file_dir,
'*' + self.animated_image_extension),
'photobooth_photo_' + file_suffix,
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Upload just the first of the photo files
[
os.path.join(self.local_file_dir,
'photobooth_thumb' + self.image_extension),
'photobooth_photo_' + file_suffix,
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Upload the HTML files to handle the animated photos
[
os.path.join('html', 'individual', 'index-animated.php'),
'index',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
[
os.path.join('html', 'individual', 'img-animated.php'), '',
os.path.join(self.remote_file_dir, remote_upload_dir), 1, True
],
# Make sure the base .htaccess and index files are in place
[
os.path.join('html', 'index.php'), '', self.remote_file_dir, 1,
True
],
[
os.path.join('html', 'redirect.html'), '',
self.remote_file_dir, 1, True
],
[
os.path.join('html', '.htaccess'), '', self.remote_file_dir, 1,
True
],
# Make sure that all common files are in place
[
os.path.join('html', 'common', '*'), '',
os.path.join(self.remote_file_dir, 'common'), 0, True
],
]
success = self.upload_photos_using_defs(file_defs)
if success:
return remote_upload_dir
else:
return None
def combine_into_gif(self):
local_file_dir = self.filehandler.get_local_file_dir()
print "Creating animated GIF ..."
# Thanks to drumminhands
graphicsmagick = ("gm convert -delay " + str(self.gif_delay) + " " +
local_file_dir + "/*.jpg " +
self.local_upload_file_dir + "/photobooth.gif")
os.system(graphicsmagick) #make the .gif