129 lines
4.6 KiB
Python
129 lines
4.6 KiB
Python
from pydantic import BaseModel
|
|
from typing import Tuple, Callable
|
|
|
|
import pygame
|
|
|
|
screen_sizes = [
|
|
(480, 360), # 360p
|
|
(640, 480), # VGA
|
|
(800, 600), # SVGA
|
|
(1024, 768), # XGA
|
|
(1280, 720), # HD 720p
|
|
(1366, 768), # HD 1366x768
|
|
(1600, 900), # HD+ 1600x900
|
|
(1920, 1080), # Full HD 1080p
|
|
(2560, 1440), # 2K / QHD 1440p
|
|
(3840, 2160) # 4K / UHD 2160p
|
|
]
|
|
|
|
|
|
|
|
class MenuItem(BaseModel):
|
|
display_text: str
|
|
on_click: Callable = None
|
|
text_color: Tuple[str, str, str] = (0, 0, 0)
|
|
|
|
|
|
|
|
|
|
class Menu:
|
|
def __init__(self, game):
|
|
pygame.font.init()
|
|
|
|
self.game = game
|
|
self.hamburger = Hamburger(game)
|
|
self.menu_width = min(max(200, self.game.screen.get_width()
|
|
* 0.8), self.game.screen.get_width())
|
|
self.menu_height = min(max(200, self.game.screen.get_height()
|
|
* 0.8), self.game.screen.get_height())
|
|
self.x = (self.game.screen.get_width() - self.menu_width) / 2
|
|
self.y = (self.game.screen.get_height() - self.menu_height) / 2
|
|
self.color = (100, 100, 100)
|
|
self.is_menu_open = False
|
|
|
|
self.surface = pygame.Surface((self.menu_width, self.menu_height))
|
|
self.font = pygame.font.SysFont("", 50)
|
|
self.screen_size_index = False
|
|
self.padding = 10
|
|
|
|
self.items = [
|
|
MenuItem(display_text='Menu'),
|
|
MenuItem(display_text='Screen Size'),
|
|
MenuItem(display_text=f'{self.game.screen.get_width()}x{self.game.screen.get_height()}', color=(50, 0, 0))
|
|
]
|
|
|
|
def render(self):
|
|
if self.is_menu_open:
|
|
self.surface.fill(self.color)
|
|
|
|
pos = (self.padding, self.padding)
|
|
for item in self.items:
|
|
text = self.font.render(item.display_text, True, item.text_color)
|
|
self.surface.blit(text, pos)
|
|
pos = (pos[0], pos[1] + 50)
|
|
|
|
# put text in the menu surface
|
|
# text = self.font.render("Menu", True, (0, 0, 0))
|
|
# self.surface.blit(text, (0, 0))
|
|
# text = self.font.render("Screen Size", True, (0, 0, 0))
|
|
# self.surface.blit(text, (0, 60))
|
|
# text = self.font.render(
|
|
# f'{self.game.screen.get_width()}x{self.game.screen.get_height()}', True, (25, 25, 25))
|
|
# self.surface.blit(text, (10, 120))
|
|
#
|
|
self.game.screen.blit(self.surface, (self.x, self.y))
|
|
|
|
self.hamburger.render()
|
|
|
|
def get_mouse_pos(self):
|
|
'get mouse position relative to self.surface'
|
|
x, y = pygame.mouse.get_pos()
|
|
return x - self.x, y - self.y
|
|
|
|
def handle_click(self):
|
|
self.hamburger.handle_click(self)
|
|
pos = self.get_mouse_pos()
|
|
print(pos)
|
|
if pos[1] > 120 and pos[1] < 180 and pos[0] > 0 and pos[0] < self.menu_width:
|
|
if self.screen_size_index is False:
|
|
self.screen = pygame.display.set_mode(screen_sizes[0])
|
|
self.screen_size_index = 0
|
|
if self.screen_size_index == len(screen_sizes) - 1:
|
|
self.screen_size_index = 0
|
|
else:
|
|
self.screen_size_index += 1
|
|
self.screen = pygame.display.set_mode(screen_sizes[self.screen_size_index])
|
|
|
|
|
|
class Hamburger:
|
|
def __init__(self, game):
|
|
|
|
self.game = game
|
|
self.hamburger_width = 50
|
|
self.bar_height = self.hamburger_width / 4
|
|
self.bar_spacing = self.hamburger_width / 20
|
|
self.hamburger_height = self.bar_height * 3 + self.bar_spacing * 2
|
|
self.x = self.game.screen.get_width() - self.hamburger_width - 20
|
|
self.y = 20
|
|
self.color = (100, 100, 100)
|
|
self.rect = pygame.Rect(self.x, self.y, self.hamburger_width,
|
|
self.hamburger_height)
|
|
self.surface = pygame.Surface(
|
|
(self.hamburger_width, self.hamburger_height))
|
|
|
|
def render(self):
|
|
pygame.draw.rect(self.surface, self.color,
|
|
(0, 0, self.hamburger_width, self.bar_height),)
|
|
pygame.draw.rect(self.surface, self.color,
|
|
(0, self.bar_height + self.bar_spacing, self.hamburger_width, self.bar_height),)
|
|
pygame.draw.rect(self.surface, self.color,
|
|
(0, 2 * (self.bar_height + self.bar_spacing), self.hamburger_width, self.bar_height),)
|
|
|
|
self.game.screen.blit(self.surface, (self.x, self.y))
|
|
|
|
|
|
def handle_click(self, menu):
|
|
pos = pygame.mouse.get_pos()
|
|
|
|
if self.rect.collidepoint(pos):
|
|
menu.is_menu_open = not menu.is_menu_open
|