X-Git-Url: https://git.sesse.net/?p=remoteglot;a=blobdiff_plain;f=remoteglot.pl;h=39a0f4ba805f4312f6a12afe69be7757a6ed0384;hp=d82c20bed25034896c341b979aa9a5b36bb0fac9;hb=1856a6b8a4c1facbb3ff1df0f240acc9d378b315;hpb=88d92e1d44ad96601281d9ce96a525ac7df90757 diff --git a/remoteglot.pl b/remoteglot.pl index d82c20b..39a0f4b 100755 --- a/remoteglot.pl +++ b/remoteglot.pl @@ -209,7 +209,6 @@ sub handle_fics { for my $pos ($pos_waiting, $pos_calculating) { next if (!defined($pos)); if ($pos->fen() eq $pos_for_movelist->fen()) { - $pos->{'history'} = \@uci_movelist; $pos->{'pretty_history'} = \@pretty_movelist; } } @@ -279,7 +278,6 @@ sub handle_pgn { ($pos, $uci_move) = $pos->make_pretty_move($move); push @uci_moves, $uci_move; } - $pos->{'history'} = \@uci_moves; $pos->{'pretty_history'} = $moves; # Sometimes, PGNs lose a move or two for a short while, @@ -711,8 +709,7 @@ sub output_json { $json->{'seldepth'} = $info->{'seldepth'}; $json->{'tablebase'} = $info->{'tablebase'}; - # single-PV only for now - $json->{'pv_uci'} = $info->{'pv'}; + $json->{'pv_uci'} = $info->{'pv'}; # Still needs to be there for the JS to calculate arrows; only for the primary PV, though! $json->{'pv_pretty'} = [ prettyprint_pv($pos_calculating, @{$info->{'pv'}}) ]; my %refutation_lines = (); @@ -736,17 +733,66 @@ sub output_json { pretty_move => $pretty_move, pv_pretty => \@pretty_pv, }; - $refutation_lines{$pv->[0]}->{'pv_uci'} = $pv; }; } } $json->{'refutation_lines'} = \%refutation_lines; - open my $fh, ">", $remoteglotconf::json_output . ".tmp" + my $encoded = JSON::XS::encode_json($json); + atomic_set_contents($remoteglotconf::json_output, $encoded); + + if (exists($pos_calculating->{'pretty_history'}) && + defined($remoteglotconf::json_history_dir)) { + my $halfmove_num = scalar @{$pos_calculating->{'pretty_history'}}; + (my $fen = $pos_calculating->fen()) =~ tr,/ ,-_,; + my $filename = $remoteglotconf::json_history_dir . "/move$halfmove_num-$fen.json"; + + # Overwrite old analysis (assuming it exists at all) if we're + # using a different engine, or if we've calculated deeper. + # nodes is used as a tiebreaker. Don't bother about Multi-PV + # data; it's not that important. + my ($old_engine, $old_depth, $old_nodes) = get_json_analysis_stats($filename); + my $new_depth = $json->{'depth'} // 0; + my $new_nodes = $json->{'nodes'} // 0; + if (!defined($old_engine) || + $old_engine ne $json->{'id'}{'name'} || + $new_depth > $old_depth || + ($new_depth == $old_depth && $new_nodes >= $old_nodes)) { + atomic_set_contents($filename, $encoded); + } + } +} + +sub atomic_set_contents { + my ($filename, $contents) = @_; + + open my $fh, ">", $filename . ".tmp" or return; - print $fh JSON::XS::encode_json($json); + print $fh $contents; + close $fh; + rename($filename . ".tmp", $filename); +} + +sub get_json_analysis_stats { + my $filename = shift; + + my ($engine, $depth, $nodes); + + open my $fh, "<", $filename + or return undef; + local $/ = undef; + eval { + my $json = JSON::XS::decode_json(<$fh>); + $engine = $json->{'id'}{'name'} // die; + $depth = $json->{'depth'} // 0; + $nodes = $json->{'nodes'} // 0; + }; close $fh; - rename($remoteglotconf::json_output . ".tmp", $remoteglotconf::json_output); + if ($@) { + warn "Error in decoding $filename: $@"; + return undef; + } + return ($engine, $depth, $nodes); } sub uciprint {