Fix mistake with TT

Bench: 1033376
Dannyj1 · Dec 13, 2023 · cbafad2 · cbafad2
1 parent d1b5c68
commit cbafad2
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/src/search.cpp b/src/search.cpp
@@ -31,7 +31,7 @@
 
 namespace Zagreus {
 MoveListPool* moveListPool = MoveListPool::getInstance();
-TranspositionTable* tt = tt;
+TranspositionTable* tt = TranspositionTable::getTT();
 
 template <PieceColor color>
 Move getBestMove(senjo::GoParams params, ZagreusEngine& engine, Bitboard& board,
@@ -132,8 +132,7 @@ int search(Bitboard& board, int alpha, int beta, int16_t depth, Move& previousMo
     }
 
     if (!IS_PV_NODE) {
-        int ttScore = tt->getScore(board.getZobristHash(), depth, alpha,
-                                   beta);
+        int ttScore = tt->getScore(board.getZobristHash(), depth, alpha, beta);
 
         if (ttScore != INT32_MIN) {
             return ttScore;
@@ -233,6 +232,7 @@ int qsearch(Bitboard& board, int alpha, int beta, int16_t depth, Move& previousM
             SearchContext& context,
             senjo::SearchStats& searchStats) {
     constexpr PieceColor OPPOSITE_COLOR = color == WHITE ? BLACK : WHITE;
+    constexpr TTNodeType IS_PV_NODE = nodeType == PV ? EXACT_NODE : FAIL_LOW_NODE;
 
     if (board.isDraw()) {
         return DRAW_SCORE;
@@ -243,6 +243,15 @@ int qsearch(Bitboard& board, int alpha, int beta, int16_t depth, Move& previousM
         return beta;
     }
 
+    if (!IS_PV_NODE) {
+        int ttScore = TranspositionTable::getTT()->getScore(board.getZobristHash(), depth, alpha,
+                                                            beta);
+
+        if (ttScore != INT32_MIN) {
+            return ttScore;
+        }
+    }
+
     searchStats.qnodes += 1;
 
     bool inCheck = board.isKingInCheck<color>();
@@ -307,8 +316,7 @@ int qsearch(Bitboard& board, int alpha, int beta, int16_t depth, Move& previousM
 
         if (score > alpha) {
             if (score >= beta) {
-                moveListPool->releaseMoveList(moves);
-                // tt->addPosition(board.getZobristHash(), depth, score, FAIL_HIGH_NODE, 0);
+                moveListPool->releaseMoveList(moves);;
                 return beta;
             }