]> git.sesse.net Git - remoteglot-book/blobdiff - www/js/book.js
Embed Stockfish (!) to get square suggestions whenever you pick up a piece.
[remoteglot-book] / www / js / book.js
index 6017fd9d0d5eb6a91a1238089c080a7af460c819..bc6adfceccca4174daf9ee78d0f274028abc050e 100644 (file)
@@ -1,9 +1,14 @@
 (function() {
 
 var board = null;
-var history = [];
+var game = new Chess();
+var fens = [];
 var move_override = 0;
 var includetransp = true;
+var stockfish = new Worker('/js/stockfish.js');
+var engine_running = false;
+var engine_replacement_callback = null;
+var recommended_dest = null;
 
 var entity_map = {
        "&": "&",
@@ -19,43 +24,39 @@ function escape_html(string) {
        });
 }
 
-var get_game = function(skip_last_move) {
-       var moves_to_drop = 0;
-       if (skip_last_move === true) {
-               moves_to_drop = 1;
-       }
-
-       var game = new Chess();
-       for (var i = 0; i < move_override - moves_to_drop; ++i) {
-               game.move(history[i]);
+var current_display_fen = function() {
+       if (move_override == 0) {
+               return 'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1';
+       } else {
+               return fens[move_override - 1];
        }
-       return game;
 }
 
 var update = function() {
        var text = "";
+       var history = game.history({ verbose: true });
        for (var i = 0; i < history.length; ++i) {
                if (i % 2 == 0) {
                        text += (i/2 + 1) + ". ";
                }
                if (i + 1 == move_override) {
-                       text += '<strong>' + history[i] + '</strong>';
+                       text += '<strong>' + history[i].san + '</strong>';
                } else {
-                       text += '<a href="javascript:set_move(' + (i + 1) + ')">' + history[i] + '</a>';
+                       text += '<a href="javascript:set_move(' + (i + 1) + ')">' + history[i].san + '</a>';
                }
                text += " ";
        }
        $('#gamehistory').html(text);
 
-       var game = get_game();
-       board.position(game.fen());
+       if (board.fen() != current_display_fen()) {
+               board.position(current_display_fen());
+       }
 
-       var all_moves = game.history({ verbose: true });
-       if (all_moves.length > 0) {
-               var last_move = all_moves.pop();
+       $("#board").find('.square-55d63').removeClass('nonuglyhighlight');
+       if (move_override > 0) {
+               var last_move = history[move_override - 1];
                var highlight_from = last_move.from;
                var highlight_to = last_move.to;
-               $("#board").find('.square-55d63').removeClass('nonuglyhighlight');
                $("#board").find('.square-' + highlight_from).addClass('nonuglyhighlight');
                $("#board").find('.square-' + highlight_to).addClass('nonuglyhighlight');
        }
@@ -63,16 +64,16 @@ var update = function() {
        fetch_analysis();
 }
 
+var get_history_url = function() {
+       var history = game.history({ verbose: true }).map(function(x) { return x.san; });
+       history.length = move_override;
+       return '/?' + history.join(',');
+}
+
 var fetch_analysis = function() {
-       var game = get_game();
-       var fen = game.fen();
-       var prevfen = "";
-       if (move_override > 0) {
-               prevfen = get_game(true).fen();
-       }
+       var fen = current_display_fen();
        $.ajax({
                url: "/opening-stats.pl?fen=" + encodeURIComponent(fen) +
-                       ";prevfen=" + encodeURIComponent(prevfen) +
                        ";includetransp=" + (includetransp ? 1 : 0)
        }).done(function(data, textstatus, xhr) {
                show_lines(data, game);
@@ -95,6 +96,8 @@ var headings = [
        [ "Move", TYPE_MOVE ],
        [ "Games", TYPE_INTEGER ],
        [ "%", TYPE_RATIO ],
+       [ "CGames", TYPE_INTEGER ],
+       [ "Hum", TYPE_RATIO ],
        [ "Win%", TYPE_RATIO ],
        [ "WWin", TYPE_INTEGER ],
        [ "%WW", TYPE_RATIO ],
@@ -162,6 +165,7 @@ var show_lines = function(data, game) {
        }
 
        var lines = [];
+       var transpose_only = [];
        for (var i = 0; i < moves.length; ++i) {
                var move = moves[i];
                var line = [];
@@ -169,15 +173,22 @@ var show_lines = function(data, game) {
                var white = parseInt(move['white']);
                var draw = parseInt(move['draw']);
                var black = parseInt(move['black']);
+               var computer = parseInt(move['computer']);
 
                line.push(move['move']);  // Move.
+               transpose_only.push(move['transpose_only']);
                var num = white + draw + black;
                line.push(num);  // N.
                line.push(num / total_num);  // %.
+               line.push(computer);  // CGames.
+
+               // Adjust so that the human index is 50% overall.
+               var exp = Math.log(0.5) / Math.log(data['computer_games'] / data['total_games']);
+               line.push(1.0 - Math.pow(computer / num, exp));  // Hum.
 
                // Win%.
                var white_win_ratio = (white + 0.5 * draw) / num;
-               var win_ratio = (game.turn() == 'w') ? white_win_ratio : 1.0 - white_win_ratio;
+               var win_ratio = (move_override % 2 == 0) ? white_win_ratio : 1.0 - white_win_ratio;
                line.push(win_ratio);
 
                line.push(white);        // WWin.
@@ -197,7 +208,7 @@ var show_lines = function(data, game) {
                        var win_elo = -400.0 * Math.log(1.0 / white_win_ratio - 1.0) / Math.LN10;
                        win_elo -= (move['white_avg_elo'] - move['black_avg_elo']);
                        white_win_ratio = 1.0 / (1.0 + Math.pow(10, win_elo / -400.0));
-                       win_ratio = (game.turn() == 'w') ? white_win_ratio : 1.0 - white_win_ratio;
+                       win_ratio = (move_override % 2 == 0) ? white_win_ratio : 1.0 - white_win_ratio;
                        line.push(win_ratio);
                } else {
                        line.push(null);
@@ -219,6 +230,8 @@ var show_lines = function(data, game) {
 
                if (line[0] === undefined) {
                        $(tr).addClass("totals");
+               } else if (transpose_only[i]) {
+                       $(tr).addClass("transponly");
                }
 
                for (var j = 0; j < line.length; ++j) {
@@ -241,11 +254,19 @@ var show_lines = function(data, game) {
                                td.appendChild(move_a);
                                $(move_a).text(line[j]);
                        } else if (headings[j][1] == TYPE_INTEGER) {
-                               add_td(tr, line[j]);
+                               add_td(tr, line[j] || 0);
                        } else if (headings[j][1] == TYPE_FLOAT) {
-                               add_td(tr, line[j].toFixed(1));
+                               if (isNaN(line[j]) || !isFinite(line[j])) {
+                                       add_td(tr, '');
+                               } else {
+                                       add_td(tr, line[j].toFixed(1));
+                               }
                        } else {
-                               add_td(tr, (100.0 * line[j]).toFixed(1) + "%");
+                               if (isNaN(line[j]) || !isFinite(line[j])) {
+                                       add_td(tr, '');
+                               } else {
+                                       add_td(tr, (100.0 * line[j]).toFixed(1) + "%");
+                               }
                        }
                }
 
@@ -259,16 +280,30 @@ var set_includetransp = function(value) {
 }
 window['set_includetransp'] = set_includetransp;
 
-var make_move = function(move) {
-       if (move_override < history.length && history[move_override] == move) {
+var make_move = function(move, do_update) {
+       var history = game.history({ verbose: true });
+       if (move_override < history.length && history[move_override].san == move) {
                // User effectively only moved forward in history.
                ++move_override;
        } else {
-               history.length = move_override;
-               history.push(move);
-               move_override = history.length;
+               var moves = game.history();
+               // Truncate the history if needed.
+               if (move_override < moves.length) {
+                       game = new Chess();
+                       for (var i = 0; i < move_override; ++i) {
+                               game.move(moves[i]);
+                       }
+                       fens.length = move_override;
+               }
+               game.move(move);
+               fens.push(game.fen());
+               ++move_override;
+       }
+
+       if (do_update !== false) {
+               update();
+               window.history.pushState(null, null, get_history_url());
        }
-       update();
 }
 window['make_move'] = make_move;
 
@@ -276,38 +311,109 @@ var prev_move = function() {
        if (move_override > 0) {
                --move_override;
                update();
+               window.history.replaceState(null, null, get_history_url());
        }
 }
 window['prev_move'] = prev_move;
 
 var next_move = function() {
-       if (move_override < history.length) {
+       if (move_override < game.history().length) {
                ++move_override;
                update();
+               window.history.replaceState(null, null, get_history_url());
        }
 }
 window['next_move'] = next_move;
 
-var set_move = function(n) {
+var set_move = function(n, do_update) {
        move_override = n;
-       update();
+       if (do_update !== false) {
+               update();
+               window.history.replaceState(null, null, get_history_url());
+       }
 }
 window['set_move'] = set_move;
 
 // almost all of this stuff comes from the chessboard.js example page
 var onDragStart = function(source, piece, position, orientation) {
-       var game = get_game();
-       if (game.game_over() === true ||
-           (game.turn() === 'w' && piece.search(/^b/) !== -1) ||
-           (game.turn() === 'b' && piece.search(/^w/) !== -1)) {
+       var pseudogame = new Chess(current_display_fen());
+       if (pseudogame.game_over() === true ||
+           (pseudogame.turn() === 'w' && piece.search(/^b/) !== -1) ||
+           (pseudogame.turn() === 'b' && piece.search(/^w/) !== -1)) {
                return false;
        }
+
+       recommended_dest = null;
+       get_best_dest(pseudogame, source, function(dest) {
+               $("#board").find('.square-55d63').removeClass('nonuglyhighlight');
+               if (dest !== null) {
+                       var squareEl = $('#board .square-' + dest);
+                       squareEl.addClass('highlight1-32417');
+               }
+               recommended_dest = dest;
+       });
+}
+
+var get_best_dest = function(game, source, cb) {
+       var moves = game.moves({ square: source, verbose: true });
+       if (moves.length == 0) {
+               cb(null);
+               return;
+       }
+       if (moves.length == 1) {
+               cb(moves[0].to);
+               return;
+       }
+
+       // More than one move. Ask the engine to disambiguate.
+       var uci_moves = moves.map(function(m) { return m.from + m.to; });
+       var when_engine_is_ready = function() {
+               engine_running = true;
+               stockfish.onmessage = function(event) {
+                       var res = event.data.match(/^bestmove \S\S(\S\S)/);
+                       if (res !== null) {
+                               engine_running = false;
+                               if (engine_replacement_callback !== null) {
+                                       // We are no longer interested in this query,
+                                       // so just discard it and call this other callback.
+                                       engine_replacement_callback();
+                                       engine_replacement_callback = null;
+                               } else {
+                                       cb(res[1]);
+                               }
+                       }
+               };
+               stockfish.postMessage("position fen " + game.fen());
+               stockfish.postMessage("go depth 6 searchmoves " + uci_moves.join(" "));
+       };
+       if (engine_running) {
+               engine_replacement_callback = when_engine_is_ready;
+       } else {
+               when_engine_is_ready();
+       }
 }
 
 var onDrop = function(source, target) {
+       if (engine_running) {
+               // Snap end before the engine came back.
+               // Discard the result when it does.
+               engine_replacement_callback = function() {};
+       }
+       if (source == target) {
+               if (recommended_dest === null) {
+                       return 'snapback';
+               } else {
+                       // Accept the move. It will be changed in onSnapEnd.
+                       return;
+               }
+       } else {
+               // Suggestion not asked for.
+               recommended_dest = null;
+       }
+
        // see if the move is legal
-       var game = get_game();
-       var move = game.move({
+       var pseudogame = new Chess(current_display_fen());
+       var move = pseudogame.move({
                from: source,
                to: target,
                promotion: 'q' // NOTE: always promote to a queen for example simplicity
@@ -315,22 +421,45 @@ var onDrop = function(source, target) {
 
        // illegal move
        if (move === null) return 'snapback';
+}
 
-       var new_history = game.history({ verbose: true });
-       history = [];
-       for (var i = 0; i < new_history.length; ++i) {
-               history.push(new_history[i].san);
+var onSnapEnd = function(source, target) {
+       if (source == target && recommended_dest !== null) {
+               target = recommended_dest;
+       }
+       recommended_dest = null;
+       var pseudogame = new Chess(current_display_fen());
+       var move = pseudogame.move({
+               from: source,
+               to: target,
+               promotion: 'q' // NOTE: always promote to a queen for example simplicity
+       });
+
+       make_move(pseudogame.history({ verbose: true }).pop().san);
+}
+
+var onpopstate = function() {
+       var old_moves = game.history({ verbose: true }).map(function(x) { return x.san; });
+       var new_moves = document.location.search.replace(/^\?/, "").split(",");
+
+       if (new_moves.length == 1 && new_moves[0] == "") {
+               new_moves = [];
        }
-       move_override = history.length;
-};
 
-// update the board position after the piece snap 
-// for castling, en passant, pawn promotion
-var onSnapEnd = function() {
-       var game = get_game();
-       board.position(game.fen());
+       var num_shared_moves;
+       for (num_shared_moves = 0; num_shared_moves < Math.min(old_moves.length, new_moves.length); ++num_shared_moves) {
+               if (old_moves[i] != new_moves[i]) {
+                       break;
+               }
+       }
+
+       set_move(num_shared_moves, false);
+       for (var i = num_shared_moves; i < new_moves.length; ++i) {
+               make_move(new_moves[i], false);
+       }
        update();
-};
+       window.history.replaceState(null, null, get_history_url());
+}
 
 var init = function() {
        // Create board.
@@ -341,7 +470,9 @@ var init = function() {
                onDrop: onDrop,
                onSnapEnd: onSnapEnd
        });
-       update();
+
+       window.onpopstate = onpopstate;
+       onpopstate();
 
        $(window).keyup(function(event) {
                if (event.which == 39) {