summaryrefslogtreecommitdiff
path: root/cameltris/screens/pause.py
blob: 72ff51abe6babe73af73c91e2db8405034f964f5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import contextlib
import sys
from typing import Callable, NoReturn

import pygame

from .base_screen import Screen
from ..controller import Controller, KeyboardController, JoystickController, Input
from ..misc import UnPause


def handle_input_pressed(controller: Controller, event: pygame.event.Event):
    if controller.get_input_down(event) == Input.PAUSE:
        raise UnPause()


class Pause(Screen):
    def __init__(self, screen: pygame.surface.Surface):
        self.event_handler: dict[int, Callable[[pygame.event.Event], None]] = {}

        def exit(_) -> NoReturn:
            sys.exit()

        self.event_handler[pygame.QUIT] = exit
        self.event_handler[pygame.JOYBUTTONDOWN] = lambda evt: handle_input_pressed(JoystickController(evt.joy), evt)
        self.event_handler[pygame.KEYDOWN] = lambda evt: handle_input_pressed(KeyboardController(pygame.key), evt)

    def refresh(self) -> None:
        pass

    def oneframe(self) -> None:
        try:
            for event in pygame.event.get():
                with contextlib.suppress(KeyError):
                    self.event_handler[event.type](event)
        except UnPause:
            pygame.event.clear()
            raise