Add optional argument to define score being optimized for

A third CLI argument is now accepted, describing a score goal.
The program will then treat reaching this score as a win and
calculate lines optimized for this score, ignoring any (possible)
higher scores.
This commit is contained in:
Maximilian Keßler 2023-11-04 13:02:16 +01:00
parent 87c429e586
commit 4e67ffa9ee
Signed by: max
GPG key ID: BCC5A619923C0BA5
5 changed files with 52 additions and 35 deletions

View file

@ -45,7 +45,7 @@ namespace Download {
*
* @note Turns start counting at 1, since this is also the way hanab.live does it.
*/
std::unique_ptr<Hanabi::HanabiStateIF> get_game(std::variant<int, const char*> game_spec, unsigned turn = 1, size_t draw_pile_break = 0);
std::unique_ptr<Hanabi::HanabiStateIF> get_game(std::variant<int, const char*> game_spec, unsigned turn = 1, size_t draw_pile_break = 0, std::optional<uint8_t> score_goal = std::nullopt);
} // namespace Download

View file

@ -257,7 +257,7 @@ template <suit_t num_suits, player_t num_players, hand_index_t hand_size>
class HanabiState : public HanabiStateIF {
public:
HanabiState() = default;
explicit HanabiState(const std::vector<Card>& deck);
explicit HanabiState(const std::vector<Card>& deck, uint8_t score_goal = 5 * num_suits);
void give_clue() final;
void discard(hand_index_t index) final;
@ -374,6 +374,7 @@ private:
// further values of game state that are technically determined, but we update them anyway
int8_t _pace{};
uint8_t _score{};
uint8_t _score_goal{};
// For reverting the current game
std::stack<BacktrackAction> _actions_log;

View file

@ -111,7 +111,7 @@ namespace Hanabi {
}
template<suit_t num_suits, player_t num_players, hand_index_t hand_size>
HanabiState<num_suits, num_players, hand_size>::HanabiState(const std::vector<Card> &deck):
HanabiState<num_suits, num_players, hand_size>::HanabiState(const std::vector<Card> &deck, uint8_t score_goal):
_turn(0),
_num_clues(max_num_clues),
_weighted_draw_pile_size(deck.size()),
@ -119,8 +119,9 @@ namespace Hanabi {
_hands(),
_draw_pile(),
_endgame_turns_left(no_endgame),
_pace(deck.size() - 5 * num_suits - num_players * (hand_size - 1)),
_pace(deck.size() - score_goal - num_players * (hand_size - 1)),
_score(0),
_score_goal(score_goal),
_actions_log(),
_relative_representation(),
_position_tablebase(),
@ -561,7 +562,7 @@ namespace Hanabi {
std::vector<std::pair<Action, std::optional<probability_t>>> HanabiState<num_suits, num_players, hand_size>::get_reasonable_actions() {
std::vector<std::pair<Action, std::optional<probability_t>>> reasonable_actions {};
if(_score == 5 * num_suits or _pace < 0 or _endgame_turns_left == 0) {
if(_score == _score_goal or _pace < 0 or _endgame_turns_left == 0) {
return reasonable_actions;
}
@ -663,7 +664,7 @@ namespace Hanabi {
_enumerated_states++;
const unsigned long id_of_state = unique_id();
if (_score == 5 * num_suits) {
if (_score == _score_goal) {
return 1;
}
if(_pace < 0 || _endgame_turns_left == 0) {

View file

@ -100,9 +100,11 @@ namespace Download {
const std::vector<Hanabi::Card>& deck,
const std::vector<Action>& actions,
size_t start_turn,
size_t draw_pile_break = 0
size_t draw_pile_break = 0,
std::optional<uint8_t> score_goal = std::nullopt
) {
auto game = std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<num_suits, num_players, hand_size>(deck));
uint8_t actual_score_goal = score_goal.value_or(5 * num_suits);
auto game = std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<num_suits, num_players, hand_size>(deck, actual_score_goal));
std::uint8_t index;
for (size_t i = 0; i < std::min(start_turn - 1, actions.size()); i++) {
if (game->draw_pile_size() == draw_pile_break) {
@ -133,7 +135,7 @@ namespace Download {
return game;
}
std::unique_ptr<Hanabi::HanabiStateIF> get_game(std::variant<int, const char*> game_spec, unsigned turn, size_t draw_pile_break) {
std::unique_ptr<Hanabi::HanabiStateIF> get_game(std::variant<int, const char*> game_spec, unsigned turn, size_t draw_pile_break, std::optional<uint8_t> score_goal) {
const std::optional<boost::json::object> game_json_opt = [&game_spec]() {
if (game_spec.index() == 0) {
return download_game_json(std::get<int>(game_spec));
@ -156,65 +158,65 @@ namespace Download {
case 2:
switch(num_suits) {
case 3:
return produce_state<3,2,5>(deck, actions, turn, draw_pile_break);
return produce_state<3,2,5>(deck, actions, turn, draw_pile_break, score_goal);
case 4:
return produce_state<4,2,5>(deck, actions, turn, draw_pile_break);
return produce_state<4,2,5>(deck, actions, turn, draw_pile_break, score_goal);
case 5:
return produce_state<5,2,5>(deck, actions, turn, draw_pile_break);
return produce_state<5,2,5>(deck, actions, turn, draw_pile_break, score_goal);
case 6:
return produce_state<6,2,5>(deck, actions, turn, draw_pile_break);
return produce_state<6,2,5>(deck, actions, turn, draw_pile_break, score_goal);
default:
throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
}
case 3:
switch(num_suits) {
case 3:
return produce_state<3,3,5>(deck, actions, turn, draw_pile_break);
return produce_state<3,3,5>(deck, actions, turn, draw_pile_break, score_goal);
case 4:
return produce_state<4,3,5>(deck, actions, turn, draw_pile_break);
return produce_state<4,3,5>(deck, actions, turn, draw_pile_break, score_goal);
case 5:
return produce_state<5,3,5>(deck, actions, turn, draw_pile_break);
return produce_state<5,3,5>(deck, actions, turn, draw_pile_break, score_goal);
case 6:
return produce_state<6,3,5>(deck, actions, turn, draw_pile_break);
return produce_state<6,3,5>(deck, actions, turn, draw_pile_break, score_goal);
default:
throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
}
case 4:
switch(num_suits) {
case 3:
return produce_state<3,4,4>(deck, actions, turn, draw_pile_break);
return produce_state<3,4,4>(deck, actions, turn, draw_pile_break, score_goal);
case 4:
return produce_state<4,4,4>(deck, actions, turn, draw_pile_break);
return produce_state<4,4,4>(deck, actions, turn, draw_pile_break, score_goal);
case 5:
return produce_state<5,4,4>(deck, actions, turn, draw_pile_break);
return produce_state<5,4,4>(deck, actions, turn, draw_pile_break, score_goal);
case 6:
return produce_state<6,4,4>(deck, actions, turn, draw_pile_break);
return produce_state<6,4,4>(deck, actions, turn, draw_pile_break, score_goal);
default:
throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
}
case 5:
switch(num_suits) {
case 3:
return produce_state<3,5,4>(deck, actions, turn, draw_pile_break);
return produce_state<3,5,4>(deck, actions, turn, draw_pile_break, score_goal);
case 4:
return produce_state<4,5,4>(deck, actions, turn, draw_pile_break);
return produce_state<4,5,4>(deck, actions, turn, draw_pile_break, score_goal);
case 5:
return produce_state<5,5,4>(deck, actions, turn, draw_pile_break);
return produce_state<5,5,4>(deck, actions, turn, draw_pile_break, score_goal);
case 6:
return produce_state<6,5,4>(deck, actions, turn, draw_pile_break);
return produce_state<6,5,4>(deck, actions, turn, draw_pile_break, score_goal);
default:
throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
}
case 6:
switch(num_suits) {
case 3:
return produce_state<3,6,3>(deck, actions, turn, draw_pile_break);
return produce_state<3,6,3>(deck, actions, turn, draw_pile_break, score_goal);
case 4:
return produce_state<4,6,3>(deck, actions, turn, draw_pile_break);
return produce_state<4,6,3>(deck, actions, turn, draw_pile_break, score_goal);
case 5:
return produce_state<5,6,3>(deck, actions, turn, draw_pile_break);
return produce_state<5,6,3>(deck, actions, turn, draw_pile_break, score_goal);
case 6:
return produce_state<6,6,3>(deck, actions, turn, draw_pile_break);
return produce_state<6,6,3>(deck, actions, turn, draw_pile_break, score_goal);
default:
throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
}

View file

@ -9,8 +9,8 @@
namespace Hanabi {
void analyze_game_and_start_cli(std::variant<int, const char*> game_id, int turn) {
auto game = Download::get_game(game_id, turn);
void analyze_game_and_start_cli(std::variant<int, const char*> game_id, int turn, std::optional<uint8_t> score_goal) {
auto game = Download::get_game(game_id, turn, 0, score_goal);
if (game == nullptr) {
if(game_id.index() == 0) {
std::cout << "Failed to download game " << std::get<int>(game_id) << " from hanab.live." << std::endl;
@ -41,17 +41,19 @@ namespace Hanabi {
}
void print_usage(const char* program_name) {
std::cout << "Usage: " << program_name << "(GAME_ID | GAME_FILE) TURN" << std::endl;
std::cout << "Usage: " << program_name << "(GAME_ID | GAME_FILE) TURN [SCORE_GOAL]" << std::endl;
std::cout << " GAME_ID A game id from hanab.live" << std::endl;
std::cout << " GAME_FILE A path to a file describing the game in hanab.live json format." << std::endl;
std::cout << " TURN Turn number of state to analyze. Turn 1 means no actions have been taken." << std::endl;
std::cout << " TURN Turn number of state to analyze. Turn 1 means no actions have been taken." << std::endl;
std::cout << " SCORE_GOAL Score that counts as a win, i.e. is optimized for achieving." << std::endl;
}
}
int main(int argc, char *argv[]) {
if(argc == 3) {
if(argc == 3 or argc == 4) {
std::string game_str(argv[1]);
std::string turn_str (argv[2]);
@ -63,10 +65,21 @@ int main(int argc, char *argv[]) {
return EXIT_FAILURE;
}
std::optional<uint8_t> score_goal = std::nullopt;
if (argc == 4) {
std::string score_goal_str(argv[3]);
try {
Hanabi::analyze_game_and_start_cli(std::stoi(game_str), turn);
score_goal = std::stoi(score_goal_str);
} catch(std::invalid_argument&) {
Hanabi::analyze_game_and_start_cli(game_str.c_str(), turn);
std::cout << "Could not parse score goal number " << score_goal_str;
return EXIT_FAILURE;
}
}
try {
Hanabi::analyze_game_and_start_cli(std::stoi(game_str), turn, score_goal);
} catch(std::invalid_argument&) {
Hanabi::analyze_game_and_start_cli(game_str.c_str(), turn, score_goal);
}
}
else {