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

package/esm/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 { DataView } from '@visactor/vdataset';
+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;
+    protected _arrowData?: DataView;
+    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;
+    getSubNodeDatum(filter: (n: Datum) => boolean): IElement[];
+    getNodeDatumInTree(filter: (n: any) => boolean): any[];
+    getNodeDatumInSubTree(filter: (n: any) => boolean): {
+        [key: string]: any;
+        type: string;
+        datum: Datum[];
+    }[];
+}
+export declare const registerCompareSankeyChart: (option?: {
+    VChart?: typeof VChart;
+}) => void;
+export {};

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

@@ -0,0 +1,220 @@
+var __rest = this && this.__rest || function(s, e) {
+    var t = {};
+    for (var p in s) Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0 && (t[p] = s[p]);
+    if (null != s && "function" == typeof Object.getOwnPropertySymbols) {
+        var i = 0;
+        for (p = Object.getOwnPropertySymbols(s); i < p.length; i++) e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]) && (t[p[i]] = s[p[i]]);
+    }
+    return t;
+};
+
+import { VChart, SankeyChart, SankeySeries, registerDataSetInstanceTransform, STATE_VALUE_ENUM, AttributeLevel, DEFAULT_DATA_INDEX, DEFAULT_DATA_KEY } from "@visactor/vchart";
+
+import { DataView } from "@visactor/vdataset";
+
+import { CompareSankeyChartSpecTransformer } from "./compare-sankey-transformer";
+
+import { compareSankeySubNodes } from "./compare-sankey-sub-nodes";
+
+import { compareSankeySubData } from "./compare-sankey-sub-data";
+
+import { getDatumOfGraphic } from "../../utils/mark";
+
+import { traverseTree } from "./utils";
+
+export class CompareSankeyChart extends SankeyChart {
+    constructor() {
+        super(...arguments), this.type = "compareSankey", this.seriesType = CompareSankeySeries.type, 
+        this.transformerConstructor = CompareSankeyChartSpecTransformer;
+    }
+}
+
+CompareSankeyChart.type = "compareSankey", CompareSankeyChart.transformerConstructor = CompareSankeyChartSpecTransformer;
+
+export class CompareSankeySeries extends 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 DataView(dataSet, {
+            name: `compare-sankey-sub-data-${this.id}-data`
+        });
+        compareSubData.parse([ this.getViewData() ], {
+            type: "dataview"
+        }), this._subData = compareSubData;
+        const compareNodeData = new DataView(dataSet, {
+            name: `compare-sankey-node-${this.id}-data`
+        });
+        compareNodeData.parse([ compareSubData ], {
+            type: "dataview"
+        }), registerDataSetInstanceTransform(this._dataSet, "compareSankeySubData", 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
+            }
+        }), registerDataSetInstanceTransform(this._dataSet, "compareSankeySubNodes", compareSankeySubNodes), 
+        compareNodeData.transform({
+            type: "compareSankeySubNodes"
+        }), this._nodesSeriesData.getDataView().parse([ compareNodeData ], {
+            type: "dataview"
+        }), this._activeLinkData = new 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({}, 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
+        }, STATE_VALUE_ENUM.STATE_NORMAL, AttributeLevel.Series), this.setMarkStyle(this._activeLinkMark, {
+            fill: this._fillActiveLink
+        }, STATE_VALUE_ENUM.STATE_NORMAL, AttributeLevel.User_Mark);
+    }
+    _initNodeMarkStyle() {
+        super._initNodeMarkStyle(), this._spec.compareNodeColor && this.setMarkStyle(this._nodeMark, {
+            fill: this._fillCompareNode
+        }, "normal", 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([ STATE_VALUE_ENUM.STATE_SANKEY_EMPHASIS ]);
+        })), this._highLightElements(allLinkElements, [], highlightState, blurState);
+        const nodeDatums = allNodeElements.map((el => 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,
+                    [DEFAULT_DATA_INDEX]: highlightLinksData.length,
+                    [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,
+                        [DEFAULT_DATA_INDEX]: highlightLinksData.length,
+                        [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(STATE_VALUE_ENUM.STATE_HOVER);
+        }));
+        const allLinkElements = this._linkMark.getProductElements();
+        !allLinkElements && allLinkElements.length || allLinkElements.forEach((el => {
+            el.removeState(STATE_VALUE_ENUM.STATE_HOVER);
+        }));
+    }
+    getSubNodeDatum(filter) {
+        return this._nodeMark.getProductElements().filter((n => filter(n.data[0])));
+    }
+    getNodeDatumInTree(filter) {
+        const result = [];
+        return this._rawData.latestData[0].nodes.forEach((node => {
+            traverseTree([ node ], (node => {
+                filter(node) && result.push(node);
+            }));
+        })), result;
+    }
+    getNodeDatumInSubTree(filter) {
+        const list = [];
+        return this._rawData.latestData[0].subNode.forEach((subNode => {
+            const {nodes: nodes} = subNode, result = __rest(subNode, [ "nodes" ]);
+            result.datum = [], nodes.forEach((node => {
+                filter(node) && result.datum.push(node);
+            })), list.push(result);
+        })), list;
+    }
+}
+
+CompareSankeySeries.type = "compareSankey";
+
+export const registerCompareSankeyChart = option => {
+    const vchartConstructor = (null == option ? void 0 : option.VChart) || VChart;
+    vchartConstructor && (vchartConstructor.useChart([ CompareSankeyChart ]), vchartConstructor.useSeries([ CompareSankeySeries ]));
+};
+//# sourceMappingURL=compare-sankey.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/compare-sankey.ts"],"names":[],"mappings":";;;;;;;;;;;AAGA,OAAO,EACL,MAAM,EACN,WAAW,EACX,YAAY,EACZ,gCAAgC,EAChC,gBAAgB,EAChB,cAAc,EACd,kBAAkB,EAClB,gBAAgB,EACjB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AACjF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAIvC,MAAM,OAAO,kBAAmB,SAAQ,WAAwC;IAAhF;;QACE,SAAI,GAAG,eAAe,CAAC;QAEd,eAAU,GAAG,mBAAmB,CAAC,IAAI,CAAC;QAKtC,2BAAsB,GAAG,iCAAiC,CAAC;IACtE,CAAC;;AAPQ,uBAAI,GAAG,eAAe,CAAC;AAKd,yCAAsB,GAAG,iCAAiC,CAAC;AAI7E,MAAM,OAAO,mBAAoB,SAAQ,YAA0C;IAAnF;;QAEE,SAAI,GAAG,eAAe,CAAC;QA2Hb,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;IA8NJ,CAAC;IA3VC,QAAQ;;QACN,KAAK,CAAC,QAAQ,EAAE,CAAC;QAIjB,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAEjC,MAAM,cAAc,GAAG,IAAI,QAAQ,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,QAAQ,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,gCAAgC,CAAC,IAAI,CAAC,QAAQ,EAAE,sBAAsB,EAAE,oBAAoB,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,gCAAgC,CAAC,IAAI,CAAC,QAAQ,EAAE,uBAAuB,EAAE,qBAAqB,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,QAAQ,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,YAAY,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,gBAAgB,CAAC,YAAY,EAC7B,cAAc,CAAC,MAAM,CACtB,CAAC;QAEF,IAAI,CAAC,YAAY,CACf,IAAI,CAAC,eAAe,EACpB;YACE,IAAI,EAAE,IAAI,CAAC,eAAe;SAC3B,EACD,gBAAgB,CAAC,YAAY,EAC7B,cAAc,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,cAAc,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,gBAAgB,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,iBAAiB,CAAC,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,kBAAkB,CAAC,EAAE,kBAAkB,CAAC,MAAM,EAC/C,CAAC,gBAAgB,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,kBAAkB,CAAC,EAAE,kBAAkB,CAAC,MAAM,EAC/C,CAAC,gBAAgB,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,gBAAgB,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,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;IAEM,eAAe,CAAC,MAA6B;QAClD,OAAO,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC,MAAM,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrF,CAAC;IAEM,kBAAkB,CAAC,MAA2B;QACnD,MAAM,MAAM,GAAY,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAEhD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;YAC1B,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAS,EAAE,EAAE;gBACjC,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE;oBAChB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBACnB;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,qBAAqB,CAAC,MAA2B;QACtD,MAAM,IAAI,GAIJ,EAAE,CAAC;QACT,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACrD,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAY,EAAE,EAAE;YAChC,MAAM,EAAE,KAAK,KAAgB,OAAO,EAAlB,MAAM,UAAK,OAAO,EAA9B,SAAoB,CAAU,CAAC;YACrC,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC;YAClB,KAAK,CAAC,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;gBAC1B,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE;oBAChB,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBACzB;YACH,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;;AAnWM,wBAAI,GAAG,eAAe,CAAC;AAsWhC,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,MAAmC,EAAE,EAAE;IAChF,MAAM,iBAAiB,GAAG,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,KAAI,MAAM,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","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 { traverseTree } 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  protected _arrowData?: DataView;\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  public getSubNodeDatum(filter: (n: Datum) => boolean) {\n    return this._nodeMark.getProductElements().filter((n: Datum) => filter(n.data[0]));\n  }\n\n  public getNodeDatumInTree(filter: (n: any) => boolean) {\n    const result: Datum[] = [];\n    const nodes = this._rawData.latestData[0].nodes;\n\n    nodes.forEach((node: any) => {\n      traverseTree([node], (node: any) => {\n        if (filter(node)) {\n          result.push(node);\n        }\n      });\n    });\n    return result;\n  }\n\n  public getNodeDatumInSubTree(filter: (n: any) => boolean) {\n    const list: {\n      type: string;\n      datum: Datum[];\n      [key: string]: any;\n    }[] = [];\n    const subNodes = this._rawData.latestData[0].subNode;\n    subNodes.forEach((subNode: any) => {\n      const { nodes, ...result } = subNode;\n      result.datum = [];\n      nodes.forEach((node: any) => {\n        if (filter(node)) {\n          result.datum.push(node);\n        }\n      });\n      list.push(result);\n    });\n    return list;\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/esm/charts/compare-sankey/index.d.ts

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

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

