bridge_learning/playcc/alpha_mu_bot.h at main · hanyuu11/bridge_learning · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
//
// Created by qzz on 2023/12/7.
//

#ifndef PLAYCC_ALPHA_MU_BOT_H
#define PLAYCC_ALPHA_MU_BOT_H
#include <utility>
#include "absl/strings/str_format.h"

#include "bridge_state_without_hidden_info.h"
#include "pareto_front.h"
#include "play_bot.h"
#include "playcc/dds_evaluator.h"
#include "playcc/trajectory_bidding_bot.h"
#include "playcc/worlds.h"
#include "resampler.h"
#include "transposition_table.h"

struct AlphaMuConfig {
  int num_max_moves;
  int num_worlds;
  bool search_with_one_legal_move = false;
  bool use_transportation_table = true;
  bool root_cut = true;
  bool early_cut = true;
  RolloutResult rollout_result=kWinLose;

  bool verbose = false;
};

class AlphaMuBot final : public PlayBot {
  public:
    AlphaMuBot(const std::shared_ptr<Resampler>& resampler, const AlphaMuConfig cfg) : tt_(), resampler_(resampler),
      cfg_(cfg) {
    }

    AlphaMuBot(const std::shared_ptr<Resampler>& resampler,
               const AlphaMuConfig cfg,
               const ble::Player player_id) : tt_(), resampler_(resampler),
                                              cfg_(cfg), player_id_(player_id) {
    }

    ble::BridgeMove Step(const ble::BridgeState& state) override;

    void Restart() override{
      tt_.Clear();
      last_iteration_front_ = {};

    }

    [[nodiscard]] ParetoFront Search(const ble::BridgeStateWithoutHiddenInfo& state,
                                     int num_max_moves,
                                     const Worlds& worlds,
                                     const ParetoFront& alpha);

    std::pair<bool, ParetoFront> Stop(
        const ble::BridgeStateWithoutHiddenInfo& state, int num_max_moves,
        const Worlds& worlds);

    const TranspositionTable& GetTT() const { return tt_; }

    void SetTT(const TranspositionTable& tt) { tt_ = tt; }

    std::string Name() const override {
      return absl::StrFormat("AlphaMu, %d worlds, %d max moves", cfg_.num_worlds, cfg_.num_max_moves);
    }

    void SetBiddingBot(const std::shared_ptr<TrajectoryBiddingBot>& bot) {
      bidding_bot_ = bot;
    }

  private:
    TranspositionTable tt_;
    std::shared_ptr<Resampler> resampler_;
    const AlphaMuConfig cfg_;
    DDSEvaluator dds_evaluator_{};
    const ble::Player player_id_{};
    std::optional<ParetoFront> last_iteration_front_{};
    std::shared_ptr<TrajectoryBiddingBot> bidding_bot_ = nullptr;
};

std::unique_ptr<PlayBot> MakeAlphaMuBot(ble::Player player_id, AlphaMuConfig cfg);

#endif /* PLAYCC_ALPHA_MU_BOT_H */