core: Tree container impl. Partially broken, fix on laptop

1 files changed, 42 insertions, 10 deletions

diff --git a/c++/core/mcts.h b/c++/core/mcts.h
index f0f3ba5..8a8f5ea 100644
--- a/c++/core/mcts.h
+++ b/c++/core/mcts.h
@@ -3,18 +3,50 @@
 #include "tree.h"
 
 namespace saw {
-template<typename State>
-struct mcts_value {
-		State state;
-};
-
-template<typename State>
+template<typename T>
 class mcts_tree {
 private:
-		/**
-		 * @todo
-		 * Basically I need to a tree with tree<mcts_value<State>> 
-		 */
+	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<value, mcts_tree<T>> 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<T>& get_tree(size_t i){
+		return data_.at(i+1).get_tree();
+	}
+
+	const mcts_tree<T>& get_tree(size_t i) const {
+		return data_.at(i+1).get_tree();
+	}
 };
 }