]> git.sesse.net Git - remoteglot-book/blobdiff - www/js/book.js
Move stats derivation out into its own function.
[remoteglot-book] / www / js / book.js
index cc948fdc3b2152c50014b7eb8dcaee63e818d1b3..c0227d4646665b8a548e602504e50156c3f30043 100644 (file)
@@ -5,6 +5,11 @@ 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_move = null;
+var reverse_dragging_from = null;
 
 var entity_map = {
        "&": "&",
@@ -48,11 +53,11 @@ var update = function() {
                board.position(current_display_fen());
        }
 
+       $("#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');
        }
@@ -93,7 +98,7 @@ var headings = [
        [ "Games", TYPE_INTEGER ],
        [ "%", TYPE_RATIO ],
        [ "CGames", TYPE_INTEGER ],
-       [ "Comp%", TYPE_RATIO ],
+       [ "Hum", TYPE_RATIO ],
        [ "Win%", TYPE_RATIO ],
        [ "WWin", TYPE_INTEGER ],
        [ "%WW", TYPE_RATIO ],
@@ -131,15 +136,7 @@ var show_lines = function(data, game) {
                $('#gamesummary').html(text);
        }
 
-       var total_num = 0;
-       for (var i = 0; i < moves.length; ++i) {
-               var move = moves[i];
-               if (move['move']) {
-                       total_num += parseInt(move['white']);
-                       total_num += parseInt(move['draw']);
-                       total_num += parseInt(move['black']);
-               }
-       }
+       var total_num = find_total_games(moves);
 
        var headings_tr = $("#headings");
        headings_tr.empty();
@@ -166,49 +163,40 @@ var show_lines = function(data, game) {
                var move = moves[i];
                var line = [];
 
-               var white = parseInt(move['white']);
-               var draw = parseInt(move['draw']);
-               var black = parseInt(move['black']);
-               var computer = parseInt(move['computer']);
+               var derived = calc_move_derived_data(move, total_num, data, (move_override % 2 == 0));
+
+               var white = move['white'];
+               var draw = move['draw'];
+               var black = move['black'];
+               var computer = 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(derived['num']);  // N.
+               line.push(derived['fraction']);  // %.
                line.push(computer);  // CGames.
-               line.push(computer / num);  // Comp%.
-
-               // Win%.
-               var white_win_ratio = (white + 0.5 * draw) / num;
-               var win_ratio = (move_override % 2 == 0) ? white_win_ratio : 1.0 - white_win_ratio;
-               line.push(win_ratio);
+               line.push(derived['human_index']);  // Hum.
+               line.push(derived['win_ratio']);  // Win%.
 
-               line.push(white);        // WWin.
-               line.push(white / num);  // %WW.
-               line.push(black);        // BWin.
-               line.push(black / num);  // %BW.
-               line.push(draw);         // Draw.
-               line.push(draw / num);   // %Draw.
+               line.push(white);                   // WWin.
+               line.push(white / derived['num']);  // %WW.
+               line.push(black);                   // BWin.
+               line.push(black / derived['num']);  // %BW.
+               line.push(draw);                    // Draw.
+               line.push(draw / derived['num']);   // %Draw.
 
                if (move['num_elo'] >= 10) {
                        // Elo.
                        line.push(move['white_avg_elo']);
                        line.push(move['black_avg_elo']);
                        line.push(move['white_avg_elo'] - move['black_avg_elo']);
-
-                       // Win% corrected for Elo.
-                       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 = (move_override % 2 == 0) ? white_win_ratio : 1.0 - white_win_ratio;
-                       line.push(win_ratio);
                } else {
                        line.push(null);
                        line.push(null);
                        line.push(null);
-                       line.push(null);
                }
+
+               line.push(derived['corrected_win_ratio'] || null);
                lines.push(line);
        }
 
@@ -242,10 +230,14 @@ var show_lines = function(data, game) {
                                        }
                                }
 
-                               var move_a = document.createElement("a");
-                               move_a.href = "javascript:make_move('" + line[j] + "')";
-                               td.appendChild(move_a);
-                               $(move_a).text(line[j]);
+                               if (line[j] === '1-0' || line[j] === '1/2-1/2' || line[j] === '0-1') {
+                                       $(td).text($(td).text() + line[j]);
+                               } else {
+                                       var move_a = document.createElement("a");
+                                       move_a.href = "javascript:make_move('" + line[j] + "')";
+                                       td.appendChild(move_a);
+                                       $(move_a).text(line[j]);
+                               }
                        } else if (headings[j][1] == TYPE_INTEGER) {
                                add_td(tr, line[j] || 0);
                        } else if (headings[j][1] == TYPE_FLOAT) {
@@ -267,12 +259,62 @@ var show_lines = function(data, game) {
        }
 }
 
