RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-26 15:17:45 +00:00
Code Issues Activity Actions

Chess: Port to Core::Stream

Browse source
This commit is contained in:
Lucas CHOLLET 2023-01-07 10:30:35 -05:00 • committed by Sam Atkins
parent 9a05175a88
commit a5d5b970ff
3 changed files with 43 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

57
Userland/Games/Chess/ChessWidget.cpp
View file

@ -8,8 +8,9 @@
#include "PromotionDialog.h"
#include <AK/DeprecatedString.h>
#include <AK/Random.h>
#include <AK/String.h>
#include <LibCore/DateTime.h>
#include <LibCore/File.h>
#include <LibCore/Stream.h>
#include <LibGUI/MessageBox.h>
#include <LibGUI/Painter.h>
#include <LibGfx/AntiAliasingPainter.h>
@ -527,11 +528,11 @@ DeprecatedString ChessWidget::get_fen() const
return m_playback ? m_board_playback.to_fen() : m_board.to_fen();
}
void ChessWidget::import_pgn(Core::File& file)
ErrorOr<void> ChessWidget::import_pgn(Core::Stream::File& file)
{
m_board = Chess::Board();
ByteBuffer bytes = file.read_all();
ByteBuffer bytes = TRY(file.read_until_eof());
StringView content = bytes;
auto lines = content.lines();
StringView line;
@ -620,29 +621,31 @@ void ChessWidget::import_pgn(Core::File& file)
m_playback_move_number = m_board_playback.moves().size();
m_playback = true;
update();
return {};
}
void ChessWidget::export_pgn(Core::File& file) const
ErrorOr<void> ChessWidget::export_pgn(Core::Stream::File& file) const
{
// Tag Pair Section
file.write("[Event \"Casual Game\"]\n"sv);
file.write("[Site \"SerenityOS Chess\"]\n"sv);
file.write(DeprecatedString::formatted("[Date \"{}\"]\n", Core::DateTime::now().to_deprecated_string("%Y.%m.%d"sv)));
file.write("[Round \"1\"]\n"sv);
TRY(file.write("[Event \"Casual Game\"]\n"sv.bytes()));
TRY(file.write("[Site \"SerenityOS Chess\"]\n"sv.bytes()));
TRY(file.write(DeprecatedString::formatted("[Date \"{}\"]\n", Core::DateTime::now().to_deprecated_string("%Y.%m.%d"sv)).bytes()));
TRY(file.write("[Round \"1\"]\n"sv.bytes()));
DeprecatedString username(getlogin());
const DeprecatedString player1 = (!username.is_empty() ? username.view() : "?"sv);
const DeprecatedString player2 = (!m_engine.is_null() ? "SerenityOS ChessEngine"sv : "?"sv);
file.write(DeprecatedString::formatted("[White \"{}\"]\n", m_side == Chess::Color::White ? player1 : player2));
file.write(DeprecatedString::formatted("[Black \"{}\"]\n", m_side == Chess::Color::Black ? player1 : player2));
auto const player1 = (!username.is_empty() ? username.view() : "?"sv.bytes());
auto const player2 = (!m_engine.is_null() ? "SerenityOS ChessEngine"sv.bytes() : "?"sv.bytes());
TRY(file.write(DeprecatedString::formatted("[White \"{}\"]\n", m_side == Chess::Color::White ? player1 : player2).bytes()));
TRY(file.write(DeprecatedString::formatted("[Black \"{}\"]\n", m_side == Chess::Color::Black ? player1 : player2).bytes()));
file.write(DeprecatedString::formatted("[Result \"{}\"]\n", Chess::Board::result_to_points_deprecated_string(m_board.game_result(), m_board.turn())));
file.write("[WhiteElo \"?\"]\n"sv);
file.write("[BlackElo \"?\"]\n"sv);
file.write("[Variant \"Standard\"]\n"sv);
file.write("[TimeControl \"-\"]\n"sv);
file.write("[Annotator \"SerenityOS Chess\"]\n"sv);
file.write("\n"sv);
TRY(file.write(DeprecatedString::formatted("[Result \"{}\"]\n", Chess::Board::result_to_points_deprecated_string(m_board.game_result(), m_board.turn())).bytes()));
TRY(file.write("[WhiteElo \"?\"]\n"sv.bytes()));
TRY(file.write("[BlackElo \"?\"]\n"sv.bytes()));
TRY(file.write("[Variant \"Standard\"]\n"sv.bytes()));
TRY(file.write("[TimeControl \"-\"]\n"sv.bytes()));
TRY(file.write("[Annotator \"SerenityOS Chess\"]\n"sv.bytes()));
TRY(file.write("\n"sv.bytes()));
// Movetext Section
for (size_t i = 0, move_no = 1; i < m_board.moves().size(); i += 2, move_no++) {
@ -650,17 +653,19 @@ void ChessWidget::export_pgn(Core::File& file) const
if (i + 1 < m_board.moves().size()) {
const DeprecatedString black = m_board.moves().at(i + 1).to_algebraic();
file.write(DeprecatedString::formatted("{}. {} {} ", move_no, white, black));
TRY(file.write(DeprecatedString::formatted("{}. {} {} ", move_no, white, black).bytes()));
} else {
file.write(DeprecatedString::formatted("{}. {} ", move_no, white));
TRY(file.write(DeprecatedString::formatted("{}. {} ", move_no, white).bytes()));
}
}
file.write("{ "sv);
file.write(Chess::Board::result_to_deprecated_string(m_board.game_result(), m_board.turn()));
file.write(" } "sv);
file.write(Chess::Board::result_to_points_deprecated_string(m_board.game_result(), m_board.turn()));
file.write("\n"sv);
TRY(file.write("{ "sv.bytes()));
TRY(file.write(Chess::Board::result_to_deprecated_string(m_board.game_result(), m_board.turn()).bytes()));
TRY(file.write(" } "sv.bytes()));
TRY(file.write(Chess::Board::result_to_points_deprecated_string(m_board.game_result(), m_board.turn()).bytes()));
TRY(file.write("\n"sv.bytes()));
return {};
}
void ChessWidget::flip_board()

