def main_loop(self): """Run the main game loop. """ try: clock = pygame.time.Clock() self.led_startup.switch_on() self.initialize() menu = None fps = 40 while True: events = list(pygame.event.get()) if self.find_quit_event(events): break if self.find_fullscreen_event(events): self.window.toggle_fullscreen() event = self.find_resize_event(events) if event: self.window.resize(event.size) if not menu and self.find_settings_event(events): menu = PiConfigMenu(self.window, self.config, fps) menu.show() if menu and menu.is_shown(): # Convert HW button events to keyboard events for menu if self.find_settings_event(events, BUTTON_DOWN): events.insert(0, menu.create_back_event()) if self.find_capture_event(events, BUTTON_DOWN): events.insert(0, menu.create_next_event()) elif self.find_print_event(events, BUTTON_DOWN): events.insert(0, menu.create_click_event()) menu.process(events) elif menu and not menu.is_shown(): self.initialize() menu = None else: self.state_machine.process(events) pygame.display.update() clock.tick( fps ) # Ensure the program will never run at more than x frames per second finally: self.makers_pool.quit() self.led_startup.quit() self.led_preview.quit() self.led_capture.quit() self.led_print.quit() GPIO.cleanup() self.camera.quit() self.printer.quit() pygame.quit()
def __init__(self, pin, bouncetime=0.1): self.pin = pin # Use internal pull up/down resistors GPIO.setup(pin, GPIO.IN, pull_up_down=GPIO.PUD_UP) GPIO.add_event_detect(self.pin, GPIO.FALLING, callback=self.on_button_down, bouncetime=int(bouncetime * 1000))
def __init__(self, config): self.config = config # Clean directory where pictures are saved self.savedir = config.getpath('GENERAL', 'directory') if not osp.isdir(self.savedir): os.makedirs(self.savedir) if osp.isdir(self.savedir) and config.getboolean( 'GENERAL', 'clear_on_startup'): shutil.rmtree(self.savedir) os.makedirs(self.savedir) # Prepare GPIO, physical pins mode GPIO.setmode(GPIO.BOARD) # Prepare the pygame module for use os.environ['SDL_VIDEO_CENTERED'] = '1' pygame.init() # Dont catch mouse motion to avoid filling the queue during long actions pygame.event.set_blocked(pygame.MOUSEMOTION) # Create window of (width, height) init_size = self.config.gettyped('WINDOW', 'size') init_debug = self.config.getboolean('GENERAL', 'debug') init_color = self.config.gettyped('WINDOW', 'background') init_text_color = self.config.gettyped('WINDOW', 'text_color') if not isinstance(init_color, (tuple, list)): init_color = self.config.getpath('WINDOW', 'background') if not isinstance(init_size, str): self.window = PtbWindow('Pibooth', init_size, color=init_color, text_color=init_text_color, debug=init_debug) else: self.window = PtbWindow('Pibooth', color=init_color, text_color=init_text_color, debug=init_debug) self.state_machine = StateMachine(self) self.state_machine.add_state(StateWait()) self.state_machine.add_state( StateChoose(30)) # 30s before going back to the start self.state_machine.add_state(StateChosen(4)) self.state_machine.add_state(StateCapture()) self.state_machine.add_state(StateProcessing()) self.state_machine.add_state(StatePrint()) self.state_machine.add_state(StateFinish(0.5)) self.camera = camera.get_camera( config.getint('CAMERA', 'iso'), config.gettyped('CAMERA', 'resolution'), config.getint('CAMERA', 'rotation'), config.getboolean('CAMERA', 'flip'), config.getboolean('CAMERA', 'delete_internal_memory')) # Initialize the hardware buttons self.led_capture = PtbLed(config.getint('CONTROLS', 'picture_led_pin')) self.button_capture = PtbButton( config.getint('CONTROLS', 'picture_btn_pin'), config.getfloat('CONTROLS', 'debounce_delay')) self.led_print = PtbLed(config.getint('CONTROLS', 'print_led_pin')) self.button_print = PtbButton( config.getint('CONTROLS', 'print_btn_pin'), config.getfloat('CONTROLS', 'debounce_delay')) self.led_startup = PtbLed(config.getint('CONTROLS', 'startup_led_pin')) self.led_preview = PtbLed(config.getint('CONTROLS', 'preview_led_pin')) # Initialize the printer self.printer = PtbPrinter(config.get('PRINTER', 'printer_name')) # Variables shared between states self.dirname = None self.makers_pool = PicturesMakersPool() self.capture_nbr = None self.capture_choices = (4, 1) self.nbr_duplicates = 0 self.previous_picture = None self.previous_animated = [] self.previous_picture_file = None
def switch_off(self): """Switch off the LED. """ if threading.current_thread() != self._blinking_thread: self._blinking_thread.unregister(self) GPIO.output(self.pin, GPIO.LOW)
def __init__(self, pin): self.pin = pin GPIO.setup(pin, GPIO.OUT)
def __init__(self, config): self.config = config # Clean directory where pictures are saved self.savedir = config.getpath('GENERAL', 'directory') if not osp.isdir(self.savedir): os.makedirs(self.savedir) if osp.isdir(self.savedir) and config.getboolean( 'GENERAL', 'clear_on_startup'): shutil.rmtree(self.savedir) os.makedirs(self.savedir) # Prepare GPIO, physical pins mode GPIO.setmode(GPIO.BOARD) # Prepare the pygame module for use os.environ['SDL_VIDEO_CENTERED'] = '1' pygame.init() # Dont catch mouse motion to avoid filling the queue during long actions pygame.event.set_blocked(pygame.MOUSEMOTION) # Create window of (width, height) init_size = self.config.gettyped('WINDOW', 'size') if not isinstance(init_size, str): self.window = PtbWindow('Pibooth', init_size) else: self.window = PtbWindow('Pibooth') self.state_machine = StateMachine(self) self.state_machine.add_state(StateWait()) self.state_machine.add_state( StateChoose(30)) # 30s before going back to the start self.state_machine.add_state(StateChosen(4)) self.state_machine.add_state(StateCapture()) self.state_machine.add_state(StateProcessing()) self.state_machine.add_state(StatePrint()) self.state_machine.add_state(StateFinish(0.5)) # Initialize the camera. If gPhoto2 camera is used, try to kill # any process using gPhoto2 as it may block camera access if camera.gp_camera_connected() and camera.rpi_camera_connected(): cam_class = camera.HybridCamera pkill('*gphoto2*') elif camera.gp_camera_connected(): cam_class = camera.GpCamera pkill('*gphoto2*') elif camera.rpi_camera_connected(): cam_class = camera.RpiCamera elif camera.cv_camera_connected(): cam_class = camera.CvCamera else: raise EnvironmentError( "Neither Raspberry Pi nor GPhoto2 nor OpenCV camera detected") self.camera = cam_class(config.getint('CAMERA', 'iso'), config.gettyped('CAMERA', 'resolution'), config.getint('CAMERA', 'rotation'), config.getboolean('CAMERA', 'flip')) # Initialize the hardware buttons self.led_capture = PtbLed(config.getint('CONTROLS', 'picture_led_pin')) self.button_capture = PtbButton( config.getint('CONTROLS', 'picture_btn_pin'), config.getfloat('CONTROLS', 'debounce_delay')) self.led_print = PtbLed(config.getint('CONTROLS', 'print_led_pin')) self.button_print = PtbButton( config.getint('CONTROLS', 'print_btn_pin'), config.getfloat('CONTROLS', 'debounce_delay')) self.led_startup = PtbLed(config.getint('CONTROLS', 'startup_led_pin')) self.led_preview = PtbLed(config.getint('CONTROLS', 'preview_led_pin')) # Initialize the printer self.printer = PtbPrinter(config.get('PRINTER', 'printer_name')) # Variables shared between states self.dirname = None self.makers_pool = PicturesMakersPool() self.capture_nbr = None self.capture_choices = (4, 1) self.nbr_duplicates = 0 self.previous_picture = None self.previous_animated = [] self.previous_picture_file = None