Function getGradient

tensorboard/plugins/graph/tf_graph_common/node.ts:732–769

(
  id: string,
  colors: Array<{
    color: string;
    proportion: number;
  }>,
  svgRoot?: SVGElement
)

}

function getGradient(
  id: string,
  colors: Array<{
    color: string;
    proportion: number;
  }>,
  svgRoot?: SVGElement
): string {
  let escapedId = tf_graph_util.escapeQuerySelector(id);
  if (!svgRoot) return `url(#${escapedId})`;
  let $svgRoot = d3.select(svgRoot);
  let gradientDefs = $svgRoot.select('defs#_graph-gradients');
  if (gradientDefs.empty()) {
    gradientDefs = $svgRoot.append('defs').attr('id', '_graph-gradients');
  }
  let linearGradient = gradientDefs.select('linearGradient#' + escapedId);
  // If the linear gradient is not there yet, create it.
  if (linearGradient.empty()) {
    linearGradient = gradientDefs.append('linearGradient').attr('id', id);
    // Re-create the stops of the linear gradient.
    linearGradient.selectAll('*').remove();
    let cumulativeProportion = 0;
    // For each color, create a stop using the proportion of that device.
    _.each(colors, (d) => {
      let color = d.color;
      linearGradient
        .append('stop')
        .attr('offset', cumulativeProportion)
        .attr('stop-color', color);
      linearGradient
        .append('stop')
        .attr('offset', cumulativeProportion + d.proportion)
        .attr('stop-color', color);
      cumulativeProportion += d.proportion;
    });
  }
  return `url(#${escapedId})`;
}
export function removeGradientDefinitions(svgRoot: SVGElement) {
  d3.select(svgRoot).select('defs#_graph-gradients').remove();
}