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

modified picture taking

Browse Source
This commit is contained in:
Balthasar Reuter
2015-05-06 22:29:55 +02:00
parent 6eb6c4b036
commit dc7502783e
1 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
photobooth.py
View File

@@ -56,27 +56,24 @@ 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() self.reset()
# self.screen = pygame.display.set_mode(size)
#self.screen = pygame.display.set_mode(size, pygame.FULLSCREEN)
# Set background
# self.background = pygame.Surface(self.screen.get_size())
# self.background = self.background.convert()
# self.background.fill((250, 250, 250))
def reset(self): 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)) self.screen.fill((255,255,255))
def apply(self):
pygame.display.update()
def get_size(self): def get_size(self):
return self.size return self.size
def show_picture(self, filename, size=(0,0), offset=(0,0)): def show_picture(self, filename, size=(0,0), offset=(0,0)):
if size == (0,0): if size == (0,0):
size = self.get_size() size = self.size
image = pygame.image.load(filename) image = pygame.image.load(filename)
image = pygame.transform.scale(image, size).convert() image = pygame.transform.scale(image, size).convert()
# self.sprite.image = image.convert()
self.screen.blit(image, offset) self.screen.blit(image, offset)
def show_message(self, msg): def show_message(self, msg):
@@ -86,10 +83,6 @@ class GUI_PyGame:
textpos.centerx = self.screen.get_rect().centerx textpos.centerx = self.screen.get_rect().centerx
self.screen.blit(text, textpos) self.screen.blit(text, textpos)
def apply(self):
# pygame.display.flip()
pygame.display.update()
def mainloop(self, filename): def mainloop(self, filename):
while True: while True:
for event in pygame.event.get(): for event in pygame.event.get():
@@ -141,23 +134,26 @@ class Camera:
### Functions ### ### Functions ###
################# #################
def error(msg, exit_code=1): # def error(msg, exit_code=1):
print "ERROR: " + msg # print "ERROR: " + msg
teardown(exit_code) # teardown(exit_code)
def teardown(exit_code=0):
display.teardown()
sys.exit(exit_code)
def handle_keypress(key): def handle_keypress(key):
if key == ord('q'): if key == ord('q'):
teardown() teardown()
elif key == ord('c'): elif key == ord('c'):
print "Taking 3 pictures" print "Taking 4 pictures"
for x in xrange(3): size = display.get_size()
filename = camera.take_picture(images.get_next()) image_size = (int(size[0]/2), int(size[1]/2))
display.show_picture(filename) filenames = []
time.sleep(2) for x in range(4):
filenames[x] = camera.take_picture(images.get_next())
display.show_picture(filenames[0], image_size, (0,0))
display.show_picture(filenames[1], image_size, (image_size[0],0))
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.apply()
timer.sleep(5)
def handle_exception(msg): def handle_exception(msg):
display.reset() display.reset()
@@ -165,6 +161,10 @@ def handle_exception(msg):
display.apply() display.apply()
time.sleep(3) time.sleep(3)
def teardown(exit_code=0):
display.teardown()
sys.exit(exit_code)
def main(): def main():
while True: while True:
try: try: