From 0c0cb0ee2339afa1fa2079cffb5cb1c7e8c82c39 Mon Sep 17 00:00:00 2001
From: Waqas Hussain <waqas20@gmail.com>
Date: Sat, 27 Dec 2008 08:27:26 +0500
Subject: [PATCH] Added util.multitable.set

---
 util/multitable.lua | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/util/multitable.lua b/util/multitable.lua
index ce38e04d..414f6cb5 100644
--- a/util/multitable.lua
+++ b/util/multitable.lua
@@ -47,14 +47,25 @@ local function add(self, ...)
 	t_insert(t, (select(count, ...)));
 end
 
+local function set(self, ...)
+	local t = self.data;
+	local count = select('#', ...);
+	for n = 1,count-2 do
+		local key = select(n, ...);
+		local tab = t[key];
+		if not tab then tab = {}; t[key] = tab; end
+		t = tab;
+	end
+	t[(select(count-1, ...))] = (select(count, ...));
+end
+
 local function r(t, n, _end, ...)
 	if t == nil then return; end
-	if n > _end then
-		for key in pairs(t) do
-			t[key] = nil;
-		end
-	end
 	local k = select(n, ...);
+	if n == _end then
+		t[k] = nil;
+		return;
+	end
 	if k then
 		v = t[k];
 		if v then
@@ -87,6 +98,7 @@ function new()
 		data = {};
 		get = get;
 		add = add;
+		set = set;
 		remove = remove;
 	};
 end
-- 
2.30.2