@@ -0,0 +1,4 @@
+export * from "./interface";
+
+export * from "./compare-sankey";
+//# sourceMappingURL=index.js.map

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

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

package/esm/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/esm/charts/compare-sankey/interface.js

@@ -0,0 +1,2 @@
+export { };
+//# sourceMappingURL=interface.js.map

package/esm/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/esm/charts/compare-sankey/utils.d.ts

@@ -0,0 +1 @@
+export declare const traverseTree: (subTree: any[], callback: (node: any) => void) => void;

package/esm/charts/compare-sankey/utils.js

@@ -0,0 +1,6 @@
+export const traverseTree = (subTree, callback) => {
+    subTree.forEach((node => {
+        callback(node), node.children && node.children.length && traverseTree(node.children, callback);
+    }));
+};
+//# sourceMappingURL=utils.js.map

package/esm/charts/compare-sankey/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/compare-sankey/utils.ts"],"names":[],"mappings":"AACA,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,OAAc,EAAE,QAA6B,EAAE,EAAE;IAC5E,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;QACrB,QAAQ,CAAC,IAAI,CAAC,CAAC;QACf,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE;YACzC,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;SACvC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC","file":"utils.js","sourcesContent":["// 遍历树，桑基图节点通过 callback  返回到调用方\nexport const traverseTree = (subTree: any[], callback: (node: any) => void) => {\n  subTree.forEach(node => {\n    callback(node);\n    if (node.children && node.children.length) {\n      traverseTree(node.children, callback);\n    }\n  });\n};\n"]}

