X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=www%2Fjs%2Fbook.js;h=aacd98030c516bef45fa7bd0d380884e16f90674;hb=b6effd41313938e70b383f9d6fa61ff53806cf56;hp=f6b0365993500449cc304a221a71c6ea39bc94a8;hpb=e6e40d99dbc3c1cab0c4f98d677e72220d0567e1;p=remoteglot-book diff --git a/www/js/book.js b/www/js/book.js index f6b0365..aacd980 100644 --- a/www/js/book.js +++ b/www/js/book.js @@ -1,32 +1,242 @@ (function() { var board = null; -var moves = []; +var game = new Chess(); +var fens = []; // Position after each. 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 practice_mode = false; +var practice_side = 'W'; + +// TODO: Make this configurable. +var practice_top_moves_limit = 5; +var practice_minimum_move_fraction_start = 0.05; +var practice_minimum_move_fraction_move5 = 0.30; + +var entity_map = { + "&": "&", + "<": "<", + ">": ">", + '"': '"', + "'": ''', +}; + +function escape_html(string) { + return String(string).replace(/[&<>"']/g, function (s) { + return entity_map[s]; + }); +} + +var current_display_fen = function() { + return fen_before_move(move_override); +} -var get_game = function() { - var game = new Chess(); - for (var i = 0; i < move_override; ++i) { - game.move(moves[i]); +var fen_before_move = function(move_num) { + if (move_num == 0) { + return 'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1'; + } else { + return fens[move_num - 1]; } - return game; } var update = function() { - var game = get_game(); - board.position(game.fen()); + 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 += '' + history[i].san + ''; + } else { + text += '' + history[i].san + ''; + } + text += " "; + } + $('#gamehistory').html(text); + + if (board.fen() != current_display_fen()) { + 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-' + highlight_from).addClass('nonuglyhighlight'); + $("#board").find('.square-' + highlight_to).addClass('nonuglyhighlight'); + } + + if (practice_mode) { + find_last_move_score(); + var side_to_move = (move_override % 2 == 0) ? 'W' : 'B'; + if (side_to_move !== practice_side) { + find_computer_move(); + } + // Fall through to get the line name and such. + } + 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 = current_display_fen(); $.ajax({ - url: "/opening-stats.pl?fen=" + encodeURIComponent(game.fen()) + url: "/opening-stats.pl?fen=" + encodeURIComponent(fen) + + ";includetransp=" + (includetransp ? 1 : 0) }).done(function(data, textstatus, xhr) { show_lines(data, game); }); } +var find_last_move_score = function() { + var history = game.history({ verbose: true }); + var side_to_move = (move_override % 2 == 0) ? 'W' : 'B'; + var move_num = (side_to_move === practice_side) ? (move_override - 2) : (move_override - 1); + if (move_num < 0) { + $("#yourmove").text("(none)"); + $("#yourfraction").text("N/A"); + $("#yourrank").text("N/A"); + $("#yourawin").text("??.?%"); + $("#yourawindiff").text("+?.?%"); + return; + } + + var chosen_move = history[move_num].san; + $("#yourmove").text(chosen_move); + $.ajax({ + url: "/opening-stats.pl?fen=" + encodeURIComponent(fen_before_move(move_num)) + + ";includetransp=0" + }).done(function(data, textstatus, xhr) { + var moves = data['moves']; + var root_move = sort_move_by_frequency(moves, data); + var your_move, your_index; + + for (var i = 0; i < moves.length; ++i) { + var move = moves[i]; + if (move['move'] === chosen_move) { + your_move = move; + your_index = i; + } + } + + if (your_move) { + $("#yourfraction").text(format_fraction(your_move['fraction'])); + $("#yourawin").text(format_fraction(your_move['corrected_win_ratio'])); + $("#yourrank").text(format_ordinal(your_index + 1) + ", based on " + root_move['num'] + " games"); + var diff = your_move['corrected_win_ratio'] - root_move['corrected_win_ratio']; + $("#yourawindiff").css("color", "black"); + if (diff === 0) { + $("#yourawindiff").text("0.0%"); + } else if (diff > 0) { + $("#yourawindiff").text("+" + format_fraction(diff)); + $("#yourawindiff").css("color", "green"); + } else { + $("#yourawindiff").text(format_fraction(diff)); + if (diff < -0.02) { + $("#yourawindiff").css("color", "red"); + } + } + } else { + $("#yourfraction").text("??.?%"); + $("#yourrank").text("?th"); + $("#yourawin").text("??.?%"); + $("#yourawindiff").text("+?.?%"); + } + }); +} + +var candidate_moves = []; +var chosen_index = null; + +var find_computer_move = function() { + var fen = current_display_fen(); + $.ajax({ + url: "/opening-stats.pl?fen=" + encodeURIComponent(fen) + ";includetransp=0" + }).done(function(data, textstatus, xhr) { + candidate_moves = []; + + var moves = data['moves']; + var root_move = sort_move_by_frequency(moves, data); + + var practice_minimum_move_fraction; + if (move_override > 20) { + practice_minimum_move_fraction = practice_minimum_move_fraction_move5; + } else { + practice_minimum_move_fraction = practice_minimum_move_fraction_start + + ((move_override-1)/10.0) * (practice_minimum_move_fraction_move5 - practice_minimum_move_fraction_start); + } + console.log(practice_minimum_move_fraction); + + for (var i = 0; i < Math.min(moves.length, practice_top_moves_limit); ++i) { + var move = moves[i]; + if (i == 0 || move['fraction'] >= practice_minimum_move_fraction) { + candidate_moves.push(move); + } + } + + // Pick one at random. + choose_move(Math.floor(Math.random() * candidate_moves.length)); + }); +} + +var choose_move = function(idx) { + chosen_index = idx; + var chosen = candidate_moves[chosen_index]; + $("#compmove").text(chosen['move']); + $("#compfraction").text((100.0 * chosen['fraction']).toFixed(1) + "%"); + if (candidate_moves.length == 1) { + $("#comprank").text("only candidate move"); + } else { + $("#comprank").text(format_ordinal(chosen_index + 1) + " out of " + candidate_moves.length + " candidate moves, j/k to switch"); + } + make_move(chosen['move']); +} + +var prev_variant = function() { + if (chosen_index !== null) { + --move_override; + choose_move((chosen_index + candidate_moves.length - 1) % candidate_moves.length); + } +} + +var next_variant = function() { + if (chosen_index !== null) { + --move_override; + choose_move((chosen_index + 1) % candidate_moves.length); + } +} + +// Add deried data and then sort moves to get the most common ones (in-place). +// Remove the root mode and return it. Currently usable for practice mode only! +var sort_move_by_frequency = function(moves, data) +{ + var total_num = find_total_games(moves); + var root_move; + for (var i = 0; i < moves.length; ++i) { + var move = moves[i]; + calc_move_derived_data(move, total_num, data, (practice_side === 'W')); + if (!move['move']) { + root_move = (moves.splice(i, 1))[0]; + --i; + } + } + moves.sort(function(a, b) { return b['num'] - a['num'] }); + return root_move; +} + var add_td = function(tr, value) { var td = document.createElement("td"); tr.appendChild(td); @@ -34,6 +244,23 @@ var add_td = function(tr, value) { $(td).text(value); } +var format_ordinal = function(x) { + var tens = Math.floor(x / 10) % 10; + if (tens == 1) { + return x + "th"; + } else { + var ones = x % 10; + if (ones == 1) return x + "st"; + if (ones == 2) return x + "nd"; + if (ones == 3) return x + "rd"; + return x + "th"; + } +} + +var format_fraction = function(x) { + return (100.0 * x).toFixed(1) + '%'; +} + var TYPE_MOVE = 0; var TYPE_INTEGER = 1; var TYPE_FLOAT = 2; @@ -43,6 +270,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 ], @@ -55,73 +284,97 @@ var headings = [ [ "EloVar", TYPE_FLOAT ], [ "AWin%", TYPE_RATIO ], ]; +var sort_by = 1; +var direction = 1; var show_lines = function(data, game) { var moves = data['moves']; $('#numviewers').text(data['opening']); - var total_num = 0; - for (var i = 0; i < moves.length; ++i) { - var move = moves[i]; - total_num += parseInt(move['white']); - total_num += parseInt(move['draw']); - total_num += parseInt(move['black']); + + if (data['root_game']) { + var text = escape_html(data['root_game']['white']); + if (data['root_game']['white_elo']) { + text += " (" + escape_html(data['root_game']['white_elo']) + ")"; + } + text += " – " + escape_html(data['root_game']['black']); + if (data['root_game']['black_elo']) { + text += " (" + escape_html(data['root_game']['black_elo']) + ")"; + } + text += "  " + escape_html(data['root_game']['result']).replace(/-/, "–") + "
"; + if (data['root_game']['eco']) { + text += "[" + escape_html(data['root_game']['eco']) + "] "; + } + text += "(" + data['root_game']['moves'] + ") "; + text += escape_html(data['root_game']['event']) + "  " + escape_html(data['root_game']['date']); + $('#gamesummary').html(text); } + var total_num = find_total_games(moves); + var headings_tr = $("#headings"); headings_tr.empty(); for (var i = 0; i < headings.length; ++i) { var th = document.createElement("th"); headings_tr.append(th); $(th).text(headings[i][0]); + (function(new_sort_by) { + $(th).click(function() { + if (sort_by == new_sort_by) { + direction = -direction; + } else { + sort_by = new_sort_by; + direction = 1; + } + show_lines(data, game); + }); + })(i); } var lines = []; + var transpose_only = []; for (var i = 0; i < moves.length; ++i) { var move = moves[i]; var line = []; - var white = parseInt(move['white']); - var draw = parseInt(move['draw']); - var black = parseInt(move['black']); + 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. - var num = white + draw + black; - line.push(num); // N. - line.push(num / total_num); // %. - - // Win%. - var white_win_ratio = (white + 0.5 * draw) / num; - var win_ratio = (game.turn() == 'w') ? white_win_ratio : 1.0 - white_win_ratio; - line.push(win_ratio); - - 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. + transpose_only.push(move['transpose_only']); + line.push(move['num']); // N. + line.push(move['fraction']); // %. + line.push(computer); // CGames. + line.push(move['human_index']); // Hum. + line.push(move['win_ratio']); // Win%. + + line.push(white); // WWin. + line.push(white / move['num']); // %WW. + line.push(black); // BWin. + line.push(black / move['num']); // %BW. + line.push(draw); // Draw. + line.push(draw / move['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 = (game.turn() == 'w') ? 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(move['corrected_win_ratio'] || null); lines.push(line); } + lines.sort(function(a, b) { return direction * ( b[sort_by] - a[sort_by]); }); + var tbl = $("#lines"); tbl.empty(); @@ -129,6 +382,12 @@ var show_lines = function(data, game) { var line = lines[i]; var tr = document.createElement("tr"); + if (line[0] === undefined) { + $(tr).addClass("totals"); + } else if (transpose_only[i]) { + $(tr).addClass("transponly"); + } + for (var j = 0; j < line.length; ++j) { if (line[j] === null) { add_td(tr, ""); @@ -136,16 +395,36 @@ var show_lines = function(data, game) { var td = document.createElement("td"); tr.appendChild(td); $(td).addClass("move"); - 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] !== undefined) { + if (move_override % 2 == 0) { + $(td).text(((move_override / 2) + 1) + ". "); + } else { + $(td).text(((move_override / 2) + 0.5) + "…"); + } + } + + 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]); + 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, format_fraction(line[j])); + } } } @@ -153,44 +432,260 @@ var show_lines = function(data, game) { } } -var make_move = function(move) { - moves.length = move_override; - moves.push(move); - move_override = moves.length; +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 white = move['white']; + var draw = move['draw']; + var black = move['black']; + var computer = move['computer']; + + var num = white + draw + black; + move['num'] = num; + move['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']); + move['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; + move['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; + move['corrected_win_ratio'] = win_ratio; + } +}; + +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 set_practice = function(value) { + practice_mode = value; + if (practice_mode) { + practice_side = (move_override % 2 == 0) ? 'W' : 'B'; + find_last_move_score(); + $("#stats").hide(); + $("#practiceoutput").show(); + document.getElementById("includetransp").checked = false; + set_includetransp(false); + } else { + $("#stats").show(); + $("#practiceoutput").hide(); + } + update(); +} +window['set_practice'] = set_practice; + +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 { + 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()); + } +} window['make_move'] = make_move; var prev_move = function() { - if (move_override > 0) { - --move_override; + var moves_to_skip = practice_mode ? 2 : 1; + if (move_override >= moves_to_skip) { + move_override -= moves_to_skip; update(); + window.history.replaceState(null, null, get_history_url()); } } window['prev_move'] = prev_move; var next_move = function() { - if (move_override < moves.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, do_update) { + move_override = n; + 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_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 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 @@ -198,18 +693,46 @@ var onDrop = function(source, target) { // illegal move if (move === null) return 'snapback'; +} - moves = game.history({ verbose: true }); - move_override = moves.length; -}; +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, + to: target, + promotion: 'q' // NOTE: always promote to a queen for example simplicity + }); -// 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()); - fetch_analysis(); -}; + 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 = []; + } + + 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. @@ -220,17 +743,28 @@ var init = function() { onDrop: onDrop, onSnapEnd: onSnapEnd }); - update(); + $("#board").on('mousedown', '.square-55d63', mousedownSquare); + $("#board").on('mouseup', '.square-55d63', mouseupSquare); + + window.onpopstate = onpopstate; + onpopstate(); + set_practice(false); $(window).keyup(function(event) { if (event.which == 39) { next_move(); } else if (event.which == 37) { prev_move(); + } else if (event.which == 74) { // j + if (practice_mode) next_variant(); + } else if (event.which == 75) { // k + if (practice_mode) prev_variant(); } }); -} + // Seemingly the web worker is not started before we send it a message. + stockfish.postMessage("uci"); +} $(document).ready(init);