From a14896f9ed209dd3f9597722e5a5697bd7dbf531 Mon Sep 17 00:00:00 2001 From: "Claudius \"keldu\" Holeksa" Date: Mon, 4 Dec 2023 12:18:14 +0100 Subject: meta: Renamed folder containing source --- modules/core/mcts.h | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 modules/core/mcts.h (limited to 'modules/core/mcts.h') diff --git a/modules/core/mcts.h b/modules/core/mcts.h new file mode 100644 index 0000000..8a8f5ea --- /dev/null +++ b/modules/core/mcts.h @@ -0,0 +1,52 @@ +#pragma once + +#include "tree.h" + +namespace saw { +template +class mcts_tree { +private: + struct value { + uint64_t numerater; + uint64_t denominater; + T state; + + value() = default; + value(T st): + numerater{0}, + denominater{0}, + state{std::move(st)} + {} + }; + + tree_container> data_; +public: + mcts_tree() = default{ + data_.add(value{}); + } + + mcts_tree(T state){ + data_.add(value{std::move(state)}); + } + + size_t size() const { + return data_.size() - 1; + } + + T& get_state(){ + return data_.at(0).get_value().state; + } + + const T& get_state() const { + return data_.at(0).get_value().state; + } + + mcts_tree& get_tree(size_t i){ + return data_.at(i+1).get_tree(); + } + + const mcts_tree& get_tree(size_t i) const { + return data_.at(i+1).get_tree(); + } +}; +} -- cgit v1.2.3