force-graph/src/color-utils.js

2D force-directed graph rendered on HTML5 canvas

import { scaleOrdinal } from 'd3-scale';
import { schemePaired } from 'd3-scale-chromatic';

const autoColorScale = scaleOrdinal(schemePaired);

// Autoset attribute colorField by colorByAccessor property
// If an object has already a color, don't set it
// Objects can be nodes or links
function autoColorObjects(objects, colorByAccessor, colorField) {
  if (!colorByAccessor || typeof colorField !== 'string') return;

  objects.filter(obj => !obj[colorField]).forEach(obj => {
    obj[colorField] = autoColorScale(colorByAccessor(obj));
  });
}

export { autoColorObjects };