Function stylize

tensorboard/plugins/graph/tf_graph_common/node.ts:871–907

(
  nodeGroup,
  renderInfo: render.RenderNodeInfo,
  sceneElement: TfGraphScene,
  nodeClass?
)

 * that can't be done in css).
 */
export function stylize(
  nodeGroup,
  renderInfo: render.RenderNodeInfo,
  sceneElement: TfGraphScene,
  nodeClass?
) {
  nodeClass = nodeClass || Class.Node.SHAPE;
  const isHighlighted = sceneElement.isNodeHighlighted(renderInfo.node.name);
  const isSelected = sceneElement.isNodeSelected(renderInfo.node.name);
  const isExtract =
    renderInfo.isInExtract ||
    renderInfo.isOutExtract ||
    renderInfo.isLibraryFunction;
  const isExpanded = renderInfo.expanded && nodeClass !== Class.Annotation.NODE;
  const isFadedOut = renderInfo.isFadedOut;
  nodeGroup.classed('highlighted', isHighlighted);
  nodeGroup.classed('selected', isSelected);
  nodeGroup.classed('extract', isExtract);
  nodeGroup.classed('expanded', isExpanded);
  nodeGroup.classed('faded', isFadedOut);
  // Main node always exists here and it will be reached before subscene,
  // so d3 selection is fine here.
  const node = nodeGroup.select(
    '.' + nodeClass + ' .' + Class.Node.COLOR_TARGET
  );
  const fillColor = getFillForNode(
    sceneElement.templateIndex,
    sceneElement.colorBy,
    renderInfo,
    isExpanded,
    sceneElement.getGraphSvgRoot()
  );
  node.style('fill', fillColor);
  // Choose outline to be darker version of node color if the node is a single
  // color and is not selected.
  node.style('stroke', isSelected ? null : getStrokeForFill(fillColor));
}
/**
 * Given a node's fill color/gradient, determine the stroke for the node.
 */