145 lines
4.4 KiB
Python
Executable File
145 lines
4.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import os, sys, time, pygame
|
|
|
|
class Button:
|
|
def __init__(self, size, topleft):
|
|
self.up_image = pygame.image.load('button-up.png').convert_alpha()
|
|
self.up_image = pygame.transform.smoothscale(self.up_image, (size, size))
|
|
|
|
self.down_image = pygame.image.load('button-down.png').convert_alpha()
|
|
self.down_image = pygame.transform.smoothscale(self.down_image, (size, size))
|
|
|
|
self.rect = self.up_image.get_rect()
|
|
self.rect.topleft = topleft
|
|
|
|
self.pressed = False
|
|
|
|
def draw_on(self, screen):
|
|
if self.pressed:
|
|
screen.blit(self.down_image, self.rect)
|
|
else:
|
|
screen.blit(self.up_image, self.rect)
|
|
|
|
def is_pressed(self, pos):
|
|
return self.rect.collidepoint(pos)
|
|
|
|
def press(self):
|
|
self.pressed = True
|
|
|
|
def unpress(self):
|
|
self.pressed = False
|
|
self.run()
|
|
|
|
def run(self):
|
|
print('click')
|
|
|
|
class CaptureButton(Button):
|
|
def __init__(self, size, topleft, photoboite):
|
|
super(CaptureButton, self).__init__(size, topleft)
|
|
self.photoboite = photoboite
|
|
self.elapsed = 0
|
|
self.capture = False
|
|
|
|
def draw_on(self, screen):
|
|
super(CaptureButton, self).draw_on(screen)
|
|
if self.capture:
|
|
text = str(self.elapsed + 1)
|
|
size = self.photoboite.font.size(text)
|
|
ren = self.photoboite.font.render(text, 0, (255, 255, 0))
|
|
screen.blit(ren, (400, 400))
|
|
|
|
def run(self):
|
|
pygame.event.set_blocked((pygame.MOUSEBUTTONDOWN, pygame.MOUSEBUTTONUP))
|
|
self.capture = True
|
|
for i in range(4):
|
|
wait = True
|
|
start = time.time()
|
|
while wait:
|
|
self.elapsed = int(time.time() - start)
|
|
if self.elapsed >= 3:
|
|
wait = False
|
|
else:
|
|
photoboite.event()
|
|
photoboite.draw()
|
|
pygame.display.flip()
|
|
|
|
print('Capture')
|
|
pygame.time.wait(1000)
|
|
pygame.event.set_allowed((pygame.MOUSEBUTTONDOWN, pygame.MOUSEBUTTONUP))
|
|
self.capture = False
|
|
|
|
|
|
class Photoboite:
|
|
def __init__(self):
|
|
self.screen = self.screen()
|
|
pygame.font.init()
|
|
|
|
self.background = 255, 255, 255
|
|
self.buttons = []
|
|
|
|
button_size = int(self.screen.get_width() / 4)
|
|
topleft = self.screen.get_width() - button_size - 20, self.screen.get_height() / 4 - button_size / 2
|
|
|
|
self.buttons.append(CaptureButton(button_size, topleft, self))
|
|
|
|
self.camera = pygame.Rect(20, 20, self.screen.get_width() - self.screen.get_width() / 4 - 60, self.screen.get_height() - 40)
|
|
|
|
self.font = pygame.font.Font(None, 80)
|
|
|
|
|
|
def screen(self):
|
|
if os.getenv('DISPLAY'):
|
|
pygame.display.init()
|
|
return pygame.display.set_mode((1280, 768))
|
|
else:
|
|
drivers = ['directfb', 'fbcon', 'svgalib']
|
|
found = False
|
|
for driver in drivers:
|
|
if not os.getenv('SDL_VIDEODRIVER'):
|
|
os.putenv('SDL_VIDEODRIVER', driver)
|
|
try:
|
|
pygame.display.init()
|
|
except pygame.error as e:
|
|
print('Driver: {0} failed.'.format(driver))
|
|
continue
|
|
found = True
|
|
break
|
|
|
|
if not found:
|
|
raise Exception('No suitable video driver found!')
|
|
|
|
return pygame.display.set_mode()
|
|
|
|
def draw(self):
|
|
self.screen.fill(self.background)
|
|
|
|
self.screen.fill((0,0,0), rect=self.camera)
|
|
|
|
for button in self.buttons:
|
|
button.draw_on(self.screen)
|
|
|
|
def event(self):
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT or event.type == pygame.KEYDOWN and event.unicode == 'q':
|
|
sys.exit()
|
|
elif event.type == pygame.MOUSEBUTTONDOWN:
|
|
for button in self.buttons:
|
|
if button.is_pressed(event.pos):
|
|
button.press()
|
|
elif event.type == pygame.MOUSEBUTTONUP:
|
|
for button in self.buttons:
|
|
if button.is_pressed(event.pos):
|
|
button.unpress()
|
|
|
|
def run(self):
|
|
pressed = False
|
|
while True:
|
|
self.event()
|
|
self.draw()
|
|
|
|
pygame.display.flip()
|
|
|
|
photoboite = Photoboite()
|
|
photoboite.run()
|