force-directed-graph-components/lib/util/colors.js

force directed graph components

import { scaleOrdinal } from "d3"

export const transformColors = (colors, property) => {
  const res = {}
  if (typeof colors === 'function') {
    res.content = colors
  } else if (colors.valueList && colors.valueList.length && colors.colorList && colors.colorList.length && colors.property) {
    res.content = scaleOrdinal(colors.valueList, colors.colorList)
  }

  res.property = property
  
  return res
}