path: root/pygame/main.py

#!/usr/bin/python3

from dataclasses import dataclass
from pathlib import Path

#import pygame_sdl2 as pygame
import pygame
import yaml


#WINDOW_RESOLUTION = (1280, 720)
WINDOW_RESOLUTION = (800, 600)

if pygame.ver[0] != '2':
    raise ValueError(f'pygame2 required, got {pygame.ver}')


class Event(Exception): pass
class QuitEvent(Event): pass


@dataclass
class AppState:
    menu_selected_index: int = 0


class GuiApp:

    def __init__(self):
        pygame.init()
        #print(pygame.display.list_modes())
        self.window = pygame.display.set_mode(WINDOW_RESOLUTION)
        self.clock = pygame.time.Clock()
        self.menu_font = pygame.font.Font(pygame.font.get_default_font(), 32)
        self.font_antialias = True
        #self.font_antialias = False
        self.menu_color = (0, 0, 200)

    def process_input(self, event=None):
        if event is None:
            event = pygame.event.poll()
        if event.type == pygame.KEYDOWN:
            print('event', event, 'key name', pygame.key.name(event.key))
            if event.key in (pygame.K_q, pygame.K_ESCAPE):
                raise QuitEvent()
        if event.type == pygame.QUIT:
            raise QuitEvent()

    def update(self):
        pass

    def render(self):
        self.window.fill((0, 0, 0))

        # Initial y
        y = 50

        # Title
        #surface = self.titleFont.render("TANK BATTLEGROUNDS !!", True, (200, 0, 0))
        #x = (self.window.get_width() - surface.get_width()) // 2
        #self.window.blit(surface, (x, y))
        #y += (200 * surface.get_height()) // 100

        x = 50

        # Compute menu width
        #menuWidth = 0
        #for item in self.menuItems:
        #    surface = self.itemFont.render(item['title'], True, (200, 0, 0))
        #    menuWidth = max(menuWidth, surface.get_width())
        #    item['surface'] = surface

        surface = self.menu_font.render("I love my cat !",
                                        self.font_antialias, self.menu_color)
        self.window.blit(surface, (x, y))

        ## Draw menu items
        #x = (self.window.get_width() - menuWidth) // 2
        #for index, item in enumerate(self.menuItems):
        #    # Item text
        #    surface = item['surface']
        #    self.window.blit(surface, (x, y))

        #    # Cursor
        #    if index == self.currentMenuItem:
        #        cursorX = x - self.menuCursor.get_width() - 10
        #        cursorY = y + (surface.get_height() - self.menuCursor.get_height()) // 2
        #        self.window.blit(self.menuCursor, (cursorX, cursorY))

        #    y += (120 * surface.get_height()) // 100           


        #pygame.draw.rect(self.window,
        #                (0,0,255),
        #                (120,120,400,240))

        pygame.display.update()


    def loop(self):

        try:
            while 1:
                self.process_input()
                self.update()
                self.render()
                self.clock.tick(30) # 30 fps
        except QuitEvent:
            pass

        print('quitting')
        pygame.quit()


