Implement board's tensor representation extraction

Author: crybot

board.cpp

@@ -128,6 +128,22 @@ namespace Napoleon
         EmptySquares = ~OccupiedSquares;
     }
 
+    std::string Board::ToCsv() const 
+    {
+        std::string csv;
+
+        for (auto c = PieceColor::White; c < PieceColor::None; c++) {
+          for (Type p = PieceType::Pawn; p < PieceType::None; p++) {
+            csv += Utils::BitBoard::ToString(Pieces(c, p)) += ",";
+          }
+        }
+
+        csv += std::to_string(sideToMove) + ",";
+        csv += std::to_string(enPassantSquare) + ",";
+        csv += std::to_string(castlingStatus);
+        return csv;
+    }
+
     void Board::Display() const
     {
         Piece piece;

board.h

@@ -33,6 +33,7 @@ namespace Napoleon
 
             void LoadGame(std::string = Constants::StartPosition);
 
+            std::string ToCsv() const;
             void Display() const;
             void AddPiece(Piece, Square);
             BitBoard PlayerPieces() const;

search.cpp

@@ -38,6 +38,8 @@ namespace Napoleon
     int Age = 0;
 
     std::atomic<unsigned long> node_count(0);
+    std::ofstream* Search::positions_dataset;
+    bool Search::record_positions = false;
 
     //int Search::param[Parameters::MAX] = { 25 , 1 , 150 , 3, 250, 50, 6 , 3 , 2 , 500 , 250 , 5 , 3 , 5 , 4 , 2 , 1 , 2 , 9}; // original
     int Search::param[Parameters::MAX] = { 91 , 1 , 18 , 9 , 118 , 36 , 2 , 3 , 2 , 490 , 72 , 4 , 3 , 1 , 4 , 2 , 1 , 2 , 8 ,  }; // tuned
@@ -209,13 +211,17 @@ namespace Napoleon
     {
         Move move = Constants::NullMove;
         Move toMake = Constants::NullMove;
+        int move_score = Constants::Unknown;
         int score;
         int temp;
 
         Age = (Age + 1) % 64;
         score = searchRoot(searchInfo.MaxDepth(), -Constants::Infinity, Constants::Infinity, move, board);
         searchInfo.IncrementDepth();
-        if (score != Constants::Unknown) toMake = move;
+        if (score != Constants::Unknown) {
+          toMake = move;
+          move_score = score;
+        }
 
         while ((searchInfo.MaxDepth() < 100 && !searchInfo.TimeOver()) || pondering)
         {
@@ -248,14 +254,24 @@ namespace Napoleon
 
             score = temp;
 
-            if (score != Constants::Unknown)
+            if (score != Constants::Unknown) {
                 toMake = move;
+                move_score = score;
+            }
 
             searchInfo.IncrementDepth();
         }
 
         StopThinking();
 
+        if (record_positions)
+        {
+          (*positions_dataset) << board.GetFen() << ","
+            << board.ToCsv() << ","
+            << searchInfo.MaxDepth()-1 << "," 
+            << move_score << std::endl;
+        }
+
         return toMake;
     }
 

search.h

@@ -10,6 +10,7 @@
 #include <vector>
 #include <thread>
 #include <atomic>
+#include <fstream>
 
 namespace Napoleon
 {
@@ -43,6 +44,8 @@ namespace Napoleon
         extern int cores;
         extern std::atomic<bool> quit;
         extern const int default_cores;
+        extern std::ofstream* positions_dataset;
+        extern bool record_positions;
 
         enum Parameters : int {
             RAZOR1=0, RAZOR2, RAZOR3, 

uci.cpp

@@ -73,7 +73,12 @@ namespace Napoleon
                     while (stream >> val)
                         PieceSquareTable[PieceType::Pawn][Opening][i++] = val;
                 }
-
+                else if (token == "Record") 
+                {
+                  Search::positions_dataset = new ofstream();
+                  Search::positions_dataset->open("positions_dataset.csv", ios::out | ios::app);
+                  Search::record_positions = true;
+                }
                 else
                 {
                     for (auto i=0; i<Search::Parameters::MAX; i++)
@@ -92,6 +97,9 @@ namespace Napoleon
                 SendCommand<Command::Generic>("Bye Bye");
                 Search::KillThreads();
                 exit = true;
+                if (Search::record_positions) {
+                  Search::positions_dataset->close();
+                }
             }
             else if (cmd == "isready")
             {
@@ -170,6 +178,10 @@ namespace Napoleon
             {
                 Search::PonderHit = true;
             }
+            else if (cmd == "csv")
+            {
+                std::cout << board.ToCsv() << std::endl;
+            }
             /*
             else if (cmd == "tune")
             {

utils.cpp

@@ -67,6 +67,26 @@ namespace Napoleon
                 }
                 std::cout << "\n    A  B  C  D  E  F  G  H\n";
             }
+
+            std::string ToString(Napoleon::BitBoard bitBoard)
+            {
+                std::string flattened;
+                for (int r = 7; r >= 0; r--)
+                {
+                    for (int c = 0; c <= 7; c++)
+                    {
+                        if (IsBitSet(bitBoard, Square::GetSquareIndex(c, r)))
+                        {
+                            flattened += "1";
+                        }
+                        else
+                        {
+                            flattened += "0";
+                        }
+                    }
+                }
+                return flattened;
+            }
         }
 
         namespace Piece

utils.h

@@ -30,6 +30,7 @@ namespace Napoleon
             bool IsBitSet(Napoleon::BitBoard, int);
             void Display(Napoleon::BitBoard);
             void SetBit(Napoleon::BitBoard&, unsigned int);
+            std::string ToString(Napoleon::BitBoard);
 
             int PopCount(Napoleon::BitBoard);
             int BitScanForward(Napoleon::BitBoard);