@visactor/vchart-extension 1.13.22 → 1.13.23-alpha.0

package/cjs/charts/compare-sankey/compare-sankey-sub-data.d.ts

@@ -0,0 +1,21 @@
+import type { DataView } from '@visactor/vdataset';
+import { HierarchyNodeDatum, SankeyLinkDatum, SankeyLinkElement, SankeyNodeElement } from '@visactor/vgrammar-sankey';
+export interface ICompareSankeyLayoutOpt {
+    rawData: () => DataView;
+    nodeKey: string;
+    subNodeGap: number;
+    subNodeMinSize?: number;
+}
+export declare const calculateNodeValue: (subTree: HierarchyNodeDatum[]) => number;
+export declare function makeHierarchicNodes(originalNodes: HierarchyNodeDatum[], nodeKeyFunc: (node: HierarchyNodeDatum) => string, nodes?: SankeyNodeElement[], nodeMap?: Record<string | number, SankeyNodeElement>, originalLinks?: (SankeyLinkDatum & {
+    parents?: SankeyNodeElement[];
+})[]): SankeyNodeElement[];
+export declare function computeHierarchicNodeLinks(originalNodes: HierarchyNodeDatum[], nodeKeyFunc: (node: HierarchyNodeDatum) => string): {
+    nodes: SankeyNodeElement[];
+    links: SankeyLinkElement[];
+    nodeMap: Record<string | number, SankeyNodeElement>;
+};
+export declare function computeNodeValues(nodes: SankeyNodeElement[]): void;
+export declare const compareSankeySubData: (data: Array<DataView>, opt: ICompareSankeyLayoutOpt) => {
+    [key: string]: any;
+};

package/cjs/charts/compare-sankey/compare-sankey-sub-data.js

@@ -0,0 +1,118 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+}), exports.compareSankeySubData = exports.computeNodeValues = exports.computeHierarchicNodeLinks = exports.makeHierarchicNodes = exports.calculateNodeValue = void 0;
+
+const vutils_1 = require("@visactor/vutils"), calculateNodeValue = subTree => {
+    let sum = 0;
+    return subTree.forEach((node => {
+        var _a;
+        (0, vutils_1.isNil)(node.value) && ((null === (_a = node.children) || void 0 === _a ? void 0 : _a.length) ? node.value = (0, 
+        exports.calculateNodeValue)(node.children) : node.value = 0), sum += Math.abs(node.value);
+    })), sum;
+};
+
+function makeHierarchicNodes(originalNodes, nodeKeyFunc, nodes = [], nodeMap = {}, originalLinks) {
+    (0, exports.calculateNodeValue)(originalNodes);
+    const doSubTree = (subTree, depth, parents) => {
+        subTree.forEach(((node, index) => {
+            const nodeKey = nodeKeyFunc ? nodeKeyFunc(node) : parents ? `${parents[parents.length - 1].key}-${index}` : `${depth}-${index}`, nodeValue = (0, 
+            vutils_1.isNil)(node.value) ? 0 : (0, vutils_1.toValidNumber)(node.value);
+            if (nodeMap[nodeKey]) nodeMap[nodeKey].value = void 0; else {
+                const nodeElement = {
+                    depth: depth,
+                    datum: node,
+                    index: index,
+                    key: nodeKey,
+                    value: nodeValue,
+                    sourceLinks: [],
+                    targetLinks: []
+                };
+                nodeMap[nodeKey] = nodeElement, nodes.push(nodeElement);
+            }
+            parents && originalLinks && originalLinks.push({
+                source: parents[parents.length - 1].key,
+                target: nodeKey,
+                value: nodeValue,
+                parents: parents
+            }), node.children && node.children.length && doSubTree(node.children, depth + 1, parents ? parents.concat([ nodeMap[nodeKey] ]) : [ nodeMap[nodeKey] ]);
+        }));
+    };
+    return doSubTree(originalNodes, 0, null), nodes;
+}
+
+function computeHierarchicNodeLinks(originalNodes, nodeKeyFunc) {
+    const nodes = [], links = [], nodeMap = {}, linkMap = {}, originalLinks = [];
+    return makeHierarchicNodes(originalNodes, nodeKeyFunc, nodes, nodeMap, originalLinks), 
+    originalLinks.forEach(((link, index) => {
+        const key = `${link.source}-${link.target}`, linkDatum = (0, vutils_1.pickWithout)(link, [ "parents" ]);
+        if (linkDatum.parents = link.parents.map((node => (0, vutils_1.pickWithout)(node, [ "sourceLinks", "targetLinks" ]))), 
+        linkMap[key]) return linkMap[key].value += (0, vutils_1.toValidNumber)(link.value), 
+        void linkMap[key].datum.push(linkDatum);
+        const linkElement = {
+            index: index,
+            key: `${link.source}-${link.target}`,
+            source: link.source,
+            target: link.target,
+            datum: [ linkDatum ],
+            value: link.value,
+            parents: link.parents.map((parent => parent.key))
+        };
+        links.push(linkElement), nodeMap[link.source].sourceLinks.push(linkElement), nodeMap[link.target].targetLinks.push(linkElement), 
+        linkMap[key] = linkElement;
+    })), {
+        nodes: nodes,
+        links: links,
+        nodeMap: nodeMap
+    };
+}
+
+function computeNodeValues(nodes) {
+    for (let i = 0, len = nodes.length; i < len; i++) {
+        const node = nodes[i];
+        node.value = Math.max((0, vutils_1.isNil)(node.value) ? 0 : (0, vutils_1.toValidNumber)(node.value), node.sourceLinks.reduce(((sum, link) => {
+            var _a;
+            return sum + (null !== (_a = (0, vutils_1.toValidNumber)(link.value)) && void 0 !== _a ? _a : 0);
+        }), 0), node.targetLinks.reduce(((sum, link) => {
+            var _a;
+            return sum + (null !== (_a = (0, vutils_1.toValidNumber)(link.value)) && void 0 !== _a ? _a : 0);
+        }), 0));
+    }
+}
+
+exports.calculateNodeValue = calculateNodeValue, exports.makeHierarchicNodes = makeHierarchicNodes, 
+exports.computeHierarchicNodeLinks = computeHierarchicNodeLinks, exports.computeNodeValues = computeNodeValues;
+
+const compareSankeySubData = (data, opt) => {
+    var _a, _b;
+    const viewData = data[0];
+    if (!(null === (_a = viewData.latestData) || void 0 === _a ? void 0 : _a.length)) return {};
+    const subNodeMinSize = null !== (_b = opt.subNodeMinSize) && void 0 !== _b ? _b : 0, rawDataTree = opt.rawData().latestData[0], subNodeGap = opt.subNodeGap, keyFunc = (0, 
+    vutils_1.isFunction)(opt.nodeKey) ? opt.nodeKey : opt.nodeKey ? (0, vutils_1.field)(opt.nodeKey) : null, subNodeMap = {};
+    rawDataTree.subNode.forEach((sunGroup => {
+        subNodeMap[sunGroup.type] = computeHierarchicNodeLinks(sunGroup.nodes, keyFunc), 
+        computeNodeValues(subNodeMap[sunGroup.type].nodes);
+    }));
+    const subCount = Object.keys(subNodeMap).length, subNodes = [];
+    return viewData.latestData[0].nodes.forEach((n => {
+        let path = [];
+        if (n.targetLinks.length) {
+            path = [ ...n.targetLinks[0].parents ];
+        }
+        path.push(n.key);
+        let currentY = n.y0;
+        const totalSize = n.y1 - n.y0 - (subCount - 1) * subNodeGap, totalValue = n.value;
+        rawDataTree.subNode.forEach((sunGroup => {
+            const subNode = subNodeMap[sunGroup.type].nodes.find((subN => subN.key === n.key));
+            if (!subNode) return;
+            const percent = 0 === totalValue ? 0 : subNode.value / totalValue, dis = Math.max(subNodeMinSize, totalSize * percent);
+            subNode.x0 = n.x0, subNode.x1 = n.x1, subNode.y0 = currentY, subNode.y1 = currentY + dis, 
+            subNode.type = sunGroup.type, subNode.sourceNode = n, currentY += dis + subNodeGap, 
+            subNodes.push(subNode);
+        }));
+    })), subNodeMap;
+};
+
+exports.compareSankeySubData = compareSankeySubData;
+//# sourceMappingURL=compare-sankey-sub-data.js.map

