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

package/build/index.js

@@ -4471,6 +4471,34 @@
 	        }
 	    });
 	};
+	function getSubNodeDatum(allNodeElements, filter) {
+	    return allNodeElements.filter((n) => filter(n.data[0]));
+	}
+	function getNodeDatumInTree(nodes, filter) {
+	    const result = [];
+	    nodes.forEach((node) => {
+	        traverseTree([node], (node) => {
+	            if (filter(node)) {
+	                result.push(node);
+	            }
+	        });
+	    });
+	    return result;
+	}
+	function getNodeDatumInSubTree(subNodes, filter) {
+	    const list = [];
+	    subNodes.forEach((subNode) => {
+	        const { nodes } = subNode, result = __rest(subNode, ["nodes"]);
+	        result.datum = [];
+	        nodes.forEach((node) => {
+	            if (filter(node)) {
+	                result.datum.push(node);
+	            }
+	        });
+	        list.push(result);
+	    });
+	    return list;
+	}
 
 	class CompareSankeyChart extends vchart.SankeyChart {
 	    constructor() {
@@ -4682,34 +4710,13 @@
 	        }
 	    }
 	    getSubNodeDatum(filter) {
-	        return this._nodeMark.getProductElements().filter((n) => filter(n.data[0]));
+	        return getSubNodeDatum(this._nodeMark.getProductElements(), filter);
 	    }
 	    getNodeDatumInTree(filter) {
-	        const result = [];
-	        const nodes = this._rawData.latestData[0].nodes;
-	        nodes.forEach((node) => {
-	            traverseTree([node], (node) => {
-	                if (filter(node)) {
-	                    result.push(node);
-	                }
-	            });
-	        });
-	        return result;
+	        return getNodeDatumInTree(this._rawData.latestData[0].nodes, filter);
 	    }
 	    getNodeDatumInSubTree(filter) {
-	        const list = [];
-	        const subNodes = this._rawData.latestData[0].subNode;
-	        subNodes.forEach((subNode) => {
-	            const { nodes } = subNode, result = __rest(subNode, ["nodes"]);
-	            result.datum = [];
-	            nodes.forEach((node) => {
-	                if (filter(node)) {
-	                    result.datum.push(node);
-	                }
-	            });
-	            list.push(result);
-	        });
-	        return list;
+	        return getNodeDatumInSubTree(this._rawData.latestData[0].subNode, filter);
 	    }
 	}
 	CompareSankeySeries.type = 'compareSankey';
@@ -4721,6 +4728,228 @@
 	    }
 	};
 
