sankey: reset to initial view (x, y and groups)

Author: antoinerg

src/components/modebar/buttons.js

@@ -518,9 +518,18 @@ modeBarButtons.resetSankeyGroup = {
     title: function(gd) { return _(gd, 'Ungroup all nodes'); },
     icon: Icons.home,
     click: function(gd) {
-        Registry.call('restyle', gd, {
-            'node.groups': [[]],
-        });
+        var aObj = {
+            'node.groups': [],
+            'node.x': [],
+            'node.y': []
+        };
+        for(var i = 0; i < gd._fullData.length; i++) {
+            var viewInitial = gd._fullData[i]._viewInitial;
+            aObj['node.groups'].push(viewInitial.node.groups.slice());
+            aObj['node.x'].push(viewInitial.node.x.slice());
+            aObj['node.y'].push(viewInitial.node.y.slice());
+        }
+        Registry.call('restyle', gd, aObj);
     }
 };
 

src/traces/sankey/plot.js

@@ -127,6 +127,20 @@ module.exports = function plot(gd, calcData) {
     var svg = fullLayout._paper;
     var size = fullLayout._size;
 
+    // stash initial view
+    for(var i = 0; i < calcData.length; i++) {
+        if(!gd._fullData[i]._viewInitial) {
+            var node = gd._fullData[i].node;
+            gd._fullData[i]._viewInitial = {
+                node: {
+                    groups: node.groups.slice(),
+                    x: node.x.slice(),
+                    y: node.y.slice()
+                }
+            };
+        }
+    }
+
     var linkSelect = function(element, d) {
         var evt = d.link;
         evt.originalEvent = d3.event;