@visactor/vchart 1.3.1-test.2 → 1.3.1-test.3

package/cjs/series/sankey/sankey.js

@@ -0,0 +1,665 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+}), exports.SankeySeries = void 0;
+
+const cartesian_1 = require("../cartesian/cartesian"), interface_1 = require("../interface"), vgrammar_sankey_1 = require("@visactor/vgrammar-sankey"), utils_1 = require("../../animation/utils"), config_1 = require("../../animation/config"), register_1 = require("../../data/register"), sankey_1 = require("../../data/transforms/sankey"), sankey_nodes_1 = require("../../data/transforms/sankey-nodes"), sankey_links_1 = require("../../data/transforms/sankey-links"), mark_1 = require("../../compile/mark"), vdataset_1 = require("@visactor/vdataset"), constant_1 = require("../../constant"), series_data_1 = require("../base/series-data"), add_property_1 = require("../../data/transforms/add-property"), data_key_1 = require("../../data/transforms/data-key"), util_1 = require("../../theme/color-scheme/util"), tooltip_helper_1 = require("./tooltip-helper"), vutils_1 = require("@visactor/vutils"), util_2 = require("../../util"), color_ordinal_scale_1 = require("../../scale/color-ordinal-scale"), vchart_1 = require("../../core/vchart"), rect_1 = require("../../mark/rect"), text_1 = require("../../mark/text"), link_path_1 = require("../../mark/link-path"), constant_2 = require("./constant"), flatten_1 = require("../../data/transforms/flatten");
+
+vchart_1.VChart.useMark([ rect_1.RectMark, link_path_1.LinkPathMark, text_1.TextMark ]), 
+(0, vgrammar_sankey_1.registerSankeyTransforms)();
+
+class SankeySeries extends cartesian_1.CartesianSeries {
+    constructor() {
+        super(...arguments), this.type = interface_1.SeriesTypeEnum.sankey, this._nodeLayoutZIndex = constant_1.LayoutZIndex.Node, 
+        this._labelLayoutZIndex = constant_1.LayoutZIndex.Label, this._viewBox = new vutils_1.Bounds, 
+        this._handleAdjacencyClick = params => {
+            const element = params.item;
+            element && element.mark.id().includes("node") ? this._handleNodeAdjacencyClick(element) : element && element.mark.id().includes("link") ? this._handleLinkAdjacencyClick(element) : this._handleClearEmpty();
+        }, this._handleRelatedClick = params => {
+            const element = params.item;
+            element && element.mark.id().includes("node") ? this._handleNodeRelatedClick(element) : element && element.mark.id().includes("link") ? this._handleLinkRelatedClick(element) : this._handleClearEmpty();
+        }, this._handleClearEmpty = () => {
+            const nodeVGrammarMark = this._nodeMark.getProduct();
+            if (!nodeVGrammarMark || !nodeVGrammarMark.elements || !nodeVGrammarMark.elements.length) return;
+            const allNodeElements = nodeVGrammarMark.elements, linkVGrammarMark = this._linkMark.getProduct();
+            if (!linkVGrammarMark || !linkVGrammarMark.elements || !linkVGrammarMark.elements.length) return;
+            const allLinkElements = linkVGrammarMark.elements;
+            allNodeElements.forEach((el => {
+                el.clearStates();
+            })), allLinkElements.forEach((el => {
+                el.clearStates();
+            }));
+        }, this._handleNodeAdjacencyClick = element => {
+            const nodeDatum = element.getDatum(), highlightNodes = [ nodeDatum.key ];
+            [ this._linkMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach(((linkEl, i) => {
+                    linkEl.clearStates();
+                    const linkDatum = linkEl.getDatum(), father = (null == linkDatum ? void 0 : linkDatum.parents) ? "parents" : "source";
+                    if ((0, util_2.array)(linkDatum[father]).includes(nodeDatum.key)) {
+                        let ratio;
+                        if (highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                        highlightNodes.includes(linkDatum.target) || highlightNodes.push(linkDatum.target), 
+                        "parents" === father) {
+                            const originalDatum = linkDatum.datum;
+                            ratio = (originalDatum ? originalDatum.filter((entry => entry.parents.some((par => par.key === nodeDatum.key)))).reduce(((sum, d) => sum + d.value), 0) : 0) / linkDatum.value;
+                        }
+                        linkEl.addState("selected", {
+                            ratio: ratio
+                        });
+                    } else linkDatum.target === nodeDatum.key ? highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source) : linkEl.useStates([ "blur" ]);
+                }));
+            })), [ this._nodeMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((el => {
+                    el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                }));
+            })), [ this._labelMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((el => {
+                    el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                }));
+            }));
+        }, this._handleLinkAdjacencyClick = element => {
+            const curLinkDatum = element.getDatum(), highlightNodes = [ curLinkDatum.source, curLinkDatum.target ];
+            [ this._linkMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((linkEl => {
+                    linkEl.clearStates();
+                    const linkDatum = linkEl.getDatum(), father = (null == linkDatum ? void 0 : linkDatum.parents) ? "parents" : "source";
+                    if (linkDatum.source === curLinkDatum.target || (0, util_2.array)(linkDatum[father]).includes(curLinkDatum.source) && (0, 
+                    util_2.array)(linkDatum[father]).includes(curLinkDatum.target)) {
+                        let ratio;
+                        if (highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                        highlightNodes.includes(linkDatum.target) || highlightNodes.push(linkDatum.target), 
+                        "parents" === father) {
+                            const originalDatum = linkDatum.datum;
+                            ratio = (originalDatum ? originalDatum.filter((entry => entry.parents.some(((par, index) => {
+                                var _a;
+                                return par.key === curLinkDatum.source && (null === (_a = entry.parents[index + 1]) || void 0 === _a ? void 0 : _a.key) === curLinkDatum.target;
+                            })))).reduce(((sum, d) => sum + d.value), 0) : 0) / linkDatum.value;
+                        }
+                        linkEl.addState("selected", {
+                            ratio: ratio
+                        });
+                    } else linkEl === element || linkEl.useStates([ "blur" ]);
+                }));
+            })), [ this._nodeMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((el => {
+                    el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                }));
+            })), [ this._labelMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((el => {
+                    el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                }));
+            }));
+        }, this._handleNodeRelatedClick = element => {
+            var _a;
+            const nodeDatum = element.getDatum(), nodeVGrammarMark = this._nodeMark.getProduct();
+            if (!nodeVGrammarMark || !nodeVGrammarMark.elements || !nodeVGrammarMark.elements.length) return;
+            const allNodeElements = nodeVGrammarMark.elements, linkVGrammarMark = this._linkMark.getProduct();
+            if (!linkVGrammarMark || !linkVGrammarMark.elements || !linkVGrammarMark.elements.length) return;
+            const allLinkElements = linkVGrammarMark.elements;
+            if ("source" === ((null === (_a = allLinkElements[0].getDatum()) || void 0 === _a ? void 0 : _a.parents) ? "parents" : "source")) {
+                const highlightNodes = [ nodeDatum.key ], highlightLinks = [];
+                allLinkElements.forEach(((linkEl, i) => {
+                    var _a, _b, _c, _d;
+                    linkEl.clearStates();
+                    const linkDatum = linkEl.getDatum(), father = (null == linkDatum ? void 0 : linkDatum.parents) ? "parents" : "source";
+                    if ((0, util_2.array)(linkDatum[father]).includes(nodeDatum.key)) {
+                        if (highlightLinks.includes(null !== (_a = linkDatum.key) && void 0 !== _a ? _a : linkDatum.index) || highlightLinks.push(null !== (_b = linkDatum.key) && void 0 !== _b ? _b : linkDatum.index), 
+                        highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                        !highlightNodes.includes(linkDatum.target)) {
+                            highlightNodes.push(linkDatum.target);
+                            let targetNodeSourceLinks = allNodeElements.find((nodeElement => nodeElement.data[0].key === linkDatum.target)).data[0].sourceLinks;
+                            for (;(null == targetNodeSourceLinks ? void 0 : targetNodeSourceLinks.length) > 0; ) {
+                                const newTargetNodeSourceLinks = [];
+                                return targetNodeSourceLinks.forEach((targetNodeSourceLinkDatum => {
+                                    var _a, _b;
+                                    if (!highlightLinks.includes(null !== (_a = targetNodeSourceLinkDatum.key) && void 0 !== _a ? _a : targetNodeSourceLinkDatum.index) && (highlightLinks.push(null !== (_b = targetNodeSourceLinkDatum.key) && void 0 !== _b ? _b : targetNodeSourceLinkDatum.index), 
+                                    !highlightNodes.includes(targetNodeSourceLinkDatum.target))) {
+                                        highlightNodes.push(targetNodeSourceLinkDatum.target);
+                                        const sourceNodeTemp = allNodeElements.find((nodeElement => nodeElement.data[0].key === targetNodeSourceLinkDatum.target));
+                                        newTargetNodeSourceLinks.push(sourceNodeTemp.data[0].targetLinks);
+                                    }
+                                })), void (targetNodeSourceLinks = newTargetNodeSourceLinks);
+                            }
+                        }
+                    } else if (linkDatum.target === nodeDatum.key && (highlightLinks.includes(null !== (_c = linkDatum.key) && void 0 !== _c ? _c : linkDatum.index) || highlightLinks.push(null !== (_d = linkDatum.key) && void 0 !== _d ? _d : linkDatum.index), 
+                    !highlightNodes.includes(linkDatum.source))) {
+                        highlightNodes.push(linkDatum.source);
+                        let sourceNodeTargetLinks = allNodeElements.find((nodeElement => nodeElement.data[0].key === linkDatum.source)).data[0].targetLinks;
+                        for (;(null == sourceNodeTargetLinks ? void 0 : sourceNodeTargetLinks.length) > 0; ) {
+                            const newSourceNodeTargetLinks = [];
+                            return sourceNodeTargetLinks.forEach((sourceNodeTargetLinkDatum => {
+                                var _a, _b;
+                                if (!highlightLinks.includes(null !== (_a = sourceNodeTargetLinkDatum.key) && void 0 !== _a ? _a : sourceNodeTargetLinkDatum.index) && (highlightLinks.push(null !== (_b = sourceNodeTargetLinkDatum.key) && void 0 !== _b ? _b : sourceNodeTargetLinkDatum.index), 
+                                !highlightNodes.includes(sourceNodeTargetLinkDatum.source))) {
+                                    highlightNodes.push(sourceNodeTargetLinkDatum.source);
+                                    const sourceNodeTemp = allNodeElements.find((nodeElement => nodeElement.data[0].key === sourceNodeTargetLinkDatum.source));
+                                    newSourceNodeTargetLinks.push(sourceNodeTemp.data[0].targetLinks);
+                                }
+                            })), void (sourceNodeTargetLinks = newSourceNodeTargetLinks);
+                        }
+                    }
+                })), [ this._linkMark ].forEach((mark => {
+                    const vGrammarMark = mark.getProduct();
+                    if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                    vGrammarMark.elements.forEach(((linkEl, i) => {
+                        var _a;
+                        linkEl.clearStates(), highlightLinks.includes(null !== (_a = linkEl.getDatum().key) && void 0 !== _a ? _a : linkEl.getDatum().index) ? linkEl.useStates([ "selected" ]) : linkEl.useStates([ "blur" ]);
+                    }));
+                })), [ this._nodeMark ].forEach((mark => {
+                    const vGrammarMark = mark.getProduct();
+                    if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                    vGrammarMark.elements.forEach((el => {
+                        el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                    }));
+                })), [ this._labelMark ].forEach((mark => {
+                    const vGrammarMark = mark.getProduct();
+                    if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                    vGrammarMark.elements.forEach((el => {
+                        el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                    }));
+                }));
+            } else {
+                const highlightNodes = [ nodeDatum.key ], upstreamLinks = nodeDatum.targetLinks.reduce(((res, link) => ((0, 
+                util_2.array)(link.datum).forEach((dividedLink => {
+                    const parents = dividedLink.parents, len = parents.length;
+                    for (let i = 0; i < len; i++) {
+                        const source = parents[i].key, target = parents[i + 1] ? parents[i + 1].key : nodeDatum.key, value = dividedLink.value, existingItem = res.find((item => item.source === source && item.target === target));
+                        existingItem ? existingItem.value += value : res.push({
+                            source: source,
+                            target: target,
+                            value: value
+                        });
+                    }
+                })), res)), []);
+                allLinkElements.forEach(((linkEl, i) => {
+                    linkEl.clearStates();
+                    const linkDatum = linkEl.getDatum(), father = (null == linkDatum ? void 0 : linkDatum.parents) ? "parents" : "source", originalDatum = linkDatum.datum, selectedDatum = originalDatum ? originalDatum.filter((entry => entry[father].some((par => par.key === nodeDatum.key)))) : null, upSelectedLink = upstreamLinks.find((upLink => upLink.source === linkDatum.source && upLink.target === linkDatum.target));
+                    if (selectedDatum && selectedDatum.length) {
+                        highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                        highlightNodes.includes(linkDatum.target) || highlightNodes.push(linkDatum.target);
+                        const ratio = selectedDatum.reduce(((sum, d) => sum + d.value), 0) / linkDatum.value;
+                        return linkEl.useStates([ "selected" ]), void linkEl.addState("selected", {
+                            ratio: ratio
+                        });
+                    }
+                    if (upSelectedLink) return highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                    highlightNodes.includes(linkDatum.target) || highlightNodes.push(linkDatum.target), 
+                    linkEl.useStates([ "selected" ]), void linkEl.addState("selected", {
+                        ratio: upSelectedLink.value / linkDatum.value
+                    });
+                    linkEl.useStates([ "blur" ]);
+                })), [ this._nodeMark ].forEach((mark => {
+                    const vGrammarMark = mark.getProduct();
+                    if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                    vGrammarMark.elements.forEach((el => {
+                        el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                    }));
+                })), [ this._labelMark ].forEach((mark => {
+                    const vGrammarMark = mark.getProduct();
+                    if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                    vGrammarMark.elements.forEach((el => {
+                        el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                    }));
+                }));
+            }
+        }, this._handleLinkRelatedClick = element => {
+            var _a;
+            const nodeVGrammarMark = this._nodeMark.getProduct();
+            if (!nodeVGrammarMark || !nodeVGrammarMark.elements || !nodeVGrammarMark.elements.length) return;
+            const allNodeElements = nodeVGrammarMark.elements, linkVGrammarMark = this._linkMark.getProduct();
+            if (!linkVGrammarMark || !linkVGrammarMark.elements || !linkVGrammarMark.elements.length) return;
+            const allLinkElements = linkVGrammarMark.elements;
+            if ("source" === ((null === (_a = element.getDatum()) || void 0 === _a ? void 0 : _a.parents) ? "parents" : "source")) [ this._linkMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((linkEl => {
+                    linkEl.clearStates();
+                }));
+            })), [ this._nodeMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((el => {
+                    el.clearStates();
+                }));
+            })), [ this._labelMark ].forEach((mark => {
+                const vGrammarMark = mark.getProduct();
+                if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                vGrammarMark.elements.forEach((el => {
+                    el.clearStates();
+                }));
+            })); else {
+                const curLinkDatum = element.getDatum(), highlightNodes = [ curLinkDatum.source, curLinkDatum.target ], upstreamLinks = [];
+                (0, util_2.array)(curLinkDatum.datum).forEach((dividedLink => {
+                    const parents = dividedLink.parents, len = parents.length;
+                    for (let i = 0; i < len - 1; i++) {
+                        const source = parents[i].key, target = parents[i + 1].key, value = dividedLink.value, existingItem = upstreamLinks.find((item => item.source === source && item.target === target));
+                        upstreamLinks.push({
+                            source: parents[i].key,
+                            target: parents[i + 1].key,
+                            value: dividedLink.value
+                        }), existingItem ? existingItem.value += value : upstreamLinks.push({
+                            source: source,
+                            target: target,
+                            value: value
+                        });
+                    }
+                })), allLinkElements.forEach((linkEl => {
+                    linkEl.clearStates();
+                    const linkDatum = linkEl.getDatum(), originalDatum = linkDatum.datum;
+                    if (linkDatum.source === curLinkDatum.source && linkDatum.target === curLinkDatum.target) return linkEl.useStates([ "selected" ]), 
+                    void linkEl.addState("selected", {
+                        ratio: 1
+                    });
+                    const selectedDatum = originalDatum ? originalDatum.filter(((entry, index) => entry.parents.some((par => {
+                        var _a;
+                        return par.key === curLinkDatum.target && (null === (_a = entry.parents[index - 1]) || void 0 === _a ? void 0 : _a.key) === curLinkDatum.source;
+                    })))) : null;
+                    if (selectedDatum && selectedDatum.length) {
+                        highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                        highlightNodes.includes(linkDatum.target) || highlightNodes.push(linkDatum.target);
+                        const val = selectedDatum.filter((entry => entry.parents.some(((par, index) => {
+                            var _a;
+                            return par.key === curLinkDatum.source && (null === (_a = entry.parents[index + 1]) || void 0 === _a ? void 0 : _a.key) === curLinkDatum.target;
+                        })))).reduce(((sum, d) => sum + d.value), 0), ratio = val / linkDatum.value;
+                        return linkEl.useStates([ "selected" ]), void linkEl.addState("selected", {
+                            ratio: ratio
+                        });
+                    }
+                    const upSelectedLink = upstreamLinks.find((upLink => upLink.source === linkDatum.source && upLink.target === linkDatum.target));
+                    if (upSelectedLink) return highlightNodes.includes(linkDatum.source) || highlightNodes.push(linkDatum.source), 
+                    highlightNodes.includes(linkDatum.target) || highlightNodes.push(linkDatum.target), 
+                    linkEl.useStates([ "selected" ]), void linkEl.addState("selected", {
+                        ratio: upSelectedLink.value / linkDatum.value
+                    });
+                    linkEl.useStates([ "blur" ]);
+                })), allNodeElements.forEach((el => {
+                    el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                })), [ this._labelMark ].forEach((mark => {
+                    const vGrammarMark = mark.getProduct();
+                    if (!vGrammarMark || !vGrammarMark.elements || !vGrammarMark.elements.length) return;
+                    vGrammarMark.elements.forEach((el => {
+                        el.clearStates(), highlightNodes.includes(el.getDatum().key) || el.useStates([ "blur" ]);
+                    }));
+                }));
+            }
+        };
+    }
+    getCategoryField() {
+        return this._categoryField;
+    }
+    setCategoryField(f) {
+        return this._categoryField = f, this._categoryField;
+    }
+    getValueField() {
+        return this._valueField;
+    }
+    setValueField(f) {
+        return this._valueField = f, this._valueField;
+    }
+    setAttrFromSpec() {
+        var _a, _b, _c;
+        super.setAttrFromSpec(), this.setCategoryField(this._spec.categoryField), this.setValueField(this._spec.valueField), 
+        this.setSeriesField(null !== (_a = this._spec.seriesField) && void 0 !== _a ? _a : this._spec.categoryField), 
+        this._labelLimit = null !== (_c = null === (_b = this._spec.label) || void 0 === _b ? void 0 : _b.limit) && void 0 !== _c ? _c : 100;
+    }
+    initData() {
+        var _a, _b, _c, _d, _e, _f;
+        if (super.initData(), this._viewDataFilter) {
+            (0, register_1.registerDataSetInstanceTransform)(this._dataSet, "sankey", sankey_1.sankey), 
+            this.addViewDataFilter({
+                type: "sankey",
+                options: {
+                    view: () => ({
+                        x0: this._viewBox.x1,
+                        x1: this._viewBox.x2,
+                        y0: this._viewBox.y1,
+                        y1: this._viewBox.y2
+                    }),
+                    sourceField: this._spec.sourceField,
+                    targetField: this._spec.targetField,
+                    direction: this._spec.direction,
+                    nodeAlign: null !== (_a = this._spec.nodeAlign) && void 0 !== _a ? _a : "justify",
+                    nodeGap: null !== (_b = this._spec.nodeGap) && void 0 !== _b ? _b : 8,
+                    nodeWidth: null !== (_c = this._spec.nodeWidth) && void 0 !== _c ? _c : 10,
+                    linkWidth: this._spec.linkWidth,
+                    minStepWidth: this._spec.minStepWidth,
+                    minNodeHeight: null !== (_d = this._spec.minNodeHeight) && void 0 !== _d ? _d : 4,
+                    minLinkHeight: this._spec.minLinkHeight,
+                    iterations: this._spec.iterations,
+                    nodeKey: this._spec.nodeKey,
+                    linkSortBy: this._spec.linkSortBy,
+                    nodeSortBy: this._spec.nodeSortBy,
+                    setNodeLayer: this._spec.setNodeLayer
+                }
+            });
+            const nodesDataSet = new vdataset_1.DataSet;
+            (0, register_1.registerDataSetInstanceParser)(nodesDataSet, "dataview", vdataset_1.dataViewParser), 
+            (0, register_1.registerDataSetInstanceTransform)(nodesDataSet, "sankeyNodes", sankey_nodes_1.sankeyNodes), 
+            (0, register_1.registerDataSetInstanceTransform)(nodesDataSet, "addVChartProperty", add_property_1.addVChartProperty), 
+            (0, register_1.registerDataSetInstanceTransform)(nodesDataSet, "flatten", flatten_1.flatten);
+            const nodesDataView = new vdataset_1.DataView(nodesDataSet);
+            nodesDataView.parse([ this.getViewData() ], {
+                type: "dataview"
+            }), nodesDataView.transform({
+                type: "sankeyNodes"
+            }), nodesDataView.transform({
+                type: "flatten",
+                options: {
+                    callback: node => {
+                        if (node.datum) {
+                            const nodeData = node.datum[node.depth];
+                            return Object.assign(Object.assign({}, node), nodeData);
+                        }
+                        return node;
+                    }
+                }
+            }), nodesDataView.transform({
+                type: "addVChartProperty",
+                options: {
+                    beforeCall: data_key_1.initKeyMap,
+                    call: data_key_1.addDataKey.bind(this)
+                }
+            }, !1), null === (_e = this._data) || void 0 === _e || _e.getDataView().target.addListener("change", nodesDataView.reRunAllTransform), 
+            this._nodesSeriesData = new series_data_1.SeriesData(this._option, nodesDataView);
+            const linksDataSet = new vdataset_1.DataSet;
+            (0, register_1.registerDataSetInstanceParser)(linksDataSet, "dataview", vdataset_1.dataViewParser), 
+            (0, register_1.registerDataSetInstanceTransform)(linksDataSet, "sankeyLinks", sankey_links_1.sankeyLinks), 
+            (0, register_1.registerDataSetInstanceTransform)(linksDataSet, "addVChartProperty", add_property_1.addVChartProperty);
+            const linksDataView = new vdataset_1.DataView(linksDataSet);
+            linksDataView.parse([ this.getViewData() ], {
+                type: "dataview"
+            }), linksDataView.transform({
+                type: "sankeyLinks"
+            }), linksDataView.transform({
+                type: "addVChartProperty",
+                options: {
+                    beforeCall: data_key_1.initKeyMap,
+                    call: data_key_1.addDataKey.bind(this)
+                }
+            }, !1), null === (_f = this._data) || void 0 === _f || _f.getDataView().target.addListener("change", linksDataView.reRunAllTransform), 
+            this._linksSeriesData = new series_data_1.SeriesData(this._option, linksDataView);
+        }
+    }
+    initMark() {
+        var _a;
+        const nodeMark = this._createMark(SankeySeries.mark.node, {
+            isSeriesMark: !0,
+            key: constant_1.DEFAULT_DATA_INDEX,
+            dataView: this._nodesSeriesData.getDataView(),
+            dataProductId: this._nodesSeriesData.getProductId()
+        });
+        nodeMark && (nodeMark.setZIndex(this._nodeLayoutZIndex), this._nodeMark = nodeMark);
+        const linkMark = this._createMark(SankeySeries.mark.link, {
+            key: constant_1.DEFAULT_DATA_INDEX,
+            dataView: this._linksSeriesData.getDataView(),
+            dataProductId: this._linksSeriesData.getProductId()
+        });
+        if (linkMark && (this._linkMark = linkMark), null === (_a = this._spec.label) || void 0 === _a ? void 0 : _a.visible) {
+            const labelMark = this._createMark(SankeySeries.mark.label, {
+                key: constant_1.DEFAULT_DATA_INDEX,
+                dataView: this._nodesSeriesData.getDataView(),
+                dataProductId: this._nodesSeriesData.getProductId()
+            });
+            labelMark && (this._labelMark = labelMark);
+        }
+    }
+    initMarkStyle() {
+        this._initNodeMarkStyle(), this._initLinkMarkStyle(), this._initLabelMarkStyle();
+    }
+    _initNodeMarkStyle() {
+        var _a;
+        const nodeMark = this._nodeMark;
+        nodeMark && (this.setMarkStyle(nodeMark, {
+            x: datum => datum.x0,
+            x1: datum => datum.x1,
+            y: datum => datum.y0,
+            y1: datum => datum.y1,
+            fill: datum => {
+                var _a, _b, _c, _d, _e;
+                const nodeName = null !== (_b = null !== (_a = datum.key) && void 0 !== _a ? _a : datum[this._spec.categoryField]) && void 0 !== _b ? _b : "";
+                return null !== (_e = null === (_d = null === (_c = this._spec.node) || void 0 === _c ? void 0 : _c.style) || void 0 === _d ? void 0 : _d.fill) && void 0 !== _e ? _e : this.getNodeOrdinalColorScale(nodeName);
+            }
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Mark), this._trigger.registerMark(nodeMark), 
+        null === (_a = this._tooltipHelper) || void 0 === _a || _a.activeTriggerSet.mark.add(nodeMark));
+    }
+    _initLinkMarkStyle() {
+        var _a, _b;
+        const linkMark = this._linkMark;
+        linkMark && (this.setMarkStyle(linkMark, {
+            x0: datum => datum.x0,
+            x1: datum => datum.x1,
+            y0: datum => datum.y0,
+            y1: datum => datum.y1,
+            thickness: datum => datum.thickness,
+            fill: datum => {
+                var _a, _b, _c;
+                return null !== (_c = null === (_b = null === (_a = this._spec.link) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(datum.source);
+            },
+            direction: null !== (_a = this._spec.direction) && void 0 !== _a ? _a : "horizontal"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series), this._trigger.registerMark(linkMark), 
+        null === (_b = this._tooltipHelper) || void 0 === _b || _b.activeTriggerSet.mark.add(linkMark));
+    }
+    _initLabelMarkStyle() {
+        var _a;
+        this._labelMark && ("vertical" === this._spec.direction ? "inside-start" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x0,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: "#ffffff",
+            text: datum => this._createText(datum),
+            limit: datum => {
+                var _a;
+                return null !== (_a = this._spec.label.limit) && void 0 !== _a ? _a : datum.x1 - datum.x0;
+            },
+            textAlign: "left",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "inside-middle" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => (datum.x0 + datum.x1) / 2,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: "#ffffff",
+            text: datum => this._createText(datum),
+            limit: datum => {
+                var _a;
+                return null !== (_a = this._spec.label.limit) && void 0 !== _a ? _a : datum.x1 - datum.x0;
+            },
+            textAlign: "center",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "inside-end" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x1,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: "#ffffff",
+            text: datum => this._createText(datum),
+            limit: datum => {
+                var _a;
+                return null !== (_a = this._spec.label.limit) && void 0 !== _a ? _a : datum.x1 - datum.x0;
+            },
+            textAlign: "right",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : this.setMarkStyle(this._labelMark, {
+            x: datum => (datum.x0 + datum.x1) / 2,
+            y: datum => datum.y1 >= this._viewBox.y2 ? datum.y0 : datum.y1,
+            fill: datum => {
+                var _a, _b, _c;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(datum.key);
+            },
+            text: datum => this._createText(datum),
+            limit: this._labelLimit,
+            textAlign: "center",
+            textBaseline: datum => datum.y1 >= this._viewBox.y2 ? "bottom" : "top"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "inside-start" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x0,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: "#ffffff",
+            text: datum => this._createText(datum),
+            limit: datum => {
+                var _a;
+                return null !== (_a = this._spec.label.limit) && void 0 !== _a ? _a : datum.x1 - datum.x0;
+            },
+            textAlign: "left",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "inside-middle" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => (datum.x0 + datum.x1) / 2,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: "#ffffff",
+            text: datum => this._createText(datum),
+            limit: datum => {
+                var _a;
+                return null !== (_a = this._spec.label.limit) && void 0 !== _a ? _a : datum.x1 - datum.x0;
+            },
+            textAlign: "center",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "inside-end" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x1,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: "#ffffff",
+            text: datum => this._createText(datum),
+            limit: datum => {
+                var _a;
+                return null !== (_a = this._spec.label.limit) && void 0 !== _a ? _a : datum.x1 - datum.x0;
+            },
+            textAlign: "right",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "left" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x0,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: datum => {
+                var _a, _b, _c;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(datum.key);
+            },
+            text: datum => this._createText(datum),
+            limit: this._labelLimit,
+            textAlign: "right",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : "right" === this._spec.label.position ? this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x1,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: datum => {
+                var _a, _b, _c;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(datum.key);
+            },
+            text: datum => this._createText(datum),
+            limit: this._labelLimit,
+            textAlign: "left",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series) : this.setMarkStyle(this._labelMark, {
+            x: datum => datum.x1 >= this._viewBox.x2 ? datum.x0 : datum.x1,
+            y: datum => (datum.y0 + datum.y1) / 2,
+            fill: datum => {
+                var _a, _b, _c;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(datum.key);
+            },
+            text: datum => this._createText(datum),
+            limit: this._labelLimit,
+            textAlign: datum => datum.x1 >= this._viewBox.x2 ? "right" : "left",
+            textBaseline: "middle"
+        }, mark_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series), this._labelMark.setZIndex(this._labelLayoutZIndex), 
+        this._trigger.registerMark(this._labelMark), null === (_a = this._tooltipHelper) || void 0 === _a || _a.activeTriggerSet.mark.add(this._labelMark));
+    }
+    _createText(datum) {
+        var _a;
+        if ((0, util_2.isNil)(datum) || (0, util_2.isNil)(datum.datum)) return "";
+        let text = datum.datum[this._spec.categoryField] || "";
+        return (null === (_a = this._spec.label) || void 0 === _a ? void 0 : _a.formatMethod) && (text = this._spec.label.formatMethod(text, datum.datum)), 
+        text;
+    }
+    initAnimation() {
+        var _a, _b;
+        const animationParams = {
+            direction: this.direction,
+            growFrom: () => {
+                var _a, _b;
+                return "horizontal" === this.direction ? null === (_a = this._xAxisHelper) || void 0 === _a ? void 0 : _a.getScale(0).scale(0) : null === (_b = this._yAxisHelper) || void 0 === _b ? void 0 : _b.getScale(0).scale(0);
+            }
+        }, appearPreset = null === (_b = null === (_a = this._spec) || void 0 === _a ? void 0 : _a.animationAppear) || void 0 === _b ? void 0 : _b.preset;
+        this._nodeMark && this._nodeMark.setAnimationConfig((0, utils_1.animationConfig)(config_1.DEFAULT_MARK_ANIMATION.sankeyNode(animationParams, appearPreset), (0, 
+        utils_1.userAnimationConfig)(interface_1.SeriesMarkNameEnum.node, this._spec))), 
+        this._linkMark && this._linkMark.setAnimationConfig((0, utils_1.animationConfig)(config_1.DEFAULT_MARK_ANIMATION.sankeyLinkPath(), (0, 
+        utils_1.userAnimationConfig)(interface_1.SeriesMarkNameEnum.link, this._spec))), 
+        this._labelMark && this._labelMark.setAnimationConfig((0, utils_1.animationConfig)(config_1.DEFAULT_MARK_ANIMATION.label(), (0, 
+        utils_1.userAnimationConfig)(interface_1.SeriesMarkNameEnum.label, this._spec)));
+    }
+    initEvent() {
+        var _a, _b, _c, _d, _e, _f, _g, _h;
+        super.initEvent(), null === (_a = this._nodesSeriesData.getDataView()) || void 0 === _a || _a.target.addListener("change", this.nodesSeriesDataUpdate.bind(this)), 
+        null === (_b = this._linksSeriesData.getDataView()) || void 0 === _b || _b.target.addListener("change", this.linksSeriesDataUpdate.bind(this)), 
+        (null === (_c = this._spec.emphasis) || void 0 === _c ? void 0 : _c.enable) && "adjacency" === (null === (_d = this._spec.emphasis) || void 0 === _d ? void 0 : _d.effect) && ("hover" === (null === (_e = this._spec.emphasis) || void 0 === _e ? void 0 : _e.trigger) ? this.event.on("pointerover", {
+            level: constant_1.Event_Bubble_Level.mark
+        }, this._handleAdjacencyClick) : this.event.on("pointerdown", {
+            level: constant_1.Event_Bubble_Level.mark
+        }, this._handleAdjacencyClick)), (null === (_f = this._spec.emphasis) || void 0 === _f ? void 0 : _f.enable) && "related" === (null === (_g = this._spec.emphasis) || void 0 === _g ? void 0 : _g.effect) && ("hover" === (null === (_h = this._spec.emphasis) || void 0 === _h ? void 0 : _h.trigger) ? this.event.on("pointerover", {
+            level: constant_1.Event_Bubble_Level.mark
+        }, this._handleRelatedClick) : this.event.on("pointerdown", {
+            level: constant_1.Event_Bubble_Level.mark
+        }, this._handleRelatedClick));
+    }
+    nodesSeriesDataUpdate() {
+        this.event.emit(constant_1.ChartEvent.legendFilter, {
+            model: this
+        }), this._nodesSeriesData.updateData();
+    }
+    linksSeriesDataUpdate() {
+        this.event.emit(constant_1.ChartEvent.legendFilter, {
+            model: this
+        }), this._linksSeriesData.updateData();
+    }
+    initTooltip() {
+        this._tooltipHelper = new tooltip_helper_1.SankeySeriesTooltipHelper(this);
+    }
+    getNodeOrdinalColorScale(item) {
+        var _a, _b, _c, _d, _e, _f, _g;
+        const colorDomain = (null === (_a = this._rawData.latestData[0]) || void 0 === _a ? void 0 : _a.nodes) ? (null === (_b = this._rawData.latestData[0].nodes[0]) || void 0 === _b ? void 0 : _b.children) ? Array.from(this.extractNamesFromTree(this._rawData.latestData[0].nodes)) : this._rawData.latestData[0].nodes.map(((datum, index) => this._spec.nodeKey ? datum[this._spec.categoryField] : index)) : null === (_c = this._rawData.latestData[0]) || void 0 === _c ? void 0 : _c.values.map(((datum, index) => this._spec.nodeKey ? datum[this._spec.categoryField] : index)), colorRange = null !== (_e = null === (_d = this._option.globalScale.color) || void 0 === _d ? void 0 : _d.range()) && void 0 !== _e ? _e : (0, 
+        util_1.getDataScheme)(this._option.getTheme().colorScheme, this.type), ordinalScale = new color_ordinal_scale_1.ColorOrdinalScale;
+        return null === (_g = (_f = ordinalScale.domain(colorDomain)).range) || void 0 === _g || _g.call(_f, colorRange), 
+        ordinalScale.scale(item);
+    }
+    extractNamesFromTree(tree) {
+        const uniqueNames = new Set;
+        return tree.forEach((node => {
+            if (uniqueNames.add(node.name), node.children) {
+                this.extractNamesFromTree(node.children).forEach((name => uniqueNames.add(name)));
+            }
+        })), uniqueNames;
+    }
+    getDimensionField() {
+        return [ this._spec.categoryField ];
+    }
+    getMeasureField() {
+        return [ this._valueField ];
+    }
+    getSeriesKeys() {
+        var _a;
+        if (this._seriesField) {
+            const keyArray = [];
+            return null === (_a = this._nodesSeriesData) || void 0 === _a || _a.getDataView().latestData.forEach((datum => {
+                var _a;
+                keyArray.push(null !== (_a = datum[this._seriesField]) && void 0 !== _a ? _a : datum.datum[this._seriesField]);
+            })), keyArray;
+        }
+        return [];
+    }
+    onLayoutEnd(ctx) {
+        super.onLayoutEnd(ctx), this._viewBox.set(0, 0, this._region.getLayoutRect().width, this._region.getLayoutRect().height), 
+        this._rawData.reRunAllTransform(), this.getViewData().reRunAllTransform(), this._nodesSeriesData.updateData(), 
+        this._linksSeriesData.updateData();
+    }
+    getDefaultShapeType() {
+        return "square";
+    }
+    _noAnimationDataKey(datum, index) {}
+}
+
+exports.SankeySeries = SankeySeries, SankeySeries.type = interface_1.SeriesTypeEnum.sankey, 
+SankeySeries.mark = constant_2.sankeySeriesMark;
+//# sourceMappingURL=sankey.js.map