package/cjs/charts/compare-sankey/compare-sankey-sub-data.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/compare-sankey-sub-data.ts"],"names":[],"mappings":";;;AACA,6CAAwF;AAUjF,MAAM,kBAAkB,GAAG,CAAC,OAA6B,EAAE,EAAE;IAClE,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;;QACrB,IAAI,IAAA,cAAK,EAAC,IAAI,CAAC,KAAK,CAAC,EAAE;YACrB,IAAI,MAAA,IAAI,CAAC,QAAQ,0CAAE,MAAM,EAAE;gBACzB,IAAI,CAAC,KAAK,GAAG,IAAA,0BAAkB,EAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;aAChD;iBAAM;gBACL,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;aAChB;SACF;QAED,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAfW,QAAA,kBAAkB,sBAe7B;AAEF,SAAgB,mBAAmB,CACjC,aAAmC,EACnC,WAAiD,EACjD,QAA6B,EAAE,EAC/B,UAAsD,EAAE,EACxD,aAAuE;IAEvE,IAAA,0BAAkB,EAAC,aAAa,CAAC,CAAC;IAElC,MAAM,SAAS,GAAG,CAAC,OAA6B,EAAE,KAAa,EAAE,OAA6B,EAAE,EAAE;QAChG,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC9B,MAAM,OAAO,GAAG,WAAW;gBACzB,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC;gBACnB,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,KAAK,EAAE;oBAC/C,CAAC,CAAC,GAAG,KAAK,IAAI,KAAK,EAAE,CAAC;YACxB,MAAM,SAAS,GAAW,CAAC,IAAA,cAAK,EAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAA,sBAAa,EAAC,IAAI,CAAC,KAAK,CAAC,CAAW,CAAC;YAExF,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;gBACpB,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,GAAG,SAAS,CAAC;aACpC;iBAAM;gBACL,MAAM,WAAW,GAAsB;oBACrC,KAAK;oBACL,KAAK,EAAE,IAAI;oBACX,KAAK,EAAE,KAAK;oBACZ,GAAG,EAAE,OAAO;oBACZ,KAAK,EAAE,SAAS;oBAChB,WAAW,EAAE,EAAyB;oBACtC,WAAW,EAAE,EAAyB;iBACvC,CAAC;gBAEF,OAAO,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC;gBAC/B,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;aACzB;YACD,IAAI,OAAO,IAAI,aAAa,EAAE;gBAC5B,aAAa,CAAC,IAAI,CAAC;oBACjB,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG;oBACvC,MAAM,EAAE,OAAO;oBACf,KAAK,EAAE,SAAS;oBAChB,OAAO;iBACR,CAAC,CAAC;aACJ;YAED,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE;gBACzC,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;aACxG;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,SAAS,CAAC,aAAa,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IAClC,OAAO,KAAK,CAAC;AACf,CAAC;AAnDD,kDAmDC;AAED,SAAgB,0BAA0B,CACxC,aAAmC,EACnC,WAAiD;IAEjD,MAAM,KAAK,GAAwB,EAAE,CAAC;IACtC,MAAM,KAAK,GAAwB,EAAE,CAAC;IACtC,MAAM,OAAO,GAA+C,EAAE,CAAC;IAC/D,MAAM,OAAO,GAA+C,EAAE,CAAC;IAC/D,MAAM,aAAa,GAA4D,EAAE,CAAC;IAElF,mBAAmB,CAAC,aAAa,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;IAE/E,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACpC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAC5C,MAAM,SAAS,GAAG,IAAA,oBAAW,EAAC,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAEhD,SAAiB,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YACnD,OAAO,IAAA,oBAAW,EAAC,IAAI,EAAE,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE;YAChB,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,IAAA,sBAAa,EAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE/C,OAAO,CAAC,GAAG,CAAC,CAAC,KAA2B,CAAC,IAAI,CAAC,SAA4B,CAAC,CAAC;YAE7E,OAAO;SACR;QACD,MAAM,WAAW,GAAG;YAClB,KAAK;YACL,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE;YACpC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,CAAC,SAAS,CAAQ;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC;SAChD,CAAC;QAEF,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnD,OAAO,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AACnC,CAAC;AA5CD,gEA4CC;AAED,SAAgB,iBAAiB,CAAC,KAA0B;IAC1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE;QAChD,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAEtB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CACnB,IAAA,cAAK,EAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAA,sBAAa,EAAC,IAAI,CAAC,KAAK,CAAC,EACjD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAuB,EAAE,EAAE;;YACvD,OAAO,GAAG,GAAG,CAAC,MAAA,IAAA,sBAAa,EAAC,IAAI,CAAC,KAAK,CAAC,mCAAI,CAAC,CAAC,CAAC;QAChD,CAAC,EAAE,CAAC,CAAC,EACL,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAuB,EAAE,EAAE;;YACvD,OAAO,GAAG,GAAG,CAAC,MAAA,IAAA,sBAAa,EAAC,IAAI,CAAC,KAAK,CAAC,mCAAI,CAAC,CAAC,CAAC;QAChD,CAAC,EAAE,CAAC,CAAC,CACN,CAAC;KACH;AACH,CAAC;AAdD,8CAcC;AAEM,MAAM,oBAAoB,GAAG,CAAC,IAAqB,EAAE,GAA4B,EAAE,EAAE;;IAC1F,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAa,CAAC;IACrC,IAAI,CAAC,CAAA,MAAA,QAAQ,CAAC,UAAU,0CAAE,MAAM,CAAA,EAAE;QAChC,OAAO,EAAE,CAAC;KACX;IACD,MAAM,cAAc,GAAG,MAAA,GAAG,CAAC,cAAc,mCAAI,CAAC,CAAC;IAE/C,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAChD,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC;IAClC,MAAM,OAAO,GAAG,IAAA,mBAAU,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAA,cAAK,EAAC,GAAG,CAAC,OAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAE1G,MAAM,UAAU,GAA2B,EAAE,CAAC;IAC9C,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,QAAa,EAAE,EAAE;QAC5C,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,0BAA0B,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChF,iBAAiB,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC;IAEhD,MAAM,QAAQ,GAAU,EAAE,CAAC;IAC3B,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAoB,EAAE,EAAE;QAC5D,IAAI,IAAI,GAAwB,EAAE,CAAC;QACnC,IAAI,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE;YACxB,MAAM,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YAC9B,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;SAC1B;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAGjB,IAAI,QAAQ,GAAG,CAAC,CAAC,EAAE,CAAC;QACpB,MAAM,SAAS,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,QAAQ,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC;QAC5D,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC;QAC3B,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,QAAa,EAAE,EAAE;YAC5C,MAAM,OAAO,GAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAA6B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC;YAC1G,IAAI,CAAC,OAAO,EAAE;gBACZ,OAAO;aACR;YACD,MAAM,OAAO,GAAG,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,GAAG,UAAU,CAAC;YAClE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC;YAC1D,OAAO,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;YAClB,OAAO,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;YAClB,OAAO,CAAC,EAAE,GAAG,QAAQ,CAAC;YACtB,OAAO,CAAC,EAAE,GAAG,QAAQ,GAAG,GAAG,CAAC;YAE5B,OAAO,CAAC,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;YAE7B,OAAO,CAAC,UAAU,GAAG,CAAC,CAAC;YACvB,QAAQ,IAAI,GAAG,GAAG,UAAU,CAAC;YAC7B,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IACH,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAnDW,QAAA,oBAAoB,wBAmD/B","file":"compare-sankey-sub-data.js","sourcesContent":["import type { DataView } from '@visactor/vdataset';\nimport { field, isFunction, isNil, pickWithout, toValidNumber } from '@visactor/vutils';\nimport { HierarchyNodeDatum, SankeyLinkDatum, SankeyLinkElement, SankeyNodeElement } from '@visactor/vgrammar-sankey';\n\nexport interface ICompareSankeyLayoutOpt {\n  rawData: () => DataView;\n  nodeKey: string;\n  subNodeGap: number;\n  subNodeMinSize?: number;\n}\n\nexport const calculateNodeValue = (subTree: HierarchyNodeDatum[]) => {\n  let sum = 0;\n  subTree.forEach(node => {\n    if (isNil(node.value)) {\n      if (node.children?.length) {\n        node.value = calculateNodeValue(node.children);\n      } else {\n        node.value = 0;\n      }\n    }\n\n    sum += Math.abs(node.value);\n  });\n\n  return sum;\n};\n\nexport function makeHierarchicNodes(\n  originalNodes: HierarchyNodeDatum[],\n  nodeKeyFunc: (node: HierarchyNodeDatum) => string,\n  nodes: SankeyNodeElement[] = [],\n  nodeMap: Record<string | number, SankeyNodeElement> = {},\n  originalLinks?: (SankeyLinkDatum & { parents?: SankeyNodeElement[] })[]\n) {\n  calculateNodeValue(originalNodes);\n\n  const doSubTree = (subTree: HierarchyNodeDatum[], depth: number, parents?: SankeyNodeElement[]) => {\n    subTree.forEach((node, index) => {\n      const nodeKey = nodeKeyFunc\n        ? nodeKeyFunc(node)\n        : parents\n        ? `${parents[parents.length - 1].key}-${index}`\n        : `${depth}-${index}`;\n      const nodeValue: number = (isNil(node.value) ? 0 : toValidNumber(node.value)) as number;\n\n      if (nodeMap[nodeKey]) {\n        nodeMap[nodeKey].value = undefined;\n      } else {\n        const nodeElement: SankeyNodeElement = {\n          depth,\n          datum: node,\n          index: index,\n          key: nodeKey,\n          value: nodeValue,\n          sourceLinks: [] as SankeyLinkElement[],\n          targetLinks: [] as SankeyLinkElement[]\n        };\n\n        nodeMap[nodeKey] = nodeElement;\n        nodes.push(nodeElement);\n      }\n      if (parents && originalLinks) {\n        originalLinks.push({\n          source: parents[parents.length - 1].key,\n          target: nodeKey,\n          value: nodeValue,\n          parents\n        });\n      }\n\n      if (node.children && node.children.length) {\n        doSubTree(node.children, depth + 1, parents ? parents.concat([nodeMap[nodeKey]]) : [nodeMap[nodeKey]]);\n      }\n    });\n  };\n\n  doSubTree(originalNodes, 0, null);\n  return nodes;\n}\n\nexport function computeHierarchicNodeLinks(\n  originalNodes: HierarchyNodeDatum[],\n  nodeKeyFunc: (node: HierarchyNodeDatum) => string\n) {\n  const nodes: SankeyNodeElement[] = [];\n  const links: SankeyLinkElement[] = [];\n  const nodeMap: Record<string | number, SankeyNodeElement> = {};\n  const linkMap: Record<string | number, SankeyLinkElement> = {};\n  const originalLinks: (SankeyLinkDatum & { parents?: SankeyNodeElement[] })[] = [];\n\n  makeHierarchicNodes(originalNodes, nodeKeyFunc, nodes, nodeMap, originalLinks);\n\n  originalLinks.forEach((link, index) => {\n    const key = `${link.source}-${link.target}`;\n    const linkDatum = pickWithout(link, ['parents']);\n\n    (linkDatum as any).parents = link.parents.map(node => {\n      return pickWithout(node, ['sourceLinks', 'targetLinks']);\n    });\n\n    if (linkMap[key]) {\n      linkMap[key].value += toValidNumber(link.value);\n\n      (linkMap[key].datum as SankeyLinkDatum[]).push(linkDatum as SankeyLinkDatum);\n\n      return;\n    }\n    const linkElement = {\n      index,\n      key: `${link.source}-${link.target}`,\n      source: link.source,\n      target: link.target,\n      datum: [linkDatum] as any,\n      value: link.value,\n      parents: link.parents.map(parent => parent.key)\n    };\n\n    links.push(linkElement);\n    nodeMap[link.source].sourceLinks.push(linkElement);\n    nodeMap[link.target].targetLinks.push(linkElement);\n    linkMap[key] = linkElement;\n  });\n\n  return { nodes, links, nodeMap };\n}\n\nexport function computeNodeValues(nodes: SankeyNodeElement[]) {\n  for (let i = 0, len = nodes.length; i < len; i++) {\n    const node = nodes[i];\n\n    node.value = Math.max(\n      isNil(node.value) ? 0 : toValidNumber(node.value),\n      node.sourceLinks.reduce((sum, link: SankeyLinkElement) => {\n        return sum + (toValidNumber(link.value) ?? 0);\n      }, 0),\n      node.targetLinks.reduce((sum, link: SankeyLinkElement) => {\n        return sum + (toValidNumber(link.value) ?? 0);\n      }, 0)\n    );\n  }\n}\n\nexport const compareSankeySubData = (data: Array<DataView>, opt: ICompareSankeyLayoutOpt) => {\n  const viewData = data[0] as DataView;\n  if (!viewData.latestData?.length) {\n    return {};\n  }\n  const subNodeMinSize = opt.subNodeMinSize ?? 0;\n  // 读取参数\n  const rawDataTree = opt.rawData().latestData[0];\n  const subNodeGap = opt.subNodeGap;\n  const keyFunc = isFunction(opt.nodeKey) ? opt.nodeKey : opt.nodeKey ? field(opt.nodeKey as string) : null;\n\n  const subNodeMap: { [key: string]: any } = {};\n  rawDataTree.subNode.forEach((sunGroup: any) => {\n    subNodeMap[sunGroup.type] = computeHierarchicNodeLinks(sunGroup.nodes, keyFunc);\n    computeNodeValues(subNodeMap[sunGroup.type].nodes);\n  });\n  const subCount = Object.keys(subNodeMap).length;\n\n  const subNodes: any[] = [];\n  viewData.latestData[0].nodes.forEach((n: SankeyNodeElement) => {\n    let path: (string | number)[] = [];\n    if (n.targetLinks.length) {\n      const link = n.targetLinks[0];\n      path = [...link.parents];\n    }\n    path.push(n.key);\n    // 根据path获取sub的节点\n    // 当前已使用比例\n    let currentY = n.y0;\n    const totalSize = n.y1 - n.y0 - (subCount - 1) * subNodeGap;\n    const totalValue = n.value;\n    rawDataTree.subNode.forEach((sunGroup: any) => {\n      const subNode = (subNodeMap[sunGroup.type].nodes as SankeyNodeElement[]).find(subN => subN.key === n.key);\n      if (!subNode) {\n        return;\n      }\n      const percent = totalValue === 0 ? 0 : subNode.value / totalValue;\n      const dis = Math.max(subNodeMinSize, totalSize * percent);\n      subNode.x0 = n.x0;\n      subNode.x1 = n.x1;\n      subNode.y0 = currentY;\n      subNode.y1 = currentY + dis;\n      // @ts-ignore\n      subNode.type = sunGroup.type;\n      // @ts-ignore\n      subNode.sourceNode = n;\n      currentY += dis + subNodeGap;\n      subNodes.push(subNode);\n    });\n  });\n  return subNodeMap;\n};\n"]}

package/cjs/charts/compare-sankey/compare-sankey-sub-nodes.d.ts

@@ -0,0 +1,2 @@
+import type { DataView } from '@visactor/vdataset';
+export declare const compareSankeySubNodes: (data: Array<DataView>) => {};

package/cjs/charts/compare-sankey/compare-sankey-sub-nodes.js

@@ -0,0 +1,21 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+}), exports.compareSankeySubNodes = void 0;
+
+const compareSankeySubNodes = data => {
+    const viewData = data[0];
+    if (!viewData.latestData) return [];
+    const subData = Object.keys(viewData.latestData);
+    if (!subData.length) return {};
+    const subNodes = [];
+    return subData.forEach((key => {
+        subNodes.push(...viewData.latestData[key].nodes);
+    })), [ {
+        nodes: subNodes
+    } ];
+};
+
+exports.compareSankeySubNodes = compareSankeySubNodes;
+//# sourceMappingURL=compare-sankey-sub-nodes.js.map

