This commit is contained in:
waylon 2023-06-18 15:36:26 -05:00
parent a96b1cc5a9
commit 78cc5ff0cd
2 changed files with 137 additions and 112 deletions

View file

@ -4,6 +4,7 @@ import pygame
from typer import Typer
from websocket import create_connection
from learn_sql_model.game.menu import Menu
from learn_sql_model.config import get_config
from learn_sql_model.console import console
from learn_sql_model.factories.hero import HeroFactory
@ -11,119 +12,13 @@ from learn_sql_model.models.hero import HeroCreate, HeroDelete, HeroUpdate, Hero
speed = 10
pygame.font.init() # you have to call this at the start,
# if you want to use this module.
my_font = pygame.font.SysFont("Comic Sans MS", 30)
config = get_config()
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 Menu:
def __init__(self, game):
self.game = 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
def render(self):
if self.is_menu_open:
self.surface.fill(self.color)
# 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))
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):
print('checking click on menu')
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.menu_width = 50
self.bar_height = self.menu_width / 4
self.bar_spacing = self.menu_width / 20
self.menu_height = self.bar_height * 3 + self.bar_spacing * 2
self.x = self.game.screen.get_width() - self.menu_width - 20
self.y = 20
self.color = (100, 100, 100)
self.menu = Menu(self.game)
self.rect = pygame.Rect(self.x, self.y, self.menu_width,
self.menu_height)
self.surface = pygame.Surface(
(self.menu_width, self.menu_height))
def render(self):
pygame.draw.rect(self.surface, self.color,
(0, 0, self.menu_width, self.bar_height),)
pygame.draw.rect(self.surface, self.color,
(0, self.bar_height + self.bar_spacing, self.menu_width, self.bar_height),)
pygame.draw.rect(self.surface, self.color,
(0, 2 * (self.bar_height + self.bar_spacing), self.menu_width, self.bar_height),)
self.game.screen.blit(self.surface, (self.x, self.y))
self.menu.render()
def handle_click(self):
pos = pygame.mouse.get_pos()
if self.rect.collidepoint(pos):
self.menu.is_menu_open = not self.menu.is_menu_open
self.menu.handle_click()
class Client:
def __init__(self):
hero = HeroFactory().build(size=50, x=100, y=100)
self.hero = HeroCreate(**hero.dict()).post()
# self.screen = pygame.display.set_mode((800, 600))
self.screen = pygame.display.set_mode((0, 0), pygame.FULLSCREEN)
pygame.display.set_caption("Learn SQL Model")
self.clock = pygame.time.Clock()
@ -136,7 +31,8 @@ class Client:
self.moving_right = False
self.ticks = 0
self.others = []
self.hamburger = Hamburger(self)
self.menu = Menu(self)
self.font = pygame.font.SysFont("", 50)
atexit.register(self.quit)
@ -144,7 +40,7 @@ class Client:
def ws(self):
def connect():
self._ws = create_connection(
f"ws://{config.api_client.url.replace('https://', '')}/wsecho"
f"wss://{config.api_client.url.replace('https://', '')}/wsecho"
)
if not hasattr(self, "_ws"):
@ -213,7 +109,7 @@ class Client:
self.screen, (255, 0, 0), (other.x, other.y), other.size
)
self.screen.blit(
my_font.render(other.name, False, (255, 255, 255), 1),
self.font.render(other.name, False, (255, 255, 255), 1),
(other.x, other.y),
)
@ -221,12 +117,12 @@ class Client:
self.screen, (0, 0, 255), (self.hero.x, self.hero.y), self.hero.size
)
self.screen.blit(
my_font.render(self.hero.name, False, (255, 255, 255)),
self.font.render(self.hero.name, False, (255, 255, 255)),
(self.hero.x, self.hero.y),
)
# update the screen
self.hamburger.render()
self.menu.render()
pygame.display.flip()
def handle_events(self):
@ -276,7 +172,7 @@ class Client:
self.moving_right = False
if event.type == pygame.MOUSEBUTTONDOWN:
if event.button == 1: # Left mouse button
self.hamburger.handle_click()
self.menu.handle_click()
def check_events(self):
pass

View file

@ -0,0 +1,129 @@
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