]> git.sesse.net Git - remoteglot/blobdiff - www/js/remoteglot.js
Add a SCORE_NONE, and fix a typo.
[remoteglot] / www / js / remoteglot.js
index 486ca8de6c1bb45c755de95a6de78d62ff38d831..21a995eea87a4073c7a92062ab8964e069baf340 100644 (file)
@@ -158,10 +158,12 @@ var display_fen = null;
  *    pretty_pv: Array.<string>,
  *    move_num: number,
  *    toplay: string,
+ *    score: string,
  *    start_display_move_num: number
  * }} DisplayLine
  *
  * "start_display_move_num" is the (half-)move number to start displaying the PV at.
+ * "score" is also evaluated at this point.
  */
 
 /** All PVs that we currently know of.
@@ -546,7 +548,8 @@ var create_arrow = function(from_square, to_square, fg_color, line_width, arrow_
        arrows.push(arrow);
 }
 
-var compare_by_sort_key = function(refutation_lines, a, b) {
+// Note: invert is ignored.
+var compare_by_sort_key = function(refutation_lines, invert, a, b) {
        var ska = refutation_lines[a]['sort_key'];
        var skb = refutation_lines[b]['sort_key'];
        if (ska < skb) return -1;
@@ -554,9 +557,9 @@ var compare_by_sort_key = function(refutation_lines, a, b) {
        return 0;
 };
 
-var compare_by_score = function(refutation_lines, a, b) {
-       var sa = parseInt(refutation_lines[b]['score_sort_key'], 10);
-       var sb = parseInt(refutation_lines[a]['score_sort_key'], 10);
+var compare_by_score = function(refutation_lines, invert, a, b) {
+       var sa = compute_score_sort_key(refutation_lines[b]['score'], invert);
+       var sb = compute_score_sort_key(refutation_lines[a]['score'], invert);
        return sa - sb;
 }
 
@@ -567,17 +570,18 @@ var compare_by_score = function(refutation_lines, a, b) {
  * @param {!Object} data
  * @param {number} margin The maximum number of centipawns worse than the
  *     best move can be and still be included.
+ * @param {boolean} margin Whether black is to play.
  * @return {Array.<string>} The UCI representation (e.g. e1g1) of all
  *     moves, in score order.
  */
