From 4077858c9c3e4cb7ca3fde821bb6b859831ef8c3 Mon Sep 17 00:00:00 2001
From: aah13002 <aaron.huang@uconn.edu>
Date: Sat, 30 Apr 2016 01:16:42 -0400
Subject: [PATCH] Added piece count tracker and simple heuristic function

---
 src/model/Board.java | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/model/Board.java b/src/model/Board.java
index 46f658a..0306d84 100755
--- a/src/model/Board.java
+++ b/src/model/Board.java
@@ -17,6 +17,9 @@ public class Board {
 	
 	private Piece lastPieceMoved;
 	private Move lastMove;
+	
+	private int blackPieces;
+	private int whitePieces;
 
 	// Move properties
 	private int movesSinceCapture;
@@ -28,7 +31,9 @@ public Board() {
 		movesSinceCapture = 0;
 		init();
 		lastPieceMoved = null;
-		lastMove = null;
+		lastMove = null;
+		blackPieces = 12;
+		whitePieces = 12;
 	}
 	
 	/**
@@ -46,6 +51,8 @@ public Board(Board other) {
 		movesSinceCapture = other.getMovesSinceCapture();
 		lastPieceMoved = other.getLastPieceMoved();
 		lastMove = other.getLastMove();
+		blackPieces = other.blackPieces;
+		whitePieces = other.whitePieces;
 	}
 
 	/**
@@ -87,6 +94,10 @@ public void movePiece(Move move) {
 			int monkeyRow = (destRow + sourceRow)/2;
 			int monkeyCol = (destCol + sourceCol)/2;
 	
+			Color color_removed = representation[monkeyRow][monkeyCol].getColor();
+			
+			if (color_removed == Color.BLACK) --blackPieces; else --whitePieces;
+			
 			/* Remove the piece being jumped ("monkey in the middle") */
 			representation[monkeyRow][monkeyCol] = null;
 		} 
@@ -355,10 +366,13 @@ public boolean canPromote(Piece p) {
 						(row == BOARD_SIZE - 1 && p.getColor() == Color.BLACK));
 	}
 	
+	public int getHeuristic(Color color) {
+		return color == Color.BLACK ? this.blackPieces : this.whitePieces;
+	}
+	
 	public Piece getPiece(Location location) {
 		return representation[location.row][location.column];
 	}
-
 	
 	public Piece[][] getRepresentation() {
 		return this.representation;