sankey: introduce node.customdata

Author: antoinerg

src/traces/sankey/attributes.js

@@ -128,6 +128,13 @@ var attrs = module.exports = overrideAll({
                 'what is beneath the node.'
             ].join(' ')
         },
+        customdata: {
+            valType: 'data_array',
+            editType: 'calc',
+            description: [
+                'Assigns extra data each node.'
+            ].join(' ')
+        },
         line: {
             color: {
                 valType: 'color',

src/traces/sankey/calc.js

@@ -116,6 +116,7 @@ function convertToD3Sankey(trace) {
     // Process nodes
     var totalCount = nodeCount + groups.length;
     var hasNodeColorArray = isArrayOrTypedArray(nodeSpec.color);
+    var hasNodeCustomdataArray = isArrayOrTypedArray(nodeSpec.customdata);
     var nodes = [];
     for(i = 0; i < totalCount; i++) {
         if(!linkedNodes[i]) continue;
@@ -126,7 +127,8 @@ function convertToD3Sankey(trace) {
             childrenNodes: [],
             pointNumber: i,
             label: l,
-            color: hasNodeColorArray ? nodeSpec.color[i] : nodeSpec.color
+            color: hasNodeColorArray ? nodeSpec.color[i] : nodeSpec.color,
+            customdata: hasNodeCustomdataArray ? nodeSpec.customdata[i] : nodeSpec.customdata
         });
     }
 

src/traces/sankey/defaults.js

@@ -50,6 +50,7 @@ module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout
     coerceNode('color', nodeOut.label.map(function(d, i) {
         return Color.addOpacity(defaultNodePalette(i), 0.8);
     }));
+    coerceNode('customdata');
 
     // link attributes
     var linkIn = traceIn.link || {};

test/jasmine/tests/sankey_test.js

@@ -816,6 +816,7 @@ describe('sankey tests', function() {
         it('should show the correct hover labels when hovertemplate is specified', function(done) {
             var gd = createGraphDiv();
             var mockCopy = Lib.extendDeep({}, mock);
+            mockCopy.data[0].node.customdata = [0, 0, 0, 0, '15']
 
             Plotly.plot(gd, mockCopy).then(function() {
                 _hover(404, 302);
@@ -836,15 +837,15 @@ describe('sankey tests', function() {
             // Test (node|link).hovertemplate
             .then(function() {
                 return Plotly.restyle(gd, {
-                    'node.hovertemplate': 'hovertemplate<br>%{value}<br>%{value:0.2f}<extra>%{fullData.name}</extra>',
+                    'node.hovertemplate': 'hovertemplate<br>%{value}<br>%{value:0.2f}<br>%{customdata}<extra>%{fullData.name}</extra>',
                     'link.hovertemplate': 'hovertemplate<br>source: %{source.label}<br>target: %{target.label}<br>size: %{value:0.0f}TWh<extra>%{fullData.name}</extra>'
                 });
             })
             .then(function() {
                 _hover(404, 302);
 
                 assertLabel(
-                    [ 'hovertemplate', '447TWh', '447.48', 'trace 0'],
+                    [ 'hovertemplate', '447TWh', '447.48', '15', 'trace 0'],
                     ['rgb(148, 103, 189)', 'rgb(255, 255, 255)', 13, 'Arial', 'rgb(255, 255, 255)']
                 );
             })