"""
-----------------------------------------
Project: Escape Drake
Standard: 91896 (2.7)
School: Tauranga Boys' Collage
Auther: Hunter Moss
Date: May 2024
Python: 3.11.2
-----------------------------------------
"""
import pygame
import os
import pickle
pygame.init()

# Screen Size
screen_width = 1000
screen_height = 1000
screen = pygame.display.set_mode((screen_width, screen_height))
pygame.display.set_caption('Escape Drake')

# Load images
background_img = pygame.image.load('Images/Background.png')

# Constants
tile_size = 50

# Load player images function
def load_images(folder, prefix, count):
    return [pygame.transform.scale(pygame.image.load(os.path.join(folder, f"{prefix} ({i}).png")), (tile_size, tile_size)) for i in range(1, count + 1)]

# Player class
class Player(pygame.sprite.Sprite):
    def __init__(self):
        super().__init__()
        self.images = {
            'idle': load_images('Images/Player', 'idle', 16),
            'run': load_images('Images/Player', 'run', 20),
            'jump': load_images('Images/Player', 'jump', 30)
        }
        self.image = self.images['idle'][0]
        self.rect = self.image.get_rect(topleft=(screen_width // 2, screen_height // 2))  # Adjust topleft position
        self.vel = pygame.math.Vector2(0, 0)
        self.speed = 5
        self.gravity = 0.5
        self.jump_power = -10
        self.on_ground = False
        self.jumping = False
        self.facing_right = True
        self.frame = 0
        self.idle_animation_speed = 0.2  # Adjust idle animation speed
        self.animation_speed = 0.6  # Adjust animation speed

    def update(self):
        keys = pygame.key.get_pressed()

        # Horizontal movement
        if keys[pygame.K_LEFT]:
            self.vel.x = -self.speed
            self.facing_right = False
        elif keys[pygame.K_RIGHT]:
            self.vel.x = self.speed
            self.facing_right = True
        else:
            self.vel.x = 0

        # Jumping
        if keys[pygame.K_UP] and self.on_ground:
            self.vel.y = self.jump_power
            self.on_ground = False
            self.jumping = True

        # Apply gravity
        self.vel.y += self.gravity

        # Update player position
        self.rect.x += self.vel.x
        self.rect.y += self.vel.y

        # Screen collision
        self.rect.clamp_ip(pygame.Rect(0, 0, screen_width, screen_height))

        # Animation handling
        if self.jumping:
            self.image = self.images['jump'][0]
        elif self.vel.x != 0:
            self.frame = (self.frame + self.animation_speed) % len(self.images['run'])
            self.image = self.images['run'][int(self.frame)]
        else:
            self.frame = (self.frame + self.idle_animation_speed) % len(self.images['idle'])
            self.image = self.images['idle'][int(self.frame)]

        # Flip sprite
        if not self.facing_right:
            self.image = pygame.transform.flip(self.image, True, False)

        # Reset jumping state if on ground
        if self.rect.bottom >= screen_height:
            self.on_ground = True
            self.jumping = False
            self.vel.y = 0

        # Draw rectangle around sprite for debugging
       # pygame.draw.rect(screen, (255, 0, 0), self.rect, 2)

# Main game loop
def main():
    run = True
    clock = pygame.time.Clock()
    player = Player()

    while run:
        screen.blit(background_img, (0, 0))

        # Event handling
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False

        # Update player
        player.update()

        # Draw player
        screen.blit(player.image, player.rect)

        pygame.display.update()
        clock.tick(60)  # Adjusted frame rate

    pygame.quit()

if __name__ == "__main__":
    main()