WinScreen.java

/*
* Authors: Jerry Li & Victor Jiang
* Date: June 13, 2025
* Description: This class renders a win screen when all waves are cleared
*/

import java.awt.*;
import java.awt.event.ActionListener;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import javax.imageio.ImageIO;
import javax.swing.*;

public class WinScreen extends JPanel {

    public final GamePanel PARENT;
    public final Font UI_FONT;
    public BufferedImage backgroundImage;
    public JLabel messageLabel;

    public WinScreen(GamePanel parent) {
        PARENT = parent;
        setLayout(null);
        setOpaque(false);
        UI_FONT = parent.getGameFont();

        loadBackgroundImage();
        setupComponents();
    }

    public void loadBackgroundImage() {
        try {
            backgroundImage = ImageIO.read(new File("./assets/images/backgrounds/mainmenubg.jpg"));
        } catch (IOException e) {
            System.err.println("Error loading win screen background image: " + e.getMessage());
            backgroundImage = null;
        }
    }

    public void setupComponents() {
        JLabel titleLabel;
        GameButton playButton;
        GameButton menuButton;

        // Victory title
        titleLabel = new JLabel("YOU WIN!");
        titleLabel.setForeground(new Color(0, 220, 0));
        titleLabel.setFont(UI_FONT.deriveFont(Font.BOLD, 72f));
        titleLabel.setHorizontalAlignment(SwingConstants.CENTER);
        add(titleLabel);

        // Message display
        messageLabel = new JLabel();
        messageLabel.setForeground(Color.WHITE);
        messageLabel.setFont(UI_FONT.deriveFont(Font.BOLD, 32f));
        messageLabel.setHorizontalAlignment(SwingConstants.CENTER);
        add(messageLabel);

        // Play again button
        playButton = makeMenuButton("PLAY AGAIN", 36f, e -> {
            PARENT.resetGame();
            if (Settings.DEBUG_MODE) {
                PARENT.setCurrentWave(DebugConfig.startWave);
                PARENT.playScreen.applyDebugSettings();
            }
            PARENT.showScreen("PLAYING");
        });
        add(playButton);

        // Main menu button
        menuButton = makeMenuButton("MAIN MENU", 36f, e -> {
            PARENT.showScreen("MAIN_MENU");
        });
        add(menuButton);

        addComponentListener(new java.awt.event.ComponentAdapter() {
            @Override
            public void componentResized(java.awt.event.ComponentEvent e) {
                positionComponents();
            }
        });
    }

    public GameButton makeMenuButton(String text, float size, ActionListener action) {
        GameButton b;

        b = new GameButton(text);
        b.setFont(UI_FONT.deriveFont(Font.BOLD, size));
        b.addActionListener(action);
        b.setColor(new Color(76, 72, 144));
        b.addMouseListener(new java.awt.event.MouseAdapter() {
            @Override
            public void mouseEntered(java.awt.event.MouseEvent e) {
                b.setFont(UI_FONT.deriveFont(Font.BOLD, size * 1.15f));
            }
            @Override
            public void mouseExited(java.awt.event.MouseEvent e) {
                b.setFont(UI_FONT.deriveFont(Font.BOLD, size));
            }
        });
        return b;
    }

    public void positionComponents() {
        int w;
        int h;
        Component[] components;
        int btnWidth = 0;
        int btnHeight = 0;

        w = getWidth() > 0 ? getWidth() : GamePanel.GAME_WIDTH;
        h = getHeight() > 0 ? getHeight() : GamePanel.GAME_HEIGHT;

        components = getComponents();
        if (components.length >= 4) {
            components[0].setBounds(0, h/5, w, 80);        // Title
            components[1].setBounds(0, h/5 + 100, w, 40);   // Message
            btnWidth = 300;
            btnHeight = 80;
            components[2].setBounds(w/2 - btnWidth/2, h/5 + 180, btnWidth, btnHeight);
            components[3].setBounds(w/2 - btnWidth/2, h/5 + 280, btnWidth, btnHeight);
        }
    }

    @Override
    protected void paintComponent(Graphics g) {
        Graphics2D g2d;
        g2d = (Graphics2D) g.create();

        g2d.setColor(Color.BLACK);
        g2d.fillRect(0, 0, getWidth(), getHeight());

        if (backgroundImage != null) {
            g2d.setComposite(AlphaComposite.SrcOver.derive(0.3f));
            g2d.drawImage(backgroundImage, 0, 0, getWidth(), getHeight(), null);
        }

        g2d.dispose();
        super.paintComponent(g);
    }

    @Override
    public void invalidate() {
        super.invalidate();
        positionComponents();
    }

    @Override
    public void addNotify() {
        super.addNotify();
        updateWaveDisplay();
    }

    public void updateWaveDisplay() {
        int total;
        String plural;

        total = 0;
        plural = null;
        if (messageLabel != null) {
            total = PARENT.waves != null ? PARENT.waves.size() : 0;
            plural = total == 1 ? " wave" : " waves";
            messageLabel.setText("You survived " + total + plural + "!");
        }
    }
}