menu manager to make cleaner

This commit is contained in:
Vincent Rodley 2025-08-21 11:49:03 +12:00
parent f17c76d072
commit 18bde75910
2 changed files with 127 additions and 85 deletions

View File

@ -1,69 +1,28 @@
# Imports
# imports
import pygame
from button import Button
from menu_manager import MenuManager
# some pygame constst
# consts
WINDOW_SIZE = (768, 768)
TARGET_FPS = 60
COLS, ROWS = 7, 6
TILE_SIZE, TILE_SPACING = 50, 20
# pygame inits
GRID_WIDTH = COLS * TILE_SIZE + (COLS - 1) * TILE_SPACING
GRID_HEIGHT = ROWS * TILE_SIZE + (ROWS - 1) * TILE_SPACING
# init the pygame
pygame.init()
font = pygame.font.Font("Baloo2-Bold.ttf", 40)
display = pygame.display.set_mode(WINDOW_SIZE)
clock = pygame.time.Clock()
# more variable inits
menu = ""
# variables
tiles = []
COLS = 7
ROWS = 6
# menu functions
def change_menu(targetMenu):
global menu
menu = targetMenu
display.fill(bg_color)
def start_game_func(*_):
change_menu("game")
create_tiles()
def settings_menu(*_):
change_menu("settings")
def go_back(*_):
change_menu("start")
def create_tiles():
global tiles
tiles = []
for c in range(COLS):
for r in range(ROWS):
tile = Button(
50*c+50, 50*r+50, 30, 30, str(len(tiles)),
tile_color, tile_hover, tile_text, tile_press, None, 30, (len(tiles),c,r),
rounding=5
)
tiles.append(tile)
# gets called when you click on a tile
def tile_press(tile):
tile_id,x,y = tile.extra_data
print(f"TILE {tile_id} at {x},{y} PRESSED")
# Main block
if __name__ == "__main__":
# You're running the game, therefore running = True
running = True
# Button inits
width = 280
height = 75
x = WINDOW_SIZE[0] / 2 - width / 2 # center of the screen horizontally
y = WINDOW_SIZE[1] / 2 - height / 2 # center of the screen vertically
menu_manager = MenuManager(display, (30, 30, 40)) # background color
# colorss
primary_color = (70, 130, 180) # button background
hover_color = (51, 102, 145) # button hover
text_color = (245, 245, 245) # button text
@ -72,58 +31,110 @@ if __name__ == "__main__":
tile_text = (50, 50, 50) # tile text
bg_color = (30, 30, 40) # main background
start_button = Button(x, y - 100, width, height, "Start Game", primary_color, hover_color, text_color, start_game_func, "Baloo2-Bold.ttf", 50, rounding=8)
settings_button = Button(x, y-100+height*2, width, height, "Settings", primary_color, hover_color, text_color, settings_menu, "Baloo2-Bold.ttf", 50, rounding=8)
go_back_button = Button(x, y, width, height, "Go back", primary_color, hover_color, text_color, go_back, "Baloo2-Bold.ttf", 50, rounding=8)
# menu functions
def start_game_func(*_):
create_tiles()
menu_manager.change_menu("game")
change_menu("start")
def settings_menu(*_):
menu_manager.change_menu("settings")
# Game loop
while running:
def go_back(*_):
menu_manager.change_menu("start")
# handles user input
for event in pygame.event.get():
# Lets you actually close the game, or ESC out
if event.type == pygame.QUIT:
running = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
running = False
# tile stuff
def create_tiles():
global tiles
tiles = []
# Handles all the inputs for the buttons
if menu == "start":
start_x = (WINDOW_SIZE[0] - GRID_WIDTH) // 2
start_y = (WINDOW_SIZE[1] - GRID_HEIGHT) // 2
for c in range(COLS):
for r in range(ROWS):
x = start_x + c * (TILE_SIZE + TILE_SPACING)
y = start_y + r * (TILE_SIZE + TILE_SPACING)
tile = Button(
x, y, TILE_SIZE, TILE_SIZE, str(len(tiles)),
tile_color, tile_hover, tile_text, tile_press, None, 30, (len(tiles), c, r),
rounding=5
)
tiles.append(tile)
def tile_press(tile):
tile_id, x, y = tile.extra_data
print(f"TILE {tile_id} at {x},{y} PRESSED")
# button stuff
width, height = 280, 75
x = WINDOW_SIZE[0] / 2 - width / 2
y = WINDOW_SIZE[1] / 2 - height / 2
start_button = Button(x, y - 100, width, height, "Start Game",
primary_color, hover_color, text_color,
start_game_func, "Baloo2-Bold.ttf", 50, rounding=8)
settings_button = Button(x, y - 100 + height * 2, width, height, "Settings",
primary_color, hover_color, text_color,
settings_menu, "Baloo2-Bold.ttf", 50, rounding=8)
go_back_button = Button(x, y, width, height, "Go back",
primary_color, hover_color, text_color,
go_back, "Baloo2-Bold.ttf", 50, rounding=8)
# menu handlers
# start
def start_menu_events(event):
start_button.handle_event(event)
settings_button.handle_event(event)
elif menu == "settings":
go_back_button.handle_event(event)
elif menu == "game":
for tile in tiles:
tile.handle_event(event)
else:
# very descriptive error message
print("You broke smth idek what tbh")
running = False
# Display stuff!
# so depending on what menu you're in, draw different stuff
if menu == "start":
def start_menu_draw():
start_button.draw(display)
settings_button.draw(display)
elif menu == "settings":
# settings
def settings_menu_events(event):
go_back_button.handle_event(event)
def settings_menu_draw():
text_surface = font.render("No settings yet :(", True, text_color)
text_rect = text_surface.get_rect(center=(WINDOW_SIZE[0] / 2, WINDOW_SIZE[1] / 2 - 100))
display.blit(text_surface, text_rect)
go_back_button.draw(display)
# basic connect-4 ahh grid
elif menu == "game":
# game
def game_menu_events(event):
for tile in tiles:
tile.handle_event(event)
def game_menu_draw():
for tile in tiles:
tile.draw(display)
else:
# very descriptive error msg
print("you broke smth.")
# register the menus
menu_manager.register_menu("start", start_menu_events, start_menu_draw)
menu_manager.register_menu("settings", settings_menu_events, settings_menu_draw)
menu_manager.register_menu("game", game_menu_events, game_menu_draw)
menu_manager.change_menu("start")
# main loopy loopy
if __name__ == "__main__":
running = True
while running:
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
if event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE:
running = False
# flip the display and clock the tick so stuff actually updates
menu_manager.handle_event(event)
menu_manager.draw()
pygame.display.flip()
clock.tick(TARGET_FPS)

31
gui/menu_manager.py Normal file
View File

@ -0,0 +1,31 @@
import pygame
class MenuManager:
def __init__(self, display, bg_color):
self.display = display
self.bg_color = bg_color
self.current_menu = None
self.event_handlers = {}
self.draw_handlers = {}
# switch menu, and clear the screen
def change_menu(self, name):
if name not in self.draw_handlers:
raise ValueError(f"Menu: '{name} not registered")
self.current_menu = name
self.display.fill(self.bg_color)
# register a menus event and draw functions
def register_menu(self, name, event_handler, draw_handler):
self.event_handlers[name] = event_handler
self.draw_handlers[name] = draw_handler
# pass event to the menu handler
def handle_event(self, event):
if self.current_menu in self.event_handlers:
self.event_handlers[self.current_menu](event)
# draw the current menu
def draw(self):
if self.current_menu in self.draw_handlers:
self.draw_handlers[self.current_menu]()