]> git.sesse.net Git - remoteglot/blobdiff - remoteglot.pl
Print number annotations in the PVs. Not happy, will revert.
[remoteglot] / remoteglot.pl
index 1561ac338d0a615eef815bc0e324ea6e3b736ef2..1b2e49a8d89057d6cbe0fc3edaadf21ee8582747 100755 (executable)
@@ -18,11 +18,17 @@ use warnings;
 
 # Configuration
 my $server = "freechess.org";
-my $target = "Sesse";
+my $target = "224";
 # my $engine = "/usr/games/toga2";
 my $engine = "wine Rybkav2.3.2a.mp.w32.exe";
 my $telltarget = undef;   # undef to be silent
 my @tell_intervals = (5, 20, 60, 120, 240, 480, 960);  # after each move
+my $uci_assume_full_compliance = 1;                    # dangerous :-)
+my @masters = (
+       'Sesse',
+       'Sessse',
+       'Sesssse'
+);
 
 # Program starts here
 $SIG{ALRM} = sub { output_screen(); };
@@ -46,10 +52,10 @@ select(STDOUT);
 my $pid = IPC::Open2::open2(*UCIREAD, *UCIWRITE, $engine);
 my %uciinfo = ();
 my %uciid = ();
-my %ficsinfo = ();
 my ($last_move, $last_tell);
 my $last_text = '';
 my $last_told_text = '';
+my ($pos_waiting, $pos_calculating);
 
 uciprint("uci");
 
@@ -60,6 +66,7 @@ while (<UCIREAD>) {
 }
 
 uciprint("setoption name UCI_AnalyseMode value true");
+# uciprint("setoption name Preserve Analysis value true");
 uciprint("setoption name NalimovPath value c:\\nalimov");
 uciprint("setoption name NalimovUsage value Rarely");
 uciprint("setoption name Hash value 1024");
