A force directed graph algorithm written in Lua.
Graphoon emerged from the graph calculation code used in both LoGiVi and LoFiVi.
A force directed graph layout is achieved by simulating physical forces, which push and pull each node in the graph until a nice layout is found.
The basic idea is that you create a new graph object, to which you can then add nodes and edges.
local GraphLibrary = require('Graphoon').Graph
graph = GraphLibrary.new()
graph:addNode( "Ash Williams" )
graph:addNode( "Necronomicon" )
graph:connectIDs( "Ash Williams", "Necronomicon" )
By itself Graphoon only provides functionality for creating the graph and calculating the layout based on physical attraction and repulsion forces.
It provides a draw
and update
function, which can be used to easily write your own rendering code.
The draw
function should be called with two callback functions. The first callback will be used for all nodes and the second one for all the edges.
graph:draw( function( node )
local x, y = node:getPosition()
drawCircle( 'fill', x, y, 10 )
end,
function( edge )
local ox, oy = edge.origin:getPosition()
local tx, ty = edge.target:getPosition()
drawLine( ox, oy, tx, ty )
end)
At its simplest the force calculations can be updated via graph:update( dt )
, but the update
function also can receive optional callbacks for both nodes and edges.
Anchors can be used to attach a node to a certain position on the screen. This can be useful if you want to center a certain node for example.
This can either be done directly via the constructor of the node:
-- Anchor the node to the center of the screen.
graph:addNode( "Ash Williams", screenX * 0.5, screenY * 0.5, true )
Or by using the setAnchor
function:
-- Invert anchor status
node:setAnchor( not node:isAnchor(), mouseX, mouseY )
If you prefer to not touch the default classes, you can simply inherit from them and tell Graphoon to use your custom classes instead.
local GraphLibraryNode = require('lib.libfdgraph.fd').Node
local CustomNodeClass = {}
-- You can pass additional arguments to your custom class. Just make sure the
-- default parameters ar in the right order.
function CustomNodeClass.new( id, x, y, anchor, ... )
local self = GraphLibraryNode.new( id, x, y, anchor )
-- ... Custom code
end
return CustomNodeClass
local GraphLibrary = require('Graphoon').Graph
GraphLibrary.setNodeClass( require('CustomNodeClass') )