print promotions and check

1 files changed, 20 insertions, 3 deletions

diff --git a/chess.c b/chess.c
index 024f489..0977877 100644
--- a/chess.c
+++ b/chess.c
@@ -6,7 +6,6 @@
 #include <keyboard.h>
 
 // Missing features
-// underpromotion
 // display checkmate and stalemate
 // arrows to go back and forward in history
 // show moves better
@@ -405,6 +404,8 @@ islegal(int board[8][8], Point src, Point dst, Move prev_move, CastleRights cr)
 void print_move(char* movestring, int board[8][8], Move move, Move prev_move, MoveType type, int was_capture, CastleRights cr) {
 	char capturestr[2] = "x";
 	char piece_id[3] = "";
+	char promotion[3] = "";
+	char check[2] = "";
 
 	if (!was_capture) {
 		capturestr[0] = 0;
@@ -439,13 +440,29 @@ void print_move(char* movestring, int board[8][8], Move move, Move prev_move, Mo
 		}
 		sprint(piece_id, "%c%s%s", PIECE_STR[move.piece][1], src_file, src_rank);
 	}
+	if (type == PromotionQ) {
+		sprint(promotion, "=Q");
+	} else if (type == PromotionN) {
+		sprint(promotion, "=N");
+	} else if (type == PromotionB) {
+		sprint(promotion, "=B");
+	} else if (type == PromotionR) {
+		sprint(promotion, "=R");
+	}
+	// test if it will be check
+	int tmpboard[8][8];
+	memcpy(tmpboard, board, sizeof(int)*8*8);
+	do_move(tmpboard, move.src, move.dst, type);
+	if (is_in_check(tmpboard, !owner(move.piece))) {
+		sprint(check, "+");
+	}
 	if (type == WhiteLongCastle || type == BlackLongCastle) {
 		sprint(movestring, "O-O-O");
 	} else if (type == WhiteShortCastle || type == BlackShortCastle) {
 		sprint(movestring, "O-O");
 	} else {
-		sprint(movestring, "%s%s%c%d", piece_id,
-			   capturestr, 'a'+move.dst.x, 8-move.dst.y);
+		sprint(movestring, "%s%s%c%d%s%s", piece_id,
+			   capturestr, 'a'+move.dst.x, 8-move.dst.y, promotion, check);
 	}
 }