@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>
281 lines (280 loc) • 12 kB
JavaScript
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 _defineProperty(obj, key, value) {
if (key in obj) {
Object.defineProperty(obj, key, {
value: value,
enumerable: true,
configurable: true,
writable: true
});
} else {
obj[key] = value;
}
return obj;
}
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 _objectSpread(target) {
for(var i = 1; i < arguments.length; i++){
var source = arguments[i] != null ? arguments[i] : {};
var ownKeys = Object.keys(source);
if (typeof Object.getOwnPropertySymbols === "function") {
ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function(sym) {
return Object.getOwnPropertyDescriptor(source, sym).enumerable;
}));
}
ownKeys.forEach(function(key) {
_defineProperty(target, key, source[key]);
});
}
return target;
}
function ownKeys(object, enumerableOnly) {
var keys = Object.keys(object);
if (Object.getOwnPropertySymbols) {
var symbols = Object.getOwnPropertySymbols(object);
if (enumerableOnly) {
symbols = symbols.filter(function(sym) {
return Object.getOwnPropertyDescriptor(object, sym).enumerable;
});
}
keys.push.apply(keys, symbols);
}
return keys;
}
function _objectSpreadProps(target, source) {
source = source != null ? source : {};
if (Object.getOwnPropertyDescriptors) {
Object.defineProperties(target, Object.getOwnPropertyDescriptors(source));
} else {
ownKeys(Object(source)).forEach(function(key) {
Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key));
});
}
return target;
}
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);
}
import { useState, useMemo } from "react";
import { cloneDeep } from "lodash-es";
import { sankey as d3Sankey } from "d3-sankey";
import { useTheme, usePropertyAccessor, useValueFormatter } from "@dfnivo/core";
import { useOrdinalColorScale, useInheritedColor } from "@dfnivo/colors";
import { sankeyAlignmentFromProp } from "./props";
var getId = function(node) {
return node.id;
};
export 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 = d3Sankey().nodeAlign(alignFunction).nodeWidth(nodeThickness).nodePadding(nodeSpacing).size(layout === "horizontal" ? [
width,
height
] : [
height,
width
]).nodeId(getId);
if (sortFunction) {
sankey.nodeSort(sortFunction);
}
if (linkSortMode) {
sankey.linkSort(linkSortMode);
}
// deep clone is required as the sankey diagram mutates data
// we need a different identity for correct updates
var _sankey = sankey(cloneDeep(_data)), nodes = _sankey.nodes, links = _sankey.links;
return {
nodes: nodes.map(function(node) {
var cloneNode = _objectSpreadProps(_objectSpread({}, node), {
color: "",
label: "",
formattedValue: "",
x: 0,
y: 0,
height: 0,
width: 0,
layer: 0
});
cloneNode.color = getColor(cloneNode);
cloneNode.label = getLabel(cloneNode);
var _cloneNode_value;
cloneNode.formattedValue = formatValue((_cloneNode_value = cloneNode.value) !== null && _cloneNode_value !== void 0 ? _cloneNode_value : 0);
if (layout === "horizontal") {
var _cloneNode_x0;
cloneNode.x = ((_cloneNode_x0 = cloneNode.x0) !== null && _cloneNode_x0 !== void 0 ? _cloneNode_x0 : 0) + nodeInnerPadding;
var _cloneNode_y0;
cloneNode.y = (_cloneNode_y0 = cloneNode.y0) !== null && _cloneNode_y0 !== void 0 ? _cloneNode_y0 : 0;
var _cloneNode_x1, _cloneNode_x01;
cloneNode.width = Math.max(((_cloneNode_x1 = cloneNode.x1) !== null && _cloneNode_x1 !== void 0 ? _cloneNode_x1 : 0) - ((_cloneNode_x01 = cloneNode.x0) !== null && _cloneNode_x01 !== void 0 ? _cloneNode_x01 : 0) - nodeInnerPadding * 2, 0);
var _cloneNode_y1, _cloneNode_y01;
cloneNode.height = Math.max(((_cloneNode_y1 = cloneNode.y1) !== null && _cloneNode_y1 !== void 0 ? _cloneNode_y1 : 0) - ((_cloneNode_y01 = cloneNode.y0) !== null && _cloneNode_y01 !== void 0 ? _cloneNode_y01 : 0), 0);
} else {
var _cloneNode_y02;
cloneNode.x = (_cloneNode_y02 = cloneNode.y0) !== null && _cloneNode_y02 !== void 0 ? _cloneNode_y02 : 0;
var _cloneNode_x02;
cloneNode.y = ((_cloneNode_x02 = cloneNode.x0) !== null && _cloneNode_x02 !== void 0 ? _cloneNode_x02 : 0) + nodeInnerPadding;
var _cloneNode_y11, _cloneNode_y03;
cloneNode.width = Math.max(((_cloneNode_y11 = cloneNode.y1) !== null && _cloneNode_y11 !== void 0 ? _cloneNode_y11 : 0) - ((_cloneNode_y03 = cloneNode.y0) !== null && _cloneNode_y03 !== void 0 ? _cloneNode_y03 : 0), 0);
var _cloneNode_x11, _cloneNode_x03;
cloneNode.height = Math.max(((_cloneNode_x11 = cloneNode.x1) !== null && _cloneNode_x11 !== void 0 ? _cloneNode_x11 : 0) - ((_cloneNode_x03 = cloneNode.x0) !== null && _cloneNode_x03 !== void 0 ? _cloneNode_x03 : 0) - nodeInnerPadding * 2, 0);
var oldX0 = cloneNode.x0;
var oldX1 = cloneNode.x1;
cloneNode.x0 = cloneNode.y0;
cloneNode.x1 = cloneNode.y1;
cloneNode.y0 = oldX0;
cloneNode.y1 = oldX1;
}
return cloneNode;
}),
links: links.map(function(link) {
var clonedLink = _objectSpreadProps(_objectSpread({}, link), {
thickness: 1,
color: "",
formattedValue: ""
});
clonedLink.formattedValue = formatValue(clonedLink.value);
if (typeof clonedLink.source === "object") {
clonedLink.color = clonedLink.source.color || "";
}
clonedLink.pos0 = clonedLink.y0;
clonedLink.pos1 = clonedLink.y1;
var _clonedLink_width;
clonedLink.thickness = (_clonedLink_width = clonedLink.width) !== null && _clonedLink_width !== void 0 ? _clonedLink_width : 0;
delete clonedLink.y0;
delete clonedLink.y1;
delete clonedLink.width;
return clonedLink;
})
};
};
export 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(useState(null), 2), currentNode = _useState[0], setCurrentNode = _useState[1];
var _useState1 = _slicedToArray(useState(null), 2), currentLink = _useState1[0], setCurrentLink = _useState1[1];
var sortFunction = useMemo(function() {
if (sort === "auto") return undefined;
if (sort === "input") return null;
if (sort === "ascending") {
return function(a, b) {
return Number(a.value) - Number(b.value);
};
}
if (sort === "descending") {
return function(a, b) {
return Number(b.value) - Number(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 = useMemo(function() {
if (typeof align === "function") return align;
return sankeyAlignmentFromProp(align);
}, [
align
]);
var theme = useTheme();
var getColor = useOrdinalColorScale(colors, "id");
var getNodeBorderColor = useInheritedColor(nodeBorderColor, theme);
var getLabel = usePropertyAccessor(label);
var getLabelTextColor = useInheritedColor(labelTextColor, theme);
var formatValue = useValueFormatter(valueFormat);
var _useMemo = 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 = 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