08-27-周三_17-09-29

node_modules/dagre-d3-renderer/lib/label/add-label.js

@@ -0,0 +1,37 @@
+import addTextLabel from './add-text-label'
+import addHtmlLabel from './add-html-label'
+import addSVGLabel from './add-svg-label'
+
+function addLabel (root, node, location) {
+  const label = node.label
+  const labelSvg = root.append('g')
+
+  // Allow the label to be a string, a function that returns a DOM element, or
+  // a DOM element itself.
+  if (node.labelType === 'svg') {
+    addSVGLabel(labelSvg, node)
+  } else if (typeof label !== 'string' || node.labelType === 'html') {
+    addHtmlLabel(labelSvg, node)
+  } else {
+    addTextLabel(labelSvg, node)
+  }
+
+  const labelBBox = labelSvg.node().getBBox()
+  let y
+  switch (location) {
+    case 'top':
+      y = (-node.height / 2)
+      break
+    case 'bottom':
+      y = (node.height / 2) - labelBBox.height
+      break
+    default:
+      y = (-labelBBox.height / 2)
+  }
+  labelSvg.attr('transform',
+                'translate(' + (-labelBBox.width / 2) + ',' + y + ')')
+
+  return labelSvg
+}
+
+export default addLabel