BoardPanel.java

import java.awt.*;
import javax.swing.*;
import java.io.File;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.IOException;
import ChessEngine.*;
import java.util.*;

public class BoardPanel extends JPanel{

   public SquarePanel[][] squares = new SquarePanel[8][8];
   public SquarePanel activeSquare = null;
   public ArrayList<SquarePanel> allowedSquares = new ArrayList<SquarePanel>();
   public Board board;
   public EngineInterface eInterface;
   
   
   public BoardPanel(EngineInterface e)
   {
      board = new Board();
      eInterface = e;
      
      setPreferredSize( new Dimension(500, 500) );
      setLayout(new GridLayout(8,8));
      
      for(int i = 0; i < 8; i++)
      {
         for(int j = 0; j < 8; j++)
         {
         
            SquarePanel square = new SquarePanel(7 - i, j);
            squares[7 - i][j] = square;
            
            if(i%2 == 0 ^ j%2 == 0)
            {
               square.setDefaultColor(Color.GRAY);
            }
            else
            {
               square.setDefaultColor(Color.WHITE);     
            }
                           
            add(square);
         }
         
      }
   
      setBoard();
      paintImmediately(0,0,getWidth(),getHeight());

   }
   
   public void handleMouseClick(SquarePanel sq)
   {
            
      //check that it's the correct person's turn
      if(activeSquare == null && sq.getImage() != null && board.checkSquare(sq.row(), sq.col()).color == board.curPlayer)
      {
         setActiveSquare(sq);
         sq.setBackground(Color.YELLOW);
         if(SettingsMenu.settings.getBoolean("showPossibleMoves", false))
         {
            for(SquarePanel s : allowedSquares)
            {
               s.setBorder(BorderFactory.createLineBorder(Color.GREEN, 5));
            }    
         }
      }
      else if(activeSquare != null)
      {
          //attempt to movePiece
         try
         {
            movePiece(activeSquare, sq);    
         }
         catch (InvalidMoveException e)
         {
            return;
         }
                       
         paintImmediately(0,0,getWidth(),getHeight());
         eInterface.getEngineMove();
         
      }
 
   }         
  
    
   
   public void movePiece(SquarePanel a, SquarePanel b) throws InvalidMoveException{
      
      //remove all the coloring
      activeSquare.resetBackground();   
      for(SquarePanel panel : allowedSquares)
      {
         panel.resetBackground();
      }           
      activeSquare = null;
   
      
      if(!allowedSquares.contains(b))
      {     
         throw new InvalidMoveException();
      }
     
      
      //update internal board
      Move m = new Move(a.toString() + b.toString());
      MoveHistory mh = board.movePiece(m);
      
      //just look at mh.getEnPassant instead. math was already handled !!!!!!!!!!! FIX
      Piece p = mh.getCaptured();
      if(p != null && p.getPos()/16 != b.row())
      {
         SquarePanel sq = squares[p.getPos()/16][p.getPos()%16];
         sq.setImage(null);
         sq.repaint();
      }
     
      //check for castle
      switch(mh.didCastle())
      {
         case -1:
            break;
         case 1:
            squares[0][3].setImage(squares[0][0].getImage());
            squares[0][0].setImage(null);
            squares[0][0].repaint();
            break;
         case 2:
            squares[0][5].setImage(squares[0][7].getImage());
            squares[0][7].setImage(null);
            squares[0][7].repaint();
            break;    
         case 3:    
            squares[7][3].setImage(squares[7][0].getImage());
            squares[7][0].setImage(null);
            squares[7][0].repaint();
            break;     
         case 4:
            squares[7][5].setImage(squares[7][7].getImage());
            squares[7][7].setImage(null);
            squares[7][7].repaint();
            break;    
      }
     
     //check for pawn promotion
      if(mh.getPromo())
      {
         try
         {
            char c = board.checkSquare(b.row(), b.col()).color;
            File file = new File("PieceImages/" + c + "Queen.png");
            Image img =  ImageIO.read(file);
            b.setImage(img);
            System.out.println("image changed to queen!");
         }
         catch(IOException e)
         {
            System.out.println(e);
         }
         catch(NullPointerException e)
         {
            System.out.println(e);
         }
              
      }
                
      
      b.setImage(a.getImage());
      a.setImage(null);   
      b.repaint();
      a.repaint();
                                    
   }
      
   public void setActiveSquare(SquarePanel sq)
   {
      activeSquare = sq;
      
      //find legal moves
      allowedSquares.clear();  
      Move[] moves = new Move[30];
      MoveHistory mh;
      SquarePanel panel;
      
      
      Piece p = board.checkSquare(activeSquare.row(), activeSquare.col());
      int n = p.genMoves(board, moves, 0);
      
            
      for(int i = 0; i < n; i++)
      {
         mh = board.movePiece(moves[i]);
         if(board.curPlayer == 'b' ? ((King)(board.wKing)).isChecked(board) : ((King)(board.bKing)).isChecked(board))
         {
            board.unmovePiece(mh);
            continue;
         }
        
         board.unmovePiece(mh); 
         panel = squares[moves[i].s2()/16][moves[i].s2()%16];
         allowedSquares.add(panel);
               
      }

   }
   
   public void setBoard()
   {
         
      board.setup(Board.STARTFEN);
            
      for(int i = 0; i < 8; i++)
      {
         for(int j = 0; j < 8; j++)
         {
                                
            Piece p = board.checkSquare(7 - i, j);         
            try
            {
               File file = new File("PieceImages/" + p.color + p.getClass().getSimpleName() + ".png");
               Image img =  ImageIO.read(file);
               squares[7 - i][j].setImage(img);
            }
            catch(IOException e){
               System.out.println(e);
               throw new RuntimeException("Could not load piece images");
            }
            catch(NullPointerException e){
               squares[7 - i][j].setImage(null);
            }
            
         }
         
      }
      repaint();
   
   }
   
   public void attatchEngineInterface(EngineInterface e)
   {
      eInterface = e;
   }
   
   
   private class InvalidMoveException extends Exception{}
}