# HG changeset patch # User Mike Becker # Date 1440593202 -7200 # Node ID eef745ba3774befefdaac0038c6b844e37de990f # Parent 78b580bfde33111cbf0c89bb06abace84fdaf83d implemented FEN diff -r 78b580bfde33 -r eef745ba3774 src/chess/pgn.c --- a/src/chess/pgn.c Wed Aug 26 12:59:30 2015 +0200 +++ b/src/chess/pgn.c Wed Aug 26 14:46:42 2015 +0200 @@ -194,3 +194,158 @@ return bytes; } + +static size_t fen_pieces(char *str, GameState *gamestate) { + size_t i = 0; + for (int row = 7 ; row >= 0 ; row--) { + unsigned int skip = 0; + for (int file = 0 ; file < 8 ; file++) { + if (gamestate->board[row][file]) { + if (skip > 0) { + str[i++] = '0'+skip; + skip = 0; + } + switch (gamestate->board[row][file] & ~ENPASSANT_THREAT) { + case WHITE|KING: str[i++] = 'K'; break; + case WHITE|QUEEN: str[i++] = 'Q'; break; + case WHITE|BISHOP: str[i++] = 'B'; break; + case WHITE|KNIGHT: str[i++] = 'N'; break; + case WHITE|ROOK: str[i++] = 'R'; break; + case WHITE|PAWN: str[i++] = 'P'; break; + case BLACK|KING: str[i++] = 'k'; break; + case BLACK|QUEEN: str[i++] = 'q'; break; + case BLACK|BISHOP: str[i++] = 'b'; break; + case BLACK|KNIGHT: str[i++] = 'n'; break; + case BLACK|ROOK: str[i++] = 'r'; break; + case BLACK|PAWN: str[i++] = 'p'; break; + } + } else { + skip++; + } + } + if (skip > 0) { + str[i++] = '0'+skip; + } + if (row > 0) { + str[i++] = '/'; + } + } + + return i; +} + +static size_t fen_color(char *str, GameState *gamestate) { + uint8_t color = opponent_color(gamestate->lastmove ? + (gamestate->lastmove->move.piece & COLOR_MASK) : BLACK); + + str[0] = color == WHITE ? 'w' : 'b'; + + return 1; +} + +static _Bool fen_castling_chkmoved(GameState *gamestate, + uint8_t row, uint8_t file) { + + MoveList *ml = gamestate->movelist; + while (ml) { + if (ml->move.fromfile == file && ml->move.fromrow == row) { + return 1; + } + ml = ml->next; + } + + return 0; +} + +static size_t fen_castling(char *str, GameState *gamestate) { + _Bool K, Q, k, q; + + if (fen_castling_chkmoved(gamestate, rowidx('1'), fileidx('e'))) { + K = Q = 0; + } else { + K = !fen_castling_chkmoved(gamestate, rowidx('1'), fileidx('h')); + Q = !fen_castling_chkmoved(gamestate, rowidx('1'), fileidx('a')); + } + if (fen_castling_chkmoved(gamestate, rowidx('8'), fileidx('e'))) { + k = q = 0; + } else { + k = !fen_castling_chkmoved(gamestate, rowidx('8'), fileidx('h')); + q = !fen_castling_chkmoved(gamestate, rowidx('8'), fileidx('a')); + } + + size_t i = 0; + if (K) str[i++] = 'K'; + if (Q) str[i++] = 'Q'; + if (k) str[i++] = 'k'; + if (q) str[i++] = 'q'; + if (!i) str[i++] = '-'; + + return i; +} + +static size_t fen_enpassant(char *str, GameState *gamestate) { + + str[0] = '-'; str[1] = '\0'; + + for (int file = 0 ; file < 8 ; file++) { + if (gamestate->board[3][file] & ENPASSANT_THREAT) { + str[0] = filechr(file); + str[1] = rowchr(2); + } + if (gamestate->board[4][file] & ENPASSANT_THREAT) { + str[0] = filechr(file); + str[1] = rowchr(5); + } + } + + return str[0] == '-' ? 1 : 2; +} + +static size_t fen_halfmove(char *str, GameState *gamestate) { + + unsigned int i = 0; + for (MoveList *move = gamestate->movelist ; move ; move = move->next) { + if (move->move.capture || (move->move.piece & PIECE_MASK) == PAWN) { + i = 0; + } else { + i++; + } + } + + char m[8]; + size_t len = sprintf(m, "%u", i); + memcpy(str, m, len); + + return len; +} + +static size_t fen_movenr(char *str, GameState *gamestate) { + + MoveList *move = gamestate->movelist; + unsigned int i = 1; + while (move) { + i++; + move = move->next; + } + + char m[8]; + size_t len = sprintf(m, "%u", i); + memcpy(str, m, len); + + return len; +} + +void compute_fen(char *str, GameState *gamestate) { + str += fen_pieces(str, gamestate); + *str = ' '; str++; + str += fen_color(str, gamestate); + *str = ' '; str++; + str += fen_castling(str, gamestate); + *str = ' '; str++; + str += fen_enpassant(str, gamestate); + *str = ' '; str++; + str += fen_halfmove(str, gamestate); + *str = ' '; str++; + str += fen_movenr(str, gamestate); + str[0] = '\0'; +} diff -r 78b580bfde33 -r eef745ba3774 src/chess/pgn.h --- a/src/chess/pgn.h Wed Aug 26 12:59:30 2015 +0200 +++ b/src/chess/pgn.h Wed Aug 26 14:46:42 2015 +0200 @@ -37,9 +37,10 @@ #ifdef __cplusplus extern "C" { #endif - + int read_pgn(FILE *stream, GameState *gamestate, GameInfo *gameinfo); size_t write_pgn(FILE* stream, GameState *gamestate, GameInfo *gameinfo); +void compute_fen(char *str, GameState *gamestate); #ifdef __cplusplus } diff -r 78b580bfde33 -r eef745ba3774 src/game.c --- a/src/game.c Wed Aug 26 12:59:30 2015 +0200 +++ b/src/game.c Wed Aug 26 14:46:42 2015 +0200 @@ -72,6 +72,10 @@ } static void draw_board(GameState *gamestate, uint8_t perspective) { + char fen[90]; + compute_fen(fen, gamestate); + mvaddstr(0, 0, fen); + for (uint8_t y = 0 ; y < 8 ; y++) { for (uint8_t x = 0 ; x < 8 ; x++) { uint8_t col = gamestate->board[y][x] & COLOR_MASK; @@ -109,7 +113,7 @@ /* move log */ // TODO: introduce window to avoid bugs with a long move log - uint8_t logy = 0; + uint8_t logy = 1; const uint8_t logx = boardx + 30; int logi = 1; MoveList *logelem = gamestate->movelist;