loic/chessarbiter
1
0
Fork 0
mirror of https://gitlab.com/manzerbredes/chessarbiter.git synced 2025-04-06 10:06:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add WasEnPassant() method for convenience

Browse source
This commit is contained in:
Loic Guegan 2022-03-07 11:30:55 +01:00
parent 934b00f729
commit 9025383477
3 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/ChessArbiter.cpp
View file

@ -3,7 +3,7 @@
namespace chessarbiter { namespace chessarbiter {
ChessArbiter::ChessArbiter() ChessArbiter::ChessArbiter()
: wPawn(1), wRook(5), wKnight(3), wBishop(3), wQueen(9), wKing(0), SAN(""), : wPawn(1), wRook(5), wKnight(3), wBishop(3), wQueen(9), wKing(0), SAN(""),
capture(' ') { capture(' '), was_enpassant(false) {
Setup("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1"); Setup("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1");
} }
@ -50,6 +50,7 @@ bool ChessArbiter::Play(std::string move) {
INIT_BACKUP(); INIT_BACKUP();
SAN = ""; SAN = "";
capture = ' '; capture = ' ';
was_enpassant=false;
if (IsCapture) { if (IsCapture) {
capture = board.GetPieceAt(dst).piece; capture = board.GetPieceAt(dst).piece;
@ -120,6 +121,7 @@ bool ChessArbiter::Play(std::string move) {
board.RemovePiece(dst[0] + std::string() + (char)(dst[1] - 1)); board.RemovePiece(dst[0] + std::string() + (char)(dst[1] - 1));
capture = 'p'; capture = 'p';
} }
was_enpassant=true;
} }
newFen.halfmove = 0; // Pawn moves reset half moves newFen.halfmove = 0; // Pawn moves reset half moves
} }
@ -173,6 +175,8 @@ bool ChessArbiter::Play(std::string move) {
return (false); return (false);
} }
bool ChessArbiter::WasEnPassant() { return (was_enpassant); }
bool ChessArbiter::IsAttacked(std::string square, bool by) { bool ChessArbiter::IsAttacked(std::string square, bool by) {
std::vector<std::string> moves = board.ListPossibleMoves(by); std::vector<std::string> moves = board.ListPossibleMoves(by);
for (std::string &m : moves) { for (std::string &m : moves) {

8
src/ChessArbiter.hpp
View file

@ -10,14 +10,16 @@
if (positions.find(fen.board) != positions.end()) \ if (positions.find(fen.board) != positions.end()) \
positions_backup = positions[fen.board]; \ positions_backup = positions[fen.board]; \
std::string SAN_backup = SAN; \ std::string SAN_backup = SAN; \
char capture_backup = capture; char capture_backup = capture; \
bool was_enpassant_backup = was_enpassant;
#define RESTORE_BACKUP() \ #define RESTORE_BACKUP() \
SetFEN(fen_backup); \ SetFEN(fen_backup); \
if (positions_backup != 0) \ if (positions_backup != 0) \
positions[fen.board] = positions_backup; \ positions[fen.board] = positions_backup; \
SAN = SAN_backup; \ SAN = SAN_backup; \
capture = capture_backup; capture = capture_backup; \
was_enpassant = was_enpassant_backup;
namespace chessarbiter { namespace chessarbiter {
class ChessArbiter { class ChessArbiter {
@ -31,6 +33,7 @@ class ChessArbiter {
void SetFEN(FEN); void SetFEN(FEN);
std::string SAN, SAN_last; std::string SAN, SAN_last;
char capture; char capture;
bool was_enpassant;
public: public:
ChessArbiter(); ChessArbiter();
@ -65,6 +68,7 @@ public:
bool IsDrawByNoMoves(); bool IsDrawByNoMoves();
bool IsDrawByRepetitions(); bool IsDrawByRepetitions();
bool IsDraw(); bool IsDraw();
bool WasEnPassant();
std::string ParseSAN(std::string SANMove); std::string ParseSAN(std::string SANMove);
}; };
} // namespace chessarbiter } // namespace chessarbiter

4
tests/chessarbiter.cpp
View file

@ -368,6 +368,7 @@ TEST_CASE("SimpleCapture", "[SimplePieceCapture]") {
CHECK(a.GetFEN() == CHECK(a.GetFEN() ==
"rnbqkbnr/ppp1pppp/8/3P4/8/8/PPPP1PPP/RNBQKBNR b KQkq - 0 2"); "rnbqkbnr/ppp1pppp/8/3P4/8/8/PPPP1PPP/RNBQKBNR b KQkq - 0 2");
CHECK(a.GetCapture() == 'p'); CHECK(a.GetCapture() == 'p');
CHECK_FALSE(a.WasEnPassant());
} }
TEST_CASE("SimpleCastle", "[SimpleCastle]") { TEST_CASE("SimpleCastle", "[SimpleCastle]") {
@ -407,6 +408,7 @@ TEST_CASE("SimpleEnPassant", "[SimpleEnPassant]") {
CHECK(a.GetFEN() == CHECK(a.GetFEN() ==
"rnbqkbnr/ppppp1pp/5P2/8/8/8/PPPP1PPP/RNBQKBNR b KQkq - 0 2"); "rnbqkbnr/ppppp1pp/5P2/8/8/8/PPPP1PPP/RNBQKBNR b KQkq - 0 2");
CHECK(a.GetCapture() == 'p'); CHECK(a.GetCapture() == 'p');
CHECK(a.WasEnPassant());
// Black capture // Black capture
a.Setup("rnbqkbnr/ppppp1pp/8/8/4Pp2/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1"); a.Setup("rnbqkbnr/ppppp1pp/8/8/4Pp2/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1");
@ -414,6 +416,7 @@ TEST_CASE("SimpleEnPassant", "[SimpleEnPassant]") {
CHECK(a.GetFEN() == CHECK(a.GetFEN() ==
"rnbqkbnr/ppppp1pp/8/8/8/4p3/PPPP1PPP/RNBQKBNR w KQkq - 0 2"); "rnbqkbnr/ppppp1pp/8/8/8/4p3/PPPP1PPP/RNBQKBNR w KQkq - 0 2");
CHECK(a.GetCapture() == 'P'); CHECK(a.GetCapture() == 'P');
CHECK(a.WasEnPassant());
// Check en_passant is set // Check en_passant is set
a.Setup( a.Setup(
@ -421,6 +424,7 @@ TEST_CASE("SimpleEnPassant", "[SimpleEnPassant]") {
CHECK(a.Play("f7f5")); CHECK(a.Play("f7f5"));
CHECK(a.GetFEN() == "r3k2r/1pqbb2p/1nn1p1p1/p2pPp2/3P1PP1/PP1B4/1B1NN2P/" CHECK(a.GetFEN() == "r3k2r/1pqbb2p/1nn1p1p1/p2pPp2/3P1PP1/PP1B4/1B1NN2P/"
"R2Q1RK1 w kq f6 0 15"); "R2Q1RK1 w kq f6 0 15");
CHECK_FALSE(a.WasEnPassant());
} }
TEST_CASE("ParseSAN", "[ParseSAN]") { TEST_CASE("ParseSAN", "[ParseSAN]") {