-var find_nonstupid_moves = function(data, margin) {
+var find_nonstupid_moves = function(data, margin, invert) {
        // First of all, if there are any moves that are more than 0.5 ahead of
        // the primary move, the refutation lines are probably bunk, so just
        // kill them all. 
        var best_score = undefined;
        var pv_score = undefined;
        for (var move in data['refutation_lines']) {
-               var score = parseInt(data['refutation_lines'][move]['score_sort_key'], 10);
+               var score = compute_score_sort_key(data['refutation_lines'][move]['score'], invert);
                if (move == data['pv_uci'][0]) {
                        pv_score = score;
                }
@@ -597,12 +601,12 @@ var find_nonstupid_moves = function(data, margin) {
        // The PV move will always be first.
        var moves = [];
        for (var move in data['refutation_lines']) {
-               var score = parseInt(data['refutation_lines'][move]['score_sort_key'], 10);
+               var score = compute_score_sort_key(data['refutation_lines'][move]['score'], invert);
                if (move != data['pv_uci'][0] && best_score - score <= margin) {
                        moves.push(move);
                }
        }
-       moves = moves.sort(function(a, b) { return compare_by_score(data['refutation_lines'], a, b) });
+       moves = moves.sort(function(a, b) { return compare_by_score(data['refutation_lines'], data['position']['toplay'] === 'B', a, b) });
        moves.unshift(data['pv_uci'][0]);
 
        return moves;
@@ -621,17 +625,19 @@ var thousands = function(x) {
  * @param {Array.<string>} pretty_pv
  * @param {number} move_num
  * @param {!string} toplay
+ * @param {!string} score
  * @param {number} start_display_move_num
  * @param {number=} opt_limit
  * @param {boolean=} opt_showlast
  */
-var add_pv = function(start_fen, pretty_pv, move_num, toplay, start_display_move_num, opt_limit, opt_showlast) {
+var add_pv = function(start_fen, pretty_pv, move_num, toplay, score, start_display_move_num, opt_limit, opt_showlast) {
        display_lines.push({
                start_fen: start_fen,
                pretty_pv: pretty_pv,
                move_num: parseInt(move_num),
                toplay: toplay,
-               start_display_move_num: start_display_move_num,
+               score: score,
+               start_display_move_num: start_display_move_num
        });
        return print_pv(display_lines.length - 1, opt_limit, opt_showlast);
 }
@@ -765,8 +771,13 @@ var update_refutation_lines = function() {
        for (var move in refutation_lines) {
                moves.push(move);
        }
+
+       var invert = (toplay === 'B');
+       if (current_display_line && current_display_move % 2 == 0) {
+               invert = !invert;
+       }
        var compare = sort_refutation_lines_by_score ? compare_by_score : compare_by_sort_key;
-       moves = moves.sort(function(a, b) { return compare(refutation_lines, a, b) });
+       moves = moves.sort(function(a, b) { return compare(refutation_lines, invert, a, b) });
        for (var i = 0; i < moves.length; ++i) {
                var line = refutation_lines[moves[i]];
 
@@ -794,7 +805,7 @@ var update_refutation_lines = function() {
                        var pv_td = document.createElement("td");
                        tr.appendChild(pv_td);
                        $(pv_td).addClass("pv");
-                       $(pv_td).html(add_pv(base_fen, base_line.concat([ line['pretty_move'] ]), move_num, toplay, start_display_move_num));
+                       $(pv_td).html(add_pv(base_fen, base_line.concat([ line['pretty_move'] ]), move_num, toplay, line['score'], start_display_move_num));
 
                        tbl.append(tr);
                        continue;
@@ -806,7 +817,7 @@ var update_refutation_lines = function() {
                var score_td = document.createElement("td");
                tr.appendChild(score_td);
                $(score_td).addClass("score");
-               $(score_td).text(line['pretty_score']);
+               $(score_td).text(format_short_score(line['score']));
 
                var depth_td = document.createElement("td");
                tr.appendChild(depth_td);
@@ -816,7 +827,7 @@ var update_refutation_lines = function() {
                var pv_td = document.createElement("td");
                tr.appendChild(pv_td);
                $(pv_td).addClass("pv");
-               $(pv_td).html(add_pv(base_fen, base_line.concat(line['pv_pretty']), move_num, toplay, start_display_move_num, 10));
+               $(pv_td).html(add_pv(base_fen, base_line.concat(line['pv_pretty']), move_num, toplay, line['score'], start_display_move_num, 10));
 
                tbl.append(tr);
        }
@@ -916,7 +927,7 @@ var update_board = function() {
        // unconditionally taken from current_data (we're not interested in
        // historic history).
        if (current_data['position']['pretty_history']) {
-               add_pv('start', current_data['position']['pretty_history'], 1, 'W', 0, 8, true);
+               add_pv('start', current_data['position']['pretty_history'], 1, 'W', null, 0, 8, true);
        } else {
                display_lines.push(null);
        }
@@ -1008,8 +1019,8 @@ var update_board = function() {
 
        // The <title> contains a very brief headline.
        var title_elems = [];
-       if (data['short_score'] !== undefined && data['short_score'] !== null) {
-               title_elems.push(data['short_score'].replace(/^ /, ""));
+       if (data['score']) {
+               title_elems.push(format_short_score(data['score']).replace(/^ /, ""));
        }
        if (last_move !== null) {
                title_elems.push(last_move);
@@ -1058,8 +1069,14 @@ var update_board = function() {
        update_clock();
 
        // The score.
-       if (data['score'] !== null) {
-               $("#score").text(data['score']);
+       if (current_display_line) {
+               if (current_display_line.score) {
+                       $("#score").text(format_long_score(current_display_line.score));
+               } else {
+                       $("#score").text("No score for this move");
+               }
+       } else if (data['score']) {
+               $("#score").text(format_long_score(data['score']));
        }
 
        // The search stats.
@@ -1091,7 +1108,7 @@ var update_board = function() {
 
        // Print the PV.
        $("#pvtitle").text("PV:");
-       $("#pv").html(add_pv(data['position']['fen'], data['pv_pretty'], data['position']['move_num'], data['position']['toplay'], 0));
+       $("#pv").html(add_pv(data['position']['fen'], data['pv_pretty'], data['position']['move_num'], data['position']['toplay'], data['score'], 0));
 
        // Update the PV arrow.
        clear_arrows();
@@ -1107,7 +1124,7 @@ var update_board = function() {
                        create_arrow(from, to, '#f66', 6, 20);
                }
 
-               var alt_moves = find_nonstupid_moves(data, 30);
+               var alt_moves = find_nonstupid_moves(data, 30, data['position']['toplay'] === 'B');
                for (var i = 1; i < alt_moves.length && i < 3; ++i) {
                        create_arrow(alt_moves[i].substr(0, 2),
                                     alt_moves[i].substr(2, 2), '#f66', 1, 10);
@@ -1116,7 +1133,7 @@ var update_board = function() {
 
        // See if all semi-reasonable moves have only one possible response.
        if (data['pv_uci'].length >= 2) {
-               var nonstupid_moves = find_nonstupid_moves(data, 300);
+               var nonstupid_moves = find_nonstupid_moves(data, 300, data['position']['toplay'] === 'B');
                var response = data['pv_uci'][1];
                for (var i = 0; i < nonstupid_moves.length; ++i) {
                        if (nonstupid_moves[i] == data['pv_uci'][0]) {
@@ -1187,15 +1204,15 @@ var update_sparkline = function(data) {
                        var scores = [];
                        for (var halfmove_num = first_move_num; halfmove_num <= last_move_num; ++halfmove_num) {
                                if (data['score_history'][halfmove_num]) {
-                                       var score = data['score_history'][halfmove_num][0];
+                                       var score = compute_plot_score(data['score_history'][halfmove_num]);
+                                       last_score = score;
                                        if (score < min_score) min_score = score;
                                        if (score > max_score) max_score = score;
-                                       last_score = data['score_history'][halfmove_num][0];
                                }
                                scores.push(last_score);
                        }
-                       if (data['plot_score']) {
-                               scores.push(data['plot_score']);
+                       if (data['score']) {
+                               scores.push(compute_plot_score(data['score']));
                        }
                        // FIXME: at some widths, calling sparkline() seems to push
                        // #scorecontainer under the board.
@@ -1394,10 +1411,10 @@ var format_tooltip = function(data, halfmove_num) {
                var short_score;
                if (halfmove_num === data['position']['pretty_history'].length) {
                        move = data['position']['last_move'];
-                       short_score = data['short_score'];
+                       short_score = format_short_score(data['score']);
                } else {
                        move = data['position']['pretty_history'][halfmove_num];
-                       short_score = data['score_history'][halfmove_num][1];
+                       short_score = format_short_score(data['score_history'][halfmove_num]);
                }
                var move_with_number = format_halfmove_with_number(move, halfmove_num);
 
@@ -1559,28 +1576,7 @@ var update_move_highlight = function() {
                highlighted_move.removeClass('highlight'); 
        }
        if (current_display_line) {
-               // See if the current displayed line is identical to any of the ones
-               // we have on screen. (It might not be if e.g. the analysis reloaded
-               // since we started looking.)
-               var display_line_num = null;
-               for (var i = 0; i < display_lines.length; ++i) {
-                       var line = display_lines[i];
-                       if (line.start_display_move_num > 0) continue;
-                       if (current_display_line.start_fen !== line.start_fen) continue;
-                       if (current_display_line.pretty_pv.length !== line.pretty_pv.length) continue;
-                       var ok = true;
-                       for (var j = 0; j < line.pretty_pv.length; ++j) {
-                               if (current_display_line.pretty_pv[j] !== line.pretty_pv[j]) {
-                                       ok = false;
-                                       break;
-                               }
-                       }
-                       if (ok) {
-                               display_line_num = i;
-                               break;
-                       }
-               }
-
+               var display_line_num = find_display_line_matching_num();
                if (display_line_num === null) {
                        // Replace the PV with the (complete) line.
                        $("#pvtitle").text("Exploring:");
@@ -1595,6 +1591,33 @@ var update_move_highlight = function() {
        }
 }
 
+/**
+ * See if the current displayed line is identical to any of the ones
+ * we have on screen. (It might not be if e.g. the analysis reloaded
+ * since we started looking.)
+ *
+ * @return {?number}
+ */
+var find_display_line_matching_num = function() {
+       for (var i = 0; i < display_lines.length; ++i) {
+               var line = display_lines[i];
+               if (line.start_display_move_num > 0) continue;
+               if (current_display_line.start_fen !== line.start_fen) continue;
+               if (current_display_line.pretty_pv.length !== line.pretty_pv.length) continue;
+               var ok = true;
+               for (var j = 0; j < line.pretty_pv.length; ++j) {
+                       if (current_display_line.pretty_pv[j] !== line.pretty_pv[j]) {
+                               ok = false;
+                               break;
+                       }
+               }
+               if (ok) {
+                       return i;
+               }
+       }
+       return null;
+}
+
 var update_displayed_line = function() {
        if (current_display_line === null) {
                $("#linenav").hide();
@@ -1672,6 +1695,7 @@ var explore_hash = function(fen) {
                clearTimeout(current_hash_display_timer);
                current_hash_display_timer = null;
        }
+       $("#refutationlines").empty();
        current_hash_xhr = $.ajax({
                url: backend_hash_url + "?fen=" + fen
        }).done(function(data, textstatus, xhr) {
@@ -1704,7 +1728,7 @@ var onDragStart = function(source, piece, position, orientation) {
                return false;
        }
 
-       recommended_move = get_best_move(pseudogame, source, null);
+       recommended_move = get_best_move(pseudogame, source, null, pseudogame.turn() === 'b');
        if (recommended_move) {
                var squareEl = $('#board .square-' + recommended_move.to);
                squareEl.addClass('highlight1-32417');
@@ -1728,7 +1752,7 @@ var mousedownSquare = function(e) {
            (pseudogame.turn() === 'w' && position[square].search(/^b/) !== -1) ||
            (pseudogame.turn() === 'b' && position[square].search(/^w/) !== -1)) {
                reverse_dragging_from = square;
-               recommended_move = get_best_move(pseudogame, null, square);
+               recommended_move = get_best_move(pseudogame, null, square, pseudogame.turn() === 'b');
                if (recommended_move) {
                        var squareEl = $('#board .square-' + recommended_move.from);
                        squareEl.addClass('highlight1-32417');
@@ -1751,7 +1775,7 @@ var mouseupSquare = function(e) {
        $("#board").find('.square-55d63').removeClass('highlight1-32417');
 }
 
-var get_best_move = function(game, source, target) {
+var get_best_move = function(game, source, target, invert) {
        var moves = game.moves({ verbose: true });
        if (source !== null) {
                moves = moves.filter(function(move) { return move.from == source; });
@@ -1773,20 +1797,29 @@ var get_best_move = function(game, source, target) {
                move_hash[moves[i].san] = moves[i];
        }
 
+       // History and PV take priority over the display lines.
+       for (var i = 0; i < 2; ++i) {
+               var line = display_lines[i];
+               var first_move = line.pretty_pv[line.start_display_move_num];
+               if (move_hash[first_move]) {
+                       return move_hash[first_move];
+               }
+       }
+
        var best_move = null;
        var best_move_score = null;
 
        for (var move in refutation_lines) {
                var line = refutation_lines[move];
-               var score = parseInt(line['score_sort_key'], 10);
-               if (score < -1000000) {  // Two zeros less than in the server (just some margin).
+               if (!line['score']) {
                        continue;
                }
                var first_move = line['pv_pretty'][0];
                if (move_hash[first_move]) {
-                       if (best_move_score === null || line['score_sort_key'] > best_move_score) {
+                       var score = compute_score_sort_key(line['score'], invert);
+                       if (best_move_score === null || score > best_move_score) {
                                best_move = move_hash[first_move];
-                               best_move_score = line['score_sort_key'];
+                               best_move_score = score;
                        }
                }
        }
@@ -1833,7 +1866,7 @@ var onSnapEnd = function(source, target) {
 
        if (current_display_line &&
            current_display_move < current_display_line.pretty_pv.length - 1 &&
-           current_display_line.pretty_pv[current_display_move] === move.san) {
+           current_display_line.pretty_pv[current_display_move + 1] === move.san) {
                next_move();
                return;
        }
@@ -1854,6 +1887,111 @@ var onSnapEnd = function(source, target) {
 }
 // End of dragging-related code.
 
+var pad = function(val, num_digits) {
+       var s = val.toString();
+       while (s.length < num_digits) {
+               s = " " + s;
+       }
+       return s;
+}
+
+var fmt_cp = function(v) {
+       if (v === 0) {
+               return "0.00";
+       } else if (v > 0) {
+               return "+" + (v / 100).toFixed(2);
+       } else {
+               v = -v;
+               return "-" + (v / 100).toFixed(2);
+       }
+}
+
+var format_short_score = function(score) {
+       if (!score) {
+               return "???";
+       }
+       if (score[0] === 'm') {
+               if (score[2]) {  // Is a bound.
+                       return score[2] + "\u00a0M" + pad(score[1], 3);
+               } else {
+                       return "M" + pad(score[1], 3);
+               }
+       } else if (score[0] === 'd') {
+               return "TB draw";
+       } else if (score[0] === 'cp') {
+               if (score[2]) {  // Is a bound.
+                       return score[2] + "\u00a0" + fmt_cp(score[1]);
+               } else {
+                       return pad(fmt_cp(score[1]), 5);
+               }
+       }
+       return null;
+}
+
+var format_long_score = function(score) {
+       if (score[0] === 'm') {
+               if (score[1] > 0) {
+                       return "White mates in " + score[1];
+               } else {
+                       return "Black mates in " + (-score[1]);
+               }
+       } else if (score[0] === 'd') {
+               return "Theoretical draw";
+       } else if (score[0] === 'cp') {
+               return "Score: " + format_short_score(score);
+       }
+       return null;
+}
+
+var compute_plot_score = function(score) {
+       if (score[0] === 'm') {
+               if (score[1] > 0) {
+                       return 500;
+               } else {
+                       return -500;
+               }
+       } else if (score[0] === 'd') {
+               return 0;
+       } else if (score[0] === 'cp') {
+               if (score[1] > 500) {
+                       return 500;
+               } else if (score[1] < -500) {
+                       return -500;
+               } else {
+                       return score[1];
+               }
+       }
+       return null;
+}
+
+/**
+ * @param score The score digest tuple.
+ * @param {boolean} invert Whether black is to play.
+ * @return {number}
+ */
+var compute_score_sort_key = function(score, invert) {
+       var s;
+       if (!score) {
+               return -10000000;
+       }
+       if (score[0] === 'm') {
+               if (score[1] > 0) {
+                       // White mates.
+                       s = 99999 - score[1];
+               } else {
+                       // Black mates (note the double negative for score[1]).
+                       s = -99999 - score[1];
+               }
+               if (invert) s = -s;
+               return s;
+       } else if (score[0] === 'd') {
+               return 0;
+       } else if (score[0] === 'cp') {
+               return invert ? -score[1] : score[1];
+       }
+       return null;
+}
+
 /**
  * @param {string} new_backend_url
  */