git.sesse.net Git - remoteglot-book/blob - www/opening-stats.pl

   1 #! /usr/bin/perl
   2 use strict;
   3 use warnings;
   4 use CGI;
   5 use JSON::XS;
   6 use lib '..';
   7 use Position;
   8 use IPC::Open2;
   9 use Chess::PGN::Parse;
  10
  11 our %openings = ();
  12 read_openings();
  13
  14 my $cgi = CGI->new;
  15 my ($chld_out, $chld_in);
  16 my $pid = IPC::Open2::open2($chld_out, $chld_in, "../binlookup", "../open.mtbl", "40");
  17
  18 # Root position. Basically ignore everything except the opening (and later some root game stuff).
  19 my $fen = $cgi->param('fen') // 'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1';
  20 my $includetransp = $cgi->param('includetransp') // 1;
  21
  22 my $pos = Position->from_fen($fen);
  23 my ($json_root_pos, $root_aux_data) = get_json_move($pos, undef, $chld_in, $chld_out);
  24
  25 my $opening = $openings{$json_root_pos->{'opening_num'}} // 'A00: Start position';
  26 my @json_moves = ($json_root_pos);
  27
  28 my $root_game;
  29 eval {
  30         if (!exists($root_aux_data->{'pgn_file_number'}) ||
  31             !exists($root_aux_data->{'pgn_start_position'})) {
  32                 die "Missing PGN position data."
  33         }
  34         my $pgntext = read_root_pgn($root_aux_data->{'pgn_file_number'}, $root_aux_data->{'pgn_start_position'});
  35         my $pgn = Chess::PGN::Parse->new(undef, $pgntext);
  36         $pgn->read_game() or die;
  37         $pgn->parse_game() or die;
  38
  39         my $tags = $pgn->tags;
  40         $root_game = {};
  41         $root_game->{'white'} = $pgn->white;
  42         $root_game->{'white_elo'} = $tags->{'WhiteElo'};
  43         $root_game->{'black'} = $pgn->black;
  44         $root_game->{'black_elo'} = $tags->{'BlackElo'};
  45         $root_game->{'event'} = $pgn->event;
  46         $root_game->{'date'} = $pgn->date;
  47         $root_game->{'result'} = $pgn->result;
  48         $root_game->{'eco'} = $pgn->eco;
  49         $root_game->{'moves'} = int(((scalar @{$pgn->moves}) + 1) / 2);
  50 };
  51 if ($@) {
  52        print STDERR "Error while getting root move: $@\n";
  53 }
  54
  55 # Explore one move out.
  56 my $white_left = $json_root_pos->{'white'};
  57 my $draw_left = $json_root_pos->{'draw'};
  58 my $black_left = $json_root_pos->{'black'};
  59 for my $move (@{$root_aux_data->{'moves'}}) {
  60         my ($np, $uci_move) = $pos->make_pretty_move($move);
  61         my $json_pos;
  62
  63         my ($json_pos_only_this_root, undef) = get_json_move($np, $root_aux_data->{'pos_hash'}, $chld_in, $chld_out);
  64         $white_left -= $json_pos_only_this_root->{'white'};
  65         $draw_left -= $json_pos_only_this_root->{'draw'};
  66         $black_left -= $json_pos_only_this_root->{'black'};
  67
  68         if ($includetransp) {
  69                 ($json_pos, undef) = get_json_move($np, undef, $chld_in, $chld_out);
  70
  71                 # See if this move exists only due to transpositions.
  72                 if (!defined($json_pos_only_this_root)) {
  73                         $json_pos->{'transpose_only'} = 1;
  74                 }
  75         } else {
  76                 $json_pos = $json_pos_only_this_root;
  77         }
  78         $json_pos->{'move'} = $move;
  79         push @json_moves, $json_pos;
  80 }
  81
  82 # If there are any positions that are not accounted for by any moves,
  83 # these have to be games that end here. Add them as pseudo-moves so as
  84 # not to confuse the user.
  85 for my $result (['1-0', 'white', $white_left], ['1/2-1/2', 'draw', $draw_left], ['0-1', 'black', $black_left]) {
  86         next if ($result->[2] == 0);
  87         my $move = {
  88                 move => $result->[0],
  89                 white => 0,
  90                 draw => 0,
  91                 black => 0
  92         };
  93         $move->{$result->[1]} = $result->[2];
  94         push @json_moves, $move;
  95 }
  96
  97 # Get stats for the root position, for the human index.
  98 my $start_pos = Position->start_pos("white", "black");
  99 my ($json_start_pos, undef) = get_json_move($start_pos, 0, $chld_in, $chld_out);
 100 my $total_games = $json_start_pos->{'white'} + $json_start_pos->{'draw'} + $json_start_pos->{'black'};
 101 my $computer_games = $json_start_pos->{'computer'} * 1;
 102
 103 print $cgi->header(-type=>'application/json');
 104 print JSON::XS::encode_json({
 105         moves => \@json_moves,
 106         opening => $opening,
 107         root_game => $root_game,
 108         total_games => $total_games,
 109         computer_games => $computer_games
 110 });
 111
 112 sub read_openings {
 113         open my $fh, "../openings.txt"
 114                 or die "../openings.txt: $!";
 115         for my $line (<$fh>) {
 116                 chomp $line;
 117                 my ($hash, $eco, $opening, $variation, $subvariation) = split /\t/, $line;
 118                 if ($variation eq '') {
 119                         $openings{$hash} = $eco . ": " . $opening;
 120                 } else {
 121                         $openings{$hash} = $eco . ": " . $opening . ": " . $variation;
 122                 }
 123         }
 124         close $fh;
 125 }
 126
 127 sub read_root_pgn {
 128         my ($pgn_file_number, $pgn_start_position) = @_;
 129         my @pgnnames;
 130         open my $pgnnamesfh, "<", "../pgnnames.txt"
 131                 or die "../pgnnames.txt: $!";
 132         while (<$pgnnamesfh>) {
 133                 chomp;
 134                 s/^comp://;
 135                 push @pgnnames, $_;
 136         }
 137         close $pgnnamesfh;
 138
 139         if ($pgn_file_number > $#pgnnames) {
 140                 die "Unknown PGN file number $pgn_file_number";
 141         }
 142
 143         my $root_pgn;
 144         open my $pgnfh, "<", "../" . $pgnnames[$pgn_file_number]
 145                 or die $pgnnames[$pgn_file_number] . ": $!";
 146         sysseek($pgnfh, $pgn_start_position, 0)
 147                 or die "Could not seek to $pgn_start_position: $!";
 148         sysread($pgnfh, $root_pgn, 32768)
 149                 or die "Could not read PGN from $pgn_start_position at $pgnnames[$pgn_file_number]: $!";
 150         close $pgnfh;
 151         $root_pgn =~ s/^.*?(\[Event )/$1/s;
 152         $root_pgn =~ s/^(.+?)\[Event .*/$1/s;
 153
 154         return $root_pgn;
 155 }
 156
 157 sub get_json_move {
 158         my ($pos, $filter_prev_pos_hash, $chld_in, $chld_out) = @_;
 159         my $bpfen_hex = unpack('H*', $pos->bitpacked_fen);
 160         my $prev_pos_hash_hex = '';
 161         if (defined($filter_prev_pos_hash)) {
 162                 $prev_pos_hash_hex .= unpack('H*', pack('S', $filter_prev_pos_hash));
 163         }
 164         print $chld_in $bpfen_hex, "\n", $prev_pos_hash_hex, "\n";
 165
 166         # Read the hash of this position.
 167         chomp (my $pos_hash = <$chld_out>);
 168
 169         chomp (my $line = <$chld_out>);
 170         if ($line eq '-') {
 171                 warn "Missing pos '" . $pos->fen . "' " . $filter_prev_pos_hash;
 172                 return (undef, undef);
 173         }
 174
 175         my ($white, $draw, $black, $computer, $opening_num, $white_sum_elo, $black_sum_elo, $num_elo, $timestamp, $pgn_file_number, $pgn_start_position, @moves) = split / /, $line;
 176         my $json_pos = {
 177                 white => int($white),
 178                 draw => int($draw),
 179                 black => int($black),
 180                 computer => int($computer),
 181                 white_avg_elo => $num_elo == 0 ? undef : $white_sum_elo / $num_elo,
 182                 black_avg_elo => $num_elo == 0 ? undef : $black_sum_elo / $num_elo,
 183                 num_elo => int($num_elo),
 184                 opening_num => $opening_num,  # Keep as string.
 185         };
 186         my $aux_data = {  # Only relevant for the root.
 187                 pos_hash => $pos_hash * 1,
 188                 moves => \@moves,
 189                 pgn_file_number => $pgn_file_number,
 190                 pgn_start_position => $pgn_start_position,
 191         };
 192         return ($json_pos, $aux_data);
 193 }