package/cjs/charts/compare-sankey/compare-sankey-sub-nodes.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/compare-sankey-sub-nodes.ts"],"names":[],"mappings":";;;AAEO,MAAM,qBAAqB,GAAG,CAAC,IAAqB,EAAE,EAAE;IAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAa,CAAC;IACrC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE;QACxB,OAAO,EAAE,CAAC;KACX;IACD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACjD,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;QACnB,OAAO,EAAE,CAAC;KACX;IACD,MAAM,QAAQ,GAAU,EAAE,CAAC;IAC3B,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QACpB,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;AAC/B,CAAC,CAAC;AAdW,QAAA,qBAAqB,yBAchC","file":"compare-sankey-sub-nodes.js","sourcesContent":["import type { DataView } from '@visactor/vdataset';\n\nexport const compareSankeySubNodes = (data: Array<DataView>) => {\n  const viewData = data[0] as DataView;\n  if (!viewData.latestData) {\n    return [];\n  }\n  const subData = Object.keys(viewData.latestData);\n  if (!subData.length) {\n    return {};\n  }\n  const subNodes: any[] = [];\n  subData.forEach(key => {\n    subNodes.push(...viewData.latestData[key].nodes);\n  });\n  return [{ nodes: subNodes }];\n};\n"]}

package/cjs/charts/compare-sankey/compare-sankey-transformer.d.ts

@@ -0,0 +1,7 @@
+import type { ICompareSankeyChartSpecBase } from './interface';
+import { SankeyChartSpecTransformer } from '@visactor/vchart';
+export declare class CompareSankeyChartSpecTransformer extends SankeyChartSpecTransformer<ICompareSankeyChartSpecBase> {
+    seriesType: string;
+    transformSpec(spec: ICompareSankeyChartSpecBase): void;
+    _getDefaultSeriesSpec(spec: ICompareSankeyChartSpecBase): any;
+}

package/cjs/charts/compare-sankey/compare-sankey-transformer.js

@@ -0,0 +1,25 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+}), exports.CompareSankeyChartSpecTransformer = void 0;
+
+const vchart_1 = require("@visactor/vchart");
+
+class CompareSankeyChartSpecTransformer extends vchart_1.SankeyChartSpecTransformer {
+    constructor() {
+        super(...arguments), this.seriesType = "compareSankey";
+    }
+    transformSpec(spec) {
+        super.transformSpec(spec);
+    }
+    _getDefaultSeriesSpec(spec) {
+        const seriesSpec = super._getDefaultSeriesSpec(spec);
+        return seriesSpec.subNodeGap = spec.subNodeGap, seriesSpec.subNodeMinSize = spec.subNodeMinSize, 
+        seriesSpec.compareNodeColor = spec.compareNodeColor, seriesSpec.compareLinkColor = spec.compareLinkColor, 
+        seriesSpec.activeLink = spec.activeLink, seriesSpec;
+    }
+}
+
+exports.CompareSankeyChartSpecTransformer = CompareSankeyChartSpecTransformer;
+//# sourceMappingURL=compare-sankey-transformer.js.map

package/cjs/charts/compare-sankey/compare-sankey-transformer.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/compare-sankey-transformer.ts"],"names":[],"mappings":";;;AACA,6CAA8D;AAE9D,MAAa,iCAAkC,SAAQ,mCAAuD;IAA9G;;QACE,eAAU,GAAG,eAAe,CAAC;IAc/B,CAAC;IAbC,aAAa,CAAC,IAAiC;QAC7C,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAED,qBAAqB,CAAC,IAAiC;QACrD,MAAM,UAAU,GAAG,KAAK,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACpD,UAA0C,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACxE,UAA0C,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAChF,UAA0C,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QACpF,UAA0C,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QACpF,UAA0C,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACzE,OAAO,UAAU,CAAC;IACpB,CAAC;CACF;AAfD,8EAeC","file":"compare-sankey-transformer.js","sourcesContent":["import type { ICompareSankeyChartSpecBase } from './interface';\nimport { SankeyChartSpecTransformer } from '@visactor/vchart';\n\nexport class CompareSankeyChartSpecTransformer extends SankeyChartSpecTransformer<ICompareSankeyChartSpecBase> {\n  seriesType = 'compareSankey';\n  transformSpec(spec: ICompareSankeyChartSpecBase): void {\n    super.transformSpec(spec);\n  }\n\n  _getDefaultSeriesSpec(spec: ICompareSankeyChartSpecBase) {\n    const seriesSpec = super._getDefaultSeriesSpec(spec);\n    (seriesSpec as ICompareSankeyChartSpecBase).subNodeGap = spec.subNodeGap;\n    (seriesSpec as ICompareSankeyChartSpecBase).subNodeMinSize = spec.subNodeMinSize;\n    (seriesSpec as ICompareSankeyChartSpecBase).compareNodeColor = spec.compareNodeColor;\n    (seriesSpec as ICompareSankeyChartSpecBase).compareLinkColor = spec.compareLinkColor;\n    (seriesSpec as ICompareSankeyChartSpecBase).activeLink = spec.activeLink;\n    return seriesSpec;\n  }\n}\n"]}

