[TASK] add possibility to attach data in moreInfo in sankey links and node

Sebastian Wojtowicz · Sebastian Wojtowicz · commit 40b73e94b715 · 2019-08-29T12:51:13.000+02:00
changes in src files now
diff --git a/src/traces/sankey/attributes.js b/src/traces/sankey/attributes.js
@@ -181,6 +181,10 @@ var attrs = module.exports = overrideAll({
             description: 'Variables `sourceLinks` and `targetLinks` are arrays of link objects.',
             keys: ['value', 'label']
         }),
+        moreInfo: {
+            valType: 'data_array',
+            dflt: [],
+        },
         description: 'The nodes of the Sankey plot.'
     },
 
@@ -255,6 +259,10 @@ var attrs = module.exports = overrideAll({
             description: 'Variables `source` and `target` are node objects.',
             keys: ['value', 'label']
         }),
+        moreInfo: {
+            valType: 'data_array',
+            dflt: [],
+        },
         colorscales: templatedArray('concentrationscales', {
             editType: 'calc',
             label: {
diff --git a/src/traces/sankey/defaults.js b/src/traces/sankey/defaults.js
@@ -42,6 +42,7 @@ module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout
     coerceNode('hoverinfo', traceIn.hoverinfo);
     handleHoverLabelDefaults(nodeIn, nodeOut, coerceNode, hoverlabelDefault);
     coerceNode('hovertemplate');
+    coerceNode('moreInfo');
 
     var colors = layout.colorway;
 
@@ -67,6 +68,7 @@ module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout
     coerceLink('hoverinfo', traceIn.hoverinfo);
     handleHoverLabelDefaults(linkIn, linkOut, coerceLink, hoverlabelDefault);
     coerceLink('hovertemplate');
+    coerceLink('moreInfo');
 
     var defaultLinkColor = tinycolor(layout.paper_bgcolor).getLuminance() < 0.333 ?
                 'rgba(255, 255, 255, 0.6)' :
diff --git a/src/traces/sankey/render.js b/src/traces/sankey/render.js
@@ -296,6 +296,7 @@ function linkModel(d, l, i) {
     // for event data
     l.trace = d.trace;
     l.curveNumber = d.trace.index;
+    l.moreInfo = d.trace.link.moreInfo[i];
 
     return {
         circular: d.circular,
@@ -483,6 +484,7 @@ function nodeModel(d, n) {
     // for event data
     n.trace = d.trace;
     n.curveNumber = d.trace.index;
+    n.moreInfo = d.trace.node.moreInfo[n.pointNumber];
 
     return {
         index: n.pointNumber,
