@antv/hierarchy 0.6.8 → 0.6.9

package/src/layout/indented.js

@@ -1,4 +1,6 @@
-function positionNode(node, previousNode, indent, dropCap) {
+const util = require('../util');
+
+function positionNode(node, previousNode, indent, dropCap, align) {
   //  caculate the node's horizontal offset DX, dx's type might be number or function
   const displacementX = typeof indent === 'function' ? indent(node) : indent * node.depth;
 
@@ -15,13 +17,26 @@ function positionNode(node, previousNode, indent, dropCap) {
   }
 
   node.x += displacementX;
-  node.y = previousNode ? previousNode.y + previousNode.height : 0;
+  if (previousNode) {
+    node.y = previousNode.y + util.getHeight(previousNode, node, align);
+    if (node.parent.id !== previousNode.parent?.id) {
+      // previous node has different parent
+      const index = node.parent.children.findIndex(n => n.id === node.id);
+      const preNode = node.parent.children[index - 1];
+      if (preNode) {
+        const preY = preNode.y + util.getHeight(preNode, node, align);
+        node.y = preY > node.y ? preY : node.y;
+      }
+    }
+  } else {
+    node.y = 0;
+  }
   return;
 }
-module.exports = (root, indent, dropCap) => {
+module.exports = (root, indent, dropCap, align) => {
   let previousNode = null;
   root.eachNode(node => {
-    positionNode(node, previousNode, indent, dropCap);
+    positionNode(node, previousNode, indent, dropCap, align);
     previousNode = node;
   });
 };

package/src/util.js

@@ -1,4 +1,18 @@
 const { mix } = require('@antv/util');
+
+/**
+ * Get average height or height for node's position calculation, according to align.
+ * @param {*} preNode previous node
+ * @param {*} node current node, whose position is going to be calculated
+ * @param {'center' | undefined} align 'center' means nodes align at the center, other value means align at the left-top
+ * @param {string} heightField field name for height value on preNode and node
+ * @return {number} the height for calculation
+ */
+function getHeight(preNode, node, align, heightField = 'height') {
+  return align === 'center' ? (preNode[heightField] + node[heightField]) / 2 : preNode.height;
+}
+
 module.exports = {
-  assign: mix
+  assign: mix,
+  getHeight
 };