X-Git-Url: https://git.sesse.net/?p=remoteglot;a=blobdiff_plain;f=Position.pm;h=90e7c1ccc07f649d66b081cb8e01f0a33cb70622;hp=7bdb913fa5716aedc9cd81fab34b0a1789250b30;hb=1eb8400865727a17fdfa2d6b30e86aa570862d6c;hpb=c26cad3df9d1b65031c26b65f4cfbe89aef18de7 diff --git a/Position.pm b/Position.pm index 7bdb913..90e7c1c 100644 --- a/Position.pm +++ b/Position.pm @@ -4,6 +4,7 @@ # use strict; use warnings; +use MIME::Base64; require 'Board.pm'; @@ -15,7 +16,6 @@ sub new { my $pos = {}; my (@x) = split / /, $str; - # TODO: Not all of this stuff really belongs in this module. $pos->{'board'} = Board->new(@x[1..8]); $pos->{'toplay'} = $x[9]; $pos->{'ep_file_num'} = $x[10]; @@ -23,11 +23,11 @@ sub new { $pos->{'white_castle_q'} = $x[12]; $pos->{'black_castle_k'} = $x[13]; $pos->{'black_castle_q'} = $x[14]; - $pos->{'time_to_100move_rule'} = $x[15]; + $pos->{'time_since_100move_rule_reset'} = $x[15]; $pos->{'player_w'} = $x[17]; $pos->{'player_b'} = $x[18]; - $pos->{'player_w'} =~ s/^[IG]M//; - $pos->{'player_b'} =~ s/^[IG]M//; + $pos->{'player_w'} =~ s/^W?[FCIG]M//; + $pos->{'player_b'} =~ s/^W?[FCIG]M//; $pos->{'move_num'} = $x[26]; if ($x[27] =~ /([a-h][1-8])-([a-h][1-8])/) { $pos->{'last_move_uci'} = $1 . $2; @@ -35,11 +35,19 @@ sub new { $pos->{'last_move_uci'} = undef; } $pos->{'last_move'} = $x[29]; + $pos->{'prettyprint_cache'} = {}; bless $pos, $class; return $pos; } +sub start_pos { + my ($class, $white, $black) = @_; + $white = "base64:" . MIME::Base64::encode_base64($white); + $black = "base64:" . MIME::Base64::encode_base64($black); + return $class->new("<12> rnbqkbnr pppppppp -------- -------- -------- -------- PPPPPPPP RNBQKBNR W -1 1 1 1 1 0 dummygamenum $white $black -2 dummytime dummyincrement 39 39 dummytime dummytime 1 none (0:00) none 0 0 0"); +} + sub fen { my $pos = shift; @@ -81,11 +89,11 @@ sub fen { # "right" thing as per the standard, though. # if ($pos->{'toplay'} eq 'B') { - $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'); + $ep = $nep if ($col > 0 && $pos->{'board'}[4][$col-1] eq 'p'); + $ep = $nep if ($col < 7 && $pos->{'board'}[4][$col+1] eq 'p'); } else { - $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'); + $ep = $nep if ($col > 0 && $pos->{'board'}[3][$col-1] eq 'P'); + $ep = $nep if ($col < 7 && $pos->{'board'}[3][$col+1] eq 'P'); } } $fen .= " "; @@ -93,7 +101,7 @@ sub fen { # half-move clock $fen .= " "; - $fen .= $pos->{'time_to_100move_rule'}; + $fen .= $pos->{'time_since_100move_rule_reset'}; # full-move clock $fen .= " "; @@ -104,7 +112,122 @@ sub fen { sub to_json_hash { my $pos = shift; - return { %$pos, board => undef, fen => $pos->fen() }; + my $json = { %$pos, board => undef, prettyprint_cache => undef, fen => $pos->fen() }; + if ($json->{'player_w'} =~ /^base64:(.*)$/) { + $json->{'player_w'} = MIME::Base64::decode_base64($1); + } + if ($json->{'player_b'} =~ /^base64:(.*)$/) { + $json->{'player_b'} = MIME::Base64::decode_base64($1); + } + return $json; +} + +sub parse_pretty_move { + my ($pos, $move) = @_; + return $pos->{'board'}->parse_pretty_move($move, $pos->{'toplay'}); +} + +sub num_pieces { + my ($pos) = @_; + return $pos->{'board'}->num_pieces(); +} + +# Returns a new Position object. +sub make_move { + my ($pos, $from_row, $from_col, $to_row, $to_col, $promo) = @_; + + my $from_square = _pos_to_square($from_row, $from_col); + my $to_square = _pos_to_square($to_row, $to_col); + + my $np = {}; + $np->{'board'} = $pos->{'board'}->make_move($from_row, $from_col, $to_row, $to_col, $promo); + if ($pos->{'toplay'} eq 'W') { + $np->{'toplay'} = 'B'; + $np->{'move_num'} = $pos->{'move_num'}; + } else { + $np->{'toplay'} = 'W'; + $np->{'move_num'} = $pos->{'move_num'} + 1; + } + + my $piece = $pos->{'board'}[$from_row][$from_col]; + my $dest_piece = $pos->{'board'}[$to_row][$to_col]; + + # Find out if this was a two-step pawn move. + if (lc($piece) eq 'p' && abs($from_row - $to_row) == 2) { + $np->{'ep_file_num'} = $from_col; + } else { + $np->{'ep_file_num'} = -1; + } + + # Castling rights. + $np->{'white_castle_k'} = $pos->{'white_castle_k'}; + $np->{'white_castle_q'} = $pos->{'white_castle_q'}; + $np->{'black_castle_k'} = $pos->{'black_castle_k'}; + $np->{'black_castle_q'} = $pos->{'black_castle_q'}; + if ($piece eq 'K') { + $np->{'white_castle_k'} = 0; + $np->{'white_castle_q'} = 0; + } elsif ($piece eq 'k') { + $np->{'black_castle_k'} = 0; + $np->{'black_castle_q'} = 0; + } elsif ($from_square eq 'a1' || $to_square eq 'a1') { + $np->{'white_castle_q'} = 0; + } elsif ($from_square eq 'h1' || $to_square eq 'h1') { + $np->{'white_castle_k'} = 0; + } elsif ($from_square eq 'a8' || $to_square eq 'a8') { + $np->{'black_castle_q'} = 0; + } elsif ($from_square eq 'h8' || $to_square eq 'h8') { + $np->{'black_castle_k'} = 0; + } + + # 50-move rule. + if (lc($piece) eq 'p' || $dest_piece ne '-') { + $np->{'time_since_100move_rule_reset'} = 0; + } else { + $np->{'time_since_100move_rule_reset'} = $pos->{'time_since_100move_rule_reset'} + 1; + } + $np->{'player_w'} = $pos->{'player_w'}; + $np->{'player_b'} = $pos->{'player_b'}; + my ($move, $nb) = $pos->{'board'}->prettyprint_move($from_row, $from_col, $to_row, $to_col, $promo); + $np->{'last_move'} = $move; + $np->{'last_move_uci'} = Board::move_to_uci_notation($from_row, $from_col, $to_row, $to_col, $promo); + + return bless $np; +} + +sub _pos_to_square { + my ($row, $col) = @_; + return sprintf("%c%d", ord('a') + $col, 8 - $row); +} + +sub apply_uci_pv { + my ($pos, @pv) = @_; + + my $pvpos = $pos; + for my $pv_move (@pv) { + my ($from_row, $from_col, $to_row, $to_col, $promo) = _parse_uci_move($pv_move); + $pvpos = $pvpos->make_move($from_row, $from_col, $to_row, $to_col, $promo); + } + + return $pvpos; +} + +sub _col_letter_to_num { + return ord(shift) - ord('a'); +} + +sub _row_letter_to_num { + return 7 - (ord(shift) - ord('1')); +} + +sub _parse_uci_move { + my $move = shift; + my $from_col = _col_letter_to_num(substr($move, 0, 1)); + my $from_row = _row_letter_to_num(substr($move, 1, 1)); + my $to_col = _col_letter_to_num(substr($move, 2, 1)); + my $to_row = _row_letter_to_num(substr($move, 3, 1)); + my $promo = substr($move, 4, 1); + return ($from_row, $from_col, $to_row, $to_col, $promo); } 1;