oniietzschan / ygg

Behaviour trees in Lua.

Ygg

Behaviour trees in Lua.

local isHungry = Ygg(function(this)
  return this.hunger >= 50
end)

local isSleepy = Ygg(function(this)
  return this.tiredness >= 100
end)

local eat = Ygg(function(this, dt)
  if this.hunger == 0 then
    return false
  end
  this.hunger = math.max(0, this.hunger - 10 * dt)
  return (this.hunger == 0) and true or nil
end)

local sleep = Ygg(function(this, dt)
  if this.tiredness == 0 then
    return false
  end
  this.tiredness = math.max(0, this.tiredness - 5 * dt)
  return (this.tiredness == 0) and true or nil
end)

local idle = Ygg(function(this, dt)
  this.hunger = this.hunger + dt
  this.tiredness = this.tiredness + dt
  return true
end)

local tree = Ygg.selector()
  :add(
    Ygg.sequence()
      :add(isHungry)
      :add(eat)
  )
  :add(
    Ygg.sequence()
      :add(isSleepy)
      :add(sleep)
  )
  :add(idle)

local runner = Ygg.run(tree)

local entity = {
  hunger = 25,
  tiredness = 0,
}

for _ = 1, 10 do
  print(("Hunger: %d, Tired: %d"):format(entity.hunger, entity.tiredness))
  local dt = 1
  -- :update() arguments will be passed into action functions.
  runner:update(entity, dt)
end

Documentation

Coming soon???? You're on your own for now, chief.

Debug Graph

Included in the repository is an optional file debug_graph.lua which you can use to help visualize your tree and work out issues. You can use it like so:

local YggDebugGraph = require 'debug_graph'
local graph = YggDebugGraph(yggTreeRunner)

function love.draw()
  local x, y = 10, 10
  graph:draw(x, y)
end