X-Git-Url: https://git.sesse.net/?p=remoteglot;a=blobdiff_plain;f=Position.pm;h=9d90cfc777e2576dcd275815c8089774040ecc8d;hp=5d63ae2ff662d9abce22ba9bbe40a729ea7b8f8d;hb=0a124edc28c716548327826127a9145da091a3d0;hpb=8bb77e05c46c67ba4fe237f47d0e6fe52a57b694

diff --git a/Position.pm b/Position.pm
index 5d63ae2..9d90cfc 100644
--- a/Position.pm
+++ b/Position.pm
@@ -4,6 +4,7 @@
 #
 use strict;
 use warnings;
+use MIME::Base64;
 
 require 'Board.pm';
 
@@ -27,8 +28,16 @@ sub new {
 	$pos->{'player_b'} = $x[18];
 	$pos->{'player_w'} =~ s/^W?[FCIG]M//;
 	$pos->{'player_b'} =~ s/^W?[FCIG]M//;
+	$pos->{'white_clock'} = $x[24];
+	$pos->{'black_clock'} = $x[25];
 	$pos->{'move_num'} = $x[26];
+	if ($x[27] =~ /([a-h][1-8])-([a-h][1-8])/) {
+		$pos->{'last_move_uci'} = $1 . $2;
+	} else {
+		$pos->{'last_move_uci'} = undef;
+	}
 	$pos->{'last_move'} = $x[29];
+	$pos->{'prettyprint_cache'} = {};
 
 	bless $pos, $class;
 	return $pos;
@@ -36,6 +45,8 @@ sub new {
 
 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");
 }
 
@@ -64,27 +75,12 @@ sub fen {
 	my $ep = "-";
 	if ($pos->{'ep_file_num'} != -1) {
 		my $col = $pos->{'ep_file_num'};
-		my $nep = (qw(a b c d e f g h))[$col];
+		$ep = (qw(a b c d e f g h))[$col];
 
 		if ($pos->{'toplay'} eq 'B') {
-			$nep .= "3";
+			$ep .= "3";
 		} 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
-		# a pawn of the opposite side that can do the en passant move, and if
-		# not, just lie -- it doesn't matter anyway. I'm unsure what's the
-		# "right" thing as per the standard, though.
-		#
-		if ($pos->{'toplay'} eq 'B') {
-			$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 && $pos->{'board'}[3][$col-1] eq 'P');
-			$ep = $nep if ($col < 7 && $pos->{'board'}[3][$col+1] eq 'P');
+			$ep .= "6";
 		}
 	}
 	$fen .= " ";
@@ -101,9 +97,52 @@ sub fen {
 	return $fen;
 }
 
+# Returns a compact bit string describing the same data as fen(),
+# except for the half-move and full-move clock.
+sub bitpacked_fen {
+	my $pos = shift;
+	my $board = $pos->{'board'}->bitpacked_fen();
+
+	my $bits = "";
+	if ($pos->{'toplay'} eq 'W') {
+		$bits .= "0";
+	} else {
+		$bits .= "1";
+	}
+
+	$bits .= $pos->{'white_castle_k'};
+	$bits .= $pos->{'white_castle_q'};
+	$bits .= $pos->{'black_castle_k'};
+	$bits .= $pos->{'black_castle_q'};
+
+	my $col = $pos->{'ep_file_num'};
+	if ($col == -1) {
+		$bits .= "0";
+	} else {
+		$bits .= "1";
+		$bits .= (qw(000 001 010 011 100 101 110 111))[$col];
+	}
+
+	return $board . pack('b*', $bits);
+}
+
 sub to_json_hash {
 	my $pos = shift;
-	return { %$pos, board => undef, fen => $pos->fen() };
+	my $json = { %$pos, fen => $pos->fen() };
+	delete $json->{'board'};
+	delete $json->{'prettyprint_cache'};
+	delete $json->{'black_castle_k'};
+	delete $json->{'black_castle_q'};
+	delete $json->{'white_castle_k'};
+	delete $json->{'white_castle_q'};
+	delete $json->{'time_since_100move_rule_reset'};
+	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 {
@@ -111,6 +150,11 @@ sub parse_pretty_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) = @_;
@@ -167,10 +211,23 @@ sub make_move {
 	}
 	$np->{'player_w'} = $pos->{'player_w'};
 	$np->{'player_b'} = $pos->{'player_b'};
-	$np->{'last_move'} = '(move)';  # FIXME
+	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;
 }
 
+# Returns a new Position object, and the parsed UCI move.
+sub make_pretty_move {
+	my ($pos, $move) = @_;
+
+	my ($from_row, $from_col, $to_row, $to_col, $promo) = $pos->parse_pretty_move($move);
+	my $uci_move = Board::move_to_uci_notation($from_row, $from_col, $to_row, $to_col, $promo);
+	$pos = $pos->make_move($from_row, $from_col, $to_row, $to_col, $promo);
+	return ($pos, $uci_move);
+}
+
 sub _pos_to_square {
         my ($row, $col) = @_;
         return sprintf("%c%d", ord('a') + $col, 8 - $row);