package/cjs/charts/compare-sankey/compare-sankey.d.ts

@@ -0,0 +1,46 @@
+import type { ICompareSankeyChartSpecBase, ICompareSankeySeriesSpecBase } from './interface';
+import { VChart, SankeyChart, SankeySeries } from '@visactor/vchart';
+import { CompareSankeyChartSpecTransformer } from './compare-sankey-transformer';
+import type { IElement } from '@visactor/vgrammar-core';
+type Datum = any;
+export declare class CompareSankeyChart extends SankeyChart<ICompareSankeyChartSpecBase> {
+    type: string;
+    static type: string;
+    readonly seriesType: string;
+    _spec: ICompareSankeyChartSpecBase;
+    static readonly transformerConstructor: typeof CompareSankeyChartSpecTransformer;
+    readonly transformerConstructor: typeof CompareSankeyChartSpecTransformer;
+}
+export declare class CompareSankeySeries extends SankeySeries<ICompareSankeySeriesSpecBase> {
+    type: string;
+    static type: string;
+    private _subData;
+    private _activeLinkData;
+    private _activeLinkMark;
+    initData(): void;
+    initEvent(): void;
+    initMark(): void;
+    compileData(): void;
+    _initLinkMarkStyle(): void;
+    _initNodeMarkStyle(): void;
+    private activeLinkDataUpdate;
+    protected _fillCompareNode: (datum: Datum) => any;
+    protected _fillActiveLink: (datum: Datum) => any;
+    protected _handleNodeRelatedClick(element: IElement, highlightState: string, blurState: string): void;
+    private _activeTargetLink;
+    private _activeSourceLink;
+    protected _handleClearEmpty(highlightState: string, blurState: string): void;
+    protected _handlePointerOutElement(): void;
+    protected _clearAllSelectedState(): void;
+    getSubNodeDatum(filter: (n: Datum) => boolean): IElement[];
+    getNodeDatumInTree(filter: (n: any) => boolean): any[];
+    getNodeDatumInSubTree(filter: (n: any) => boolean): {
+        [key: string]: any;
+        type: string;
+        datum: any[];
+    }[];
+}
+export declare const registerCompareSankeyChart: (option?: {
+    VChart?: typeof VChart;
+}) => void;
+export {};

package/cjs/charts/compare-sankey/compare-sankey.js