package/esm/charts/conversion-funnel/arrow-data-transform.d.ts

@@ -0,0 +1,24 @@
+import type { IPointLike } from '@visactor/vutils';
+import type { IConversionFunnelSpec, Arrow } from './interface';
+export interface ParsedArrow extends Arrow {
+    position: 'left' | 'right';
+    distance: number;
+    span: number;
+    layout: {
+        isLayout: boolean;
+        duplicateNode: ParsedArrow | null;
+        fromIndex: number;
+        toIndex: number;
+        fromTotal: number;
+        toTotal: number;
+        level: number;
+        points: IPointLike[];
+    };
+    context: {
+        field: string;
+    };
+    id: string;
+}
+export declare const conversionArrowTransform: (arrowSpec: IConversionFunnelSpec['conversionArrow'], options: {
+    categoryField: string;
+}) => ParsedArrow[];

package/esm/charts/conversion-funnel/arrow-data-transform.js

@@ -0,0 +1,94 @@
+import { isValid, isValidNumber, maxInArray } from "@visactor/vutils";
+
+import { isArrowCross, isSameArrow } from "./util";
+
+export const conversionArrowTransform = (arrowSpec, options) => {
+    var _a, _b, _c;
+    if (!arrowSpec || !(null === (_a = arrowSpec.arrows) || void 0 === _a ? void 0 : _a.length)) return [];
+    const {arrows: arrows} = arrowSpec, parsedArrows = parseArrow(arrows, options.categoryField), leftArrows = parsedArrows.filter((arrow => "left" === arrow.position)), rightArrows = parsedArrows.filter((arrow => "right" === arrow.position));
+    return [ ...null !== (_b = computeArrows(leftArrows)) && void 0 !== _b ? _b : [], ...null !== (_c = computeArrows(rightArrows)) && void 0 !== _c ? _c : [] ];
+};
+
+function parseArrow(arrows, categoryField) {
+    return arrows.filter((arrow => isValidNumber(arrow.from * arrow.to))).map(((arrow, index) => {
+        const {from: from, to: to, position: position = "right"} = arrow;
+        return Object.assign(Object.assign({}, arrow), {
+            position: position,
+            distance: arrow.distance || 40,
+            from: Math.min(from, to),
+            to: Math.max(from, to),
+            span: Math.abs(from - to),
+            isLayout: !1,
+            context: {
+                field: categoryField
+            },
+            layout: {
+                level: 0
+            },
+            id: `${from}-${to}-${position}-${index}`
+        });
+    }));
+}
+
+function computeArrows(arrows) {
+    if (0 === (null == arrows ? void 0 : arrows.length)) return null;
+    const nodeDegreeMap = new Map;
+    return arrows.forEach((arrow => {
+        const fromNodeDegree = nodeDegreeMap.get(arrow.from);
+        if (isValid(fromNodeDegree)) {
+            const firstNodeWidthSameFromTo = fromNodeDegree.fromArrows.find((node => isSameArrow(node, arrow)));
+            firstNodeWidthSameFromTo ? arrow.layout.duplicateNode = firstNodeWidthSameFromTo : fromNodeDegree.degree += 1, 
+            fromNodeDegree.fromArrows.push(arrow);
+        } else nodeDegreeMap.set(arrow.from, {
+            fromArrows: [ arrow ],
+            toArrows: [],
+            totalArrows: [],
+            degree: 1
+        });
+        const toNodeDegree = nodeDegreeMap.get(arrow.to);
+        if (isValid(toNodeDegree)) {
+            const firstNodeWidthSameFromTo = toNodeDegree.toArrows.find((node => isSameArrow(node, arrow)));
+            firstNodeWidthSameFromTo ? arrow.layout.duplicateNode = firstNodeWidthSameFromTo : toNodeDegree.degree += 1, 
+            toNodeDegree.toArrows.push(arrow);
+        } else nodeDegreeMap.set(arrow.to, {
+            toArrows: [ arrow ],
+            fromArrows: [],
+            totalArrows: [],
+            degree: 1
+        });
+    })), nodeDegreeMap.forEach((node => {
+        node.fromArrows.sort(((a, b) => b.span - a.span)), node.toArrows.sort(((a, b) => a.span - b.span)), 
+        node.totalArrows = [ ...node.toArrows, ...node.fromArrows ];
+    })), arrows.sort(((a, b) => a.span - b.span)).forEach((arrow => {
+        var _a;
+        const arrowsIsLayout = arrows.filter((arrow => arrow.layout.isLayout)), maxLevelArrow = maxInArray(arrowsIsLayout, ((cur, curMax) => cur.layout.level - curMax.layout.level));
+        let level = null !== (_a = null == maxLevelArrow ? void 0 : maxLevelArrow.layout.level) && void 0 !== _a ? _a : 0;
+        for (;level >= 0; ) {
+            if (arrowsIsLayout.some((arr => arr.layout.level === level && isArrowCross(arr, arrow)))) {
+                level += 1;
+                break;
+            }
+            --level;
+        }
+        arrow.layout.level = Math.max(0, level), arrow.layout.fromTotal = nodeDegreeMap.get(arrow.from).degree, 
+        arrow.layout.toTotal = nodeDegreeMap.get(arrow.to).degree;
+        const duplicateNode = arrow.layout.duplicateNode;
+        duplicateNode ? (arrow.layout.fromIndex = duplicateNode.layout.fromIndex, arrow.layout.toIndex = duplicateNode.layout.toIndex) : (arrow.layout.fromIndex = computeIndex(arrow, nodeDegreeMap.get(arrow.from).totalArrows), 
+        arrow.layout.toIndex = computeIndex(arrow, nodeDegreeMap.get(arrow.to).totalArrows)), 
+        arrow.layout.isLayout = !0;
+    })), arrows;
+}
+
+function computeIndex(arrow, totalArrows) {
+    let index = 0, duplicateCount = 0;
+    for (let i = 0; i < totalArrows.length; i++) {
+        const curArrow = totalArrows[i];
+        if (curArrow === arrow) {
+            index = i - duplicateCount;
+            break;
+        }
+        curArrow.layout.duplicateNode && duplicateCount++;
+    }
+    return index;
+}
+//# sourceMappingURL=arrow-data-transform.js.map

