Graphoon/fd/Node.lua

local Node = {};

local FORCE_SPRING = -0.01;
local FORCE_CHARGE = 100000;

local FORCE_MAX = 4;
local NODE_SPEED = 8;
local DAMPING_FACTOR = 0.95;

local DEFAULT_MASS = 0.05;

---
-- @param id - A unique id which will be used to reference this node.
--
function Node.new( id, name, x, y )
    local self = {};

    local name = name or id;
    local px, py = x or 0, y or 0;
    local ax, ay = 0, 0;
    local vx, vy = 0, 0;

    ---
    -- Clamps a value to a certain range.
    -- @param min
    -- @param val
    -- @param max
    --
    local function clamp( min, val, max )
        return math.max( min, math.min( val, max ) );
    end

    ---
    -- Calculates the new xy-acceleration for this node.
    -- The values are clamped to keep the graph from "exploding".
    -- @param fx - The force to apply in x-direction.
    -- @param fy - The force to apply in y-direction.
    --
    local function applyForce( fx, fy )
        ax = clamp( -FORCE_MAX, ax + fx, FORCE_MAX );
        ay = clamp( -FORCE_MAX, ay + fy, FORCE_MAX );
    end

    function self:attractTo( node )
        local dx, dy = px - node:getX(), py - node:getY();
        local distance = math.sqrt(dx * dx + dy * dy);
        dx = dx / distance;
        dy = dy / distance;

        local strength = FORCE_SPRING * distance;
        applyForce( dx * strength, dy * strength );
    end

    function self:repelFrom( node )
        local dx, dy = px - node:getX(), py - node:getY();
        local distance = math.sqrt(dx * dx + dy * dy);
        dx = dx / distance;
        dy = dy / distance;

        local strength = FORCE_CHARGE * (DEFAULT_MASS / (distance * distance));
        applyForce(dx * strength, dy * strength);
    end

    ---
    -- Update the node's position based on the calculated velocity and
    -- acceleration.
    --
    function self:move( dt )
        vx = (vx + ax * dt * NODE_SPEED) * DAMPING_FACTOR;
        vy = (vy + ay * dt * NODE_SPEED) * DAMPING_FACTOR;
        px = px + vx;
        py = py + vy;
        ax, ay = 0, 0;
    end

    ---
    -- Returns the node's unique identifier.
    --
    function self:getID()
        return id;
    end

    function self:getName()
        return name;
    end

    function self:getX()
        return px;
    end

    function self:getY()
        return py;
    end

    function self:getPosition()
        return px, py;
    end

    function self:setPosition( nx, ny )
        px, py = nx, ny;
    end

    return self;
end

return Node;