feat(treemap): fix TreeMapD3 nodes exit

plouc · May 13, 2016 · 9e02da4 · 9e02da4
1 parent 3b61508
commit 9e02da4
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/components/charts/treemap/TreeMapD3.js b/src/components/charts/treemap/TreeMapD3.js
@@ -72,10 +72,11 @@ class TreeMapD3 extends Component {
 
         const nodes = wrapper.datum(root).selectAll('.nivo_treemap_node').data(treemap.nodes);
 
-        const newNodes = nodes.enter().append('div')
+        nodes.enter().append('div')
             .classed('nivo_treemap_node', true)
             .classed('_is-parent', d => (d.children && d.children.length > 0))
             .classed('_is-child', d => (d.parent !== undefined && (!d.children || d.children.length === 0)))
+            .style('z-index', 10)
             .each(function (d) {
                 if (d.depth > 1 ) {
                     d.color = d3.rgb(d.parent.color).brighter(.2);
@@ -139,6 +140,17 @@ class TreeMapD3 extends Component {
                 ;
             })
         ;
+
+        nodes.exit()
+            .style('z-index', 5)
+            .transition()
+            .duration(transitionDuration)
+            .ease(transitionEasing)
+            .style('width', '0px')
+            .style('height', '0px')
+            .style('opacity', 0)
+            .remove()
+        ;
     }
 
     shouldComponentUpdate(nextProps) {