package/esm/charts/conversion-funnel/arrow-data-transform.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["charts/conversion-funnel/arrow-data-transform.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAuBnD,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACtC,SAAmD,EACnD,OAEC,EACD,EAAE;;IACF,IAAI,CAAC,SAAS,IAAI,CAAC,CAAA,MAAA,SAAS,CAAC,MAAM,0CAAE,MAAM,CAAA,EAAE;QAC3C,OAAO,EAAE,CAAC;KACX;IACD,MAAM,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAE7B,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC;IAC3E,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC;IAE7E,MAAM,IAAI,GAAG,MAAA,aAAa,CAAC,UAAU,CAAC,mCAAI,EAAE,CAAC;IAC7C,MAAM,KAAK,GAAG,MAAA,aAAa,CAAC,WAAW,CAAC,mCAAI,EAAE,CAAC;IAC/C,OAAO,CAAC,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC,CAAC;AAC7B,CAAC,CAAC;AAEF,SAAS,UAAU,CAAC,MAAe,EAAE,aAAqB;IACxD,OAAO,MAAM;SACV,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;SACrD,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACpB,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,GAAG,OAAO,EAAE,GAAG,KAAK,CAAC;QAC/C,OAAO,gCACF,KAAK,KACR,QAAQ,EACR,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,EAAE,EAC9B,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,EACxB,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,EACtB,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC,EACzB,QAAQ,EAAE,KAAK,EACf,OAAO,EAAE;gBACP,KAAK,EAAE,aAAa;aACrB,EACD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC;aACT,EACD,EAAE,EAAE,GAAG,IAAI,IAAI,EAAE,IAAI,QAAQ,IAAI,KAAK,EAAE,GACf,CAAC;IAC9B,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,aAAa,CAAC,MAAqB;IAC1C,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,MAAK,CAAC,EAAE;QACxB,OAAO,IAAI,CAAC;KACb;IAED,MAAM,aAAa,GAAG,IAAI,GAAG,EAQ1B,CAAC;IAEJ,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;QACrB,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,cAAc,CAAC,EAAE;YAC3B,MAAM,wBAAwB,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YAClG,IAAI,CAAC,wBAAwB,EAAE;gBAC7B,cAAc,CAAC,MAAM,IAAI,CAAC,CAAC;aAC5B;iBAAM;gBACL,KAAK,CAAC,MAAM,CAAC,aAAa,GAAG,wBAAwB,CAAC;aACvD;YACD,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SACvC;aAAM;YACL,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE;gBAC5B,UAAU,EAAE,CAAC,KAAK,CAAC;gBACnB,QAAQ,EAAE,EAAE;gBACZ,WAAW,EAAE,EAAE;gBACf,MAAM,EAAE,CAAC;aACV,CAAC,CAAC;SACJ;QAED,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,OAAO,CAAC,YAAY,CAAC,EAAE;YACzB,MAAM,wBAAwB,GAAG,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YAC9F,IAAI,CAAC,wBAAwB,EAAE;gBAC7B,YAAY,CAAC,MAAM,IAAI,CAAC,CAAC;aAC1B;iBAAM;gBACL,KAAK,CAAC,MAAM,CAAC,aAAa,GAAG,wBAAwB,CAAC;aACvD;YACD,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SACnC;aAAM;YACL,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE;gBAC1B,QAAQ,EAAE,CAAC,KAAK,CAAC;gBACjB,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,EAAE;gBACf,MAAM,EAAE,CAAC;aACV,CAAC,CAAC;SACJ;IACH,CAAC,CAAC,CAAC;IACH,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;QAE3B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;QAEhD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;QAE9C,IAAI,CAAC,WAAW,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,MAAM;SACH,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC;SAC/B,OAAO,CAAC,KAAK,CAAC,EAAE;;QACf,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAErE,MAAM,aAAa,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAE1G,IAAI,KAAK,GAAG,MAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,MAAM,CAAC,KAAK,mCAAI,CAAC,CAAC;QAC7C,OAAO,KAAK,IAAI,CAAC,EAAE;YACjB,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,KAAK,IAAI,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,EAAE;gBACtF,KAAK,IAAI,CAAC,CAAC;gBACX,MAAM;aACP;YACD,EAAE,KAAK,CAAC;SACT;QACD,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAExC,KAAK,CAAC,MAAM,CAAC,SAAS,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;QAC9D,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;QAE1D,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC;QACjD,IAAI,aAAa,EAAE;YACjB,KAAK,CAAC,MAAM,CAAC,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC;YACxD,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC;SACrD;aAAM;YACL,KAAK,CAAC,MAAM,CAAC,SAAS,GAAG,YAAY,CAAC,KAAK,EAAE,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC;YACxF,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC,KAAK,EAAE,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC;SACrF;QACD,KAAK,CAAC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEL,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,KAAkB,EAAE,WAA0B;IAClE,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QAC3C,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,QAAQ,KAAK,KAAK,EAAE;YACtB,KAAK,GAAG,CAAC,GAAG,cAAc,CAAC;YAC3B,MAAM;SACP;QACD,IAAI,QAAQ,CAAC,MAAM,CAAC,aAAa,EAAE;YACjC,cAAc,EAAE,CAAC;SAClB;KACF;IACD,OAAO,KAAK,CAAC;AACf,CAAC","file":"arrow-data-transform.js","sourcesContent":["import type { IPointLike } from '@visactor/vutils';\nimport type { IConversionFunnelSpec, Arrow } from './interface';\nimport { isValid, isValidNumber, maxInArray } from '@visactor/vutils';\nimport { isArrowCross, isSameArrow } from './util';\n\nexport interface ParsedArrow extends Arrow {\n  position: 'left' | 'right';\n  distance: number;\n  /** 层级跨度 */\n  span: number;\n  layout: {\n    isLayout: boolean;\n    duplicateNode: ParsedArrow | null;\n    fromIndex: number;\n    toIndex: number;\n    fromTotal: number;\n    toTotal: number;\n    level: number;\n    points: IPointLike[];\n  };\n  context: {\n    field: string;\n  };\n  id: string;\n}\n\nexport const conversionArrowTransform = (\n  arrowSpec: IConversionFunnelSpec['conversionArrow'],\n  options: {\n    categoryField: string;\n  }\n) => {\n  if (!arrowSpec || !arrowSpec.arrows?.length) {\n    return [];\n  }\n  const { arrows } = arrowSpec;\n\n  const parsedArrows = parseArrow(arrows, options.categoryField);\n  const leftArrows = parsedArrows.filter(arrow => arrow.position === 'left');\n  const rightArrows = parsedArrows.filter(arrow => arrow.position === 'right');\n\n  const left = computeArrows(leftArrows) ?? [];\n  const right = computeArrows(rightArrows) ?? [];\n  return [...left, ...right];\n};\n\nfunction parseArrow(arrows: Arrow[], categoryField: string): ParsedArrow[] {\n  return arrows\n    .filter(arrow => isValidNumber(arrow.from * arrow.to))\n    .map((arrow, index) => {\n      const { from, to, position = 'right' } = arrow;\n      return {\n        ...arrow,\n        position,\n        distance: arrow.distance || 40,\n        from: Math.min(from, to),\n        to: Math.max(from, to),\n        span: Math.abs(from - to),\n        isLayout: false,\n        context: {\n          field: categoryField\n        },\n        layout: {\n          level: 0\n        },\n        id: `${from}-${to}-${position}-${index}`\n      } as unknown as ParsedArrow;\n    });\n}\n\nfunction computeArrows(arrows: ParsedArrow[]) {\n  if (arrows?.length === 0) {\n    return null;\n  }\n\n  const nodeDegreeMap = new Map<\n    number,\n    {\n      fromArrows: ParsedArrow[];\n      toArrows: ParsedArrow[];\n      totalArrows: ParsedArrow[];\n      degree: number;\n    }\n  >();\n\n  arrows.forEach(arrow => {\n    const fromNodeDegree = nodeDegreeMap.get(arrow.from);\n    if (isValid(fromNodeDegree)) {\n      const firstNodeWidthSameFromTo = fromNodeDegree.fromArrows.find(node => isSameArrow(node, arrow));\n      if (!firstNodeWidthSameFromTo) {\n        fromNodeDegree.degree += 1;\n      } else {\n        arrow.layout.duplicateNode = firstNodeWidthSameFromTo;\n      }\n      fromNodeDegree.fromArrows.push(arrow);\n    } else {\n      nodeDegreeMap.set(arrow.from, {\n        fromArrows: [arrow],\n        toArrows: [],\n        totalArrows: [],\n        degree: 1\n      });\n    }\n\n    const toNodeDegree = nodeDegreeMap.get(arrow.to);\n    if (isValid(toNodeDegree)) {\n      const firstNodeWidthSameFromTo = toNodeDegree.toArrows.find(node => isSameArrow(node, arrow));\n      if (!firstNodeWidthSameFromTo) {\n        toNodeDegree.degree += 1;\n      } else {\n        arrow.layout.duplicateNode = firstNodeWidthSameFromTo;\n      }\n      toNodeDegree.toArrows.push(arrow);\n    } else {\n      nodeDegreeMap.set(arrow.to, {\n        toArrows: [arrow],\n        fromArrows: [],\n        totalArrows: [],\n        degree: 1\n      });\n    }\n  });\n  nodeDegreeMap.forEach(node => {\n    // 入边，层级跨度越大的越靠下\n    node.fromArrows.sort((a, b) => b.span - a.span);\n    // 出边，层级跨度越大的越靠上\n    node.toArrows.sort((a, b) => a.span - b.span);\n    // 入边在上，出边在下\n    node.totalArrows = [...node.toArrows, ...node.fromArrows];\n  });\n\n  arrows\n    .sort((a, b) => a.span - b.span)\n    .forEach(arrow => {\n      const arrowsIsLayout = arrows.filter(arrow => arrow.layout.isLayout);\n      // 1. 计算当前箭头所在层级\n      const maxLevelArrow = maxInArray(arrowsIsLayout, (cur, curMax) => cur.layout.level - curMax.layout.level);\n\n      let level = maxLevelArrow?.layout.level ?? 0;\n      while (level >= 0) {\n        if (arrowsIsLayout.some(arr => arr.layout.level === level && isArrowCross(arr, arrow))) {\n          level += 1;\n          break;\n        }\n        --level;\n      }\n      arrow.layout.level = Math.max(0, level);\n      // 2. 获取当前箭头所在层级的出入度信息\n      arrow.layout.fromTotal = nodeDegreeMap.get(arrow.from).degree;\n      arrow.layout.toTotal = nodeDegreeMap.get(arrow.to).degree;\n      // 3. 计算 from 和 to 的顺序\n      const duplicateNode = arrow.layout.duplicateNode;\n      if (duplicateNode) {\n        arrow.layout.fromIndex = duplicateNode.layout.fromIndex;\n        arrow.layout.toIndex = duplicateNode.layout.toIndex;\n      } else {\n        arrow.layout.fromIndex = computeIndex(arrow, nodeDegreeMap.get(arrow.from).totalArrows);\n        arrow.layout.toIndex = computeIndex(arrow, nodeDegreeMap.get(arrow.to).totalArrows);\n      }\n      arrow.layout.isLayout = true;\n    });\n\n  return arrows;\n}\n\nfunction computeIndex(arrow: ParsedArrow, totalArrows: ParsedArrow[]) {\n  let index = 0;\n  let duplicateCount = 0;\n  for (let i = 0; i < totalArrows.length; i++) {\n    const curArrow = totalArrows[i];\n    if (curArrow === arrow) {\n      index = i - duplicateCount;\n      break;\n    }\n    if (curArrow.layout.duplicateNode) {\n      duplicateCount++;\n    }\n  }\n  return index;\n}\n"]}

package/esm/charts/conversion-funnel/conversion-funnel-transformer.d.ts

@@ -0,0 +1,7 @@
+import type { IConversionFunnelChartSpecBase } from './interface';
+declare const ConversionFunnelChartSpecTransformer_base: typeof import("@visactor/vchart/esm/chart/funnel").FunnelChartSpecTransformer;
+export declare class ConversionFunnelChartSpecTransformer extends ConversionFunnelChartSpecTransformer_base<IConversionFunnelChartSpecBase> {
+    transformSpec(spec: IConversionFunnelChartSpecBase): void;
+    _getDefaultSeriesSpec(spec: IConversionFunnelChartSpecBase): import("@visactor/vchart").IFunnelSeriesSpec;
+}
+export {};