stellar

commit 564362bd70ae4b35fc956fd5c97e8dd672c4341a
parent a8feaaca3d441954200377a3aac1c374bcaa1881
Author: Dimitrije Dobrota <mail@dimitrijedobrota.com>
Date:   Fri, 30 Sep 2022 00:42:46 +0200

Support Basic UCI Instructions

Diffstat:
M
include/CBoard.h
 | 
7
+++++--
M
src/CBoard.c
 | 
2
++
M
src/engine.c
 | 
130
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--------

3 files changed, 125 insertions(+), 14 deletions(-)
diff --git a/include/CBoard.h b/include/CBoard.h
@@ -22,6 +22,7 @@ Piece_T Piece_get(ePiece piece, eColor color);
 typedef struct CBoard_T *CBoard_T;
 
 CBoard_T CBoard_new(void);
+void     CBoard_free(CBoard_T *p);
 void     CBoard_copy(CBoard_T self, CBoard_T dest);
 
 U64     CBoard_colorBB(CBoard_T self, eColor color);
@@ -35,8 +36,10 @@ void   CBoard_enpassant_set(CBoard_T self, Square target);
 
 U64  CBoard_pieceSet(CBoard_T self, Piece_T piece);
 U64  CBoard_piece_attacks(CBoard_T self, Piece_T Piece, Square src);
-void CBoard_piece_capture(CBoard_T self, Piece_T Piece, Square source, Square target);
-void CBoard_piece_move(CBoard_T self, Piece_T Piece, Square square, Square target);
+void CBoard_piece_capture(CBoard_T self, Piece_T Piece, Square source,
+                          Square target);
+void CBoard_piece_move(CBoard_T self, Piece_T Piece, Square square,
+                       Square target);
 void CBoard_piece_pop(CBoard_T self, Piece_T Piece, Square square);
 void CBoard_piece_set(CBoard_T self, Piece_T Piece, Square square);
 
diff --git a/src/CBoard.c b/src/CBoard.c
@@ -76,6 +76,8 @@ CBoard_T CBoard_new(void) {
   return p;
 }
 
+void CBoard_free(CBoard_T *p) { FREE(*p); }
+
 void CBoard_copy(CBoard_T self, CBoard_T dest) { *dest = *self; }
 
 Square  CBoard_enpassant(CBoard_T self) { return self->enpassant; }
diff --git a/src/engine.c b/src/engine.c
@@ -318,11 +318,69 @@ void init_all() {
 
 /* UCI */
 
+typedef struct Position_T *Position_T;
+struct Position_T {
+  char *command;
+  char *token;
+  char *crnt;
+};
+
+char *Position_token_next(Position_T self);
+
+Position_T Position_new(char *command) {
+  Position_T p;
+  NEW0(p);
+  p->command = ALLOC(strlen(command) + 1);
+  p->token = ALLOC(strlen(command) + 1);
+  strcpy(p->command, command);
+  p->crnt = command;
+  Position_token_next(p);
+  return p;
+}
+
+void Position_free(Position_T *p) {
+  FREE((*p)->command);
+  FREE((*p)->token);
+  FREE(*p);
+}
+
+char *Position_token(Position_T self) { return self->token; }
+char *Position_token_n(Position_T self, int n) {
+  while (isspace(*self->crnt) && *self->crnt != '\0')
+    self->crnt++;
+
+  if (*self->crnt == '\0') {
+    *self->token = '\0';
+    return NULL;
+  }
+
+  char *p = self->token;
+  while (n--) {
+    while (!isspace(*self->crnt) && *self->crnt != '\0')
+      *p++ = *self->crnt++;
+    if (*self->crnt == '\0') {
+      p++;
+      break;
+    }
+    self->crnt++;
+    *p++ = ' ';
+  }
+  *--p = '\0';
+
+  return self->token;
+}
+
+char *Position_token_next(Position_T self) { return Position_token_n(self, 1); }
+
 Move parse_move(CBoard_T self, char *move_string) {
-  MoveList_T moves = generate_moves(self, NULL);
-  Square     source = coordinates_to_square(move_string);
-  Square     target = coordinates_to_square(move_string + 2);
+  Move       result = 0;
+  MoveList_T moves;
+  Square     source, target;
 
+  source = coordinates_to_square(move_string);
+  target = coordinates_to_square(move_string + 2);
+
+  moves = generate_moves(self, NULL);
   for (int i = 0; i < moves->count; i++) {
     Move move = moves->moves[i];
     if (Move_source(move) == source && Move_target(move) == target) {
@@ -331,23 +389,71 @@ Move parse_move(CBoard_T self, char *move_string) {
         if (tolower(Piece_code(promoted)) != move_string[4])
           continue;
       }
-      return move;
+      result = move;
+      break;
     }
   }
 
-  return 0;
+  MoveList_free(&moves);
+  return result;
+}
+
+CBoard_T Position_parse(Position_T self, CBoard_T board) {
+  printf("Commands: %s\n", self->command);
+  int   count = 0;
+  char *token = Position_token(self);
+  do {
+    printf("Token %d: %s\n", ++count, token);
+
+    if (strcmp(token, "position") == 0) {
+      printf("FOUND position\n");
+      token = Position_token_next(self);
+      if (strcmp(token, "startpos") == 0) {
+        printf("FOUND startpos\n");
+        board = CBoard_fromFEN(board, start_position);
+      } else if (strcmp(token, "fen") == 0) {
+        token = Position_token_n(self, 6);
+        printf("fen: %s\n", token);
+        board = CBoard_fromFEN(board, token);
+        continue;
+      } else {
+        printf("Unknown argument after position\n");
+        assert(0);
+      }
+    }
+
+    if (strcmp(token, "moves") == 0) {
+      printf("FOUND moves\n");
+      CBoard_print(board);
+      while ((token = Position_token_next(self))) {
+        Move move = parse_move(board, token);
+        if (move) {
+          make_move(board, move, 0);
+          CBoard_print(board);
+
+        } else {
+          printf("Invalid move %s!\n", token);
+          assert(0);
+        }
+      }
+    }
+
+  } while ((token = Position_token_next(self)));
+
+  return board;
 }
 
 int main(void) {
   init_all();
 
-  CBoard_T board = CBoard_fromFEN(NULL, start_position);
-  Move     move = parse_move(board, "e2e4");
-  if (move) {
-    make_move(board, move, 0);
-    CBoard_print(board);
-  } else
-    printf("Illegal move!\n");
+  CBoard_T   board = NULL;
+  Position_T position;
+
+  position =
+      Position_new("   position   \t fen " start_position "   moves e2e4 e7e5");
+  board = Position_parse(position, board);
+  Position_free(&position);
+  CBoard_free(&board);
 
   return 0;
 }