Fixes issue with dynamic changes in depthFactor

Author: bkrem

src/Node/index.js

@@ -35,13 +35,7 @@ export default class Node extends React.Component {
   }
 
   setTransformOrientation(x, y) {
-    const { orientation, nodeData, depthFactor } = this.props;
-
-    if (depthFactor) {
-      y = nodeData.depth * depthFactor;
-    }
-
-    return orientation === 'horizontal' ?
+    return this.props.orientation === 'horizontal' ?
       `translate(${y},${x})` :
       `translate(${x},${y})`;
   }

src/Tree/index.js

@@ -192,7 +192,7 @@ export default class Tree extends React.Component {
    * @return {object} Object containing `nodes` and `links` fields.
    */
   generateTree() {
-    const { initialDepth } = this.props;
+    const { initialDepth, depthFactor } = this.props;
     const tree = layout.tree()
       .nodeSize([100 + 40, 100 + 40])
       .separation((d) => d._children ? 1.2 : 0.9)
@@ -207,6 +207,10 @@ export default class Tree extends React.Component {
       this.setInitialTreeDepth(nodes, initialDepth);
     }
 
+    if (depthFactor) {
+      nodes.forEach((node) => { node.y = node.depth * depthFactor; });
+    }
+
     return { nodes, links };
   }