+	class CompareSankeyDiffChartSpecTransformer extends vchart.SankeyChartSpecTransformer {
+	    constructor() {
+	        super(...arguments);
+	        this.seriesType = 'compareSankeyDiff';
+	    }
+	    transformSpec(spec) {
+	        super.transformSpec(spec);
+	    }
+	    _getDefaultSeriesSpec(spec) {
+	        const seriesSpec = super._getDefaultSeriesSpec(spec);
+	        seriesSpec.subNodeGap = spec.subNodeGap;
+	        seriesSpec.subNodeMinSize = spec.subNodeMinSize;
+	        seriesSpec.compareNodeColor = spec.compareNodeColor;
+	        seriesSpec.compareLinkColor = spec.compareLinkColor;
+	        return seriesSpec;
+	    }
+	}
+
+	const compareSankeyDiffSubData = (data, opt) => {
+	    var _a, _b, _c, _d, _e, _f, _g;
+	    const viewData = data[0];
+	    if (!((_a = viewData.latestData) === null || _a === void 0 ? void 0 : _a.length)) {
+	        return {};
+	    }
+	    const subNodeMinSize = (_b = opt.subNodeMinSize) !== null && _b !== void 0 ? _b : 0;
+	    const subNodeGap = opt.subNodeGap;
+	    const subDataList = [];
+	    const sankeyLayoutTransform = viewData.dataSet.getTransform('sankeyLayout');
+	    const sankeyTransformOption = Object.assign({}, viewData.transformsArr.find(t => t.type === 'sankeyLayout').options);
+	    delete sankeyTransformOption.customLayout;
+	    const viewDataLatest = viewData.latestData[0];
+	    (_g = (_f = (_e = (_d = (_c = viewData.rawData) === null || _c === void 0 ? void 0 : _c[0]) === null || _d === void 0 ? void 0 : _d.latestData) === null || _e === void 0 ? void 0 : _e[0]) === null || _f === void 0 ? void 0 : _f.subNode) === null || _g === void 0 ? void 0 : _g.forEach((subGroup) => {
+	        const { nodes } = subGroup, rest = __rest(subGroup, ["nodes"]);
+	        const data = sankeyLayoutTransform([Object.assign({ nodes }, rest)], sankeyTransformOption);
+	        Object.keys(rest).forEach(key => {
+	            data[0].nodes.forEach((n) => {
+	                n[key] = rest[key];
+	            });
+	        });
+	        subDataList.push({ data: data, external: rest });
+	    });
+	    const subCount = subDataList.length;
+	    viewData.latestData[0].nodes.forEach((n, index) => {
+	        let currentY = n.y0;
+	        const totalSize = n.y1 - n.y0 - (subCount - 1) * subNodeGap;
+	        const totalValue = n.value;
+	        subDataList.forEach((subData) => {
+	            var _a;
+	            let subN = subData.data[0].nodes[index];
+	            if (subN.key !== n.key) {
+	                subN = subData.data[0].nodes.find((n) => n.key === n.key);
+	            }
+	            if (!subN) {
+	                return;
+	            }
+	            const percent = totalValue === 0 ? 0 : subN.value / totalValue;
+	            const nodeSize = Math.max(subNodeMinSize, totalSize * percent);
+	            const diffX = n.x0 - subN.x0;
+	            const lastY0 = subN.y0;
+	            const lastSize = subN.y1 - subN.y0;
+	            const sizeMultiply = nodeSize / lastSize;
+	            subN._compare_compute_temp = {
+	                diffX,
+	                lastY0,
+	                lastSize,
+	                sizeMultiply
+	            };
+	            subN.x0 = n.x0;
+	            subN.x1 = n.x1;
+	            subN.y0 = currentY;
+	            subN.y1 = currentY + nodeSize;
+	            currentY += nodeSize + subNodeGap;
+	            (_a = subN.sourceLinks) === null || _a === void 0 ? void 0 : _a.forEach(link => {
+	                var _a;
+	                link.type = subN.type;
+	                link.x0 = ((_a = link.x0) !== null && _a !== void 0 ? _a : 0) + diffX;
+	                link.thickness = link.thickness * sizeMultiply;
+	                link.y0 = (link.y0 - lastY0) * sizeMultiply + subN.y0;
+	                if (link.sourceRect) {
+	                    link.sourceRect.x0 = subN.x0;
+	                    link.sourceRect.x1 = subN.x1;
+	                    link.sourceRect.y0 = subN.y0;
+	                    link.sourceRect.y1 = subN.y1;
+	                }
+	            });
+	        });
+	    });
+	    subDataList.forEach((subData) => {
+	        subData.data[0].nodes.forEach((n) => {
+	            var _a;
+	            const { diffX, lastY0, sizeMultiply } = n._compare_compute_temp;
+	            (_a = n.targetLinks) === null || _a === void 0 ? void 0 : _a.forEach(link => {
+	                var _a;
+	                link.x1 = ((_a = link.x1) !== null && _a !== void 0 ? _a : 0) + diffX;
+	                link.y1 = (link.y1 - lastY0) * sizeMultiply + n.y0;
+	                if (link.targetRect) {
+	                    link.targetRect.x0 = n.x0;
+	                    link.targetRect.x1 = n.x1;
+	                    link.targetRect.y0 = n.y0;
+	                    link.targetRect.y1 = n.y1;
+	                }
+	            });
+	        });
+	    });
+	    const finalViewData = {
+	        columns: [],
+	        links: [],
+	        nodes: []
+	    };
+	    finalViewData.columns = viewDataLatest.columns.reduce((prev, cur, curIndex) => {
+	        prev.push(subDataList.reduce((prev, cur) => prev.concat(cur.data[0].columns[curIndex]), []));
+	        return prev;
+	    }, []);
+	    finalViewData.links = subDataList.reduce((prev, cur) => prev.concat(cur.data[0].links), []);
+	    finalViewData.nodes = subDataList.reduce((prev, cur) => prev.concat(cur.data[0].nodes), []);
+	    return [finalViewData];
+	};
+
+	class CompareSankeyDiffChart extends vchart.SankeyChart {
+	    constructor() {
+	        super(...arguments);
+	        this.type = 'compareSankeyDiff';
+	        this.seriesType = CompareSankeyDiffSeries.type;
+	        this.transformerConstructor = CompareSankeyDiffChartSpecTransformer;
+	    }
+	}
+	CompareSankeyDiffChart.type = 'compareSankeyDiff';
+	CompareSankeyDiffChart.transformerConstructor = CompareSankeyDiffChartSpecTransformer;
+	class CompareSankeyDiffSeries extends vchart.SankeySeries {
+	    constructor() {
+	        super(...arguments);
+	        this.type = 'compareSankeyDiff';
+	        this._fillCompareNode = (datum) => {
+	            var _a, _b, _c, _d;
+	            const specAny = this._spec;
+	            if ((_a = specAny.compareNodeColor) === null || _a === void 0 ? void 0 : _a[datum.type]) {
+	                return specAny.compareNodeColor[datum.type];
+	            }
+	            return (_d = (_c = (_b = this._spec.node) === null || _b === void 0 ? void 0 : _b.style) === null || _c === void 0 ? void 0 : _c.fill) !== null && _d !== void 0 ? _d : this._fillByNode(datum);
+	        };
+	        this._fillCompareLink = (datum) => {
+	            var _a, _b, _c, _d;
+	            const specAny = this._spec;
+	            if ((_a = specAny.compareLinkColor) === null || _a === void 0 ? void 0 : _a[datum.type]) {
+	                return specAny.compareLinkColor[datum.type];
+	            }
+	            return (_d = (_c = (_b = this._spec.link) === null || _b === void 0 ? void 0 : _b.style) === null || _c === void 0 ? void 0 : _c.fill) !== null && _d !== void 0 ? _d : this._fillByLink(datum);
+	        };
+	    }
+	    initData() {
+	        var _a, _b;
+	        super.initData();
+	        const { dataSet } = this._option;
+	        const compareData = new DataView(dataSet, { name: `compare-sankey-diff-compare-data-${this.id}-data` });
+	        compareData.parse([this.getViewData()], { type: 'dataview' });
+	        vchart.registerDataSetInstanceTransform(this._dataSet, 'compareSankeyDiffSubData', compareSankeyDiffSubData);
+	        compareData.transform({
+	            type: 'compareSankeyDiffSubData',
+	            options: {
+	                rawData: () => this.getRawData(),
+	                valueField: this._valueField,
+	                nodeKey: this._spec.nodeKey,
+	                subNodeGap: (_a = this._spec.subNodeGap) !== null && _a !== void 0 ? _a : 2,
+	                subNodeMinSize: (_b = this._spec.subNodeMinSize) !== null && _b !== void 0 ? _b : 0
+	            }
+	        });
+	        this._nodesSeriesData.getDataView().parse([compareData], {
+	            type: 'dataview'
+	        });
+	        this._linksSeriesData.getDataView().parse([compareData], {
+	            type: 'dataview'
+	        });
+	    }
+	    compileData() {
+	        super.compileData();
+	    }
+	    _initLinkMarkStyle() {
+	        super._initLinkMarkStyle();
+	        if (this._spec.compareLinkColor) {
+	            this.setMarkStyle(this._linkMark, { fill: this._fillCompareLink }, 'normal', vchart.AttributeLevel.User_Mark);
+	        }
+	    }
+	    _initNodeMarkStyle() {
+	        super._initNodeMarkStyle();
+	        if (this._spec.compareNodeColor) {
+	            this.setMarkStyle(this._nodeMark, { fill: this._fillCompareNode }, 'normal', vchart.AttributeLevel.User_Mark);
+	        }
+	    }
+	    _handleClearEmpty(highlightState, blurState) {
+	        super._handleClearEmpty(highlightState, blurState);
+	        const allNodeElements = this._nodeMark.getProductElements();
+	        if (allNodeElements || !allNodeElements.length) {
+	            allNodeElements.forEach(el => {
+	                el.removeState(vchart.STATE_VALUE_ENUM.STATE_HOVER);
+	            });
+	        }
+	        const allLinkElements = this._linkMark.getProductElements();
+	        if (allLinkElements || !allLinkElements.length) {
+	            allLinkElements.forEach(el => {
+	                el.removeState(vchart.STATE_VALUE_ENUM.STATE_HOVER);
+	            });
+	        }
+	    }
+	    getSubNodeDatum(filter) {
+	        return getSubNodeDatum(this._nodeMark.getProductElements(), filter);
+	    }
+	    getNodeDatumInTree(filter) {
+	        return getNodeDatumInTree(this._rawData.latestData[0].nodes, filter);
+	    }
+	    getNodeDatumInSubTree(filter) {
+	        return getNodeDatumInSubTree(this._rawData.latestData[0].subNode, filter);
+	    }
+	}
+	CompareSankeyDiffSeries.type = 'compareSankeyDiff';
+	const registerCompareSankeyDiffChart = (option) => {
+	    const vchartConstructor = (option === null || option === void 0 ? void 0 : option.VChart) || vchart.VChart;
+	    if (vchartConstructor) {
+	        vchartConstructor.useChart([CompareSankeyDiffChart]);
+	        vchartConstructor.useSeries([CompareSankeyDiffSeries]);
+	    }
+	};
+
 	class Generator {
 	  static GenAutoIncrementId() {
 	    return Generator.auto_increment_id++;
@@ -14447,6 +14676,8 @@
 	exports.BAR_LINK = BAR_LINK;
 	exports.BarLinkComponent = BarLinkComponent;
 	exports.CompareSankeyChart = CompareSankeyChart;
+	exports.CompareSankeyDiffChart = CompareSankeyDiffChart;
+	exports.CompareSankeyDiffSeries = CompareSankeyDiffSeries;
 	exports.CompareSankeySeries = CompareSankeySeries;
 	exports.ConversionFunnelChart = ConversionFunnelChart;
 	exports.ConversionFunnelSeries = ConversionFunnelSeries;
@@ -14470,6 +14701,7 @@
 	exports.getSeriesLabelConfig = getSeriesLabelConfig;
 	exports.registerBarLink = registerBarLink;
 	exports.registerCompareSankeyChart = registerCompareSankeyChart;
+	exports.registerCompareSankeyDiffChart = registerCompareSankeyDiffChart;
 	exports.registerConversionFunnelChart = registerConversionFunnelChart;
 	exports.registerRankingBarChart = registerRankingBarChart;
 	exports.registerRankingList = registerRankingList;