Function buildGroup

tensorboard/plugins/graph/tf_graph_common/node.ts:87–194

(
  sceneGroup,
  nodeData: render.RenderNodeInfo[],
  sceneElement
)

 * @return selection of the created nodeGroups
 */
export function buildGroup(
  sceneGroup,
  nodeData: render.RenderNodeInfo[],
  sceneElement
) {
  let container = tf_graph_common.selectOrCreateChild(
    sceneGroup,
    'g',
    Class.Node.CONTAINER
  );
  // Select all children and join with data.
  // (Note that all children of g.nodes are g.node)
  let nodeGroups = (container as any)
    .selectAll(function () {
      return this.childNodes;
    })
    .data(nodeData, (d) => {
      // make sure that we don't have to swap shape type
      return d.node.name + ':' + d.node.type;
    });
  // ENTER
  nodeGroups
    .enter()
    .append('g')
    .attr('data-name', (d) => {
      return d.node.name;
    })
    .each(function (d) {
      let nodeGroup = d3.select(this);
      // index node group for quick stylizing
      sceneElement.addNodeGroup(d.node.name, nodeGroup);
    })
    .merge(nodeGroups)
    // ENTER + UPDATE
    .attr('class', (d) => {
      return Class.Node.GROUP + ' ' + nodeClass(d);
    })
    .each(function (d) {
      let nodeGroup = d3.select(this);
      // Add g.in-annotations (always add -- to keep layer order
      // consistent.)
      let inAnnotationBox = tf_graph_common.selectOrCreateChild(
        nodeGroup,
        'g',
        Class.Annotation.INBOX
      );
      buildGroupForAnnotation(
        inAnnotationBox,
        d.inAnnotations,
        d,
        sceneElement
      );
      // Add g.out-annotations  (always add -- to keep layer order
      // consistent.)
      let outAnnotationBox = tf_graph_common.selectOrCreateChild(
        nodeGroup,
        'g',
        Class.Annotation.OUTBOX