Yahtzee.java

import javax.swing.JFrame;
import java.awt.FlowLayout;
import javax.swing.BoxLayout;
import javax.swing.JLabel;
import java.awt.event.ActionListener;
import java.awt.event.ActionEvent;
import javax.swing.JButton;
import javax.swing.JTextField;
import java.util.Random;
import javax.swing.JPanel;
import java.awt.Graphics;
import java.awt.Color;
import java.awt.GridLayout;
import javax.swing.JToggleButton;
import javax.swing.Icon;
import javax.swing.ImageIcon;
import javax.swing.border.Border;
import javax.swing.BorderFactory;
import javax.swing.JComponent;
import java.awt.Component;

import java.awt.*;
import javax.swing.*;
import java.awt.event.*;
import java.lang.*;
import java.io.Serializable;
import java.io.*;

public class Yahtzee
{
    private static JMenuBar mb;
    private static JMenu options;
    static JMenuItem save, load;

   public static void main( String args[] )
   { 
      JFrame frame = new JFrame("Yahtzee");
      frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

        //menu bar
        mb = new JMenuBar();
        options = new JMenu("Options");
        save = new JMenuItem("Save");
        load = new JMenuItem("Load");
        options.add(save);
        options.add(load);
        mb.add(options);
        frame.setJMenuBar(mb);

      YahtzeePanel yp = new YahtzeePanel(); 
      save.addActionListener(yp);
      load.addActionListener(yp);
      frame.add(yp);
      frame.setSize(550, 475); // set frame size
      frame.setVisible(true); // display frame
   }
}

class YahtzeePanel extends JPanel implements ActionListener, Serializable
{
    private Scoresheet scoresheet;
    private Dice dice;
    private Integer turn, round;
    private transient JButton roll, play, newGame, loadGame;
    private boolean playable;
    private int players;
    private int curPlayer;

    public YahtzeePanel()
    {
        MainMenu();
    }

    public int promptNumplayers(){
        String num_players = JOptionPane.showInputDialog("How many people are playing?");
        if(num_players == null){
            return -1;
        }
        int players = Integer.parseInt(num_players);
        return players;
    }

    
    public int getCurPlayer(){
        return curPlayer;
    }


    public void MainMenu(){
        this.removeAll();

        setLayout(new FlowLayout());
        JLabel title = new JLabel("YAHTZEE!");
        title.setFont(new Font("Serif", Font.PLAIN, 100));
        add(title);

        newGame = new JButton("New Game");
        newGame.addActionListener(this);
        add(newGame);

        File f = new File("game_data.txt");
        loadGame = new JButton("Load Game");
        loadGame.addActionListener(this);
        if(f.exists()){
            add(loadGame);
        }
    }
    
    public void EndGame(){
        this.removeAll();

        setLayout(new BoxLayout(this, BoxLayout.PAGE_AXIS));
        JLabel winner = new JLabel("Player " + (scoresheet.getResult()[0] + 1) + " wins!");
        winner.setFont(new Font("Serif", Font.PLAIN, 64));
        add(winner);

        JLabel winnerScore = new JLabel("with a score of " + scoresheet.getResult()[1]);
        add(winnerScore);

        newGame = new JButton("New Game");
        newGame.addActionListener(this);
        add(newGame);

        File f = new File("game_data.txt");
        loadGame = new JButton("Load Game");
        loadGame.addActionListener(this);
        if(f.exists()){
            add(loadGame);
        }
        this.revalidate();
        this.repaint();
    }

    public void buildPanel(){
        this.removeAll();

        setLayout(new FlowLayout());
        add(scoresheet.panel);
        add(dice.panel);
        add(roll);
        add(play);
    }

    public void start_game(boolean newGame)
    {
        if(newGame){
            players = promptNumplayers();
        }
        else{
            players = 1;
        }

        if(players <= 0){
            return;
        }

        this.removeAll();

        dice = new Dice();
        scoresheet = new Scoresheet(players, dice, this);
        round = 0;
        curPlayer = 1;
        roll = new JButton("ROLL");
        roll.addActionListener(this);
        play = new JButton("PLAY");
        play.addActionListener(this);

        buildPanel();
        
        this.revalidate();

        turn = 0;
        dice.reset();
        scoresheet.play();
    }


    //only updates dice and buttons, not scorecards
    public void start_round()
    {
        curPlayer++;
        if(curPlayer > players){
            curPlayer = 1;
        }
        roll.setEnabled(true);
        play.setEnabled(false);

        if (curPlayer == 1){
            round++;
        }

        turn = 0;
        dice.reset();
        dice.panel.updateDiceButtons();
    }

    private void save_data(){
        try (FileOutputStream f = new FileOutputStream("game_data.txt");
            ObjectOutputStream s = new ObjectOutputStream(f)) 
        {
            s.writeObject(this);
            s.close();
            System.out.println("Successfully saved game!");
        } catch (IOException error) {
            error.printStackTrace();
            System.out.println("IO save error: file stream");
        }
    }

    private void load_data(){
        //read data
        YahtzeePanel loaded_panel = null;
        try(FileInputStream in = new FileInputStream("game_data.txt");
            ObjectInputStream s = new ObjectInputStream(in)) {
            loaded_panel = (YahtzeePanel) s.readObject();
            s.close();

            //load into current panel
            this.scoresheet.copy(loaded_panel.scoresheet);
            this.dice.copy(loaded_panel.dice);
            this.turn = loaded_panel.turn;
            this.round = loaded_panel.round;
            this.playable = loaded_panel.playable;
            this.curPlayer = loaded_panel.curPlayer;
            this.players = loaded_panel.players;

            dice.panel.updateDiceButtons();

            buildPanel();
            scoresheet.panel.update_panel();

            System.out.println("Loaded previous game!");
        } 
        catch (IOException error) {
            error.printStackTrace();
            System.out.println("IO load error");
        }
        catch (ClassNotFoundException e) {
            e.printStackTrace();
            System.out.println("class not found");
        }
    }
    
    public void set_playable(boolean b){ play.setEnabled(b); }

    public void actionPerformed(ActionEvent e)
    {
        if(e.getSource() == roll){
            dice.roll_dice();
            turn++;
            if(turn >= 3){
                roll.setEnabled(false);
            }
            dice.panel.updateDiceButtons();
        }
        else if(e.getSource() == play){
            System.out.println(curPlayer + " " + players + " " + round);
            if ( curPlayer == players && round == 12){
                EndGame();
            }
            else {
                start_round();
            }
            scoresheet.play();
        }
        else if(e.getSource() == Yahtzee.load){
            start_game(false);
            load_data();
        }
        else if(e.getSource() == Yahtzee.save){
            if(scoresheet != null){
                save_data();
            }
        }
        else if(e.getSource() == newGame){
            start_game(true);
        }
        else if(e.getSource() == loadGame){
            start_game(false);
            load_data();
        }
    }
}