main.py

import pygame
import random
import sys

# --------------------
# INITIAL SETUP
# --------------------
pygame.init()
pygame.mixer.init()

# 🎥 Dev size (recorded & upscaled later)
WIDTH, HEIGHT = 540, 960
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Santa Ice Skate - Shorts")

clock = pygame.time.Clock()
FPS = 60

# --------------------
# LOAD ASSETS
# --------------------
def load_image(path, scale=None):
    img = pygame.image.load(path).convert_alpha()
    if scale:
        img = pygame.transform.scale(img, scale)
    return img

# --------------------
# IMAGES (⬆ BIGGER)
# --------------------
bg = load_image("assets/ice-track5.png", (WIDTH, HEIGHT))

santa_img = load_image("assets/santa-on-skate.png", (150, 180))   # ⬆ BIG
ice_block = load_image("assets/ice-block2.png", (90, 70))          # ⬆ BIG
snowman = load_image("assets/snowman.png", (95, 120))             # ⬆ BIG
coin = load_image("assets/christmas-coin.png", (55, 55))          # ⬆ BIG
candy = load_image("assets/candy-cane.png", (50, 85))             # ⬆ BIG
slip_fx = load_image("assets/ice-slip-effect.png", (180, 180))    # ⬆ BIG

# --------------------
# SOUNDS
# --------------------
music = pygame.mixer.Sound("assets/music.wav")
jump_snd = pygame.mixer.Sound("assets/jump.wav")
collect_snd = pygame.mixer.Sound("assets/collect.wav")
fail_snd = pygame.mixer.Sound("assets/fail.wav")

music.set_volume(0.35)
jump_snd.set_volume(0.7)
collect_snd.set_volume(0.7)
fail_snd.set_volume(0.9)

music.play(-1)

# --------------------
# BACKGROUND SCROLL
# --------------------
bg_y1 = 0
bg_y2 = -HEIGHT
base_speed = 7.5   # slightly faster

# --------------------
# PLAYER
# --------------------
GROUND_Y = HEIGHT - 260

santa = santa_img.get_rect()
santa.midbottom = (WIDTH // 2, GROUND_Y)

# 🔥 Fair Santa hitbox (still smaller than sprite)
santa_hitbox = santa.inflate(-45, -35)

speed_x = 13
jump_power = -22
gravity = 1.2
vel_y = 0
on_ground = True

# --------------------
# OBJECTS
# --------------------
obstacles = []
coins = []
spawn_timer = 0
score = 0
game_over = False
fx_rect = None
fail_played = False

# --------------------
# FONT (⬆ BIGGER)
# --------------------
font = pygame.font.SysFont("arial", 52, bold=True)

# --------------------
# RESET FUNCTION
# --------------------
def reset_game():
    global obstacles, coins, spawn_timer, score, game_over
    global vel_y, on_ground, fail_played, fx_rect
    global bg_y1, bg_y2

    obstacles.clear()
    coins.clear()
    spawn_timer = 0
    score = 0
    game_over = False
    vel_y = 0
    on_ground = True
    fail_played = False
    fx_rect = None

    santa.midbottom = (WIDTH // 2, GROUND_Y)
    santa_hitbox.center = santa.center

    bg_y1 = 0
    bg_y2 = -HEIGHT
    music.play(-1)

# --------------------
# GAME LOOP
# --------------------
running = True
while running:
    clock.tick(FPS)

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()

        if event.type == pygame.KEYDOWN:
            if game_over and event.key == pygame.K_r:
                reset_game()

    keys = pygame.key.get_pressed()

    if not game_over:
        # --------------------
        # SPEED (DYNAMIC)
        # --------------------
        current_speed = base_speed + score * 0.08

        # --------------------
        # BACKGROUND MOVE
        # --------------------
        bg_y1 += current_speed
        bg_y2 += current_speed
        if bg_y1 >= HEIGHT:
            bg_y1 = -HEIGHT
        if bg_y2 >= HEIGHT:
            bg_y2 = -HEIGHT

        # --------------------
        # PLAYER MOVEMENT
        # --------------------
        if keys[pygame.K_LEFT]:
            santa.x -= speed_x
        if keys[pygame.K_RIGHT]:
            santa.x += speed_x

        if keys[pygame.K_SPACE] and on_ground:
            vel_y = jump_power
            on_ground = False
            jump_snd.play()

        vel_y += gravity
        santa.y += vel_y

        if santa.bottom >= GROUND_Y:
            santa.bottom = GROUND_Y
            vel_y = 0
            on_ground = True

        santa.x = max(30, min(WIDTH - santa.width - 30, santa.x))
        santa_hitbox.center = santa.center

        # --------------------
        # SPAWN OBJECTS
        # --------------------
        spawn_timer += 1
        if spawn_timer > 45:
            spawn_timer = 0
            lane_x = random.choice([WIDTH//4, WIDTH//2, WIDTH*3//4])

            if random.random() < 0.6:
                obj = random.choice([ice_block, snowman])
                rect = obj.get_rect(midtop=(lane_x, -140))
                rect.inflate_ip(-30, -30)  # fair hitbox
                obstacles.append((obj, rect))
            else:
                obj = random.choice([coin, candy])
                rect = obj.get_rect(midtop=(lane_x, -90))
                coins.append((obj, rect))

        # --------------------
        # MOVE OBJECTS
        # --------------------
        for obj, rect in obstacles:
            rect.y += current_speed

        for obj, rect in coins:
            rect.y += current_speed

        obstacles = [(o, r) for o, r in obstacles if r.y < HEIGHT + 200]
        coins = [(o, r) for o, r in coins if r.y < HEIGHT + 150]

        # --------------------
        # COLLISIONS
        # --------------------
        for obj, rect in obstacles:
            if santa_hitbox.colliderect(rect):
                game_over = True
                fx_rect = slip_fx.get_rect(center=santa.center)
                if not fail_played:
                    fail_snd.play()
                    music.stop()
                    fail_played = True

        for obj, rect in coins[:]:
            if santa_hitbox.colliderect(rect):
                coins.remove((obj, rect))
                score += 1
                collect_snd.play()

    # --------------------
    # DRAW
    # --------------------
    screen.blit(bg, (0, bg_y1))
    screen.blit(bg, (0, bg_y2))

    for obj, rect in obstacles:
        screen.blit(obj, rect)

    for obj, rect in coins:
        screen.blit(obj, rect)

    screen.blit(santa_img, santa)

    # SCORE (RED, BIG, CLEAR)
    score_text = font.render(str(score), True, (255, 0, 0))
    screen.blit(score_text, (WIDTH//2 - score_text.get_width()//2, 20))

    if game_over:
        screen.blit(slip_fx, fx_rect)
        over_text = font.render("GAME OVER", True, (255, 60, 60))
        retry_text = font.render("PRESS R", True, (0, 255, 0))
        screen.blit(over_text, (WIDTH//2 - over_text.get_width()//2, HEIGHT//2 - 50))
        screen.blit(retry_text, (WIDTH//2 - retry_text.get_width()//2, HEIGHT//2 + 10))


    pygame.display.update()