X-Git-Url: http://winboard.nl/cgi-bin?a=blobdiff_plain;f=fen.c;fp=fen.c;h=be3ec17e8fb142a43dad8185a375824f56b03919;hb=e15efca6667b2673b4c1a5879a6917eab6800e58;hp=0000000000000000000000000000000000000000;hpb=0d182b4efac85dce968068bfe4509e52e9a30051;p=polyglot.git

diff --git a/fen.c b/fen.c
new file mode 100644
index 0000000..be3ec17
--- /dev/null
+++ b/fen.c
@@ -0,0 +1,392 @@
+
+// fen.c
+
+// includes
+
+#include <ctype.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "board.h"
+#include "colour.h"
+#include "fen.h"
+#include "option.h"
+#include "piece.h"
+#include "square.h"
+#include "util.h"
+
+// "constants"
+
+// const char * StartFen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w HAha - 0 1";
+const char * StartFen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1";
+
+// variables
+
+static const bool Strict = FALSE;
+
+// macros
+
+#define skip_white_space() \
+        c=string[pos];\
+        if (c != ' ' && c!='\t') my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos); \
+        while(c==' ' || c=='\t') c=string[++pos];
+
+
+// functions
+
+// board_from_fen()
+
+bool board_from_fen(board_t * board, const char string[]) {
+
+   int pos;
+   int file, rank, sq;
+   int c;
+   int i, len;
+   int piece;
+   int king_pos[ColourNb];
+
+   ASSERT(board!=NULL);
+   ASSERT(string!=NULL);
+
+   board_clear(board);
+
+   king_pos[White] = SquareNone;
+   king_pos[Black] = SquareNone;
+
+   pos = 0;
+   c = string[pos];
+
+   // piece placement
+
+   for (rank = 7; rank >= 0; rank--) {
+
+      for (file = 0; file < 8;) {
+
+         sq = square_make(file,rank);
+
+         if (c >= '1' && c <= '8') { // empty square(s)
+
+            len = c - '0';
+            if (file + len > 8) my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+
+            for (i = 0; i < len; i++) {
+               board->square[sq++] = Empty;
+               file++;
+            }
+
+         } else { // piece
+
+            piece = piece_from_char(c);
+            if (piece == PieceNone256) my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+
+            if (piece_is_king(piece)) king_pos[piece_colour(piece)] = sq;
+
+            board->square[sq++] = piece;
+            file++;
+         }
+
+         c = string[++pos];
+      }
+
+      if (rank > 0) {
+         if (c != '/') my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+         c = string[++pos];
+     }
+   }
+
+   // active colour
+
+   skip_white_space();
+
+   switch (c) {
+   case 'w':
+      board->turn = White;
+      break;
+   case 'b':
+      board->turn = Black;
+      break;
+   default:
+      my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+      break;
+   }
+
+   c = string[++pos];
+
+   // castling
+
+   skip_white_space();
+
+   board->castle[White][SideH] = SquareNone;
+   board->castle[White][SideA] = SquareNone;
+   board->castle[Black][SideH] = SquareNone;
+   board->castle[Black][SideA] = SquareNone;
+
+   if (c == '-') { // no castling rights
+
+      c = string[++pos];
+
+   } else {
+
+      // TODO: filter out illegal rights
+
+      do {
+
+         if (FALSE) {
+
+         } else if (c == 'K') {
+
+            for (sq = H1; sq > king_pos[White]; sq--) {
+               if (board->square[sq] == WhiteRook256) {
+                  board->castle[White][SideH] = sq;
+                  break;
+               }
+            }
+
+         } else if (c == 'Q') {
+
+            for (sq = A1; sq < king_pos[White]; sq++) {
+               if (board->square[sq] == WhiteRook256) {
+                  board->castle[White][SideA] = sq;
+                  break;
+               }
+            }
+
+         } else if (c == 'k') {
+
+            for (sq = H8; sq > king_pos[Black]; sq--) {
+               if (board->square[sq] == BlackRook256) {
+                  board->castle[Black][SideH] = sq;
+                  break;
+               }
+            }
+
+         } else if (c == 'q') {
+
+            for (sq = A8; sq < king_pos[Black]; sq++) {
+               if (board->square[sq] == BlackRook256) {
+                  board->castle[Black][SideA] = sq;
+                  break;
+               }
+            }
+
+         } else if (c >= 'A' && c <= 'H') {
+
+            // white castling right
+
+            sq = square_make(file_from_char(tolower(c)),Rank1);
+
+            if (sq > king_pos[White]) { // h side
+               board->castle[White][SideH] = sq;
+            } else { // a side
+               board->castle[White][SideA] = sq;
+            }
+
+         } else if (c >= 'a' && c <= 'h') {
+
+            // black castling right
+
+            sq = square_make(file_from_char(tolower(c)),Rank8);
+
+            if (sq > king_pos[Black]) { // h side
+               board->castle[Black][SideH] = sq;
+            } else { // a side
+               board->castle[Black][SideA] = sq;
+            }
+
+         } else {
+
+            my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+         }
+
+         c = string[++pos];
+
+      } while (c != ' ');
+   }
+
+   // en-passant
+
+   skip_white_space();
+
+   if (c == '-') { // no en-passant
+
+      sq = SquareNone;
+      c = string[++pos];
+
+   } else {
+
+      if (c < 'a' || c > 'h') my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+      file = file_from_char(c);
+      c = string[++pos];
+
+      if (c < '1' || c > '8') my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+      rank = rank_from_char(c);
+      c = string[++pos];
+
+      sq = square_make(file,rank);
+   }
+
+   board->ep_square = sq;
+
+   // halfmove clock
+
+   board->ply_nb = 0;
+   board->move_nb = 0; // HACK, in case of broken syntax
+
+   if (c != ' ') {
+      if (!Strict) goto update;
+      my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+   }
+   c = string[++pos];
+
+   if (!isdigit(c)) {
+      if (!Strict) goto update;
+      my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+   }
+
+   board->ply_nb = atoi(&string[pos]);
+   do c = string[++pos]; while (isdigit(c));
+
+   // fullmove number
+
+   board->move_nb = 0;
+
+   if (c != ' ') {
+      if (!Strict) goto update;
+      my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+   }
+   c = string[++pos];
+
+   if (!isdigit(c)) {
+      if (!Strict) goto update;
+      my_fatal("board_from_fen(): bad FEN (pos=%d)\n",pos);
+   }
+
+   board->move_nb = atoi(&string[pos]) - 1;
+   do c = string[++pos]; while (isdigit(c));
+
+   // board update
+
+update:
+   board_init_list(board);
+
+   return TRUE;
+}
+
+// board_to_fen()
+
+bool board_to_fen(const board_t * board, char string[], int size) {
+
+   int pos;
+   int file, rank;
+   int sq, piece;
+   int c;
+   int len;
+   int old_pos;
+
+   ASSERT(board_is_ok(board));
+   ASSERT(string!=NULL);
+   ASSERT(size>=92);
+
+   // init
+
+   if (size < 92) return FALSE;
+
+   pos = 0;
+
+   // piece placement
+
+   for (rank = 7; rank >= 0; rank--) {
+
+      for (file = 0; file < 8;) {
+
+         sq = square_make(file,rank);
+         piece = board->square[sq];
+         ASSERT(piece==Empty||piece_is_ok(piece));
+
+         if (piece == Empty) {
+
+            len = 0;
+            for (; file < 8 && board->square[square_make(file,rank)] == Empty; file++) {
+               len++;
+            }
+
+            ASSERT(len>=1&&len<=8);
+            c = '0' + len;
+
+         } else {
+
+            c = piece_to_char(piece);
+            file++;
+         }
+
+         string[pos++] = c;
+      }
+
+      string[pos++] = '/';
+   }
+
+   string[pos-1] = ' '; // HACK: remove the last '/'
+
+   // active colour
+
+   string[pos++] = (colour_is_white(board->turn)) ? 'w' : 'b';
+   string[pos++] = ' ';
+
+   // castling
+
+   old_pos = pos;
+
+   if (option_get_bool("Chess960")) {
+
+      // FEN-960
+
+      if (board->castle[White][SideH] != SquareNone) {
+         string[pos++] = toupper(file_to_char(square_file(board->castle[White][SideH])));
+      }
+
+      if (board->castle[White][SideA] != SquareNone) {
+         string[pos++] = toupper(file_to_char(square_file(board->castle[White][SideA])));
+      }
+
+      if (board->castle[Black][SideH] != SquareNone) {
+         string[pos++] = tolower(file_to_char(square_file(board->castle[Black][SideH])));
+      }
+
+      if (board->castle[Black][SideA] != SquareNone) {
+         string[pos++] = tolower(file_to_char(square_file(board->castle[Black][SideA])));
+      }
+
+   } else {
+
+      // FEN
+
+      if (board->castle[White][SideH] != SquareNone) string[pos++] = 'K';
+      if (board->castle[White][SideA] != SquareNone) string[pos++] = 'Q';
+      if (board->castle[Black][SideH] != SquareNone) string[pos++] = 'k';
+      if (board->castle[Black][SideA] != SquareNone) string[pos++] = 'q';
+   }
+
+   if (pos == old_pos) string[pos++] = '-';
+
+   string[pos++] = ' ';
+
+   // en-passant
+
+   if (board->ep_square == SquareNone) {
+      string[pos++] = '-';
+   } else {
+      if (!square_to_string(board->ep_square,&string[pos],3)) return FALSE;
+      pos += 2;
+   }
+
+   string[pos++] = ' ';
+
+   // halfmove clock and fullmove number
+
+   sprintf(&string[pos],"%d %d",board->ply_nb,board->move_nb+1);
+
+   return TRUE;
+}
+
+// end of fen.cpp
+