]> git.sesse.net Git - remoteglot/blobdiff - remoteglot.pl
Add some primitive remote control.
[remoteglot] / remoteglot.pl
index a58168bb3347d663fb5d0c867aa0bcc4bd6090f5..8c871084ead93aca57e599d80e1a98aaaf927a8b 100755 (executable)
@@ -18,11 +18,17 @@ use warnings;
 
 # Configuration
 my $server = "freechess.org";
-my $target = "Sesse";
+my $target = "22";
 # 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,7 +373,9 @@ sub style12_to_fen {
        $fen .= " ";
        $fen .= $x[26];
 
-       return $fen;
+       $pos{'fen'} = $fen;
+
+       return \%pos;
 }
 
 sub prettyprint_pv {
@@ -459,6 +541,8 @@ sub prettyprint_pv {
 
 sub output_screen {
        #return;
+       
+       return if (!defined($pos_calculating));
 
        #
        # Check the PVs first. if they're invalid, just wait, as our data
@@ -468,16 +552,17 @@ sub output_screen {
        eval {
                my $dummy;
                if (exists($uciinfo{'pv'})) {
-                       $dummy = prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv'}});
+                       $dummy = prettyprint_pv($pos_calculating->{'board'}, @{$uciinfo{'pv'}});
                }
        
                my $mpv = 1;
                while (exists($uciinfo{'pv' . $mpv})) {
-                       $dummy = prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv' . $mpv}});
+                       $dummy = prettyprint_pv($pos_calculating->{'board'}, @{$uciinfo{'pv' . $mpv}});
                        ++$mpv;
                }
        };
        if ($@) {
+               %uciinfo = ();
                return;
        }
 
@@ -489,7 +574,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.
@@ -509,24 +594,29 @@ sub output_screen {
                my $mpv = 1;
                while (exists($uciinfo{'pv' . $mpv})) {
                        $text .= sprintf "  PV%2u", $mpv;
-                       my $score = short_score(\%uciinfo, \%ficsinfo, $mpv);
-                       $text .= "  ($score)" if (!defined($score));
+                       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 .= "  " . join(', ', prettyprint_pv($pos_calculating->{'board'}, @{$uciinfo{'pv' . $mpv}})) . "\n";
                        $text .= "\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 .=  "  PV: " . join(', ', prettyprint_pv($pos_calculating->{'board'}, @{$uciinfo{'pv'}}));
                $text .=  "\n";
 
                if (exists($uciinfo{'nodes'}) && exists($uciinfo{'nps'}) && exists($uciinfo{'depth'})) {
@@ -564,7 +654,7 @@ 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})) {
@@ -572,15 +662,15 @@ sub output_screen {
                        }
 
                        $tell_text .= ": ";
-                       $tell_text .= join(', ', prettyprint_pv($ficsinfo{'board'}, @{$uciinfo{'pv' . $mpv}}));
+                       $tell_text .= join(', ', prettyprint_pv($pos_calculating->{'board'}, @{$uciinfo{'pv' . $mpv}}));
                        $tell_text .= "\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: " . join(', ', prettyprint_pv($pos_calculating->{'board'}, @{$uciinfo{'pv'}}));
                if (exists($uciinfo{'depth'})) {
                        $tell_text .= sprintf " (depth %u ply)", $uciinfo{'depth'};
                }
@@ -797,18 +887,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 +909,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 +924,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;