[remoteglot-book] / www / js / book.js

(function() {

var board = null;
var game = new Chess();
var fens = [];
var move_override = 0;
var includetransp = true;

var entity_map = {
        "&": "&",
        "<": "&lt;",
        ">": "&gt;",
        '"': '&quot;',
        "'": '&#39;',
};

function escape_html(string) {
        return String(string).replace(/[&<>"']/g, function (s) {
                return entity_map[s];
        });
}

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];
        }
}

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].san + '</strong>';
                } else {
                        text += '<a href="javascript:set_move(' + (i + 1) + ')">' + history[i].san + '</a>';
                }
                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');
        }

        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 fen = current_display_fen();
        $.ajax({
                url: "/opening-stats.pl?fen=" + encodeURIComponent(fen) +
                        ";includetransp=" + (includetransp ? 1 : 0)
        }).done(function(data, textstatus, xhr) {
                show_lines(data, game);
        });
}

var add_td = function(tr, value) {
        var td = document.createElement("td");
        tr.appendChild(td);
        $(td).addClass("num");
        $(td).text(value);
}

var TYPE_MOVE = 0;
var TYPE_INTEGER = 1;
var TYPE_FLOAT = 2;
var TYPE_RATIO = 3;

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 ],
        [ "Bwin", TYPE_INTEGER ],
        [ "%BW", TYPE_RATIO ],
        [ "Draw", TYPE_INTEGER ],
        [ "Draw%", TYPE_RATIO ],
        [ "AvWElo", TYPE_FLOAT ],
        [ "AvBElo", TYPE_FLOAT ],
        [ "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']);

        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 += " &ndash; " + escape_html(data['root_game']['black']);
                if (data['root_game']['black_elo']) {
                        text += " (" + escape_html(data['root_game']['black_elo']) + ")";
                }
                text += " &nbsp;" + escape_html(data['root_game']['result']).replace(/-/, "&ndash;") + "<br />";
                if (data['root_game']['eco']) {
                        text += "[" + escape_html(data['root_game']['eco']) + "] ";
                }
                text += "(" + data['root_game']['moves'] + ") ";
                text += escape_html(data['root_game']['event']) + " &nbsp;" + escape_html(data['root_game']['date']);
                $('#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 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']);
                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 = (move_override % 2 == 0) ? 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.

                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);
                }
                lines.push(line);
        }

        lines.sort(function(a, b) { return direction * ( b[sort_by] - a[sort_by]); });

        var tbl = $("#lines");
        tbl.empty();

        for (var i = 0; i < moves.length; ++i) {
                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, "");
                        } else if (headings[j][1] == TYPE_MOVE) {
                                var td = document.createElement("td");
                                tr.appendChild(td);
                                $(td).addClass("move");
                                if (line[j] !== undefined) {
                                        if (move_override % 2 == 0) {
                                                $(td).text(((move_override / 2) + 1) + ". ");
                                        } else {
                                                $(td).text(((move_override / 2) + 0.5) + "…");
                                        }
                                }

                                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) {
                                if (isNaN(line[j]) || !isFinite(line[j])) {
                                        add_td(tr, '');
                                } else {
                                        add_td(tr, line[j].toFixed(1));
                                }
                        } else {
                                if (isNaN(line[j]) || !isFinite(line[j])) {
                                        add_td(tr, '');
                                } else {
                                        add_td(tr, (100.0 * line[j]).toFixed(1) + "%");
                                }
                        }
                }

                tbl.append(tr);
        }
}

var set_includetransp = function(value) {
        includetransp = value;
        update();
}
window['set_includetransp'] = set_includetransp;

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;
                update();
                window.history.replaceState(null, null, get_history_url());
        }
}
window['prev_move'] = prev_move;

var next_move = function() {
        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 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;
        }
}

var onDrop = function(source, target) {
        // see if the move is legal
        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
        });

        // illegal move
        if (move === null) return 'snapback';
}

var onSnapEnd = function(source, target) {
        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 = [];
        }

        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.
        board = new window.ChessBoard('board', {
                draggable: true,
                position: 'start',
                onDragStart: onDragStart,
                onDrop: onDrop,
                onSnapEnd: onSnapEnd
        });

        window.onpopstate = onpopstate;
        onpopstate();

        $(window).keyup(function(event) {
                if (event.which == 39) {
                        next_move();
                } else if (event.which == 37) {
                        prev_move();
                }
        });
}


$(document).ready(init);

})();