tutorial.py

"""Primary file for the platformer game"""

import os
import random
import pygame
from os import listdir
from os.path import isfile, join

pygame.init()

pygame.display.set_caption("Platformer")

BG_COLOR = (255, 255, 255)
WIDTH, HEIGHT = 800, 600
FPS = 60
PLAYER_VEL = 5

WINDOW = pygame.display.set_mode((WIDTH, HEIGHT))


def flip(sprites: list[pygame.Surface]):
    """Flip the sprites"""
    return [pygame.transform.flip(sprite, True, False) for sprite in sprites]


def load_sprite_sheets(
    dir1: str, dir2: str, width: int, height: int, direction: bool = False
):
    """Loads sprite sheets"""
    path = join("assets", dir1, dir2)
    images = [f for f in listdir(path) if isfile(join(path, f))]

    all_sprites: dict[str, list[pygame.Surface]] = {}

    for image in images:
        sprite_sheet = pygame.image.load(join(path, image)).convert_alpha()

        sprites: list[pygame.Surface] = []
        for i in range(sprite_sheet.get_width() // width):
            surface = pygame.Surface((width, height), pygame.SRCALPHA, 32)
            rect = pygame.Rect(i * width, 0, width, height)
            surface.blit(sprite_sheet, (0, 0), rect)
            sprites.append(pygame.transform.scale2x(surface))

        if direction:
            all_sprites[image.replace(".png", "") + "_right"] = sprites
            all_sprites[image.replace(".png", "") + "_left"] = flip(sprites)
        else:
            all_sprites[image.replace(".png", "")] = sprites

    return all_sprites


class Player(pygame.sprite.Sprite):
    """Player Class"""

    COLOR = (255, 0, 0)
    GRAVITY = 1

    def __init__(self, x: int, y: int, width: int, height: int) -> None:
        self.rect = pygame.Rect(x, y, width, height)
        self.x_vel = 0
        self.y_vel = 0
        self.mask = None
        self.direction = "left"
        self.animation_count = 0
        self.fall_count = 0

    def move(self, dx: int, dy: int):
        """Move the player"""
        self.rect.x += dx
        self.rect.y += dy

    def move_left(self, vel: int):
        """Change the x direction to move left"""
        self.x_vel = -vel
        if self.direction != "left":
            self.direction = "left"
            self.animation_count = 0

    def move_right(self, vel: int):
        """Change the x direction to move right"""
        self.x_vel = vel
        if self.direction != "right":
            self.direction = "right"
            self.animation_count = 0

    def loop(self, fps: int):
        """Loop the animation"""
        self.y_vel += min(1, int((self.fall_count / fps) * self.GRAVITY))
        self.move(self.x_vel, self.y_vel)

        self.fall_count += 1

    def draw(self, win: pygame.Surface):
        pygame.draw.rect(win, self.COLOR, self.rect)


def get_background(name: str):
    """Get background"""
    image: pygame.Surface = pygame.image.load(join("assets", "Background", name))
    _, _, width, height = image.get_rect()
    tiles: list[tuple[int, int]] = []

    for i in range(WIDTH // width + 1):
        for j in range(HEIGHT // height + 1):
            pos = (i * width, j * height)
            tiles.append(pos)

    return tiles, image


def draw(
    window: pygame.Surface,
    background: list[tuple[int, int]],
    bg_image: pygame.Surface,
    player: Player,
) -> None:
    """Draw the background"""
    for tile in background:
        window.blit(bg_image, tile)

    player.draw(window)

    pygame.display.update()


def handle_move(player: Player):
    """Handle player movement"""
    keys = pygame.key.get_pressed()

    player.x_vel = 0
    if keys[pygame.K_LEFT]:
        player.move_left(PLAYER_VEL)
    if keys[pygame.K_RIGHT]:
        player.move_right(PLAYER_VEL)


def main(window: pygame.Surface) -> None:
    """Main function"""
    clock = pygame.time.Clock()
    background, bg_image = get_background("Blue.png")

    player = Player(100, 100, 50, 50)

    run: bool = True
    while run:
        clock.tick(FPS)

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
                break

        player.loop(FPS)
        handle_move(player)
        draw(window, background, bg_image, player)

    pygame.quit()
    quit()


if __name__ == "__main__":
    main(WINDOW)