@@ -0,0 +1,202 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+}), exports.registerCompareSankeyChart = exports.CompareSankeySeries = exports.CompareSankeyChart = void 0;
+
+const vchart_1 = require("@visactor/vchart"), vdataset_1 = require("@visactor/vdataset"), compare_sankey_transformer_1 = require("./compare-sankey-transformer"), compare_sankey_sub_nodes_1 = require("./compare-sankey-sub-nodes"), compare_sankey_sub_data_1 = require("./compare-sankey-sub-data"), mark_1 = require("../../utils/mark"), utils_1 = require("./utils");
+
+class CompareSankeyChart extends vchart_1.SankeyChart {
+    constructor() {
+        super(...arguments), this.type = "compareSankey", this.seriesType = CompareSankeySeries.type, 
+        this.transformerConstructor = compare_sankey_transformer_1.CompareSankeyChartSpecTransformer;
+    }
+}
+
+exports.CompareSankeyChart = CompareSankeyChart, CompareSankeyChart.type = "compareSankey", 
+CompareSankeyChart.transformerConstructor = compare_sankey_transformer_1.CompareSankeyChartSpecTransformer;
+
+class CompareSankeySeries extends vchart_1.SankeySeries {
+    constructor() {
+        super(...arguments), this.type = "compareSankey", this._fillCompareNode = datum => {
+            var _a, _b, _c;
+            return (null === (_a = this._spec.compareNodeColor) || void 0 === _a ? void 0 : _a[datum.type]) ? this._spec.compareNodeColor[datum.type] : null !== (_c = null === (_b = this._spec.node.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this._fillByNode(datum);
+        }, this._fillActiveLink = datum => {
+            var _a, _b, _c;
+            return (null === (_a = this._spec.compareLinkColor) || void 0 === _a ? void 0 : _a[datum.type]) ? this._spec.compareLinkColor[datum.type] : null !== (_c = null === (_b = this._spec.link.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this._fillByLink(datum);
+        };
+    }
+    initData() {
+        var _a, _b;
+        super.initData();
+        const {dataSet: dataSet} = this._option, compareSubData = new vdataset_1.DataView(dataSet, {
+            name: `compare-sankey-sub-data-${this.id}-data`
+        });
+        compareSubData.parse([ this.getViewData() ], {
+            type: "dataview"
+        }), this._subData = compareSubData;
+        const compareNodeData = new vdataset_1.DataView(dataSet, {
+            name: `compare-sankey-node-${this.id}-data`
+        });
+        compareNodeData.parse([ compareSubData ], {
+            type: "dataview"
+        }), (0, vchart_1.registerDataSetInstanceTransform)(this._dataSet, "compareSankeySubData", compare_sankey_sub_data_1.compareSankeySubData), 
+        compareSubData.transform({
+            type: "compareSankeySubData",
+            options: {
+                rawData: () => this.getRawData(),
+                valueField: this._valueField,
+                nodeKey: this._spec.nodeKey,
+                subNodeGap: null !== (_a = this._spec.subNodeGap) && void 0 !== _a ? _a : 2,
+                subNodeMinSize: null !== (_b = this._spec.subNodeMinSize) && void 0 !== _b ? _b : 0
+            }
+        }), (0, vchart_1.registerDataSetInstanceTransform)(this._dataSet, "compareSankeySubNodes", compare_sankey_sub_nodes_1.compareSankeySubNodes), 
+        compareNodeData.transform({
+            type: "compareSankeySubNodes"
+        }), this._nodesSeriesData.getDataView().parse([ compareNodeData ], {
+            type: "dataview"
+        }), this._activeLinkData = new vdataset_1.DataView(dataSet, {
+            name: `compare-sankey-link-${this.id}-data`
+        });
+    }
+    initEvent() {
+        var _a;
+        super.initEvent(), null === (_a = this._activeLinkData) || void 0 === _a || _a.target.addListener("change", this.activeLinkDataUpdate.bind(this));
+    }
+    initMark() {
+        super.initMark();
+        const linkMark = this._createMark(Object.assign(Object.assign({}, vchart_1.SankeySeries.mark.link), {
+            name: "activeLink"
+        }), {
+            dataView: this._activeLinkData
+        });
+        linkMark && (this._activeLinkMark = linkMark);
+    }
+    compileData() {
+        var _a;
+        super.compileData(), null === (_a = this._activeLinkMark) || void 0 === _a || _a.compileData();
+    }
+    _initLinkMarkStyle() {
+        super._initLinkMarkStyle(), this._activeLinkMark.setAttribute("direction", this.direction), 
+        this.setMarkStyle(this._activeLinkMark, {
+            x0: datum => datum.x0,
+            x1: datum => datum.x1,
+            y0: datum => datum.y0,
+            y1: datum => datum.y1,
+            thickness: datum => datum.thickness
+        }, vchart_1.STATE_VALUE_ENUM.STATE_NORMAL, vchart_1.AttributeLevel.Series), this.setMarkStyle(this._activeLinkMark, {
+            fill: this._fillActiveLink
+        }, vchart_1.STATE_VALUE_ENUM.STATE_NORMAL, vchart_1.AttributeLevel.User_Mark);
+    }
+    _initNodeMarkStyle() {
+        super._initNodeMarkStyle(), this._spec.compareNodeColor && this.setMarkStyle(this._nodeMark, {
+            fill: this._fillCompareNode
+        }, "normal", vchart_1.AttributeLevel.User_Mark);
+    }
+    activeLinkDataUpdate() {
+        this._activeLinkMark.getData().updateData();
+    }
+    _handleNodeRelatedClick(element, highlightState, blurState) {
+        const nodeDatum = element.getDatum(), allNodeElements = this._nodeMark.getProductElements();
+        if (!allNodeElements || !allNodeElements.length) return;
+        const allLinkElements = this._linkMark.getProductElements();
+        if (!allLinkElements || !allLinkElements.length) return;
+        allLinkElements.forEach((el => {
+            el.removeState([ vchart_1.STATE_VALUE_ENUM.STATE_SANKEY_EMPHASIS ]);
+        })), this._highLightElements(allLinkElements, [], highlightState, blurState);
+        const nodeDatums = allNodeElements.map((el => (0, mark_1.getDatumOfGraphic)(el))), pickNodeDatums = nodeDatums.filter((d => d.key === nodeDatum.key)), highlightLinksData = [], highlightNodeKeys = pickNodeDatums.map((d => d.key));
+        this._activeTargetLink(nodeDatum, pickNodeDatums, allNodeElements, allLinkElements, highlightNodeKeys, highlightLinksData, highlightState, blurState), 
+        this._activeSourceLink(nodeDatum, pickNodeDatums, nodeDatums, allNodeElements, highlightNodeKeys, highlightLinksData, highlightState, blurState), 
+        this._highLightElements(allNodeElements, highlightNodeKeys, highlightState, blurState), 
+        this._activeLinkData.parseNewData(highlightLinksData), this._needClear = !0;
+    }
+    _activeTargetLink(nodeDatum, pickNodeDatums, allNodeElements, allLinkElements, highlightNodeKeys, highlightLinksData, highlightState, blurState) {
+        this._handleClearEmpty(highlightState, blurState);
+        const firstTarget = nodeDatum.sourceNode.targetLinks[0];
+        if (!firstTarget) return;
+        let percent = 0;
+        pickNodeDatums.forEach((n => {
+            const link = n.targetLinks.find((l => l.key === firstTarget.key));
+            if (link) {
+                const p = link.value / firstTarget.value, key = firstTarget.key + "_" + n.type, activeLink = Object.assign(Object.assign({}, firstTarget), {
+                    y0: firstTarget.y0 - firstTarget.thickness / 2 + (percent + p / 2) * firstTarget.thickness,
+                    y1: n.y0 + p * firstTarget.thickness / 2,
+                    thickness: p * firstTarget.thickness,
+                    key: key,
+                    type: n.type,
+                    [vchart_1.DEFAULT_DATA_INDEX]: highlightLinksData.length,
+                    [vchart_1.DEFAULT_DATA_KEY]: key
+                });
+                highlightLinksData.push(activeLink), percent += p;
+            }
+        })), highlightNodeKeys.push(...firstTarget.parents, nodeDatum.key);
+        const linkKeys = [];
+        for (let i = 0; i < firstTarget.parents.length - 1; i++) linkKeys.push(firstTarget.parents[i] + "-" + firstTarget.parents[i + 1]);
+        this._highLightElements(allLinkElements, linkKeys, highlightState, blurState);
+    }
+    _activeSourceLink(nodeDatum, pickNodeDatums, allNodeDatums, allNodeElements, highlightNodeKeys, highlightLinksData, highlightState, blurState) {
+        this._handleClearEmpty(highlightState, blurState);
+        const sourceLinks = nodeDatum.sourceNode.sourceLinks;
+        if (!(null == sourceLinks ? void 0 : sourceLinks.length)) return;
+        const sourceValueTemp = {};
+        sourceLinks.forEach((sourceLink => {
+            highlightNodeKeys.push(sourceLink.target), pickNodeDatums.forEach(((n, index) => {
+                var _a;
+                sourceValueTemp[n.type] = null !== (_a = sourceValueTemp[n.type]) && void 0 !== _a ? _a : 0;
+                const link = n.sourceLinks.find((l => l.key === sourceLink.key));
+                if (link) {
+                    const p = link.value / n.value, totalSize = n.y1 - n.y0, size = totalSize * p, key = sourceLink.key + "_" + n.type, activeLink = Object.assign(Object.assign({}, sourceLink), {
+                        y0: n.y0 + sourceValueTemp[n.type] * totalSize + size / 2,
+                        thickness: size,
+                        key: key,
+                        type: n.type,
+                        [vchart_1.DEFAULT_DATA_INDEX]: highlightLinksData.length,
+                        [vchart_1.DEFAULT_DATA_KEY]: key
+                    });
+                    if (0 === index) activeLink.y1 = sourceLink.y1 - sourceLink.thickness / 2 + size / 2; else {
+                        const targetNode = allNodeDatums.find((_n => _n.key === link.target && _n.type === n.type));
+                        if (!targetNode) return;
+                        activeLink.y1 = targetNode.y0 + size / 2;
+                    }
+                    highlightLinksData.push(activeLink), sourceValueTemp[n.type] += p;
+                }
+            }));
+        }));
+    }
+    _handleClearEmpty(highlightState, blurState) {
+        super._handleClearEmpty(highlightState, blurState), this._activeLinkData.parseNewData([]);
+        const allNodeElements = this._nodeMark.getProductElements();
+        !allNodeElements && allNodeElements.length || allNodeElements.forEach((el => {
+            el.removeState(vchart_1.STATE_VALUE_ENUM.STATE_HOVER);
+        }));
+        const allLinkElements = this._linkMark.getProductElements();
+        !allLinkElements && allLinkElements.length || allLinkElements.forEach((el => {
+            el.removeState(vchart_1.STATE_VALUE_ENUM.STATE_HOVER);
+        }));
+    }
+    _handlePointerOutElement() {
+        this._activeLinkData.parseNewData([]);
+    }
+    _clearAllSelectedState() {
+        super._clearAllSelectedState(), this._activeLinkData.parseNewData([]);
+    }
+    getSubNodeDatum(filter) {
+        return (0, utils_1.getSubNodeDatum)(this._nodeMark.getProductElements(), filter);
+    }
+    getNodeDatumInTree(filter) {
+        return (0, utils_1.getNodeDatumInTree)(this._rawData.latestData[0].nodes, filter);
+    }
+    getNodeDatumInSubTree(filter) {
+        return (0, utils_1.getNodeDatumInSubTree)(this._rawData.latestData[0].subNode, filter);
+    }
+}
+
+exports.CompareSankeySeries = CompareSankeySeries, CompareSankeySeries.type = "compareSankey";
+
+const registerCompareSankeyChart = option => {
+    const vchartConstructor = (null == option ? void 0 : option.VChart) || vchart_1.VChart;
+    vchartConstructor && (vchartConstructor.useChart([ CompareSankeyChart ]), vchartConstructor.useSeries([ CompareSankeySeries ]));
+};
+
+exports.registerCompareSankeyChart = registerCompareSankeyChart;
+//# sourceMappingURL=compare-sankey.js.map

package/cjs/charts/compare-sankey/compare-sankey.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/compare-sankey.ts"],"names":[],"mappings":";;;AAGA,6CAS0B;AAC1B,iDAA8C;AAC9C,6EAAiF;AACjF,yEAAmE;AAEnE,uEAAiE;AAEjE,2CAAqD;AACrD,mCAAqF;AAIrF,MAAa,kBAAmB,SAAQ,oBAAwC;IAAhF;;QACE,SAAI,GAAG,eAAe,CAAC;QAEd,eAAU,GAAG,mBAAmB,CAAC,IAAI,CAAC;QAKtC,2BAAsB,GAAG,8DAAiC,CAAC;IACtE,CAAC;;AATD,gDASC;AAPQ,uBAAI,GAAG,eAAe,CAAC;AAKd,yCAAsB,GAAG,8DAAiC,CAAC;AAI7E,MAAa,mBAAoB,SAAQ,qBAA0C;IAAnF;;QAEE,SAAI,GAAG,eAAe,CAAC;QAyHb,qBAAgB,GAAG,CAAC,KAAY,EAAE,EAAE;;YAC5C,IAAI,MAAA,IAAI,CAAC,KAAK,CAAC,gBAAgB,0CAAG,KAAK,CAAC,IAAI,CAAC,EAAE;gBAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;aAChD;YACD,OAAO,MAAA,MAAA,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,0CAAE,IAAI,mCAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC;QAEQ,oBAAe,GAAG,CAAC,KAAY,EAAE,EAAE;;YAC3C,IAAI,MAAA,IAAI,CAAC,KAAK,CAAC,gBAAgB,0CAAG,KAAK,CAAC,IAAI,CAAC,EAAE;gBAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;aAChD;YACD,OAAO,MAAA,MAAA,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,0CAAE,IAAI,mCAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC;IAgNJ,CAAC;IA7UC,QAAQ;;QACN,KAAK,CAAC,QAAQ,EAAE,CAAC;QAIjB,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAEjC,MAAM,cAAc,GAAG,IAAI,mBAAQ,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,2BAA2B,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAClG,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE;YACzC,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,cAAc,CAAC;QAE/B,MAAM,eAAe,GAAG,IAAI,mBAAQ,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,uBAAuB,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/F,eAAe,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,EAAE;YACtC,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAGH,IAAA,yCAAgC,EAAC,IAAI,CAAC,QAAQ,EAAE,sBAAsB,EAAE,8CAAoB,CAAC,CAAC;QAC9F,cAAc,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,sBAAsB;YAC5B,OAAO,EAAE;gBACP,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE;gBAChC,UAAU,EAAE,IAAI,CAAC,WAAW;gBAC5B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;gBAC3B,UAAU,EAAE,MAAA,IAAI,CAAC,KAAK,CAAC,UAAU,mCAAI,CAAC;gBACtC,cAAc,EAAE,MAAA,IAAI,CAAC,KAAK,CAAC,cAAc,mCAAI,CAAC;aAC/C;SACF,CAAC,CAAC;QACH,IAAA,yCAAgC,EAAC,IAAI,CAAC,QAAQ,EAAE,uBAAuB,EAAE,gDAAqB,CAAC,CAAC;QAChG,eAAe,CAAC,SAAS,CAAC;YACxB,IAAI,EAAE,uBAAuB;SAC9B,CAAC,CAAC;QACH,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,EAAE;YAC3D,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAGH,IAAI,CAAC,eAAe,GAAG,IAAI,mBAAQ,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,uBAAuB,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IAChG,CAAC;IAED,SAAS;;QACP,KAAK,CAAC,SAAS,EAAE,CAAC;QAClB,MAAA,IAAI,CAAC,eAAe,0CAAE,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED,QAAQ;QACN,KAAK,CAAC,QAAQ,EAAE,CAAC;QAEjB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,iCAC1B,qBAAY,CAAC,IAAI,CAAC,IAAI,KAAE,IAAI,EAAE,YAAY,KAC/C;YACE,QAAQ,EAAE,IAAI,CAAC,eAAe;SAC/B,CACe,CAAC;QACnB,IAAI,QAAQ,EAAE;YACZ,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;SACjC;IACH,CAAC;IACD,WAAW;;QACT,KAAK,CAAC,WAAW,EAAE,CAAC;QACpB,MAAA,IAAI,CAAC,eAAe,0CAAE,WAAW,EAAE,CAAC;IACtC,CAAC;IAED,kBAAkB;QAChB,KAAK,CAAC,kBAAkB,EAAE,CAAC;QAI3B,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAE/D,IAAI,CAAC,YAAY,CACf,IAAI,CAAC,eAAe,EACpB;YACE,EAAE,EAAE,CAAC,KAAY,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE;YAC9B,EAAE,EAAE,CAAC,KAAY,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE;YAC9B,EAAE,EAAE,CAAC,KAAY,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE;YAC9B,EAAE,EAAE,CAAC,KAAY,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE;YAC9B,SAAS,EAAE,CAAC,KAAY,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS;SAC7C,EACD,yBAAgB,CAAC,YAAY,EAC7B,uBAAc,CAAC,MAAM,CACtB,CAAC;QAEF,IAAI,CAAC,YAAY,CACf,IAAI,CAAC,eAAe,EACpB;YACE,IAAI,EAAE,IAAI,CAAC,eAAe;SAC3B,EACD,yBAAgB,CAAC,YAAY,EAC7B,uBAAc,CAAC,SAAS,CACzB,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,KAAK,CAAC,kBAAkB,EAAE,CAAC;QAC3B,IAAI,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE;YAC/B,IAAI,CAAC,YAAY,CACf,IAAI,CAAC,SAAS,EACd;gBACE,IAAI,EAAE,IAAI,CAAC,gBAAgB;aAC5B,EACD,QAAQ,EACR,uBAAc,CAAC,SAAS,CACzB,CAAC;SACH;IACH,CAAC;IAEO,oBAAoB;QAC1B,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC,UAAU,EAAE,CAAC;IAC9C,CAAC;IAgBS,uBAAuB,CAAC,OAAiB,EAAE,cAAsB,EAAE,SAAiB;QAC5F,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QAErC,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;QAC5D,IAAI,CAAC,eAAe,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC/C,OAAO;SACR;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;QAC5D,IAAI,CAAC,eAAe,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC/C,OAAO;SACR;QAGD,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;YAC3B,EAAE,CAAC,WAAW,CAAC,CAAC,yBAAgB,CAAC,qBAAqB,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,EAAE,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;QAExE,MAAM,UAAU,GAAG,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAA,wBAAiB,EAAC,EAAE,CAAU,CAAC,CAAC;QAC7E,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,SAAS,CAAC,GAAG,CAAC,CAAC;QAEvE,MAAM,kBAAkB,GAAU,EAAE,CAAC;QAErC,MAAM,iBAAiB,GAAwB,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAG9E,IAAI,CAAC,iBAAiB,CACpB,SAAS,EACT,cAAc,EACd,eAAe,EACf,eAAe,EACf,iBAAiB,EACjB,kBAAkB,EAClB,cAAc,EACd,SAAS,CACV,CAAC;QACF,IAAI,CAAC,iBAAiB,CACpB,SAAS,EACT,cAAc,EACd,UAAU,EACV,eAAe,EACf,iBAAiB,EACjB,kBAAkB,EAClB,cAAc,EACd,SAAS,CACV,CAAC;QACF,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,iBAA6B,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;QACnG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC;QAEtD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;IACzB,CAAC;IAEO,iBAAiB,CACvB,SAAgB,EAChB,cAAuB,EACvB,eAAsB,EACtB,eAAsB,EACtB,iBAAsC,EACtC,kBAAyB,EACzB,cAAsB,EACtB,SAAiB;QAEjB,IAAI,CAAC,iBAAiB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;QAExC,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAI,CAAC,WAAW,EAAE;YAChB,OAAO;SACR;QACD,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACzB,MAAM,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,WAAW,CAAC,GAAG,CAAC,CAAC;YACvE,IAAI,IAAI,EAAE;gBACR,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC;gBACzC,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC;gBAC3C,MAAM,UAAU,mCACX,WAAW,KAEd,EAAE,EAAE,WAAW,CAAC,EAAE,GAAG,WAAW,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC,SAAS,EAC1F,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,EAC1C,SAAS,EAAE,CAAC,GAAG,WAAW,CAAC,SAAS,EACpC,GAAG,EACH,IAAI,EAAE,CAAC,CAAC,IAAI,EACZ,CAAC,2BAAkB,CAAC,EAAE,kBAAkB,CAAC,MAAM,EAC/C,CAAC,yBAAgB,CAAC,EAAE,GAAG,GACxB,CAAC;gBACF,kBAAkB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpC,OAAO,IAAI,CAAC,CAAC;aACd;QACH,CAAC,CAAC,CAAC;QAGH,iBAAiB,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;QAC9D,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE;YACvD,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SAC1E;QACD,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,QAAQ,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;IAChF,CAAC;IAEO,iBAAiB,CACvB,SAAgB,EAChB,cAAuB,EACvB,aAAsB,EACtB,eAAsB,EACtB,iBAAsC,EACtC,kBAAyB,EACzB,cAAsB,EACtB,SAAiB;QAEjB,IAAI,CAAC,iBAAiB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;QAExC,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC3C,IAAI,CAAC,CAAA,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,MAAM,CAAA,EAAE;YACxB,OAAO;SACR;QAED,MAAM,eAAe,GAA8B,EAAE,CAAC;QACtD,WAAW,CAAC,OAAO,CAAC,CAAC,UAA6B,EAAE,EAAE;YACpD,iBAAiB,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC1C,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE;;gBAClC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,MAAA,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,mCAAI,CAAC,CAAC;gBACvD,MAAM,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAoB,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,UAAU,CAAC,GAAG,CAAC,CAAC;gBACpF,IAAI,IAAI,EAAE;oBACR,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;oBAC/B,MAAM,SAAS,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;oBAC9B,MAAM,IAAI,GAAG,SAAS,GAAG,CAAC,CAAC;oBAC3B,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC;oBAC1C,MAAM,UAAU,mCACX,UAAU,KAEb,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,SAAS,GAAG,IAAI,GAAG,CAAC,EACzD,SAAS,EAAE,IAAI,EACf,GAAG,EACH,IAAI,EAAE,CAAC,CAAC,IAAI,EACZ,CAAC,2BAAkB,CAAC,EAAE,kBAAkB,CAAC,MAAM,EAC/C,CAAC,yBAAgB,CAAC,EAAE,GAAG,GACxB,CAAC;oBAEF,IAAI,KAAK,KAAK,CAAC,EAAE;wBAEf,UAAU,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,GAAG,UAAU,CAAC,SAAS,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;qBACrE;yBAAM;wBAEL,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,KAAK,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;wBAC1F,IAAI,UAAU,EAAE;4BACd,UAAU,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,GAAG,IAAI,GAAG,CAAC,CAAC;yBAC1C;6BAAM;4BAEL,OAAO;yBACR;qBACF;oBACD,kBAAkB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;oBACpC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAES,iBAAiB,CAAC,cAAsB,EAAE,SAAiB;QACnE,KAAK,CAAC,iBAAiB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAGtC,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;QAC5D,IAAI,eAAe,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC9C,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;gBAC3B,EAAE,CAAC,WAAW,CAAC,yBAAgB,CAAC,WAAW,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;SACJ;QAGD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;QAC5D,IAAI,eAAe,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE;YAC9C,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;gBAC3B,EAAE,CAAC,WAAW,CAAC,yBAAgB,CAAC,WAAW,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;IAES,wBAAwB;QAChC,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;IAKS,sBAAsB;QAC9B,KAAK,CAAC,sBAAsB,EAAE,CAAC;QAC/B,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;IAEM,eAAe,CAAC,MAA6B;QAClD,OAAO,IAAA,uBAAe,EAAC,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,EAAE,MAAM,CAAC,CAAC;IACtE,CAAC;IAEM,kBAAkB,CAAC,MAA2B;QACnD,OAAO,IAAA,0BAAkB,EAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACvE,CAAC;IAEM,qBAAqB,CAAC,MAA2B;QACtD,OAAO,IAAA,6BAAqB,EAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;;AAtVH,kDAuVC;AApVQ,wBAAI,GAAG,eAAe,CAAC;AAsVzB,MAAM,0BAA0B,GAAG,CAAC,MAAmC,EAAE,EAAE;IAChF,MAAM,iBAAiB,GAAG,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,KAAI,eAAM,CAAC;IACnD,IAAI,iBAAiB,EAAE;QACrB,iBAAiB,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC;QACjD,iBAAiB,CAAC,SAAS,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC;KACpD;AACH,CAAC,CAAC;AANW,QAAA,0BAA0B,8BAMrC","file":"compare-sankey.js","sourcesContent":["import type { SankeyLinkElement } from '@visactor/vgrammar-sankey';\nimport type { ICompareSankeyChartSpecBase, ICompareSankeySeriesSpecBase } from './interface';\nimport type { ILinkPathMarkSpec, IRectMarkSpec } from '@visactor/vchart';\nimport {\n  VChart,\n  SankeyChart,\n  SankeySeries,\n  registerDataSetInstanceTransform,\n  STATE_VALUE_ENUM,\n  AttributeLevel,\n  DEFAULT_DATA_INDEX,\n  DEFAULT_DATA_KEY\n} from '@visactor/vchart';\nimport { DataView } from '@visactor/vdataset';\nimport { CompareSankeyChartSpecTransformer } from './compare-sankey-transformer';\nimport { compareSankeySubNodes } from './compare-sankey-sub-nodes';\nimport type { ILinkPathMark } from '@visactor/vchart/src/mark/interface';\nimport { compareSankeySubData } from './compare-sankey-sub-data';\nimport type { IElement } from '@visactor/vgrammar-core';\nimport { getDatumOfGraphic } from '../../utils/mark';\nimport { getNodeDatumInSubTree, getNodeDatumInTree, getSubNodeDatum } from './utils';\n\ntype Datum = any;\n\nexport class CompareSankeyChart extends SankeyChart<ICompareSankeyChartSpecBase> {\n  type = 'compareSankey';\n  static type = 'compareSankey';\n  readonly seriesType = CompareSankeySeries.type;\n\n  declare _spec: ICompareSankeyChartSpecBase;\n\n  static readonly transformerConstructor = CompareSankeyChartSpecTransformer;\n  readonly transformerConstructor = CompareSankeyChartSpecTransformer;\n}\n\nexport class CompareSankeySeries extends SankeySeries<ICompareSankeySeriesSpecBase> {\n  // @ts-ignore\n  type = 'compareSankey';\n  static type = 'compareSankey';\n\n  private _subData: DataView;\n\n  private _activeLinkData: DataView;\n  private _activeLinkMark: ILinkPathMark;\n\n  initData() {\n    super.initData();\n    // 此时 viewData 已经初始化完成\n    // 创建 第二层 viewDataActive\n    //\n    const { dataSet } = this._option;\n\n    const compareSubData = new DataView(dataSet, { name: `compare-sankey-sub-data-${this.id}-data` });\n    compareSubData.parse([this.getViewData()], {\n      type: 'dataview'\n    });\n    this._subData = compareSubData;\n\n    const compareNodeData = new DataView(dataSet, { name: `compare-sankey-node-${this.id}-data` });\n    compareNodeData.parse([compareSubData], {\n      type: 'dataview'\n    });\n\n    // 注册对比布局函数，计算拆解 nodes 的信息\n    registerDataSetInstanceTransform(this._dataSet, 'compareSankeySubData', compareSankeySubData);\n    compareSubData.transform({\n      type: 'compareSankeySubData',\n      options: {\n        rawData: () => this.getRawData(),\n        valueField: this._valueField,\n        nodeKey: this._spec.nodeKey,\n        subNodeGap: this._spec.subNodeGap ?? 2,\n        subNodeMinSize: this._spec.subNodeMinSize ?? 0\n      }\n    });\n    registerDataSetInstanceTransform(this._dataSet, 'compareSankeySubNodes', compareSankeySubNodes);\n    compareNodeData.transform({\n      type: 'compareSankeySubNodes'\n    });\n    this._nodesSeriesData.getDataView().parse([compareNodeData], {\n      type: 'dataview'\n    });\n\n    // 激活时的linkData\n    this._activeLinkData = new DataView(dataSet, { name: `compare-sankey-link-${this.id}-data` });\n  }\n\n  initEvent() {\n    super.initEvent();\n    this._activeLinkData?.target.addListener('change', this.activeLinkDataUpdate.bind(this));\n  }\n\n  initMark() {\n    super.initMark();\n\n    const linkMark = this._createMark(\n      { ...SankeySeries.mark.link, name: 'activeLink' },\n      {\n        dataView: this._activeLinkData\n      }\n    ) as ILinkPathMark;\n    if (linkMark) {\n      this._activeLinkMark = linkMark;\n    }\n  }\n  compileData() {\n    super.compileData();\n    this._activeLinkMark?.compileData();\n  }\n\n  _initLinkMarkStyle() {\n    super._initLinkMarkStyle();\n    // (this._activeLinkMark as any).setGlyphConfig({\n    //   direction: this.direction\n    // });\n    this._activeLinkMark.setAttribute('direction', this.direction);\n\n    this.setMarkStyle<ILinkPathMarkSpec>(\n      this._activeLinkMark,\n      {\n        x0: (datum: Datum) => datum.x0,\n        x1: (datum: Datum) => datum.x1,\n        y0: (datum: Datum) => datum.y0,\n        y1: (datum: Datum) => datum.y1,\n        thickness: (datum: Datum) => datum.thickness\n      },\n      STATE_VALUE_ENUM.STATE_NORMAL,\n      AttributeLevel.Series\n    );\n\n    this.setMarkStyle(\n      this._activeLinkMark,\n      {\n        fill: this._fillActiveLink\n      },\n      STATE_VALUE_ENUM.STATE_NORMAL,\n      AttributeLevel.User_Mark\n    );\n  }\n\n  _initNodeMarkStyle() {\n    super._initNodeMarkStyle();\n    if (this._spec.compareNodeColor) {\n      this.setMarkStyle<IRectMarkSpec>(\n        this._nodeMark,\n        {\n          fill: this._fillCompareNode\n        },\n        'normal',\n        AttributeLevel.User_Mark\n      );\n    }\n  }\n\n  private activeLinkDataUpdate() {\n    this._activeLinkMark.getData().updateData();\n  }\n\n  protected _fillCompareNode = (datum: Datum) => {\n    if (this._spec.compareNodeColor?.[datum.type]) {\n      return this._spec.compareNodeColor[datum.type];\n    }\n    return this._spec.node.style?.fill ?? this._fillByNode(datum);\n  };\n\n  protected _fillActiveLink = (datum: Datum) => {\n    if (this._spec.compareLinkColor?.[datum.type]) {\n      return this._spec.compareLinkColor[datum.type];\n    }\n    return this._spec.link.style?.fill ?? this._fillByLink(datum);\n  };\n\n  protected _handleNodeRelatedClick(element: IElement, highlightState: string, blurState: string) {\n    const nodeDatum = element.getDatum();\n    // 节点\n    const allNodeElements = this._nodeMark.getProductElements();\n    if (!allNodeElements || !allNodeElements.length) {\n      return;\n    }\n    // 边\n    const allLinkElements = this._linkMark.getProductElements();\n    if (!allLinkElements || !allLinkElements.length) {\n      return;\n    }\n\n    // 原始 link 全部进入blur状态\n    allLinkElements.forEach(el => {\n      el.removeState([STATE_VALUE_ENUM.STATE_SANKEY_EMPHASIS]);\n    });\n    this._highLightElements(allLinkElements, [], highlightState, blurState);\n\n    const nodeDatums = allNodeElements.map(el => getDatumOfGraphic(el) as Datum);\n    const pickNodeDatums = nodeDatums.filter(d => d.key === nodeDatum.key);\n    // 层级型数据\n    const highlightLinksData: any[] = [];\n    // 高亮节点key\n    const highlightNodeKeys: (string | number)[] = pickNodeDatums.map(d => d.key);\n\n    // 上游路径始终只选取第一个\n    this._activeTargetLink(\n      nodeDatum,\n      pickNodeDatums,\n      allNodeElements,\n      allLinkElements,\n      highlightNodeKeys,\n      highlightLinksData,\n      highlightState,\n      blurState\n    );\n    this._activeSourceLink(\n      nodeDatum,\n      pickNodeDatums,\n      nodeDatums,\n      allNodeElements,\n      highlightNodeKeys,\n      highlightLinksData,\n      highlightState,\n      blurState\n    );\n    this._highLightElements(allNodeElements, highlightNodeKeys as string[], highlightState, blurState);\n    this._activeLinkData.parseNewData(highlightLinksData);\n\n    this._needClear = true;\n  }\n\n  private _activeTargetLink(\n    nodeDatum: Datum,\n    pickNodeDatums: Datum[],\n    allNodeElements: any[],\n    allLinkElements: any[],\n    highlightNodeKeys: (string | number)[],\n    highlightLinksData: any[],\n    highlightState: string,\n    blurState: string\n  ) {\n    this._handleClearEmpty(highlightState, blurState);\n    const sourceNode = nodeDatum.sourceNode;\n    // 上游路径始终只选取第一个\n    const firstTarget = sourceNode.targetLinks[0];\n    if (!firstTarget) {\n      return;\n    }\n    let percent = 0;\n    pickNodeDatums.forEach(n => {\n      const link = n.targetLinks.find((l: any) => l.key === firstTarget.key);\n      if (link) {\n        const p = link.value / firstTarget.value;\n        const key = firstTarget.key + '_' + n.type;\n        const activeLink = {\n          ...firstTarget,\n          // 起点重新分配\n          y0: firstTarget.y0 - firstTarget.thickness / 2 + (percent + p / 2) * firstTarget.thickness,\n          y1: n.y0 + (p * firstTarget.thickness) / 2,\n          thickness: p * firstTarget.thickness,\n          key,\n          type: n.type,\n          [DEFAULT_DATA_INDEX]: highlightLinksData.length,\n          [DEFAULT_DATA_KEY]: key\n        };\n        highlightLinksData.push(activeLink);\n        percent += p;\n      }\n    });\n\n    // 第一个 path 中上流路径上的 node 全部高亮\n    highlightNodeKeys.push(...firstTarget.parents, nodeDatum.key);\n    const linkKeys: string[] = [];\n    // 找到对应的原始 link 也都高亮\n    for (let i = 0; i < firstTarget.parents.length - 1; i++) {\n      linkKeys.push(firstTarget.parents[i] + '-' + firstTarget.parents[i + 1]);\n    }\n    this._highLightElements(allLinkElements, linkKeys, highlightState, blurState);\n  }\n\n  private _activeSourceLink(\n    nodeDatum: Datum,\n    pickNodeDatums: Datum[],\n    allNodeDatums: Datum[],\n    allNodeElements: any[],\n    highlightNodeKeys: (string | number)[],\n    highlightLinksData: any[],\n    highlightState: string,\n    blurState: string\n  ) {\n    this._handleClearEmpty(highlightState, blurState);\n    const sourceNode = nodeDatum.sourceNode;\n    // 下游路径始需要全部处理\n    const sourceLinks = sourceNode.sourceLinks;\n    if (!sourceLinks?.length) {\n      return;\n    }\n\n    const sourceValueTemp: { [key: string]: number } = {};\n    sourceLinks.forEach((sourceLink: SankeyLinkElement) => {\n      highlightNodeKeys.push(sourceLink.target);\n      pickNodeDatums.forEach((n, index) => {\n        sourceValueTemp[n.type] = sourceValueTemp[n.type] ?? 0;\n        const link = n.sourceLinks.find((l: SankeyLinkElement) => l.key === sourceLink.key);\n        if (link) {\n          const p = link.value / n.value;\n          const totalSize = n.y1 - n.y0;\n          const size = totalSize * p;\n          const key = sourceLink.key + '_' + n.type;\n          const activeLink = {\n            ...sourceLink,\n            // 起点重新分配\n            y0: n.y0 + sourceValueTemp[n.type] * totalSize + size / 2,\n            thickness: size,\n            key,\n            type: n.type,\n            [DEFAULT_DATA_INDEX]: highlightLinksData.length,\n            [DEFAULT_DATA_KEY]: key\n          };\n          // y1 分为第一个和在其他\n          if (index === 0) {\n            // 与原始link的起点\n            activeLink.y1 = sourceLink.y1 - sourceLink.thickness / 2 + size / 2;\n          } else {\n            // 与目标子 node 的 y0 对齐\n            const targetNode = allNodeDatums.find(_n => _n.key === link.target && _n.type === n.type);\n            if (targetNode) {\n              activeLink.y1 = targetNode.y0 + size / 2;\n            } else {\n              // 错误的情况\n              return;\n            }\n          }\n          highlightLinksData.push(activeLink);\n          sourceValueTemp[n.type] += p;\n        }\n      });\n    });\n  }\n\n  protected _handleClearEmpty(highlightState: string, blurState: string) {\n    super._handleClearEmpty(highlightState, blurState);\n    this._activeLinkData.parseNewData([]);\n\n    // 同时需要清除 hover\n    const allNodeElements = this._nodeMark.getProductElements();\n    if (allNodeElements || !allNodeElements.length) {\n      allNodeElements.forEach(el => {\n        el.removeState(STATE_VALUE_ENUM.STATE_HOVER);\n      });\n    }\n\n    // 同时需要清除 hover\n    const allLinkElements = this._linkMark.getProductElements();\n    if (allLinkElements || !allLinkElements.length) {\n      allLinkElements.forEach(el => {\n        el.removeState(STATE_VALUE_ENUM.STATE_HOVER);\n      });\n    }\n  }\n\n  protected _handlePointerOutElement() {\n    this._activeLinkData.parseNewData([]);\n  }\n\n  /**\n   * 清空所有选中与 hover 相关状态（包含旧模式与单选模式的全部状态位）。\n   */\n  protected _clearAllSelectedState() {\n    super._clearAllSelectedState();\n    this._activeLinkData.parseNewData([]);\n  }\n\n  public getSubNodeDatum(filter: (n: Datum) => boolean) {\n    return getSubNodeDatum(this._nodeMark.getProductElements(), filter);\n  }\n\n  public getNodeDatumInTree(filter: (n: any) => boolean) {\n    return getNodeDatumInTree(this._rawData.latestData[0].nodes, filter);\n  }\n\n  public getNodeDatumInSubTree(filter: (n: any) => boolean) {\n    return getNodeDatumInSubTree(this._rawData.latestData[0].subNode, filter);\n  }\n}\n\nexport const registerCompareSankeyChart = (option?: { VChart?: typeof VChart }) => {\n  const vchartConstructor = option?.VChart || VChart;\n  if (vchartConstructor) {\n    vchartConstructor.useChart([CompareSankeyChart]);\n    vchartConstructor.useSeries([CompareSankeySeries]);\n  }\n};\n"]}

package/cjs/charts/compare-sankey/index.d.ts

@@ -0,0 +1,2 @@
+export * from './interface';
+export * from './compare-sankey';

package/cjs/charts/compare-sankey/index.js

@@ -0,0 +1,21 @@
+"use strict";
+
+var __createBinding = this && this.__createBinding || (Object.create ? function(o, m, k, k2) {
+    void 0 === k2 && (k2 = k);
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    desc && !("get" in desc ? !m.__esModule : desc.writable || desc.configurable) || (desc = {
+        enumerable: !0,
+        get: function() {
+            return m[k];
+        }
+    }), Object.defineProperty(o, k2, desc);
+} : function(o, m, k, k2) {
+    void 0 === k2 && (k2 = k), o[k2] = m[k];
+}), __exportStar = this && this.__exportStar || function(m, exports) {
+    for (var p in m) "default" === p || Object.prototype.hasOwnProperty.call(exports, p) || __createBinding(exports, m, p);
+};
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+}), __exportStar(require("./interface"), exports), __exportStar(require("./compare-sankey"), exports);
+//# sourceMappingURL=index.js.map

package/cjs/charts/compare-sankey/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA4B;AAC5B,mDAAiC","file":"index.js","sourcesContent":["export * from './interface';\nexport * from './compare-sankey';\n"]}

package/cjs/charts/compare-sankey/interface.d.ts

@@ -0,0 +1,24 @@
+import type { ISankeyChartSpec, ISankeySeriesSpec, ILinkPathMarkSpec } from '@visactor/vchart';
+export interface ICompareSankeySpec {
+    subNodeGap?: number;
+    subNodeMinSize?: number;
+    compareNodeColor?: {
+        [key: string]: string;
+    };
+    compareLinkColor?: {
+        [key: string]: string;
+    };
+    activeLink?: {
+        style: ILinkPathMarkSpec;
+    };
+}
+export interface ICompareSankeySeriesSpecBase extends ISankeySeriesSpec, ICompareSankeySpec {
+}
+export interface ICompareSankeyChartSpecBase extends ISankeyChartSpec, ICompareSankeySpec {
+}
+export interface ICompareSankeySeriesSpec extends Omit<ICompareSankeySeriesSpecBase, 'type' | 'series'> {
+    type: 'compareSankey';
+}
+export interface ICompareSankeyChartSpec extends Omit<ICompareSankeyChartSpecBase, 'type' | 'series'> {
+    type: 'compareSankey';
+}

package/cjs/charts/compare-sankey/interface.js

@@ -0,0 +1,6 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+    value: !0
+});
+//# sourceMappingURL=interface.js.map

package/cjs/charts/compare-sankey/interface.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/interface.ts"],"names":[],"mappings":"","file":"interface.js","sourcesContent":["import type { ISankeyChartSpec, ISankeySeriesSpec, ILinkPathMarkSpec } from '@visactor/vchart';\n\nexport interface ICompareSankeySpec {\n  /**\n   * 子节点间距\n   */\n  subNodeGap?: number;\n  /**\n   * 子节点最小高度\n   */\n  subNodeMinSize?: number;\n  /**\n   * 对比节点颜色\n   */\n  compareNodeColor?: { [key: string]: string };\n  /**\n   * 对比边点颜色\n   */\n  compareLinkColor?: { [key: string]: string };\n  /**\n   * 对比边样式\n   */\n  activeLink?: {\n    style: ILinkPathMarkSpec;\n  };\n}\n\nexport interface ICompareSankeySeriesSpecBase extends ISankeySeriesSpec, ICompareSankeySpec {}\n\nexport interface ICompareSankeyChartSpecBase extends ISankeyChartSpec, ICompareSankeySpec {}\n\nexport interface ICompareSankeySeriesSpec extends Omit<ICompareSankeySeriesSpecBase, 'type' | 'series'> {\n  type: 'compareSankey';\n}\n\nexport interface ICompareSankeyChartSpec extends Omit<ICompareSankeyChartSpecBase, 'type' | 'series'> {\n  type: 'compareSankey';\n}\n"]}

package/cjs/charts/compare-sankey/utils.d.ts

@@ -0,0 +1,9 @@
+import { Datum, IElement } from '@visactor/vgrammar-core';
+export declare const traverseTree: (subTree: any[], callback: (node: any) => void) => void;
+export declare function getSubNodeDatum(allNodeElements: IElement[], filter: (n: any) => boolean): IElement[];
+export declare function getNodeDatumInTree(nodes: any[], filter: (n: any) => boolean): any[];
+export declare function getNodeDatumInSubTree(subNodes: any[], filter: (n: any) => boolean): {
+    [key: string]: any;
+    type: string;
+    datum: Datum[];
+}[];