Added handlers and utils

Yonaba · Yonaba · commit b1e40bdef8df · 2014-03-17T16:22:13.000Z
diff --git a/Minimax/Lua/Yonaba/handlers/tree_handler.lua b/Minimax/Lua/Yonaba/handlers/tree_handler.lua
@@ -0,0 +1,52 @@
+-- Search tree class handler implementation
+
+local PATH = (...):gsub('handlers.tree_handler$','')
+local class = require (PATH .. '.utils.class')
+
+
+local tree = class ()
+
+-- Tree initialization
+function tree:initialize()
+  self.nodes = {}
+end
+
+-- Adds a node in the tree
+function tree:addNode(name, parent, value)
+  assert(not self.nodes[name], 'node already exist')
+  local node = {
+    name = name, value = value,
+    parent = parent, children = {}
+  }
+  if parent then
+    parent = self:getNode(parent)
+    assert(parent, 'parent not found')
+    table.insert(parent.children, node)
+  end
+  if not self.head then self.head = node end
+  table.insert(self.nodes, node)
+end
+
+-- Get the node by its name
+function tree:getNode(name)
+  for i, node in ipairs(self.nodes) do
+    if node.name == name then return node end
+  end
+end
+
+-- Checks if a given node is terminal (leaf) in the search tree
+function tree:isLeaf(node)
+  return #node.children == 0
+end
+
+-- Returns the node value
+function tree:heuristic(node)
+  return node.value
+end
+
+-- Returns the node's children
+function tree:children(node)
+  return node.children
+end
+
+return tree
diff --git a/Minimax/Lua/Yonaba/utils/class.lua b/Minimax/Lua/Yonaba/utils/class.lua
@@ -0,0 +1,15 @@
+-- Minimalistic class implementation
+
+local class = function(attr)
+  local klass = attr or {}
+  klass.__index = klass
+  klass.__call = function(_,...) return klass:new(...) end
+  function klass:new(...)
+    local instance = setmetatable({}, klass)
+    if klass.initialize then klass.initialize(instance, ...) end
+    return instance
+  end
+  return setmetatable(klass,{__call = klass.__call})
+end
+
+return class
