Add Expander Class

Author: rmobis

Layout.tlua

@@ -109,6 +109,8 @@ LIBS.RAPHAEL = RAPHAEL_LIB
 -- \____/\__,_/____/\__/\____/_/ /_/ /_/   \____/_/\__,_/____/____/\___/____/
 --
 
+@include(src/Classes/Expander Class.lua)
+
 @include(src/Classes/Point Class.lua)
 
 @include(src/Classes/Area Class.lua)

src/Classes/Expander Class.lua

@@ -0,0 +1,43 @@
+Expander = { __class = 'Expander' }
+ExpanderMT = { __index = Expander }
+
+function Expander:new(filter, expander)
+	local newObj = {
+		filter = filter,
+		expander = expander
+	}
+
+	setmetatable(newObj, ExpanderMT)
+	return newObj
+end
+
+function Expander:makeFilter(objType)
+	return function(v)
+		return type(v) == objType
+	end
+end
+
+function Expander:makeExpander(...)
+	local props = {...}
+	return function(v)
+		local vals = {}
+		for _, vv in ipairs(props) do
+			table.insert(vals, getvalue(v[vv]))
+		end
+
+		return table.unpack(vals)
+	end
+end
+
+function Expander:expand(...)
+	local args = {...}
+
+	for k, v in ipairs(args) do
+		if self.filter(v) then
+			args[k] = {self.expander(v)}
+		end
+	end
+
+	table.flatten(args)
+	return table.unpack()
+end