@dfnivo/sankey
Version:
<a href="https://nivo.rocks"><img alt="nivo" src="https://raw.githubusercontent.com/plouc/nivo/master/nivo.png" width="216" height="68"/></a>
220 lines (219 loc) • 8.29 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
function _export(target, all) {
for(var name in all)Object.defineProperty(target, name, {
enumerable: true,
get: all[name]
});
}
_export(exports, {
computeNodeAndLinks: function() {
return computeNodeAndLinks;
},
useSankey: function() {
return useSankey;
}
});
var _react = require("react");
var _lodashEs = require("lodash-es");
var _d3Sankey = require("d3-sankey");
var _core = require("@dfnivo/core");
var _colors = require("@dfnivo/colors");
var _props = require("./props");
function _arrayLikeToArray(arr, len) {
if (len == null || len > arr.length) len = arr.length;
for(var i = 0, arr2 = new Array(len); i < len; i++)arr2[i] = arr[i];
return arr2;
}
function _arrayWithHoles(arr) {
if (Array.isArray(arr)) return arr;
}
function _iterableToArrayLimit(arr, i) {
var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"];
if (_i == null) return;
var _arr = [];
var _n = true;
var _d = false;
var _s, _e;
try {
for(_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true){
_arr.push(_s.value);
if (i && _arr.length === i) break;
}
} catch (err) {
_d = true;
_e = err;
} finally{
try {
if (!_n && _i["return"] != null) _i["return"]();
} finally{
if (_d) throw _e;
}
}
return _arr;
}
function _nonIterableRest() {
throw new TypeError("Invalid attempt to destructure non-iterable instance.\\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
}
function _slicedToArray(arr, i) {
return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest();
}
function _unsupportedIterableToArray(o, minLen) {
if (!o) return;
if (typeof o === "string") return _arrayLikeToArray(o, minLen);
var n = Object.prototype.toString.call(o).slice(8, -1);
if (n === "Object" && o.constructor) n = o.constructor.name;
if (n === "Map" || n === "Set") return Array.from(n);
if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen);
}
var getId = function(node) {
return node.id;
};
var computeNodeAndLinks = function(param) {
var _data = param.data, formatValue = param.formatValue, layout = param.layout, alignFunction = param.alignFunction, sortFunction = param.sortFunction, linkSortMode = param.linkSortMode, nodeThickness = param.nodeThickness, nodeSpacing = param.nodeSpacing, nodeInnerPadding = param.nodeInnerPadding, width = param.width, height = param.height, getColor = param.getColor, getLabel = param.getLabel;
var sankey = (0, _d3Sankey.sankey)().nodeAlign(alignFunction)// @ts-ignore: this method signature is incorrect in current @types/d3-sankey
.nodeSort(sortFunction)// @ts-ignore: this method is not available in current @types/d3-sankey
.linkSort(linkSortMode).nodeWidth(nodeThickness).nodePadding(nodeSpacing).size(layout === "horizontal" ? [
width,
height
] : [
height,
width
]).nodeId(getId);
// deep clone is required as the sankey diagram mutates data
// we need a different identity for correct updates
var data = (0, _lodashEs.cloneDeep)(_data);
sankey(data);
data.nodes.forEach(function(node) {
node.color = getColor(node);
node.label = getLabel(node);
node.formattedValue = formatValue(node.value);
if (layout === "horizontal") {
node.x = node.x0 + nodeInnerPadding;
node.y = node.y0;
node.width = Math.max(node.x1 - node.x0 - nodeInnerPadding * 2, 0);
node.height = Math.max(node.y1 - node.y0, 0);
} else {
node.x = node.y0;
node.y = node.x0 + nodeInnerPadding;
node.width = Math.max(node.y1 - node.y0, 0);
node.height = Math.max(node.x1 - node.x0 - nodeInnerPadding * 2, 0);
var oldX0 = node.x0;
var oldX1 = node.x1;
node.x0 = node.y0;
node.x1 = node.y1;
node.y0 = oldX0;
node.y1 = oldX1;
}
});
data.links.forEach(function(link) {
link.formattedValue = formatValue(link.value);
link.color = link.source.color;
// @ts-ignore
link.pos0 = link.y0;
// @ts-ignore
link.pos1 = link.y1;
// @ts-ignore
link.thickness = link.width;
// @ts-ignore
delete link.y0;
// @ts-ignore
delete link.y1;
// @ts-ignore
delete link.width;
});
return data;
};
var useSankey = function(param) {
var data = param.data, valueFormat = param.valueFormat, layout = param.layout, width = param.width, height = param.height, sort = param.sort, align = param.align, colors = param.colors, nodeThickness = param.nodeThickness, nodeSpacing = param.nodeSpacing, nodeInnerPadding = param.nodeInnerPadding, nodeBorderColor = param.nodeBorderColor, label = param.label, labelTextColor = param.labelTextColor;
var _useState = _slicedToArray((0, _react.useState)(null), 2), currentNode = _useState[0], setCurrentNode = _useState[1];
var _useState1 = _slicedToArray((0, _react.useState)(null), 2), currentLink = _useState1[0], setCurrentLink = _useState1[1];
var sortFunction = (0, _react.useMemo)(function() {
if (sort === "auto") return undefined;
if (sort === "input") return null;
if (sort === "ascending") {
return function(a, b) {
return a.value - b.value;
};
}
if (sort === "descending") {
return function(a, b) {
return b.value - a.value;
};
}
return sort;
}, [
sort
]);
// If "input" sorting is used, apply this setting to links, too.
// (In d3, `null` means input sorting and `undefined` is the default)
var linkSortMode = sort === "input" ? null : undefined;
var alignFunction = (0, _react.useMemo)(function() {
if (typeof align === "function") return align;
return (0, _props.sankeyAlignmentFromProp)(align);
}, [
align
]);
var theme = (0, _core.useTheme)();
var getColor = (0, _colors.useOrdinalColorScale)(colors, "id");
var getNodeBorderColor = (0, _colors.useInheritedColor)(nodeBorderColor, theme);
var getLabel = (0, _core.usePropertyAccessor)(label);
var getLabelTextColor = (0, _colors.useInheritedColor)(labelTextColor, theme);
var formatValue = (0, _core.useValueFormatter)(valueFormat);
var _useMemo = (0, _react.useMemo)(function() {
return computeNodeAndLinks({
data: data,
formatValue: formatValue,
layout: layout,
alignFunction: alignFunction,
sortFunction: sortFunction,
linkSortMode: linkSortMode,
nodeThickness: nodeThickness,
nodeSpacing: nodeSpacing,
nodeInnerPadding: nodeInnerPadding,
width: width,
height: height,
getColor: getColor,
getLabel: getLabel
});
}, [
data,
formatValue,
layout,
alignFunction,
sortFunction,
linkSortMode,
nodeThickness,
nodeSpacing,
nodeInnerPadding,
width,
height,
getColor,
getLabel
]), nodes = _useMemo.nodes, links = _useMemo.links;
var legendData = (0, _react.useMemo)(function() {
return nodes.map(function(node) {
return {
id: node.id,
label: node.label,
color: node.color
};
});
}, [
nodes
]);
return {
nodes: nodes,
links: links,
legendData: legendData,
getNodeBorderColor: getNodeBorderColor,
currentNode: currentNode,
setCurrentNode: setCurrentNode,
currentLink: currentLink,
setCurrentLink: setCurrentLink,
getLabelTextColor: getLabelTextColor
};
};
//# sourceMappingURL=hooks.js.map