Super IA (#5)
Reviewed-on: #5 Co-authored-by: Persson-dev <sim16.prib@gmail.com> Co-committed-by: Persson-dev <sim16.prib@gmail.com>
This commit was merged in pull request #5.
This commit is contained in:
89
app/src/main/java/chess/ai/minimax/AlphaBetaAI.java
Normal file
89
app/src/main/java/chess/ai/minimax/AlphaBetaAI.java
Normal file
@@ -0,0 +1,89 @@
|
||||
package chess.ai.minimax;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.ExecutionException;
|
||||
import java.util.concurrent.ExecutorService;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.Future;
|
||||
|
||||
import chess.ai.AI;
|
||||
import chess.controller.CommandExecutor;
|
||||
import chess.controller.commands.MoveCommand;
|
||||
import chess.controller.commands.PromoteCommand;
|
||||
import chess.controller.commands.PromoteCommand.PromoteType;
|
||||
import chess.model.Color;
|
||||
import chess.model.Coordinate;
|
||||
import chess.model.Move;
|
||||
|
||||
public class AlphaBetaAI extends AI {
|
||||
|
||||
private final int searchDepth;
|
||||
|
||||
private static final float MAX_FLOAT = Float.MAX_VALUE;
|
||||
private static final float MIN_FLOAT = -MAX_FLOAT;
|
||||
|
||||
private final ExecutorService threadPool;
|
||||
|
||||
public AlphaBetaAI(CommandExecutor commandExecutor, Color color, int searchDepth) {
|
||||
super(commandExecutor, color);
|
||||
this.searchDepth = searchDepth;
|
||||
int threadCount = Runtime.getRuntime().availableProcessors();
|
||||
this.threadPool = Executors.newFixedThreadPool(threadCount, new AlphaBetaThreadCreator(commandExecutor, color, threadCount));
|
||||
}
|
||||
|
||||
private Move getBestMove() {
|
||||
List<Move> moves = getAllowedMoves();
|
||||
List<Future<Float>> moveEvaluations = new ArrayList<>(50);
|
||||
float bestMoveValue = MIN_FLOAT;
|
||||
Move bestMove = null;
|
||||
|
||||
System.out.println("Evaluating " + moves.size() + " moves ...");
|
||||
|
||||
for (Move move : moves) {
|
||||
moveEvaluations.add(this.threadPool.submit(() -> {
|
||||
return AlphaBetaThreadCreator.getMoveValue(move, this.searchDepth);
|
||||
}));
|
||||
}
|
||||
|
||||
for (int i = 0; i < moves.size(); i++) {
|
||||
System.out.printf("Progress : %.2f %% ", (float) (i) / moves.size() * 100.0f);
|
||||
Move move = moves.get(i);
|
||||
System.out.print("\r");
|
||||
float value = MIN_FLOAT;
|
||||
try {
|
||||
value = moveEvaluations.get(i).get();
|
||||
} catch (InterruptedException | ExecutionException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
if (value > bestMoveValue) {
|
||||
bestMoveValue = value;
|
||||
bestMove = move;
|
||||
}
|
||||
}
|
||||
|
||||
System.out.println("Best move : " + bestMoveValue + " ");
|
||||
|
||||
return bestMove;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onGameEnd() {
|
||||
this.threadPool.close();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void play() {
|
||||
long current = System.currentTimeMillis();
|
||||
Move move = getBestMove();
|
||||
long elapsed = System.currentTimeMillis() - current;
|
||||
System.out.println("Took " + elapsed + "ms");
|
||||
sendCommand(new MoveCommand(move));
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void promote(Coordinate pawnCoords) {
|
||||
sendCommand(new PromoteCommand(PromoteType.Queen));
|
||||
}
|
||||
|
||||
}
|
||||
101
app/src/main/java/chess/ai/minimax/AlphaBetaThread.java
Normal file
101
app/src/main/java/chess/ai/minimax/AlphaBetaThread.java
Normal file
@@ -0,0 +1,101 @@
|
||||
package chess.ai.minimax;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Map.Entry;
|
||||
|
||||
import chess.ai.PieceCost;
|
||||
import chess.ai.PiecePosCost;
|
||||
import chess.model.ChessBoard;
|
||||
import chess.model.Color;
|
||||
import chess.model.Coordinate;
|
||||
import chess.model.Move;
|
||||
import chess.model.Piece;
|
||||
|
||||
public class AlphaBetaThread extends Thread {
|
||||
|
||||
private final GameSimulation simulation;
|
||||
private final PieceCost pieceCost;
|
||||
private final PiecePosCost piecePosCost;
|
||||
|
||||
private static final int GREAT_MOVE = 9999;
|
||||
|
||||
private static final float MAX_FLOAT = Float.MAX_VALUE;
|
||||
private static final float MIN_FLOAT = -MAX_FLOAT;
|
||||
|
||||
public AlphaBetaThread(Runnable task, GameSimulation simulation, Color color) {
|
||||
super(task);
|
||||
this.simulation = simulation;
|
||||
this.pieceCost = new PieceCost(color);
|
||||
this.piecePosCost = new PiecePosCost(color);
|
||||
}
|
||||
|
||||
private float getEndGameEvaluation() {
|
||||
Color currentTurn = this.simulation.getPlayerTurn();
|
||||
if (this.simulation.getBoard().isKingInCheck(currentTurn))
|
||||
return GREAT_MOVE;
|
||||
return getBoardEvaluation() - PieceCost.PAWN;
|
||||
}
|
||||
|
||||
private float getBoardEvaluation() {
|
||||
final ChessBoard board = this.simulation.getBoard();
|
||||
float result = 0;
|
||||
for (int i = 0; i < Coordinate.VALUE_MAX; i++) {
|
||||
for (int j = 0; j < Coordinate.VALUE_MAX; j++) {
|
||||
Coordinate coordinate = new Coordinate(i, j);
|
||||
Piece piece = board.pieceAt(coordinate);
|
||||
result += pieceCost.getCost(piece) + piecePosCost.getEvaluation(piece, coordinate);
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
public float getMoveValue(Move move, int searchDepth) {
|
||||
this.simulation.tryMove(move);
|
||||
float value = -negaMax(searchDepth - 1, MIN_FLOAT, MAX_FLOAT);
|
||||
this.simulation.undoMove();
|
||||
return value;
|
||||
}
|
||||
|
||||
private float negaMax(int depth, float alpha, float beta) {
|
||||
float value = MIN_FLOAT;
|
||||
|
||||
List<Move> moves = this.simulation.getAllowedMoves();
|
||||
|
||||
if (moves.isEmpty())
|
||||
return -getEndGameEvaluation();
|
||||
|
||||
List<Entry<Move, Float>> movesCost = new ArrayList<>(moves.size());
|
||||
|
||||
for (Move move : moves) {
|
||||
this.simulation.tryMove(move);
|
||||
movesCost.add(Map.entry(move, -getBoardEvaluation()));
|
||||
this.simulation.undoMove();
|
||||
}
|
||||
|
||||
Collections.sort(movesCost, (first, second) -> {
|
||||
return Float.compare(first.getValue(), second.getValue());
|
||||
});
|
||||
|
||||
if (depth == 1)
|
||||
return -movesCost.getFirst().getValue();
|
||||
|
||||
for (var moveEntry : movesCost) {
|
||||
Move move = moveEntry.getKey();
|
||||
this.simulation.tryMove(move);
|
||||
value = Float.max(value, -negaMax(depth - 1, -beta, -alpha));
|
||||
this.simulation.undoMove();
|
||||
alpha = Float.max(alpha, value);
|
||||
if (alpha >= beta)
|
||||
return value;
|
||||
}
|
||||
|
||||
return value;
|
||||
}
|
||||
|
||||
public GameSimulation getSimulation() {
|
||||
return simulation;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,36 @@
|
||||
package chess.ai.minimax;
|
||||
|
||||
import java.util.concurrent.ThreadFactory;
|
||||
|
||||
import chess.controller.CommandExecutor;
|
||||
import chess.model.Color;
|
||||
import chess.model.Move;
|
||||
|
||||
public class AlphaBetaThreadCreator implements ThreadFactory{
|
||||
|
||||
private final Color color;
|
||||
private final GameSimulation simulations[];
|
||||
private int currentThread = 0;
|
||||
|
||||
public AlphaBetaThreadCreator(CommandExecutor commandExecutor, Color color, int threadCount) {
|
||||
this.color = color;
|
||||
simulations = new GameSimulation[threadCount];
|
||||
for (int i = 0; i < threadCount; i++) {
|
||||
simulations[i] = new GameSimulation();
|
||||
commandExecutor.addListener(simulations[i]);
|
||||
}
|
||||
}
|
||||
|
||||
public static float getMoveValue(Move move, int searchDepth) {
|
||||
AlphaBetaThread t = (AlphaBetaThread) Thread.currentThread();
|
||||
return t.getMoveValue(move, searchDepth);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Thread newThread(Runnable r) {
|
||||
AlphaBetaThread t = new AlphaBetaThread(r, simulations[currentThread], color);
|
||||
currentThread++;
|
||||
return t;
|
||||
}
|
||||
|
||||
}
|
||||
103
app/src/main/java/chess/ai/minimax/GameSimulation.java
Normal file
103
app/src/main/java/chess/ai/minimax/GameSimulation.java
Normal file
@@ -0,0 +1,103 @@
|
||||
package chess.ai.minimax;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
import chess.controller.Command;
|
||||
import chess.controller.Command.CommandResult;
|
||||
import chess.controller.CommandExecutor;
|
||||
import chess.controller.commands.CastlingCommand;
|
||||
import chess.controller.commands.GetPlayerMovesCommand;
|
||||
import chess.controller.commands.MoveCommand;
|
||||
import chess.controller.commands.NewGameCommand;
|
||||
import chess.controller.commands.PromoteCommand;
|
||||
import chess.controller.commands.UndoCommand;
|
||||
import chess.controller.commands.PromoteCommand.PromoteType;
|
||||
import chess.controller.event.EmptyGameDispatcher;
|
||||
import chess.controller.event.GameAdaptator;
|
||||
import chess.model.ChessBoard;
|
||||
import chess.model.Color;
|
||||
import chess.model.Game;
|
||||
import chess.model.Move;
|
||||
import chess.model.PermissiveGame;
|
||||
|
||||
public class GameSimulation extends GameAdaptator {
|
||||
|
||||
private final CommandExecutor simulation;
|
||||
private final Game gameSimulation;
|
||||
|
||||
public GameSimulation() {
|
||||
this.gameSimulation = new PermissiveGame();
|
||||
this.simulation = new CommandExecutor(gameSimulation, new EmptyGameDispatcher());
|
||||
}
|
||||
|
||||
protected CommandResult sendCommand(Command command) {
|
||||
CommandResult result = this.simulation.executeCommand(command);
|
||||
if (result == CommandResult.NotAllowed) {
|
||||
System.out.println("eeeeee");
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
public void tryMove(Move move) {
|
||||
sendCommand(new MoveCommand(move));
|
||||
if (this.gameSimulation.getBoard().pawnShouldBePromoted())
|
||||
sendCommand(new PromoteCommand(PromoteType.Queen));
|
||||
}
|
||||
|
||||
public void undoMove() {
|
||||
sendCommand(new UndoCommand());
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onPawnPromoted(PromoteType promotion) {
|
||||
sendCommand(new PromoteCommand(promotion));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onCastling(boolean bigCastling) {
|
||||
sendCommand(new CastlingCommand(bigCastling));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onMove(Move move) {
|
||||
sendCommand(new MoveCommand(move));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onGameStart() {
|
||||
sendCommand(new NewGameCommand());
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onPlayerTurn(Color color, boolean undone) {
|
||||
if (undone)
|
||||
sendCommand(new UndoCommand());
|
||||
}
|
||||
|
||||
public CommandExecutor getCommandExecutor() {
|
||||
return simulation;
|
||||
}
|
||||
|
||||
public Game getGame() {
|
||||
return gameSimulation;
|
||||
}
|
||||
|
||||
public ChessBoard getBoard() {
|
||||
return this.gameSimulation.getBoard();
|
||||
}
|
||||
|
||||
public Color getPlayerTurn() {
|
||||
return this.gameSimulation.getPlayerTurn();
|
||||
}
|
||||
|
||||
public List<Move> getAllowedMoves() {
|
||||
GetPlayerMovesCommand cmd = new GetPlayerMovesCommand();
|
||||
sendCommand(cmd);
|
||||
return cmd.getMoves();
|
||||
}
|
||||
|
||||
public void close() {
|
||||
this.simulation.close();
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user