@visactor/vchart 1.6.2 → 1.6.4

package/cjs/series/sankey/sankey.js

@@ -4,7 +4,7 @@ Object.defineProperty(exports, "__esModule", {
     value: !0
 }), exports.registerSankeySeries = exports.SankeySeries = void 0;
 
-const cartesian_1 = require("../cartesian/cartesian"), type_1 = require("../interface/type"), vgrammar_sankey_1 = require("@visactor/vgrammar-sankey"), utils_1 = require("../../animation/utils"), config_1 = require("../../animation/config"), register_1 = require("../../data/register"), sankey_1 = require("../../data/transforms/sankey"), sankey_nodes_1 = require("../../data/transforms/sankey-nodes"), sankey_links_1 = require("../../data/transforms/sankey-links"), interface_1 = require("../../compile/mark/interface"), vdataset_1 = require("@visactor/vdataset"), constant_1 = require("../../constant"), series_data_1 = require("../base/series-data"), add_property_1 = require("../../data/transforms/add-property"), data_key_1 = require("../../data/transforms/data-key"), tooltip_helper_1 = require("./tooltip-helper"), vutils_1 = require("@visactor/vutils"), animation_1 = require("./animation"), color_ordinal_scale_1 = require("../../scale/color-ordinal-scale"), rect_1 = require("../../mark/rect"), text_1 = require("../../mark/text"), link_path_1 = require("../../mark/link-path"), constant_2 = require("./constant"), flatten_1 = require("../../data/transforms/flatten"), factory_1 = require("../../core/factory"), initialize_1 = require("../../data/initialize");
+const cartesian_1 = require("../cartesian/cartesian"), type_1 = require("../interface/type"), vgrammar_sankey_1 = require("@visactor/vgrammar-sankey"), utils_1 = require("../../animation/utils"), config_1 = require("../../animation/config"), register_1 = require("../../data/register"), sankey_1 = require("../../data/transforms/sankey"), sankey_nodes_1 = require("../../data/transforms/sankey-nodes"), sankey_links_1 = require("../../data/transforms/sankey-links"), interface_1 = require("../../compile/mark/interface"), vdataset_1 = require("@visactor/vdataset"), constant_1 = require("../../constant"), series_data_1 = require("../base/series-data"), tooltip_helper_1 = require("./tooltip-helper"), vutils_1 = require("@visactor/vutils"), animation_1 = require("./animation"), color_ordinal_scale_1 = require("../../scale/color-ordinal-scale"), rect_1 = require("../../mark/rect"), text_1 = require("../../mark/text"), link_path_1 = require("../../mark/link-path"), constant_2 = require("./constant"), flatten_1 = require("../../data/transforms/flatten"), factory_1 = require("../../core/factory"), initialize_1 = require("../../data/initialize"), data_key_1 = require("../../data/transforms/data-key");
 
 class SankeySeries extends cartesian_1.CartesianSeries {
     constructor() {
@@ -313,11 +313,16 @@ class SankeySeries extends cartesian_1.CartesianSeries {
         this._labelLimit = null !== (_c = null === (_b = this._spec.label) || void 0 === _b ? void 0 : _b.limit) && void 0 !== _c ? _c : 100;
     }
     initData() {
-        var _a, _b, _c, _d, _e, _f;
-        if (super.initData(), this.getViewData()) {
-            (0, register_1.registerDataSetInstanceTransform)(this._dataSet, "sankey", sankey_1.sankey), 
-            this.addViewDataFilter({
-                type: "sankey",
+        var _a, _b, _c, _d;
+        super.initData();
+        const viewData = this.getViewData(), rawData = this.getRawData();
+        if (rawData && viewData) {
+            (0, register_1.registerDataSetInstanceTransform)(this._dataSet, "sankeyLayout", sankey_1.sankeyLayout), 
+            (0, register_1.registerDataSetInstanceTransform)(this._dataSet, "sankeyFormat", sankey_1.sankeyFormat), 
+            rawData.transform({
+                type: "sankeyFormat"
+            }, !1), viewData.transform({
+                type: "sankeyLayout",
                 options: {
                     view: () => ({
                         x0: this._viewBox.x1,
@@ -344,12 +349,10 @@ class SankeySeries extends cartesian_1.CartesianSeries {
                 },
                 level: initialize_1.TransformLevel.sankeyLayout
             });
-            const nodesDataSet = new vdataset_1.DataSet;
-            (0, register_1.registerDataSetInstanceParser)(nodesDataSet, "dataview", vdataset_1.dataViewParser), 
-            (0, register_1.registerDataSetInstanceTransform)(nodesDataSet, "sankeyNodes", sankey_nodes_1.sankeyNodes), 
-            (0, register_1.registerDataSetInstanceTransform)(nodesDataSet, "addVChartProperty", add_property_1.addVChartProperty), 
-            (0, register_1.registerDataSetInstanceTransform)(nodesDataSet, "flatten", flatten_1.flatten);
-            const nodesDataView = new vdataset_1.DataView(nodesDataSet, {
+            const {dataSet: dataSet} = this._option;
+            (0, register_1.registerDataSetInstanceTransform)(dataSet, "sankeyNodes", sankey_nodes_1.sankeyNodes), 
+            (0, register_1.registerDataSetInstanceTransform)(dataSet, "flatten", flatten_1.flatten);
+            const nodesDataView = new vdataset_1.DataView(dataSet, {
                 name: `sankey-node-${this.id}-data`
             });
             nodesDataView.parse([ this.getViewData() ], {
@@ -367,19 +370,15 @@ class SankeySeries extends cartesian_1.CartesianSeries {
                         return node;
                     }
                 }
-            }), nodesDataView.transform({
+            }, !1), nodesDataView.transform({
                 type: "addVChartProperty",
                 options: {
                     beforeCall: data_key_1.initKeyMap.bind(this),
                     call: data_key_1.addDataKey
                 }
-            }, !1), null === (_e = this._data) || void 0 === _e || _e.getDataView().target.addListener("change", nodesDataView.reRunAllTransform), 
-            this._nodesSeriesData = new series_data_1.SeriesData(this._option, nodesDataView);
-            const linksDataSet = new vdataset_1.DataSet;
-            (0, register_1.registerDataSetInstanceParser)(linksDataSet, "dataview", vdataset_1.dataViewParser), 
-            (0, register_1.registerDataSetInstanceTransform)(linksDataSet, "sankeyLinks", sankey_links_1.sankeyLinks), 
-            (0, register_1.registerDataSetInstanceTransform)(linksDataSet, "addVChartProperty", add_property_1.addVChartProperty);
-            const linksDataView = new vdataset_1.DataView(linksDataSet, {
+            }, !1), this._nodesSeriesData = new series_data_1.SeriesData(this._option, nodesDataView), 
+            (0, register_1.registerDataSetInstanceTransform)(dataSet, "sankeyLinks", sankey_links_1.sankeyLinks);
+            const linksDataView = new vdataset_1.DataView(dataSet, {
                 name: `sankey-link-${this.id}-data`
             });
             linksDataView.parse([ this.getViewData() ], {
@@ -392,8 +391,7 @@ class SankeySeries extends cartesian_1.CartesianSeries {
                     beforeCall: data_key_1.initKeyMap.bind(this),
                     call: data_key_1.addDataKey
                 }
-            }, !1), null === (_f = this._data) || void 0 === _f || _f.getDataView().target.addListener("change", linksDataView.reRunAllTransform), 
-            this._linksSeriesData = new series_data_1.SeriesData(this._option, linksDataView);
+            }, !1), this._linksSeriesData = new series_data_1.SeriesData(this._option, linksDataView);
         }
     }
     initMark() {
@@ -430,8 +428,8 @@ class SankeySeries extends cartesian_1.CartesianSeries {
             y: datum => datum.y0,
             y1: datum => datum.y1,
             fill: datum => {
-                var _a, _b, _c;
-                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(this._getNodeNameFromData(datum));
+                var _a, _b, _c, _d;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : null === (_d = this._colorScale) || void 0 === _d ? void 0 : _d.scale(this._getNodeNameFromData(datum));
             }
         }, interface_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Mark), this._trigger.registerMark(nodeMark));
     }
@@ -446,8 +444,10 @@ class SankeySeries extends cartesian_1.CartesianSeries {
             thickness: datum => datum.thickness,
             fill: datum => {
                 var _a, _b, _c, _d, _e, _f, _g;
-                const sourceName = (null === (_a = this._spec) || void 0 === _a ? void 0 : _a.nodeKey) || (null === (_d = null === (_c = null === (_b = this._rawData.latestData[0]) || void 0 === _b ? void 0 : _b.nodes) || void 0 === _c ? void 0 : _c[0]) || void 0 === _d ? void 0 : _d.children) ? datum.source : this.getNodeList()[datum.source];
-                return null !== (_g = null === (_f = null === (_e = this._spec.link) || void 0 === _e ? void 0 : _e.style) || void 0 === _f ? void 0 : _f.fill) && void 0 !== _g ? _g : this.getNodeOrdinalColorScale(sourceName);
+                const fill = null === (_b = null === (_a = this._spec.link) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill;
+                if (fill) return fill;
+                const sourceName = (null === (_c = this._spec) || void 0 === _c ? void 0 : _c.nodeKey) || (null === (_f = null === (_e = null === (_d = this._rawData.latestData[0]) || void 0 === _d ? void 0 : _d.nodes) || void 0 === _e ? void 0 : _e[0]) || void 0 === _f ? void 0 : _f.children) ? datum.source : this.getNodeList()[datum.source];
+                return null === (_g = this._colorScale) || void 0 === _g ? void 0 : _g.scale(sourceName);
             },
             direction: null !== (_a = this._spec.direction) && void 0 !== _a ? _a : "horizontal"
         }, interface_1.STATE_VALUE_ENUM.STATE_NORMAL, constant_1.AttributeLevel.Series), 
@@ -491,8 +491,8 @@ class SankeySeries extends cartesian_1.CartesianSeries {
             x: datum => (datum.x0 + datum.x1) / 2,
             y: datum => datum.y1 >= this._viewBox.y2 ? datum.y0 : datum.y1,
             fill: datum => {
-                var _a, _b, _c;
-                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(this._getNodeNameFromData(datum));
+                var _a, _b, _c, _d;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : null === (_d = this._colorScale) || void 0 === _d ? void 0 : _d.scale(this._getNodeNameFromData(datum));
             },
             text: datum => this._createText(datum),
             limit: this._labelLimit,
@@ -535,8 +535,8 @@ class SankeySeries extends cartesian_1.CartesianSeries {
             x: datum => datum.x0,
             y: datum => (datum.y0 + datum.y1) / 2,
             fill: datum => {
-                var _a, _b, _c;
-                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(this._getNodeNameFromData(datum));
+                var _a, _b, _c, _d;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : null === (_d = this._colorScale) || void 0 === _d ? void 0 : _d.scale(this._getNodeNameFromData(datum));
             },
             text: datum => this._createText(datum),
             limit: this._labelLimit,
@@ -546,8 +546,8 @@ class SankeySeries extends cartesian_1.CartesianSeries {
             x: datum => datum.x1,
             y: datum => (datum.y0 + datum.y1) / 2,
             fill: datum => {
-                var _a, _b, _c;
-                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(this._getNodeNameFromData(datum));
+                var _a, _b, _c, _d;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : null === (_d = this._colorScale) || void 0 === _d ? void 0 : _d.scale(this._getNodeNameFromData(datum));
             },
             text: datum => this._createText(datum),
             limit: this._labelLimit,
@@ -557,8 +557,8 @@ class SankeySeries extends cartesian_1.CartesianSeries {
             x: datum => datum.x1 >= this._viewBox.x2 ? datum.x0 : datum.x1,
             y: datum => (datum.y0 + datum.y1) / 2,
             fill: datum => {
-                var _a, _b, _c;
-                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : this.getNodeOrdinalColorScale(this._getNodeNameFromData(datum));
+                var _a, _b, _c, _d;
+                return null !== (_c = null === (_b = null === (_a = this._spec.node) || void 0 === _a ? void 0 : _a.style) || void 0 === _b ? void 0 : _b.fill) && void 0 !== _c ? _c : null === (_d = this._colorScale) || void 0 === _d ? void 0 : _d.scale(this._getNodeNameFromData(datum));
             },
             text: datum => this._createText(datum),
             limit: this._labelLimit,
@@ -605,36 +605,27 @@ class SankeySeries extends cartesian_1.CartesianSeries {
         }, this._handleRelatedClick));
     }
     nodesSeriesDataUpdate() {
-        this.event.emit(constant_1.ChartEvent.legendFilter, {
-            model: this
-        }), this._nodesSeriesData.updateData();
+        this._nodesSeriesData.updateData(), this._setNodeOrdinalColorScale();
     }
     linksSeriesDataUpdate() {
-        this.event.emit(constant_1.ChartEvent.legendFilter, {
-            model: this
-        }), this._linksSeriesData.updateData();
+        this._linksSeriesData.updateData();
     }
     initTooltip() {
         this._tooltipHelper = new tooltip_helper_1.SankeySeriesTooltipHelper(this), this._nodeMark && this._tooltipHelper.activeTriggerSet.mark.add(this._nodeMark), 
         this._linkMark && this._tooltipHelper.activeTriggerSet.mark.add(this._linkMark), 
         this._labelMark && this._tooltipHelper.activeTriggerSet.mark.add(this._labelMark);
     }
-    getNodeOrdinalColorScale(item) {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o;
-        if (!(0, vutils_1.isNil)(null === (_c = null === (_b = null === (_a = this._option) || void 0 === _a ? void 0 : _a.globalScale) || void 0 === _b ? void 0 : _b.getScale("color")) || void 0 === _c ? void 0 : _c._specified)) {
-            const specified = this._option.globalScale.getScale("color")._specified, colorDomain = [], colorRange = [];
-            for (const key in specified) colorDomain.push(key), colorRange.push(specified[key]);
-            const ordinalScale = new color_ordinal_scale_1.ColorOrdinalScale;
-            return null === (_e = (_d = ordinalScale.domain(colorDomain)).range) || void 0 === _e || _e.call(_d, colorRange), 
-            ordinalScale.scale(item);
-        }
-        const colorDomain = (0, vutils_1.isNil)(null === (_g = null === (_f = this._option.globalScale.getScale("color")) || void 0 === _f ? void 0 : _f.domain()) || void 0 === _g ? void 0 : _g[0]) ? this.getNodeList() : this._option.globalScale.getScale("color").domain();
-        let colorRange = null !== (_j = null === (_h = this._option.globalScale.getScale("color")) || void 0 === _h ? void 0 : _h.range()) && void 0 !== _j ? _j : this._getDataScheme();
-        (0 === (null === (_k = this._option.globalScale.getScale("color")) || void 0 === _k ? void 0 : _k.domain().length) || (0, 
-        vutils_1.isNil)(this._option.globalScale.getScale("color").domain()[0])) && colorDomain.length > 10 && (colorRange = null === (_l = this._getDataScheme()[1]) || void 0 === _l ? void 0 : _l.scheme);
+    _setNodeOrdinalColorScale() {
+        var _a, _b, _c, _d, _e;
+        const colorScale = null === (_b = null === (_a = this._option) || void 0 === _a ? void 0 : _a.globalScale) || void 0 === _b ? void 0 : _b.getScale("color");
+        if (null == colorScale ? void 0 : colorScale._specified) return void (this._colorScale = colorScale);
+        let colorDomain, colorRange;
+        colorScale && (colorDomain = colorScale.domain(), colorRange = colorScale.range()), 
+        colorRange || (colorRange = this._getDataScheme()), colorDomain && !(0, vutils_1.isNil)(colorDomain[0]) || (colorDomain = this.getNodeList(), 
+        colorDomain.length > 10 && (colorRange = null === (_c = this._getDataScheme()[1]) || void 0 === _c ? void 0 : _c.scheme));
         const ordinalScale = new color_ordinal_scale_1.ColorOrdinalScale;
-        return null === (_o = (_m = ordinalScale.domain(colorDomain)).range) || void 0 === _o || _o.call(_m, colorRange), 
-        ordinalScale.scale(item);
+        null === (_e = (_d = ordinalScale.domain(colorDomain)).range) || void 0 === _e || _e.call(_d, colorRange), 
+        this._colorScale = ordinalScale;
     }
     getNodeList() {
         var _a, _b, _c;
@@ -657,21 +648,31 @@ class SankeySeries extends cartesian_1.CartesianSeries {
     getMeasureField() {
         return [ this._valueField ];
     }
-    getSeriesKeys() {
-        var _a, _b;
-        if (this._seriesField) {
-            const keyArray = [];
-            return null === (_b = null === (_a = this._nodesSeriesData) || void 0 === _a ? void 0 : _a.getDataView()) || void 0 === _b || _b.latestData.forEach((datum => {
-                var _a;
-                keyArray.push(null !== (_a = datum[this._seriesField]) && void 0 !== _a ? _a : datum.datum[this._seriesField]);
-            })), keyArray;
+    getRawDataStatisticsByField(field, isNumeric) {
+        var _a;
+        if (this._rawStatisticsCache || (this._rawStatisticsCache = {}), !this._rawStatisticsCache[field]) {
+            this._viewDataStatistics && this.getViewData().transformsArr.length <= 1 && (null === (_a = this._viewDataStatistics.latestData) || void 0 === _a ? void 0 : _a[field]) ? this._rawStatisticsCache[field] = this._viewDataStatistics.latestData[field] : this._rawData && (this._rawStatisticsCache[field] = {
+                values: this._collectByField(field)
+            });
         }
-        return [];
+        return this._rawStatisticsCache[field];
+    }
+    _collectByField(field) {
+        var _a, _b, _c;
+        const keyArray = [], rawData = null === (_b = null === (_a = this.getRawData()) || void 0 === _a ? void 0 : _a.latestData) || void 0 === _b ? void 0 : _b[0];
+        if (!rawData) return [];
+        if (rawData.links) (null === (_c = rawData.nodes) || void 0 === _c ? void 0 : _c.length) && rawData.nodes.forEach((node => {
+            node[this._seriesField] && keyArray.push(node[this._seriesField]);
+        })); else if (rawData.nodes) {
+            const set = new Set;
+            return (0, sankey_1.collectHierarchyField)(set, rawData.nodes, this._seriesField), 
+            Array.from(set);
+        }
+        return keyArray;
     }
     onLayoutEnd(ctx) {
         super.onLayoutEnd(ctx), this._viewBox.set(0, 0, this._region.getLayoutRect().width, this._region.getLayoutRect().height), 
-        this._rawData.reRunAllTransform(), this.getViewData().reRunAllTransform(), this._nodesSeriesData.updateData(), 
-        this._linksSeriesData.updateData();
+        this.getViewData().reRunAllTransform(), this._nodesSeriesData.updateData(), this._linksSeriesData.updateData();
     }
     getDefaultShapeType() {
         return "square";