import pygame
import random
import sys

# Initialize
pygame.init()
WIDTH, HEIGHT = 800, 400
win = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("PacMania")
clock = pygame.time.Clock()

# Colors
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
GREEN = (0, 255, 0)
BLUE = (0, 0, 255)
YELLOW = (255, 255, 0)
PURPLE = (128, 0, 128)
CYAN = (0, 255, 255)

TRAIL_COLORS = {
    "White": WHITE,
    "Red": RED,
    "Green": GREEN,
    "Blue": BLUE,
    "Yellow": YELLOW,
    "Purple": PURPLE,
    "Cyan": CYAN
}

purchased_trails = {"White": True}  # White trail is always owned
selected_trail = "White"

# Load and scale background image
background_img_raw = pygame.image.load('retro-sci-fi-futuristic-background-.jpg')
background_img = pygame.transform.scale(background_img_raw, (WIDTH, HEIGHT))

# Fonts
font = pygame.font.SysFont('Arial', 30)
menu_font = pygame.font.SysFont('Arial', 50)

# Player setup
player = pygame.Rect(100, 300, 30, 30)
player_dy = 0
gravity = 0.6
jump_power = -13
on_ground = False

# Load animation frames for the player
player_images = [
    pygame.transform.scale(pygame.image.load('pacmansprite1.png'), (30, 30)),
    pygame.transform.scale(pygame.image.load('pacmansprite2.png'), (30, 30)),
    pygame.transform.scale(pygame.image.load('pacmansprite3.png'), (30, 30))
]
current_frame = 0
animation_timer = 0
frame_delay = 100  # milliseconds

# Load other images
platform_image_raw = pygame.image.load('platform3.jpg')
ground_img_raw = pygame.image.load('ground1.jpg')
ground_img = pygame.transform.scale(ground_img_raw, (100, 40))

coin_image_raw = pygame.image.load('cherry pacman.png')
coin_image = pygame.transform.scale(coin_image_raw, (40, 40))

star_image_raw = pygame.image.load('coins_for_essay-removebg-preview.png')
star_image = pygame.transform.scale(star_image_raw, (20, 20))

enemy_image_raw = pygame.image.load('enemy1.png')
enemy_image = pygame.transform.scale(enemy_image_raw, (30, 30))

# World elements
ground_height = 40
ground_y = HEIGHT - ground_height
base_speed = 3
scroll_speed = base_speed
speed_level = 1
start_time = pygame.time.get_ticks()
speed_increase_interval = 8000

platforms = []
platform_timer = 0
last_platform_x = WIDTH

cherriess = []
coin_timer = 0
coin_count = 0

coins = []
star_timer = 0
star_count = 0

enemies = []
enemy_timer = 0

distance = 0
game_over = False
total_coins = 0

num_ground_images = 15
ground_x = [i * ground_img.get_width() for i in range(num_ground_images)]

trail_history = []
trail_limit = 30

def reset_game():
    global player, player_dy, on_ground, scroll_speed, speed_level, start_time
    global platforms, cherriess, coins, enemies
    global coin_count, star_count, distance, game_over
    global platform_timer, coin_timer, star_timer, enemy_timer, last_platform_x, total_coins, trail_history

    total_coins += star_count

    player = pygame.Rect(100, 300, 30, 30)
    player_dy = 0
    on_ground = False
    scroll_speed = base_speed
    speed_level = 1
    start_time = pygame.time.get_ticks()

    platforms.clear()
    cherriess.clear()
    coins.clear()
    enemies.clear()

    coin_count = 0
    star_count = 0
    distance = 0
    game_over = False
    trail_history.clear()

    platform_timer = 0
    coin_timer = 0
    star_timer = 0
    enemy_timer = 0
    last_platform_x = WIDTH