4
Userland/Games/Chess/ChessWidget.h
View file

@ -48,8 +48,8 @@ public:
void set_show_available_moves(bool e) { m_show_available_moves = e; }
DeprecatedString get_fen() const;
void import_pgn(Core::File&);
void export_pgn(Core::File&) const;
ErrorOr<void> import_pgn(Core::Stream::File&);
ErrorOr<void> export_pgn(Core::Stream::File&) const;
int resign();
void flip_board();

16
Userland/Games/Chess/main.cpp
View file

@ -67,20 +67,24 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
TRY(game_menu->try_add_separator());
TRY(game_menu->try_add_action(GUI::Action::create("&Import PGN...", { Mod_Ctrl, Key_O }, [&](auto&) {
auto result = FileSystemAccessClient::Client::the().try_open_file_deprecated(window);
auto result = FileSystemAccessClient::Client::the().open_file(window);
if (result.is_error())
return;
widget->import_pgn(result.value());
dbgln("Imported PGN file from {}", result.value()->filename());
if (auto maybe_error = widget->import_pgn(*result.value().release_stream()); maybe_error.is_error())
dbgln("Failed to import PGN: {}", maybe_error.release_error());
else
dbgln("Imported PGN file from {}", result.value().filename());
})));
TRY(game_menu->try_add_action(GUI::Action::create("&Export PGN...", { Mod_Ctrl, Key_S }, [&](auto&) {
auto result = FileSystemAccessClient::Client::the().try_save_file_deprecated(window, "Untitled", "pgn");
auto result = FileSystemAccessClient::Client::the().save_file(window, "Untitled", "pgn");
if (result.is_error())
return;
widget->export_pgn(result.value());
dbgln("Exported PGN file to {}", result.value()->filename());
if (auto maybe_error = widget->export_pgn(*result.value().release_stream()); maybe_error.is_error())
dbgln("Failed to export PGN: {}", maybe_error.release_error());
else
dbgln("Exported PGN file to {}", result.value().filename());
})));
TRY(game_menu->try_add_action(GUI::Action::create("&Copy FEN", { Mod_Ctrl, Key_C }, [&](auto&) {
GUI::Clipboard::the().set_data(widget->get_fen().bytes());