diff --git a/src/main/java/BattleShip/AIGridGUI.java b/src/main/java/BattleShip/AIGridGUI.java index 067352c..0137f24 100644 --- a/src/main/java/BattleShip/AIGridGUI.java +++ b/src/main/java/BattleShip/AIGridGUI.java @@ -475,6 +475,17 @@ public class AIGridGUI extends JPanel { } highlightCells(cell, 0); } + + 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) {