class GuiAppSub1(GuiApp):

    def __init__(self):
        super().__init__()

        self.state = AppState()
        self.base_window_fill_color = (0, 0, 0)
        self.user_move_vector = (0, 0)
        with Path('~/games/.saves/gamedata.yaml').expanduser().open() as fp:
            self.gamedb = yaml.safe_load(fp)
        self.menu_items = [x['title'] for x in self.gamedb['games']]
        self.menu_font_cache = ({}, [])
        self.menu_font_cache_max = 40
        self.image_right_arrow = pygame.image.load('arrow-right.png')
        #self.last_movement_update_time = self.clock.
        self.joysticks = {}
        self.joysticks_axis_threshold = 0

    def menu_font_render(self, text):
        cached_surface = self.menu_font_cache[0].get(text, None)
        if cached_surface is not None:
            self.menu_font_cache[1].remove(text)
            self.menu_font_cache[1].append(text) # refresh mru place
            return cached_surface

        new_surface = self.menu_font.render(text,
                                            self.font_antialias,
                                            self.menu_color)

        self.menu_font_cache[0][text] = new_surface
        self.menu_font_cache[1].append(text)

        if len(self.menu_font_cache[1]) > self.menu_font_cache_max:
            removed_item_text = self.menu_font_cache[1].pop(0)
            del self.menu_font_cache[0][removed_item_text]

        # debug for cache miss:
        print('cache miss for', text)

        return new_surface

    def process_input(self):
        movement_up_keys = (
            pygame.K_UP,
            pygame.K_k,
        )
        movement_down_keys = (
            pygame.K_DOWN,
            pygame.K_j,
        )
        axis_change = 0
        for event in pygame.event.get():
            go_super = 1
            x, y = self.user_move_vector
            if event.type == pygame.KEYDOWN:
                if event.key in (*movement_up_keys, *movement_down_keys):
                    y += 1 if event.key in movement_down_keys else -1
                    go_super = 0
                elif event.key in (pygame.K_LEFT, pygame.K_RIGHT):
                    x += 1 if event.key == pygame.K_LEFT else -1
                    go_super = 0
            elif event.type == pygame.KEYUP:
                x, y = self.user_move_vector
                if event.key in (*movement_up_keys, *movement_down_keys):
                    y -= 1 if event.key in movement_down_keys else -1
                    go_super = 0
                elif event.key in (pygame.K_LEFT, pygame.K_RIGHT):
                    x -= 1 if event.key == pygame.K_LEFT else -1
                    go_super = 0
            elif event.type in (pygame.JOYBUTTONDOWN, pygame.JOYBUTTONUP):
                go_super = 0
                factor = 1 if pygame.JOYBUTTONDOWN else -1
                joystick = self.joysticks[event.instance_id]
                print('gamepad event', event)
                #if event.button == 0:
                    #y += 1 * factor
                if event.type == pygame.JOYBUTTONDOWN:
                    joystick.rumble(0.3, 0.8, 40)
                if event.button == 1:
                    raise QuitEvent()
                #if event.button == 1:
                    #y -= 1 * factor
            elif event.type == pygame.JOYAXISMOTION:
                go_super = 0
                if event.axis == 1:  # left stick up-down axis
                    new_axis = self.joysticks_axis_threshold
                    #print('motion event', event)
                    if event.value > 0.70:
                        new_axis = 1  # down activated
                    elif event.value < 0.5 and event.value > -0.5:
                        new_axis = 0  # down and up deactivated
                    elif event.value < -0.70:
                        new_axis = -1  # up activated
                    if new_axis != self.joysticks_axis_threshold:
                        axis_change = 1
                        self.joysticks_axis_threshold = new_axis
                        print('threshold', self.joysticks_axis_threshold)
            elif event.type == pygame.JOYDEVICEADDED:
                # This event will be generated when the program starts for every
                # joystick, filling up the list without needing to create them manually.
                joy = pygame.joystick.Joystick(event.device_index)
                self.joysticks[joy.get_instance_id()] = joy
                print(f"Gamepad {joy.get_instance_id()} connected")
            elif event.type == pygame.JOYDEVICEREMOVED:
                del self.joysticks[event.instance_id]
                print(f"Gamepad {event.instance_id} disconnected")

            if go_super == 0:
                if axis_change:
                    self.user_move_vector = (x, self.joysticks_axis_threshold)
                else:
                    self.user_move_vector = (x, y)
                print('new move vector:', self.user_move_vector)
            else:
                super().process_input(event)

    def update(self):
        menu_new_index = self.state.menu_selected_index
        if self.user_move_vector[1] < 0:
            menu_new_index -= 1
        elif self.user_move_vector[1] > 0:
            menu_new_index += 1
        if menu_new_index < 0:
            menu_new_index = 0
        elif menu_new_index > (len(self.menu_items)-1):
            menu_new_index = len(self.menu_items)-1
        self.state.menu_selected_index = menu_new_index

    def render(self):
        self.window.fill(self.base_window_fill_color)

        # Initial y
        y = 50

        # Title
        #surface = self.titleFont.render("TANK BATTLEGROUNDS !!", True, (200, 0, 0))
        #x = (self.window.get_width() - surface.get_width()) // 2
        #self.window.blit(surface, (x, y))
        #y += (200 * surface.get_height()) // 100

        x = 50 + 64 + 5

        # Compute menu width
        #menuWidth = 0
        #for item in self.menuItems:
        #    surface = self.itemFont.render(item['title'], True, (200, 0, 0))
        #    menuWidth = max(menuWidth, surface.get_width())
        #    item['surface'] = surface

        y_pad = 5
        font_selected_item_height = None
        selected_item_y = y
        for index, item in enumerate(self.menu_items):
            surface = self.menu_font_render(item)
            self.window.blit(surface, (x, y))
            if self.state.menu_selected_index == index:
                font_selected_item_height = surface.get_height()
                selected_item_y = y
            y += surface.get_height() + y_pad

        self.window.blit(self.image_right_arrow, (50, selected_item_y - 64//2
                                                  + font_selected_item_height//2))


        fps_surface = self.menu_font_render(f'{self.clock.get_fps():02.0f}')
        fps_pos = (self.window.get_width() - fps_surface.get_width(), 0)
        self.window.blit(fps_surface, fps_pos)

        ## Draw menu items
        #x = (self.window.get_width() - menuWidth) // 2
        #for index, item in enumerate(self.menuItems):
        #    # Item text
        #    surface = item['surface']
        #    self.window.blit(surface, (x, y))

        #    # Cursor
        #    if index == self.currentMenuItem:
        #        cursorX = x - self.menuCursor.get_width() - 10
        #        cursorY = y + (surface.get_height() - self.menuCursor.get_height()) // 2
        #        self.window.blit(self.menuCursor, (cursorX, cursorY))

        #    y += (120 * surface.get_height()) // 100           


        #pygame.draw.rect(self.window,
        #                (0,0,255),
        #                (120,120,400,240))

        pygame.display.update()


app = GuiAppSub1()
app.loop()