main.py

import pygame
from constants import *
from ui.colors import *
import classes

from ui.setup import *
# Initialize pygame
pygame.init()
screen = pygame.display.set_mode(size)
clock = pygame.time.Clock()
fps = 30

binary_tree = classes.BinaryTree(classes.Grid(rows, cols, cell_size), "GREEN")
wilson = classes.Wilson(classes.Grid(rows, cols, cell_size), "PURPLE_E")
side_winder = classes.SideWinder(classes.Grid(rows, cols, cell_size), "BLUE")
hunt_and_kill = classes.HuntAndKill(classes.Grid(rows, cols, cell_size), "RED")
aldous_broder = classes.AldousBroder(classes.Grid(rows, cols, cell_size), "GREEN")
recursive_backtracker = classes.RecursiveBacktracker(classes.Grid(rows, cols, cell_size), "BLUE")
kruskal = classes.Kruskals(classes.Kruskals.State(classes.Grid(rows, cols, cell_size)))
simplePrims = classes.SimplePrims(classes.Grid(rows, cols, cell_size), "CYAN")
prims = classes.Prims(classes.Grid(rows, cols, cell_size))
growingTree = classes.GrowingTree(classes.Grid(rows, cols, cell_size), "GREEN")
ellers = classes.Ellers(classes.Grid(rows, cols, cell_size), 0, "RED")

show_text = False
color_mode = False
show_path = False
start = False
run = True
while run:
    #screen.fill(black)
    # Set Caption and fps
    clock.tick(fps)
    frame_rate = int(clock.get_fps())
    pygame.display.set_caption(f"Maze Generator - FPS: {frame_rate}")

    # Handle events
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            run = False
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_ESCAPE:
                run = False
            if event.key == pygame.K_RETURN:
                start = not start
            elif event.key == pygame.K_h:
                show_text = not show_text
            elif event.key == pygame.K_SPACE:
                color_mode = not color_mode
            elif event.key == pygame.K_s:
                show_path = not show_path
        if event.type == pygame.MOUSEBUTTONDOWN:
            if event.button == 1:
                rightMouseClicked = True

    if start:
        # wilson.Generate(screen, show_text, color_mode, show_path)
        # binary_tree.Generate(screen, show_text, color_mode, show_path)
        kruskal.Generate(screen, show_text, color_mode, show_path)
        # side_winder.Generate(screen, show_text, color_mode, show_path)
        # hunt_and_kill.Generate(screen, show_text, color_mode, show_path)
        # aldous_broder.Generate(screen, show_text, color_mode, show_path)
        # recursive_backtracker.Generate(screen, show_text, color_mode, show_path)
        # simplePrims.Generate(screen, show_text, color_mode, show_path)
        # prims.Generate(screen, show_text, color_mode, show_path)
        # growingTree.Generate(screen, show_text, color_mode, show_path)
        # ellers.Generate(screen, show_text, color_mode, show_path)
    else:
        PressEnter.Render(screen)

    pygame.display.flip()

pygame.image.save(screen, "./images/path.png")
pygame.quit()