+var find_total_games = function(moves) {
+       var total_num = 0;
+       for (var i = 0; i < moves.length; ++i) {
+               var move = moves[i];
+               if (move['move']) {
+                       total_num += move['white'];
+                       total_num += move['draw'];
+                       total_num += move['black'];
+               }
+       }
+       return total_num;
+}
+
+var calc_move_derived_data = function(move, total_num, data, is_white) {
+       var derived = {};
+       var white = move['white'];
+       var draw = move['draw'];
+       var black = move['black'];
+       var computer = move['computer'];
+
+       var num = white + draw + black;
+       derived['num'] = num;
+       derived['fraction'] = num / total_num;
+
+       // Adjust so that the human index is 50% overall.
+       var exp = Math.log(0.5) / Math.log(data['computer_games'] / data['total_games']);
+       derived['human_index'] = 1.0 - Math.pow(computer / num, exp);
+
+       // Win%.
+       var white_win_ratio = (white + 0.5 * draw) / num;
+       var win_ratio = is_white ? white_win_ratio : 1.0 - white_win_ratio;
+       derived['win_ratio'] = win_ratio;
+
+       if (move['num_elo'] >= 10) {
+               // Win% corrected for Elo.
+               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 = is_white ? white_win_ratio : 1.0 - white_win_ratio;
+               derived['corrected_win_ratio'] = win_ratio;
+       }
+
+       return derived;
+};
+
 var set_includetransp = function(value) {
        includetransp = value;
        update();
 }
 window['set_includetransp'] = set_includetransp;
 
+var set_flipboard = function(value) {
+       board.orientation(value ? 'black' : 'white');
+}
+window['set_flipboard'] = set_flipboard;
+
 var make_move = function(move, do_update) {
        var history = game.history({ verbose: true });
        if (move_override < history.length && history[move_override].san == move) {
@@ -335,9 +377,124 @@ var onDragStart = function(source, piece, position, orientation) {
            (pseudogame.turn() === 'b' && piece.search(/^w/) !== -1)) {
                return false;
        }
+
+       recommended_move = null;
+       get_best_dest(pseudogame, source, null, function(src, dest) {
+               $("#board").find('.square-55d63').removeClass('nonuglyhighlight');
+               if (dest !== null) {
+                       var squareEl = $('#board .square-' + dest);
+                       squareEl.addClass('highlight1-32417');
+                       recommended_move = [src, dest];
+               }
+       });
+}
+
+var mousedownSquare = function(e) {
+       reverse_dragging_from = null;
+       var square = $(this).attr('data-square');
+
+       var pseudogame = new Chess(current_display_fen());
+       if (pseudogame.game_over() === true) {
+               return;
+       }
+
+       // If the square is empty, or has a piece of the side not to move,
+       // we handle it. If not, normal piece dragging will take it.
+       var position = board.position();
+       if (!position.hasOwnProperty(square) ||
+           (pseudogame.turn() === 'w' && position[square].search(/^b/) !== -1) ||
+           (pseudogame.turn() === 'b' && position[square].search(/^w/) !== -1)) {
+               reverse_dragging_from = square;
+               get_best_dest(pseudogame, null, square, function(src, dest) {
+                       if (src !== null) {
+                               var squareEl = $('#board .square-' + src);
+                               squareEl.addClass('highlight1-32417');
+                               squareEl = $('#board .square-' + dest);
+                               squareEl.addClass('highlight1-32417');
+                               recommended_move = [src, dest];
+                       }
+               });
+       } else {
+               recommended_src = null;
+       }
+}
+
+var mouseupSquare = function(e) {
+       if (reverse_dragging_from === null) {
+               return;
+       }
+       var source = $(this).attr('data-square');
+       var target = reverse_dragging_from;
+       reverse_dragging_from = null;
+       if (onDrop(source, target) !== 'snapback') {
+               onSnapEnd(source, target);
+       }
+       $("#board").find('.square-55d63').removeClass('highlight1-32417');
+}
+
+var get_best_dest = function(game, source, target, cb) {
+       var moves = game.moves({ verbose: true });
+       if (source !== null) {
+               moves = moves.filter(function(move) { return move.from == source; });
+       }
+       if (target !== null) {
+               moves = moves.filter(function(move) { return move.to == target; });
+       }
+       if (moves.length == 0) {
+               cb(null, null);
+               return;
+       }
+       if (moves.length == 1) {
+               cb(moves[0].from, 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], res[2]);
+                               }
+                       }
+               };
+               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_move === null) {
+                       return 'snapback';
+               } else {
+                       // Accept the move. It will be changed in onSnapEnd.
+                       return;
+               }
+       } else {
+               // Suggestion not asked for.
+               recommended_move = null;
+       }
+
        // see if the move is legal
        var pseudogame = new Chess(current_display_fen());
        var move = pseudogame.move({
@@ -351,6 +508,11 @@ var onDrop = function(source, target) {
 }
 
 var onSnapEnd = function(source, target) {
+       if (source == target && recommended_move !== null) {
+               source = recommended_move[0];
+               target = recommended_move[1];
+       }
+       recommended_move = null;
        var pseudogame = new Chess(current_display_fen());
        var move = pseudogame.move({
                from: source,
@@ -393,6 +555,8 @@ var init = function() {
                onDrop: onDrop,
                onSnapEnd: onSnapEnd
        });
+       $("#board").on('mousedown', '.square-55d63', mousedownSquare);
+       $("#board").on('mouseup', '.square-55d63', mouseupSquare);
 
        window.onpopstate = onpopstate;
        onpopstate();
@@ -404,8 +568,10 @@ var init = function() {
                        prev_move();
                }
        });
-}
 
+       // Seemingly the web worker is not started before we send it a message.
+       stockfish.postMessage("uci");
+}
 
 $(document).ready(init);