From 35f814706c4ff244bc15c4285804fc0656824c41 Mon Sep 17 00:00:00 2001
From: Adrian Kummerländer
Date: Thu, 6 Feb 2014 18:37:43 +0100
Subject: Added support for changing values * member method set changes the
 value of a given path to the provided value ** returns false if the specified
 path doesn't exist

---
 src/trie.h | 13 +++++++++++++
 test.cc    |  8 ++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/trie.h b/src/trie.h
index 46fcb4b..d4763ef 100644
--- a/src/trie.h
+++ b/src/trie.h
@@ -51,6 +51,19 @@ class Trie {
 			}
 		}
 
+		inline bool set(key_list path, Value value) {
+			std::pair<bool, Trie*> tmp(this->resolve(path));
+
+			if ( tmp.first ) {
+				tmp.second->value_.first  = true;
+				tmp.second->value_.second = value;
+
+				return true;
+			} else {
+				return false;
+			}
+		}
+
 	private:
 		std::pair<bool, Value> value_;
 		std::map<Key, Trie> children_;
diff --git a/test.cc b/test.cc
index ff0b04f..2a99619 100644
--- a/test.cc
+++ b/test.cc
@@ -61,6 +61,14 @@ TEST_F(TrieTest, Value) {
 
 	EXPECT_EQ(trie.get({1, 2}).first,        false);
 	EXPECT_EQ(trie.get({1, 2}).second,       nullptr);
+
+	trie.set({1, 2},       42);
+	trie.set({1, 1, 1, 1}, 255);
+
+	EXPECT_EQ(trie.get({1, 2}).first,         true);
+	EXPECT_EQ(*trie.get({1, 2}).second,       42);
+	EXPECT_EQ(trie.get({1, 1, 1, 1}).first,   true);
+	EXPECT_EQ(*trie.get({1, 1, 1, 1}).second, 255);
 }
 
 int main(int argc, char **argv) {
-- 
cgit v1.2.3