@@ -102,11 +109,39 @@ while (1) {
                chomp $line;
                $line =~ tr/\r//d;
                if ($line =~ /^<12> /) {
-                       my $fen = style12_to_fen($line);
-                       uciprint("stop");
-                       uciprint("position fen $fen");
-                       uciprint("go infinite");
+                       my $pos = style12_to_fen($line);
+                       
+                       # if this is already in the queue, ignore it
+                       next if (defined($pos_waiting) && $pos->{'fen'} eq $pos_waiting->{'fen'});
+
+                       # if we're already chewing on this and there's nothing else in the queue,
+                       # also ignore it
+                       next if (!defined($pos_waiting) && defined($pos_calculating) &&
+                                $pos->{'fen'} eq $pos_calculating->{'fen'});
+
+                       # if we're already thinking on something, stop and wait for the engine
+                       # to approve
+                       if (defined($pos_calculating)) {
+                               if (!defined($pos_waiting)) {
+                                       uciprint("stop");
+                               }
+                               if ($uci_assume_full_compliance) {
+                                       $pos_waiting = $pos;
+                               } else {
+                                       uciprint("position fen " . $pos->{'fen'});
+                                       uciprint("go infinite");
+                                       $pos_calculating = $pos;
+                               }
+                       } else {
+                               # it's wrong just to give the FEN (the move history is useful,
+                               # and per the UCI spec, we should really have sent "ucinewgame"),
+                               # but it's easier
+                               uciprint("position fen " . $pos->{'fen'});
+                               uciprint("go infinite");
+                               $pos_calculating = $pos;
+                       }
 
+                       %uciinfo = ();
                        $last_move = time;
 
                        # 
@@ -118,13 +153,48 @@ while (1) {
                        #
                        $t->cmd("date");
                }
+               if ($line =~ /^([A-Za-z]+)(?:\([A-Z]+\))* tells you: (.*)$/) {
+                       my ($who, $msg) = ($1, $2);
+
+                       next if (grep { $_ eq $who } (@masters) == 0);
+       
+                       if ($msg =~ /^fics (.*?)$/) {
+                               $t->cmd("tell $who Executing '$1' on FICS.");
+                               $t->cmd($1);
+                       } elsif ($msg =~ /^uci (.*?)$/) {
+                               $t->cmd("tell $who Sending '$1' to the engine.");
+                               print UCIWRITE "$1\n";
+                       } else {
+                               $t->cmd("tell $who Couldn't understand '$msg', sorry.");
+                       }
+               }
                #print "FICS: [$line]\n";
                $sleep = 0;
        }
        
        # any fun on the UCI channel?
        if ($nfound > 0 && vec($rout, fileno(UCIREAD), 1) == 1) {
-               my $line = <UCIREAD>;
+               # 
+               # Read until we've got a full line -- if the engine sends part of
+               # a line and then stops we're pretty much hosed, but that should
+               # never happen.
+               #
+               my $line = '';
+               while ($line !~ /\n/) {
+                       my $tmp;
+                       my $ret = sysread UCIREAD, $tmp, 1;
+
+                       if (!defined($ret)) {
+                               next if ($!{EINTR});
+                               die "error in reading from the UCI engine: $!";
+                       } elsif ($ret == 0) {
+                               die "EOF from UCI engine";
+                       }
+
+                       $line .= $tmp;
+               }
+
+               $line =~ tr/\r\n//d;
                handle_uci($line);
                $sleep = 0;
 
@@ -140,7 +210,7 @@ sub handle_uci {
 
        chomp $line;
        $line =~ tr/\r//d;
-       print UCILOG "<= $line\n";
+       print UCILOG localtime() . " <= $line\n";
        if ($line =~ /^info/) {
                my (@infos) = split / /, $line;
                shift @infos;
@@ -153,6 +223,15 @@ sub handle_uci {
 
                parse_ids(@ids);
        }
+       if ($line =~ /^bestmove/ && $uci_assume_full_compliance) {
+               if (defined($pos_waiting)) {
+                       uciprint("position fen " . $pos_waiting->{'fen'});
+                       uciprint("go infinite");
+
+                       $pos_calculating = $pos_waiting;
+                       $pos_waiting = undef;
+               }
+       }
 }
 
 sub parse_infos {
@@ -228,11 +307,12 @@ sub parse_ids {
 }
 
 sub style12_to_fen {
-       my $str = shift; 
+       my $str = shift;
+       my %pos = ();
        my (@x) = split / /, $str;
        
-       $ficsinfo{'board'} = [ @x[1..8] ];
-       $ficsinfo{'toplay'} = $x[9];
+       $pos{'board'} = [ @x[1..8] ];
+       $pos{'toplay'} = $x[9];
        
        # the board itself
        my (@board) = @x[1..8];
@@ -266,7 +346,7 @@ sub style12_to_fen {
                } else {
                        $nep .= "6";
                }
-               
+
                #
                # Showing the en passant square when actually no capture can be made
                # seems to confuse at least Rybka. Thus, check if there's actually
@@ -275,11 +355,11 @@ sub style12_to_fen {
                # "right" thing as per the standard, though.
                #
                if ($x[9] eq 'B') {
-                       $ep = $nep if ($col > 0 && substr($board[4], $col-1, 1) eq 'p');
-                       $ep = $nep if ($col < 7 && substr($board[4], $col+1, 1) eq 'p');
+                       $ep = $nep if ($col > 0 && substr($pos{'board'}[4], $col-1, 1) eq 'p');
+                       $ep = $nep if ($col < 7 && substr($pos{'board'}[4], $col+1, 1) eq 'p');
                } else {
-                       $ep = $nep if ($col > 0 && substr($board[3], $col-1, 1) eq 'P');
-                       $ep = $nep if ($col < 7 && substr($board[3], $col+1, 1) eq 'P');
+                       $ep = $nep if ($col > 0 && substr($pos{'board'}[3], $col-1, 1) eq 'P');
+                       $ep = $nep if ($col < 7 && substr($pos{'board'}[3], $col+1, 1) eq 'P');
                }
        }
        $fen .= " ";
@@ -293,16 +373,43 @@ sub style12_to_fen {
        $fen .= " ";
        $fen .= $x[26];
 
-       return $fen;
+       $pos{'fen'} = $fen;
+       $pos{'move_num'} = $x[26];
+
+       return \%pos;
+}
+
+sub prefix_pv {
+       my ($move_num, $toplay) = @_;
+
+       if ($toplay eq 'B') {
+               return "$move_num. ..";
+       } else {
+               return "";
+       }
 }
 
 sub prettyprint_pv {
-       my ($board, @pvs) = @_;
-       
+       my ($board, $move_num, $toplay, @pvs) = @_;
+
+       if (!defined($board) || !defined($move_num) || !defined($toplay)) {
+               die "Missing data";
+       }
        if (scalar @pvs == 0 || !defined($pvs[0])) {
                return ();
        }
-       
+
+       my ($prefix, $next_move, $next_toplay);
+       if ($toplay eq 'W') {
+               $prefix = "$move_num.";
+               $next_move = $move_num;
+               $next_toplay = 'B';
+       } else {
+               $prefix = "";
+               $next_move = $move_num + 1;
+               $next_toplay = 'W';
+       }
+
        my @nb = @$board;
 
        my $pv = shift @pvs;
@@ -317,6 +424,12 @@ sub prettyprint_pv {
        if ($piece eq '-') {
                die "Invalid move";
        }
+       if ($piece eq uc($piece) && $toplay eq 'B') {
+               die "Black tried to move a white piece";
+       }
+       if ($piece eq lc($piece) && $toplay eq 'W') {
+               die "White tried to move a black piece";
+       }
 
        # white short castling
        if ($pv eq 'e1g1' && $piece eq 'K') {
@@ -328,7 +441,7 @@ sub prettyprint_pv {
                substr($nb[7], 7, 1, '-');
                substr($nb[7], 5, 1, 'R');
                                
-               return ('0-0', prettyprint_pv(\@nb, @pvs));
+               return ($prefix . '0-0', prettyprint_pv(\@nb, $next_move, $next_toplay, @pvs));
        }
 
        # white long castling
@@ -341,7 +454,7 @@ sub prettyprint_pv {
                substr($nb[7], 0, 1, '-');
                substr($nb[7], 3, 1, 'R');
                                
-               return ('0-0-0', prettyprint_pv(\@nb, @pvs));
+               return ($prefix . '0-0-0', prettyprint_pv(\@nb, $next_move, $next_toplay, @pvs));
        }
 
        # black short castling
@@ -354,7 +467,7 @@ sub prettyprint_pv {
                substr($nb[0], 7, 1, '-');
                substr($nb[0], 5, 1, 'r');
                                
-               return ('0-0', prettyprint_pv(\@nb, @pvs));
+               return ($prefix . '0-0', prettyprint_pv(\@nb, $next_move, $next_toplay, @pvs));
        }
 
        # black long castling
@@ -367,7 +480,7 @@ sub prettyprint_pv {
                substr($nb[0], 0, 1, '-');
                substr($nb[0], 3, 1, 'r');
                                
-               return ('0-0-0', prettyprint_pv(\@nb, @pvs));
+               return ($prefix . '0-0-0', prettyprint_pv(\@nb, $next_move, $next_toplay, @pvs));
        }
 
        # check if the from-piece is a pawn
@@ -454,11 +567,13 @@ sub prettyprint_pv {
                $pretty .= '+';
        }
 
-       return ($pretty, prettyprint_pv(\@nb, @pvs));
+       return ($prefix . $pretty, prettyprint_pv(\@nb, $next_move, $next_toplay, @pvs));
 }
 
 sub output_screen {
        #return;
+       
+       return if (!defined($pos_calculating));
 
        #
        # Check the PVs first. if they're invalid, just wait, as our data
@@ -468,16 +583,23 @@ sub output_screen {
        eval {
                my $dummy;
                if (exists($uciinfo{'pv'})) {
-                       $dummy = prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv'}});
+                       $dummy = prettyprint_pv($pos_calculating->{'board'},
+                               $pos_calculating->{'move_num'},
+                               $pos_calculating->{'toplay'},
+                               @{$uciinfo{'pv'}});
                }
        
                my $mpv = 1;
                while (exists($uciinfo{'pv' . $mpv})) {
-                       $dummy = prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv' . $mpv}});
+                       $dummy = prettyprint_pv($pos_calculating->{'board'},
+                               $pos_calculating->{'move_num'},
+                               $pos_calculating->{'toplay'},
+                               @{$uciinfo{'pv' . $mpv}});
                        ++$mpv;
                }
        };
        if ($@) {
+               %uciinfo = ();
                return;
        }
 
@@ -489,7 +611,7 @@ sub output_screen {
                $text .= "Analysis:\n\n";
        }
 
-       return unless (exists($ficsinfo{'board'}));
+       return unless (exists($pos_calculating->{'board'}));
                
        #
        # Some programs _always_ report MultiPV, even with only one PV.
@@ -503,31 +625,55 @@ sub output_screen {
                        }
                }
        }
+                       
+       my @text_pvs = ();
+       if (exists($uciinfo{'pv1'}) && exists($uciinfo{'pv2'})) {
+               # multi-PV
+               my $mpv = 1;
+               while (exists($uciinfo{'pv' . $mpv})) {
+                       $text_pvs[$mpv] = prefix_pv($pos_calculating->{'move_num'}, $pos_calculating->{'toplay'}) .
+                               join(' ', prettyprint_pv($pos_calculating->{'board'},
+                                       $pos_calculating->{'move_num'},
+                                       $pos_calculating->{'toplay'},
+                                       @{$uciinfo{'pv' . $mpv}}));
+                       ++$mpv;
+               }
+       } else {
+               # single-PV
+               $text_pvs[0] = prefix_pv($pos_calculating->{'move_num'}, $pos_calculating->{'toplay'}) .
+                       join(' ', prettyprint_pv($pos_calculating->{'board'},
+                               $pos_calculating->{'move_num'},
+                               $pos_calculating->{'toplay'},
+                               @{$uciinfo{'pv'}}));
+       }
 
        if (exists($uciinfo{'pv1'}) && exists($uciinfo{'pv2'})) {
                # multi-PV
                my $mpv = 1;
                while (exists($uciinfo{'pv' . $mpv})) {
                        $text .= sprintf "  PV%2u", $mpv;
-                       my $score = short_score(\%uciinfo, \%ficsinfo, $mpv);
+                       my $score = short_score(\%uciinfo, $pos_calculating, $mpv);
                        $text .= "  ($score)" if (defined($score));
 
+                       my $tbhits = '';
+                       if (exists($uciinfo{'tbhits' . $mpv})) {
+                               $tbhits = sprintf ", %u tbhits", $uciinfo{'tbhits' . $mpv};
+                       }
+
                        if (exists($uciinfo{'nodes' . $mpv}) && exists($uciinfo{'nps' . $mpv}) && exists($uciinfo{'depth' . $mpv})) {
-                               $text .= sprintf " (%5u kn, %3u kn/s, %2u ply)",
+                               $text .= sprintf " (%5u kn, %3u kn/s, %2u ply$tbhits)",
                                        $uciinfo{'nodes' . $mpv} / 1000, $uciinfo{'nps' . $mpv} / 1000, $uciinfo{'depth' . $mpv};
                        }
 
                        $text .= ":\n";
-                       $text .= "  " . join(', ', prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv' . $mpv}})) . "\n";
-                       $text .= "\n";
+                       $text .= "  $text_pvs[$mpv]\n\n";
                        ++$mpv;
                }
        } else {
                # single-PV
-               my $score = long_score(\%uciinfo, \%ficsinfo, '');
+               my $score = long_score(\%uciinfo, $pos_calculating, '');
                $text .= "  $score\n" if defined($score);
-               $text .=  "  PV: " . join(', ', prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv'}}));
-               $text .=  "\n";
+               $text .= "  PV: $text_pvs[0]\n";
 
                if (exists($uciinfo{'nodes'}) && exists($uciinfo{'nps'}) && exists($uciinfo{'depth'})) {
                        $text .= sprintf "  %u nodes, %7u nodes/sec, depth %u ply",
@@ -564,23 +710,21 @@ sub output_screen {
                my $mpv = 1;
                while (exists($uciinfo{'pv' . $mpv})) {
                        $tell_text .= sprintf "  PV%2u", $mpv;
-                       my $score = short_score(\%uciinfo, \%ficsinfo, $mpv);
+                       my $score = short_score(\%uciinfo, $pos_calculating, $mpv);
                        $tell_text .= "  ($score)" if (defined($score));
 
                        if (exists($uciinfo{'depth' . $mpv})) {
                                $tell_text .= sprintf " (%2u ply)", $uciinfo{'depth' . $mpv};
                        }
 
-                       $tell_text .= ": ";
-                       $tell_text .= join(', ', prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv' . $mpv}}));
-                       $tell_text .= "\n";
+                       $tell_text .= ": $text_pvs[$mpv]\n";
                        ++$mpv;
                }
        } else {
                # single-PV
-               my $score = long_score(\%uciinfo, \%ficsinfo, '');
+               my $score = long_score(\%uciinfo, $pos_calculating, '');
                $tell_text .= "  $score\n" if defined($score);
-               $tell_text .= "  PV: " . join(', ', prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv'}}));
+               $tell_text .= "  PV: $text_pvs[0]";
                if (exists($uciinfo{'depth'})) {
                        $tell_text .= sprintf " (depth %u ply)", $uciinfo{'depth'};
                }
@@ -797,18 +941,18 @@ sub can_reach {
 sub uciprint {
        my $msg = shift;
        print UCIWRITE "$msg\n";
-       print UCILOG "=> $msg\n";
+       print UCILOG localtime() . " => $msg\n";
 }
 
 sub short_score {
-       my ($uciinfo, $ficsinfo, $mpv) = @_;
+       my ($uciinfo, $pos, $mpv) = @_;
 
        if (defined($uciinfo{'score_mate' . $mpv})) {
                return sprintf "M%3d", $uciinfo{'score_mate' . $mpv};
        } else {
                if (exists($uciinfo{'score_cp' . $mpv})) {
                        my $score = $uciinfo{'score_cp' . $mpv} * 0.01;
-                       if ($ficsinfo{'toplay'} eq 'B') {
+                       if ($pos->{'toplay'} eq 'B') {
                                $score = -$score;
                        }
                        return sprintf "%+5.2f", $score;
@@ -819,11 +963,11 @@ sub short_score {
 }
 
 sub long_score {
-       my ($uciinfo, $ficsinfo, $mpv) = @_;
+       my ($uciinfo, $pos, $mpv) = @_;
 
        if (defined($uciinfo{'score_mate' . $mpv})) {
                my $mate = $uciinfo{'score_mate' . $mpv};
-               if ($ficsinfo{'toplay'} eq 'B') {
+               if ($pos->{'toplay'} eq 'B') {
                        $mate = -$mate;
                }
                if ($mate > 0) {
@@ -834,7 +978,7 @@ sub long_score {
        } else {
                if (exists($uciinfo{'score_cp' . $mpv})) {
                        my $score = $uciinfo{'score_cp' . $mpv} * 0.01;
-                       if ($ficsinfo{'toplay'} eq 'B') {
+                       if ($pos->{'toplay'} eq 'B') {
                                $score = -$score;
                        }
                        return sprintf "Score: %+5.2f", $score;