Photo Booth using the Raspberry Pi and Pi Camera. Designed for 7″ touch screen. Just run python code in Raspbian.
from __future__ import print_function
import pygame
import time
from picamera import PiCamera
import sys
from PIL import Image
import os
pygame.init()
win = pygame.display.set_mode((0,0),pygame.FULLSCREEN)
win.fill((255,255,255))
camera = PiCamera()
camera.resolution = (800,800)
#camera.color_effects = (128,128) #turn camera to black and white
class button():
def __init__(self, color, x,y,width,height, text=''):
self.color = color
self.x = x
self.y = y
self.width = width
self.height = height
self.text = text
def draw(self,win,outline=None):
#Call this method to draw the button on the screen
if outline:
pygame.draw.rect(win, outline,(self.x-2,self.y-2,self.width+4,self.height+4),0)
pygame.draw.rect(win, self.color,(self.x,self.y,self.width,self.height),0)
if self.text != '':
font = pygame.font.SysFont('comicsans', 60)
text = font.render(self.text, 1, (0,0,0))
win.blit(text, (self.x + (self.width/2 - text.get_width()/2), self.y + (self.height/2 - text.get_height()/2)))
def isOver(self, pos):
#Pos is the mouse position or a tuple of (x,y) coordinates
if pos[0] > self.x and pos[0] self.y and pos[1] < self.y + self.height:
return True
return False
def redrawWindow():
win.fill((255,255,255))
greenButton.draw(win)
def counterWindow():
win.fill((255,255,255))
if(counter!=0):
counterLabel.text = str(counter)
else:
counterLabel.text = "Smile"
counterLabel.draw(win)
def loadimage():
global imagename
win.fill((0,0,0))
img = pygame.image.load(imagename)
img = pygame.transform.scale(img,(480,480))
win.blit(img,(170,0))
def takePhoto():
global showImage
global counterActive
global counter
global imagename
print("Take Photo")
for x in range(4):
camera.start_preview()
time.sleep(1)
camera.capture('image'+str(x)+'.jpg')
camera.stop_preview()
files = [
'image0.jpg',
'image1.jpg',
'image2.jpg',
'image3.jpg']
result = Image.new("RGB", (800, 800))
for index, file in enumerate(files):
path = os.path.expanduser(file)
img = Image.open(path)
img.thumbnail((400, 400), Image.ANTIALIAS)
x = index // 2 * 400
y = index % 2 * 400
w, h = img.size
print('pos {0},{1} size {2},{3}'.format(x, y, w, h))
result.paste(img, (x, y, x + w, y + h))
imagename = str(round(time.time() * 1000)) +'.jpg'
result.save(os.path.expanduser(imagename))
showImage = True
counterActive = False
counter = 6
def resetStatus():
global counter
global startProcess
global counterActive
global imageCounter
counter = 6
startProcess = True
counterActive = False
imageCounter = 0
run = True
counter = 6
startProcess = True
counterActive = False
showImage = False
greenButton = button((0,255,0),280,200,250,100,'Take Photos')
counterLabel = button((100,255,255),300,200,250,100,'0')
imageCounter = 0
imagename =""
while run:
print(counter)
if(startProcess == True):
redrawWindow()
elif(counterActive == True):
counter = counter-1
counterWindow()
time.sleep(1)
elif(showImage == True):
loadimage()
time.sleep(1)
imageCounter = imageCounter +1
if(imageCounter == 15):
resetStatus()
pygame.display.update()
if(counterActive == False):
for event in pygame.event.get():
pos = pygame.mouse.get_pos()
if event.type == pygame.QUIT:
run = False
pygame.quit()
quit()
if event.type == pygame.MOUSEBUTTONDOWN:
if greenButton.isOver(pos):
print("Clicked the button")
counterActive = True
startProcess = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
run = False
pygame.quit()
quit()
if(counter == 0):
takePhoto()
Advertisements
