After create node, set node size to max between label size and shape …

…size.
kanitw · Aug 18, 2015 · bdc4f83 · bdc4f83
1 parent 40198f9
commit bdc4f83
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/create-nodes.js b/lib/create-nodes.js
@@ -24,7 +24,9 @@ function createNodes(selection, g, shapes) {
         labelGroup = thisGroup.append("g").attr("class", "label"),
         labelDom = addLabel(labelGroup, node),
         shape = shapes[node.shape],
-        bbox = _.pick(labelDom.node().getBBox(), "width", "height");
+        bbox = _.pick(labelDom.node().getBBox(), "width", "height"),
+        labelWidth = bbox.width,
+        labelHeight = bbox.height;
 
     node.elem = this;
 
@@ -46,8 +48,8 @@ function createNodes(selection, g, shapes) {
     util.applyStyle(shapeSvg, node.style);
 
     var shapeBBox = shapeSvg.node().getBBox();
-    node.width = shapeBBox.width;
-    node.height = shapeBBox.height;
+    node.width = Math.max(shapeBBox.width, labelWidth);
+    node.height = Math.max(shapeBBox.height, labelHeight);
   });
 
   util.applyTransition(svgNodes.exit(), g)