add custom pieces

Game.java

@@ -40,8 +40,10 @@ public class Game {
                 board[i][j] = new_square;
             }
         }
-        add_piece_to_location(5, 2, "Q", false, true);
-        add_piece_to_location(6, 3, "", true, false);
+        add_piece_to_location(2, 3, "A", false, true);
+        add_piece_to_location(2, 4, "B", false, true);
+        add_piece_to_location(5, 3, "A", true, true);
+        add_piece_to_location(5, 4, "B", true, true);
     }
 
     public void add_piece_to_location(int x, int y, String name, boolean white, boolean occupied){
@@ -124,12 +126,12 @@ public class Game {
     public int make_move() throws InterruptedException {
         Square[][][] legal_moves = get_all_moves(white_turn, this.board);
 
-        /*
+
         for(int i=0; i<legal_moves.length; i++){
             TimeUnit.SECONDS.sleep(1);
             print_board(legal_moves[i]);
         }
-        */
+
 
         /* Check for end of game */
         if(legal_moves.length == 0){

Main.java

@@ -16,7 +16,7 @@ public class Main {
 	    int end_of_game = -1;
 	    while(end_of_game == -1){
 	        end_of_game = my_game.make_move();
-            //TimeUnit.SECONDS.sleep(1);
+            TimeUnit.SECONDS.sleep(5);
             my_game.print_board(my_game.board);
         }
         if(end_of_game == 0){

Square.java

@@ -61,6 +61,12 @@ public class Square {
         else if(this.name == "Q"){
             return queen_squares(board);
         }
+        else if(this.name == "A"){
+            return A_squares(board);
+        }
+        else if(this.name == "B"){
+            return B_squares(board);
+        }
         else{
             return king_squares(board);
         }
@@ -261,6 +267,43 @@ public class Square {
         return squares.toArray(new int[squares.size()][2]);
     }
 
+    /**
+     * A helper method returning the possible locations of A piece
+     * @param board the current state of the board
+     * @return an array of tuples representing x and y, the locations that can be reached
+     */
+    public int[][] A_squares(Square[][] board) {
+        List<int[]> squares = new Vector<int[]>();
+        int x = this.x;
+        int y = this.y;
+        for(int i=0; i<=7; i++){
+            for(int j=0; j<=7; j++){
+                if(!(i==x && j==y) && board[i][j].name != "K" && board[i][j].name != "A") {
+                    squares.addAll(square_checker(board, x, y, i, j));
+                }
+            }
+        }
+        return squares.toArray(new int[squares.size()][2]);
+    }
+
+    /**
+     * A helper method returning the possible locations of B piece
+     * @param board the current state of the board
+     * @return an array of tuples representing x and y, the locations that can be reached
+     */
+    public int[][] B_squares(Square[][] board) {
+        List<int[]> squares = new Vector<int[]>();
+        int x = this.x;
+        int y = this.y;
+        if(x+y % 2 == 0) {
+            squares.addAll(Arrays.asList(bishop_squares(board)));
+        }
+        else {
+            squares.addAll(Arrays.asList(rook_squares(board)));
+        }
+        return squares.toArray(new int[squares.size()][2]);
+    }
+
     /**
      * A helper method returning the possible locations of a king
      * @param board the current state of the board