def shop_menu():
    global selected_trail, total_coins
    while True:
        win.blit(background_img, (0, 0))

        title_text = menu_font.render("Shop", True, WHITE)
        win.blit(title_text, (WIDTH // 2 - title_text.get_width() // 2, 30))

        back_button_rect = pygame.Rect(WIDTH - 150, HEIGHT - 60, 120, 40)
        pygame.draw.rect(win, WHITE, back_button_rect)
        back_text = font.render("Back", True, BLACK)
        win.blit(back_text, (back_button_rect.x + 25, back_button_rect.y + 5))

        y = 100
        for trail_name, color in TRAIL_COLORS.items():
            btn_rect = pygame.Rect(300, y, 200, 40)
            owned = purchased_trails.get(trail_name, False)

            pygame.draw.rect(win, color if owned else WHITE, btn_rect)
            text = f"{trail_name} {'(Owned)' if owned else '- 5 Coins'}"
            trail_label = font.render(text, True, BLACK)
            win.blit(trail_label, (btn_rect.x + 10, btn_rect.y + 5))

            if btn_rect.collidepoint(pygame.mouse.get_pos()) and pygame.mouse.get_pressed()[0]:
                pygame.time.delay(200)
                if owned:
                    selected_trail = trail_name
                elif total_coins >= 5:
                    total_coins -= 5
                    purchased_trails[trail_name] = True
                    selected_trail = trail_name

            y += 50

        pygame.display.update()

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            if event.type == pygame.MOUSEBUTTONDOWN:
                if back_button_rect.collidepoint(event.pos):
                    return

def main_menu():
    while True:
        win.blit(background_img, (0, 0))

        title_text = menu_font.render("PacMania", True, WHITE)
        start_button_text = font.render("Start", True, BLACK)
        shop_button_text = font.render("Shop", True, BLACK)

        win.blit(title_text, (WIDTH // 2 - title_text.get_width() // 2, HEIGHT // 2 - 120))

        button_width, button_height = 200, 50

        start_button_rect = pygame.Rect(WIDTH // 2 - button_width // 2, HEIGHT // 2 - 20, button_width, button_height)
        shop_button_rect = pygame.Rect(WIDTH // 2 - button_width // 2, HEIGHT // 2 + 50, button_width, button_height)

        pygame.draw.rect(win, WHITE, start_button_rect)
        pygame.draw.rect(win, WHITE, shop_button_rect)
        win.blit(start_button_text, (start_button_rect.x + 60, start_button_rect.y + 5))
        win.blit(shop_button_text, (shop_button_rect.x + 60, shop_button_rect.y + 5))

        pygame.display.update()

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            if event.type == pygame.MOUSEBUTTONDOWN:
                if start_button_rect.collidepoint(event.pos):
                    return
                if shop_button_rect.collidepoint(event.pos):
                    shop_menu()

# Start menu
main_menu()

# Game loop
running = True
while running:
    dt = clock.tick(60)
    win.blit(background_img, (0, 0))

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

    if not game_over:
        elapsed_time = pygame.time.get_ticks() - start_time
        new_level = elapsed_time // speed_increase_interval + 1
        if new_level > speed_level:
            speed_level = new_level
            scroll_speed = base_speed + speed_level - 1

        keys = pygame.key.get_pressed()
        if keys[pygame.K_SPACE] and on_ground:
            player_dy = jump_power
            on_ground = False

        player_dy += gravity
        player.y += player_dy
        on_ground = False

        if player.bottom >= ground_y:
            player.bottom = ground_y
            player_dy = 0
            on_ground = True

        for plat in platforms:
            plat.x -= scroll_speed
            if (
                player.right > plat.left + 5 and
                player.left < plat.right - 5 and
                player.bottom <= plat.top + 5 and
                player.bottom + player_dy >= plat.top
            ):
                player.bottom = plat.top
                player_dy = 0
                on_ground = True

        for coin in cherriess[:]:
            coin.x -= scroll_speed
            if player.colliderect(coin):
                cherriess.remove(coin)
                distance += 50
                coin_count += 1

        for star in coins[:]:
            star.x -= scroll_speed
            if player.colliderect(star):
                coins.remove(star)
                distance += 25
                star_count += 1

        for enemy in enemies[:]:
            enemy.x -= scroll_speed
            if player.colliderect(enemy):
                game_over = True

        platforms = [p for p in platforms if p.right > 0]
        cherriess = [c for c in cherriess if c.right > 0]
        coins = [s for s in coins if s.right > 0]
        enemies = [e for e in enemies if e.right > 0]

        platform_timer += 1
        if platform_timer > 100:
            y = random.randint(250, 270)
            w = random.randint(80, 120)
            x = last_platform_x + random.randint(5, 20)
            last_platform_x = x
            platforms.append(pygame.Rect(x, y, w, 15))
            platform_timer = 0

        coin_timer += 1
        if coin_timer > 150:
            cherriess.append(pygame.Rect(random.randint(WIDTH, WIDTH + 200), random.randint(150, 250), 20, 20))
            coin_timer = 0

        star_timer += 1
        if star_timer > 75:
            coins.append(pygame.Rect(random.randint(WIDTH, WIDTH + 200), random.randint(100, 220), 20, 20))
            star_timer = 0

        enemy_timer += 1
        if enemy_timer > 180:
            enemies.append(pygame.Rect(WIDTH + random.randint(0, 200), ground_y - 30, 30, 30))
            enemy_timer = 0

        distance += scroll_speed

        for i in range(num_ground_images):
            ground_x[i] -= scroll_speed
            if ground_x[i] <= -ground_img.get_width():
                ground_x[i] = ground_x[(i - 1) % num_ground_images] + ground_img.get_width()

        animation_timer += dt
        if animation_timer >= frame_delay:
            current_frame = (current_frame + 1) % len(player_images)
            animation_timer = 0

        # Draw trail
        trail_history.append(player.center)
        if len(trail_history) > trail_limit:
            trail_history.pop(0)
        for pos in trail_history:
            pygame.draw.circle(win, TRAIL_COLORS[selected_trail], pos, 5)

        for x in ground_x:
            win.blit(ground_img, (x, ground_y))

        for plat in platforms:
            win.blit(pygame.transform.scale(platform_image_raw, (plat.width, plat.height)), plat.topleft)

        for coin in cherriess:
            win.blit(coin_image, coin.topleft)

        for star in coins:
            win.blit(star_image, star.topleft)

        for enemy in enemies:
            win.blit(enemy_image, enemy.topleft)

        win.blit(player_images[current_frame], player.topleft)

        win.blit(font.render(f"Distance: {distance // 10}", True, WHITE), (10, 10))
        win.blit(font.render(f"Cherries: {coin_count}", True, WHITE), (10, 40))
        win.blit(font.render(f"Coins: {star_count}", True, WHITE), (10, 70))
        win.blit(font.render(f"Speed Lv: {speed_level}", True, WHITE), (10, 100))
        win.blit(font.render(f"Total Coins: {total_coins + star_count}", True, WHITE), (WIDTH - 250, 10))

    else:
        win.blit(font.render("GAME OVER", True, BLACK), (WIDTH // 2 - 100, HEIGHT // 2 - 60))
        button_rect = pygame.Rect(WIDTH // 2 - 100, HEIGHT // 2, 200, 50)
        pygame.draw.rect(win, WHITE, button_rect)
        win.blit(font.render("Restart", True, BLACK), (button_rect.x + 50, button_rect.y + 10))

        if pygame.mouse.get_pressed()[0] and button_rect.collidepoint(pygame.mouse.get_pos()):
            pygame.time.delay(200)
            reset_game()
            main_menu()

    pygame.display.update()

pygame.quit()