DiceChooser.java

package games.strategy.triplea.ui;

import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.util.ArrayList;
import java.util.Collection;

import javax.swing.AbstractAction;
import javax.swing.BorderFactory;
import javax.swing.Box;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.ScrollPaneConstants;

import games.strategy.triplea.delegate.Die.DieType;
import games.strategy.ui.SwingAction;

public class DiceChooser extends JPanel {
  private static final long serialVersionUID = -3658408802544268998L;
  private final IUIContext m_uiContext;
  private JPanel m_dicePanel;
  private final int[] m_random;
  private int m_diceCount = 0;
  private int m_numRolls = 0;
  private int m_hitAt = 0;
  private boolean m_hitOnlyIfEquals = false;
  private final Collection<JButton> m_buttons;
  private JButton m_undoButton;
  private JLabel m_diceCountLabel;
  // private final GameData m_data;
  private int m_diceSides = 6;

  public DiceChooser(final IUIContext uiContext, final int numRolls, final int hitAt, final boolean hitOnlyIfEquals,
      final int diceSides) {
    m_uiContext = uiContext;
    m_numRolls = numRolls;
    m_diceSides = diceSides;
    m_hitAt = hitAt;
    m_hitOnlyIfEquals = hitOnlyIfEquals;
    // m_data = data;
    m_buttons = new ArrayList<>(diceSides);
    m_random = new int[numRolls];
    setLayout(new BoxLayout(this, BoxLayout.Y_AXIS));
    createComponents();
  }

  public void clear() {
    m_dicePanel.removeAll();
    for (int i = 0; i < m_diceCount; i++) {
      m_random[i] = 0;
    }
    m_diceCount = 0;
  }

  public int[] getDice() {
    if (m_diceCount < m_numRolls) {
      return null;
    }
    return m_random;
  }

  private void addDie(final int roll) {
    final boolean hit = (roll == m_hitAt || (!m_hitOnlyIfEquals && (m_hitAt > 0) && roll > m_hitAt));
    final DieType dieType = hit ? DieType.HIT : DieType.MISS;
    m_dicePanel.add(new JLabel(m_uiContext.getDiceImageFactory().getDieIcon(roll, dieType)));
    m_dicePanel.add(Box.createHorizontalStrut(2));
    m_random[m_diceCount++] = roll - 1;
    updateDiceCount();
    validate();
    invalidate();
    repaint();
  }

  private void removeLastDie() {
    // remove the strut and the component
    final int lastIndex = m_dicePanel.getComponentCount() - 1;
    m_dicePanel.remove(lastIndex);
    m_dicePanel.remove(lastIndex - 1);
    m_diceCount--;
    updateDiceCount();
    validate();
    invalidate();
    repaint();
  }

  private void updateDiceCount() {
    final boolean showButtons = (m_diceCount < m_numRolls);
    for (final JButton button : m_buttons) {
      button.setEnabled(showButtons);
    }
    m_undoButton.setEnabled((m_diceCount > 0));
    m_diceCountLabel.setText("Dice remaining: " + (m_numRolls - m_diceCount));
  }

  private void createComponents() {
    final JPanel diceButtonPanel = new JPanel();
    diceButtonPanel.setLayout(new BoxLayout(diceButtonPanel, BoxLayout.X_AXIS));
    diceButtonPanel.add(Box.createHorizontalStrut(40));
    for (int roll = 1; roll <= m_diceSides; roll++) {
      final boolean hit = (roll == m_hitAt || (!m_hitOnlyIfEquals && (m_hitAt > 0) && roll > m_hitAt));
      diceButtonPanel.add(Box.createHorizontalStrut(4));
      final int dieNum = roll;
      final DieType dieType = hit ? DieType.HIT : DieType.MISS;
      final JButton button =
          new JButton(new AbstractAction(null, m_uiContext.getDiceImageFactory().getDieIcon(roll, dieType)) {
            private static final long serialVersionUID = 8900816143434068634L;

            @Override
            public void actionPerformed(final ActionEvent event) {
              addDie(dieNum);
            }
          });
      m_buttons.add(button);
      button.setPreferredSize(new Dimension(m_uiContext.getDiceImageFactory().DIE_WIDTH + 4,
          m_uiContext.getDiceImageFactory().DIE_HEIGHT + 4));
      diceButtonPanel.add(button);
    }
    diceButtonPanel.add(Box.createHorizontalStrut(4));
    m_undoButton = new JButton(SwingAction.of("Undo", e -> removeLastDie()));
    diceButtonPanel.add(m_undoButton);
    diceButtonPanel.add(Box.createHorizontalStrut(40));
    m_diceCountLabel = new JLabel("Dice remaining:   ");
    final JPanel labelPanel = new JPanel();
    labelPanel.setLayout(new BoxLayout(labelPanel, BoxLayout.X_AXIS));
    labelPanel.add(m_diceCountLabel);
    m_dicePanel = new JPanel();
    m_dicePanel.setBorder(BorderFactory.createLoweredBevelBorder());
    m_dicePanel.setLayout(new BoxLayout(m_dicePanel, BoxLayout.X_AXIS));
    final JScrollPane scroll = new JScrollPane(m_dicePanel);
    scroll.setBorder(null);
    scroll.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_NEVER);
    // we're adding to a box layout, so to prevent the component from
    // grabbing extra space, set the max height.
    // allow room for a dice and a scrollbar
    scroll.setMinimumSize(
        new Dimension(scroll.getMinimumSize().width, m_uiContext.getDiceImageFactory().DIE_HEIGHT + 17));
    scroll.setMaximumSize(
        new Dimension(scroll.getMaximumSize().width, m_uiContext.getDiceImageFactory().DIE_HEIGHT + 17));
    scroll.setPreferredSize(
        new Dimension(scroll.getPreferredSize().width, m_uiContext.getDiceImageFactory().DIE_HEIGHT + 17));
    add(scroll);
    add(Box.createVerticalStrut(8));
    add(labelPanel);
    add(Box.createVerticalStrut(8));
    add(diceButtonPanel);
    updateDiceCount();
  }
}