From efd94d1cf1a3709365c4c144875524819a6a15a3 Mon Sep 17 00:00:00 2001
From: "Claudius \"keldu\" Holeksa" <mail@keldu.de>
Date: Sun, 3 Dec 2023 23:06:43 +0100
Subject: core: Tree container impl. Partially broken, fix on laptop

---
 c++/core/mcts.h | 52 ++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 42 insertions(+), 10 deletions(-)

(limited to 'c++/core/mcts.h')

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();
+	}
 };
 }
-- 
cgit v1.2.3