diff --git a/src/main/java/BattleShip/AIGridGUI.java b/src/main/java/BattleShip/AIGridGUI.java index 1ba45bf..067352c 100644 --- a/src/main/java/BattleShip/AIGridGUI.java +++ b/src/main/java/BattleShip/AIGridGUI.java @@ -464,6 +464,17 @@ public class AIGridGUI extends JPanel { } clear = false; } + + if(e.getButton() == MouseEvent.BUTTON3) { + //Toggle whether ship placement is vertical or horizontal. + vertical = !vertical; + for(BSButton bsb : buttons) { + if(bsb.getCellContents() == null) { + bsb.setBorder(defaultBorder); + } + } + highlightCells(cell, 0); + } } public void mouseExited(MouseEvent e) {