d3-force-attract/lib/forceAttract.js

Attraction force for d3.forceSimulation.

/**
 * Pulls nodes toward a specified `(x, y)` target point.
 */
export default function (target) {

  let nodes,
    targets,
    strength,
    strengths;

  function force (alpha) {
    let node, target, strength;
    for (let i=0; i<nodes.length; i++) {
      node = nodes[i];
      target = targets[i];
      strength = strengths[i];
      node.vx += (target[0] - node.x) * strength * alpha;
      node.vy += (target[1] - node.y) * strength * alpha;
    }
  }

  function initialize () {
    if (!nodes) return;

    // populate local `strengths` using `strength` accessor
    strengths = new Array(nodes.length);
    for (let i=0; i<nodes.length; i++) strengths[i] = strength(nodes[i], i, nodes);

    // populate local `targets` using `target` accessor
    targets = new Array(nodes.length);
    for (let i=0; i<nodes.length; i++) targets[i] = target(nodes[i], i, nodes);
  }

  force.initialize = _ => {
    nodes = _;
    initialize();
  };

  force.strength = _ => {
    // return existing value if no value passed
    if (_ == null) return strength;

    // coerce `strength` accessor into a function
    strength = typeof _ === 'function' ? _ : () => +_;

    // reinitialize
    initialize();

    // allow chaining
    return force;
  };

  force.target = _ => {
    // return existing value if no value passed
    if (_ == null) return target;

    // coerce `target` accessor into a function
    target = typeof _ === 'function' ? _ : () => _;

    // reinitialize
    initialize();

    // allow chaining
    return force;
  };

  if (!strength) force.strength(0.1);
  if (!target) force.target([ 0, 0 ]);

  return force;

}