ui.py

"""
Interface do usuário: HUD, menus e efeitos visuais.
"""

import pygame
from settings import (
    FONT_SMALL, FONT_MEDIUM, FONT_LARGE,
    WHITE, YELLOW, RED, BLUE, BLACK,
    SCREEN_W, SCREEN_H, TILE_SIZE, STATE_MENU,
    STATE_PLAYING, STATE_PAUSED, STATE_GAMEOVER, STATE_WIN
)


class UI:
    def __init__(self):
        self.flash_timer = 0
        self.high_score = 0

    def draw_hud(self, surface, score, lives, level):
        # Fundo da HUD
        hud_rect = pygame.Rect(0, 0, SCREEN_W, 50)
        pygame.draw.rect(surface, BLACK, hud_rect)
        pygame.draw.line(surface, BLUE, (0, 50), (SCREEN_W, 50), 2)

        # Score
        score_text = FONT_MEDIUM.render(f"SCORE: {score}", True, WHITE)
        surface.blit(score_text, (10, 10))

        # High Score
        hs_text = FONT_MEDIUM.render(f"HIGH: {self.high_score}", True, YELLOW)
        surface.blit(hs_text, (SCREEN_W // 2 - hs_text.get_width() // 2, 10))

        # Level
        lvl_text = FONT_MEDIUM.render(f"LEVEL: {level}", True, WHITE)
        surface.blit(lvl_text, (SCREEN_W - lvl_text.get_width() - 10, 10))

        # Vidas (desenha mini Pacmans)
        lives_text = FONT_SMALL.render("LIVES:", True, WHITE)
        surface.blit(lives_text, (10, 28))
        for i in range(lives):
            cx = 70 + i * 20
            cy = 37
            pygame.draw.circle(surface, YELLOW, (cx, cy), 6)

    def draw_menu(self, surface):
        overlay = pygame.Surface((SCREEN_W, SCREEN_H))
        overlay.set_alpha(200)
        overlay.fill(BLACK)
        surface.blit(overlay, (0, 0))

        title = FONT_LARGE.render("PACMAN", True, YELLOW)
        surface.blit(title, (SCREEN_W // 2 - title.get_width() // 2, 100))

        self.flash_timer += 1
        if self.flash_timer % 60 < 30:
            start = FONT_MEDIUM.render("PRESS ENTER TO START", True, WHITE)
            surface.blit(start, (SCREEN_W // 2 - start.get_width() // 2, 250))

        controls = FONT_SMALL.render("ARROWS / WASD to move  |  P to pause", True, WHITE)
        surface.blit(controls, (SCREEN_W // 2 - controls.get_width() // 2, 320))

        ghosts_info = [
            ("BLINKY", RED, "chases you directly"),
            ("PINKY", (255, 184, 255), "ambushes from ahead"),
            ("INKY", CYAN, "unpredictable patterns"),
            ("CLYDE", ORANGE, "switches between chase and scatter"),
        ]
        y = 380
        for name, color, desc in ghosts_info:
            pygame.draw.circle(surface, color, (SCREEN_W // 2 - 140, y), 8)
            txt = FONT_SMALL.render(f"{name}: {desc}", True, WHITE)
            surface.blit(txt, (SCREEN_W // 2 - 120, y - 7))
            y += 22

    def draw_pause(self, surface):
        overlay = pygame.Surface((SCREEN_W, SCREEN_H))
        overlay.set_alpha(180)
        overlay.fill(BLACK)
        surface.blit(overlay, (0, 0))

        pause = FONT_LARGE.render("PAUSED", True, YELLOW)
        surface.blit(pause, (SCREEN_W // 2 - pause.get_width() // 2, SCREEN_H // 2 - 40))

        self.flash_timer += 1
        if self.flash_timer % 60 < 30:
            cont = FONT_MEDIUM.render("PRESS P TO RESUME", True, WHITE)
            surface.blit(cont, (SCREEN_W // 2 - cont.get_width() // 2, SCREEN_H // 2 + 20))

    def draw_gameover(self, surface, score, new_high):
        overlay = pygame.Surface((SCREEN_W, SCREEN_H))
        overlay.set_alpha(200)
        overlay.fill(BLACK)
        surface.blit(overlay, (0, 0))

        go = FONT_LARGE.render("GAME OVER", True, RED)
        surface.blit(go, (SCREEN_W // 2 - go.get_width() // 2, 150))

        sc = FONT_MEDIUM.render(f"FINAL SCORE: {score}", True, WHITE)
        surface.blit(sc, (SCREEN_W // 2 - sc.get_width() // 2, 230))

        if new_high:
            nh = FONT_MEDIUM.render("NEW HIGH SCORE!", True, YELLOW)
            surface.blit(nh, (SCREEN_W // 2 - nh.get_width() // 2, 270))

        self.flash_timer += 1
        if self.flash_timer % 60 < 30:
            restart = FONT_MEDIUM.render("PRESS ENTER TO RESTART", True, WHITE)
            surface.blit(restart, (SCREEN_W // 2 - restart.get_width() // 2, 330))

    def draw_win(self, surface, score, level):
        overlay = pygame.Surface((SCREEN_W, SCREEN_H))
        overlay.set_alpha(200)
        overlay.fill(BLACK)
        surface.blit(overlay, (0, 0))

        win = FONT_LARGE.render("YOU WIN!", True, GREEN)
        surface.blit(win, (SCREEN_W // 2 - win.get_width() // 2, 150))

        sc = FONT_MEDIUM.render(f"SCORE: {score}", True, WHITE)
        surface.blit(sc, (SCREEN_W // 2 - sc.get_width() // 2, 230))

        nxt = FONT_MEDIUM.render(f"NEXT LEVEL: {level + 1}", True, YELLOW)
        surface.blit(nxt, (SCREEN_W // 2 - nxt.get_width() // 2, 270))

        self.flash_timer += 1
        if self.flash_timer % 60 < 30:
            cont = FONT_MEDIUM.render("PRESS ENTER TO CONTINUE", True, WHITE)
            surface.blit(cont, (SCREEN_W // 2 - cont.get_width() // 2, 330))

    def draw_ready(self, surface):
        self.flash_timer += 1
        if self.flash_timer % 60 < 30:
            ready = FONT_LARGE.render("READY!", True, YELLOW)
            surface.blit(ready, (SCREEN_W // 2 - ready.get_width() // 2, SCREEN_H // 2 - 50))

    def draw_fruit(self, surface, fruit_pos, fruit_type):
        if fruit_pos is None:
            return
        cx = fruit_pos[0] * TILE_SIZE + TILE_SIZE // 2
        cy = fruit_pos[1] * TILE_SIZE + TILE_SIZE // 2
        colors = {
            'cherry': RED,
            'strawberry': (255, 0, 100),
            'orange': ORANGE,
            'apple': (255, 0, 0),
            'melon': GREEN,
        }
        color = colors.get(fruit_type, RED)
        pygame.draw.circle(surface, color, (cx, cy), 8)
        pygame.draw.circle(surface, WHITE, (cx - 3, cy - 3), 2)
        pygame.draw.circle(surface, WHITE, (cx + 3, cy - 3), 2)


# Constante GREEN local
GREEN = (0, 255, 0)