X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fendgame.cpp;h=65f0f409de3df7daa8b38ae0636f3dc69323c236;hp=d7cb2b3c8c58541a54d0e662aff2d25110a1e07f;hb=e1919384a23fe728422f995369161efa192380db;hpb=0347339970b68b998c7202e22c01560850dcd13e

diff --git a/src/endgame.cpp b/src/endgame.cpp
index d7cb2b3c..65f0f409 100644
--- a/src/endgame.cpp
+++ b/src/endgame.cpp
@@ -22,6 +22,7 @@
 
 #include "bitcount.h"
 #include "endgame.h"
+#include "movegen.h"
 
 using std::string;
 
@@ -76,27 +77,12 @@ namespace {
     string fen =  sides[0] + char('0' + int(8 - code.length()))
                 + sides[1] + "/8/8/8/8/8/8/8 w - - 0 10";
 
-    return Position(fen, false, 0).material_key();
+    return Position(fen, false).material_key();
   }
 
   template<typename M>
   void delete_endgame(const typename M::value_type& p) { delete p.second; }
 
-  // Fast stalemate detection with lone king
-  bool is_kxk_stalemate(const Position &pos, const Color c) {
-    if ( pos.side_to_move() == c &&
-        !pos.in_check()) {
-      const Square from = pos.king_square(c);
-      Bitboard b = pos.attacks_from<KING>(from);
-      while (b) {
-        // Assume there are no pinned pieces, as it is a lone king
-        if (pos.pl_move_is_legal(make_move(from, pop_1st_bit(&b)), 0))
-          return false;
-      }
-      return true;
-    }
-    return false;
-  }
 } // namespace
 
 
@@ -130,10 +116,8 @@ Endgames::~Endgames() {
 template<EndgameType E>
 void Endgames::add(const string& code) {
 
-  typedef typename eg_family<E>::type T;
-
-  map((T*)0)[key(code, WHITE)] = new Endgame<E>(WHITE);
-  map((T*)0)[key(code, BLACK)] = new Endgame<E>(BLACK);
+  map((Endgame<E>*)0)[key(code, WHITE)] = new Endgame<E>(WHITE);
+  map((Endgame<E>*)0)[key(code, BLACK)] = new Endgame<E>(BLACK);
 }
 
 
@@ -147,10 +131,13 @@ Value Endgame<KXK>::operator()(const Position& pos) const {
   assert(pos.non_pawn_material(weakerSide) == VALUE_ZERO);
   assert(pos.piece_count(weakerSide, PAWN) == VALUE_ZERO);
 
-  if (is_kxk_stalemate(pos, weakerSide)) {
+  // Stalemate detection with lone king
+  if (    pos.side_to_move() == weakerSide
+      && !pos.in_check()
+      && !MoveList<MV_LEGAL>(pos).size()) {
     return VALUE_DRAW;
   }
-  
+
   Square winnerKSq = pos.king_square(strongerSide);
   Square loserKSq = pos.king_square(weakerSide);
 
@@ -161,7 +148,7 @@ Value Endgame<KXK>::operator()(const Position& pos) const {
 
   if (   pos.piece_count(strongerSide, QUEEN)
       || pos.piece_count(strongerSide, ROOK)
-      || pos.both_color_bishops(strongerSide)) {
+      || pos.bishop_pair(strongerSide)) {
     result += VALUE_KNOWN_WIN;
   }