Add command line argument to print the final board as a scheme object

This could be useful to create a machine learning AI, which uses the
final board for training.

Also bubble up the final score and present it at the end of the game.

1 files changed, 39 insertions, 15 deletions

diff --git a/cmd/main.c b/cmd/main.c
index 12d7f3a..5863889 100644
--- a/cmd/main.c
+++ b/cmd/main.c
@@ -37,20 +37,23 @@ int main(int argc, char **argv) {
       {"help", no_argument, NULL, 'h'},
       {"player-one", required_argument, NULL, '1'},
       {"player-two", required_argument, NULL, '2'},
+      {"scm-final-board", no_argument, NULL, 's'},
       {0, 0, 0, 0}};
 
   const char *usage =
       " -h --help              show help output\n"
       " -1 --player-one <file> specify a file to use for player one strategy\n"
-      " -2 --player-two <file> specify a file to use for player two strategy\n";
+      " -2 --player-two <file> specify a file to use for player two strategy\n"
+      " -s --scm-final-board   print the final board as a Scheme object\n";
 
   char *player_one_strategy_path = NULL;
   char *player_two_strategy_path = NULL;
+  bool should_print_as_scm = false;
 
   int c;
   while (1) {
     int option_index = 0;
-    c = getopt_long(argc, argv, "h1:2:", long_options, &option_index);
+    c = getopt_long(argc, argv, "h1:2:s", long_options, &option_index);
     if (c == -1) {
       break;
     }
@@ -73,6 +76,9 @@ int main(int argc, char **argv) {
         exit_status = EXIT_FAILURE;
       }
       break;
+    case 's':
+      should_print_as_scm = true;
+      break;
     }
   }
 
@@ -81,24 +87,42 @@ int main(int argc, char **argv) {
     exit(EXIT_FAILURE);
   }
 
+  int white_score, black_score;
   enum player_color winner =
-      game_loop(player_one_strategy_path, player_two_strategy_path);
+      game_loop(player_one_strategy_path, player_two_strategy_path,
+                &white_score, &black_score);
   free(player_one_strategy_path);
   free(player_two_strategy_path);
 
-  switch (winner) {
-  case WHITE:
-    puts("Congratulations for winning player 1!");
-    break;
-  case BLACK:
-    puts("Congratulations for winning player 2!");
-    break;
-  default:
-    puts("The game ended with no winner.");
-  }
+  if (should_print_as_scm) {
+    // Make sure guile has been initialized.
+    // It probably has, since it is unlikely that the user would want to
+    // print the final board as a Scheme object if they have no Scheme
+    // players... but it is possible.
+    scm_init_guile();
+
+    scm_c_eval_string("(use-modules (ice-9 pretty-print))"
+                      "(pretty-print (get-board))");
+    printf("%d\n", white_score);
+    printf("%d\n", black_score);
+  } else {
+    switch (winner) {
+    case WHITE:
+      puts("Congratulations for winning player 1!");
+      break;
+    case BLACK:
+      puts("Congratulations for winning player 2!");
+      break;
+    default:
+      puts("The game ended with no winner.");
+    }
 
-  puts("Here was the final board:");
-  print_board(get_board());
+    puts("Here was the final board:");
+    print_board(get_board());
+
+    printf("Final white score: %d\n", white_score);
+    printf("Final black score: %d\n", black_score);
+  }
 
   return EXIT_SUCCESS;
 }