feat(visualMap): able to set the 'dimension' to an array

Author: Justin-ZS

src/component/visualMap/ContinuousModel.ts

@@ -221,7 +221,7 @@ class ContinuousModel extends VisualMapModel<ContinousVisualMapOption> {
             const dataIndices: number[] = [];
             const data = seriesModel.getData();
 
-            data.each(this.getDataDimensionIndex(data), function (value: number, dataIndex) {
+            data.each(this.getDataDimensionIndex(data, seriesModel.seriesIndex), function (value: number, dataIndex) {
                 range[0] <= value && value <= range[1] && dataIndices.push(dataIndex);
             }, this);
 

src/component/visualMap/ContinuousView.ts

@@ -855,7 +855,8 @@ class ContinuousView extends VisualMapView {
         }
 
         const data = dataModel.getData(ecData.dataType);
-        const value = data.getStore().get(visualMapModel.getDataDimensionIndex(data), ecData.dataIndex) as number;
+        const value = data.getStore()
+            .get(visualMapModel.getDataDimensionIndex(data, dataModel.seriesIndex), ecData.dataIndex) as number;
 
         if (!isNaN(value)) {
             this._showIndicator(value, value);

src/component/visualMap/PiecewiseModel.ts

@@ -313,11 +313,13 @@ class PiecewiseModel extends VisualMapModel<PiecewiseVisualMapOption> {
             const dataIndices: number[] = [];
             const data = seriesModel.getData();
 
-            data.each(this.getDataDimensionIndex(data), function (value: number, dataIndex: number) {
-                // Should always base on model pieceList, because it is order sensitive.
-                const pIdx = VisualMapping.findPieceIndex(value, pieceList);
-                pIdx === pieceIndex && dataIndices.push(dataIndex);
-            }, this);
+            data.each(
+                this.getDataDimensionIndex(data, seriesModel.seriesIndex),
+                function (value: number, dataIndex: number) {
+                    // Should always base on model pieceList, because it is order sensitive.
+                    const pIdx = VisualMapping.findPieceIndex(value, pieceList);
+                    pIdx === pieceIndex && dataIndices.push(dataIndex);
+                }, this);
 
             result.push({seriesId: seriesModel.id, dataIndex: dataIndices});
         }, this);

src/component/visualMap/VisualMapModel.ts

@@ -91,7 +91,7 @@ export interface VisualMapOption<T extends VisualOptionBase = VisualOptionBase>
     /**
      * Dimension to be encoded
      */
-    dimension?: number
+    dimension?: number | number[]
 
     /**
      * Visual configuration for the data in selection
@@ -406,8 +406,17 @@ class VisualMapModel<Opts extends VisualMapOption = VisualMapOption> extends Com
     //     }
     // }
 
-    getDataDimensionIndex(data: SeriesData): DimensionIndex {
+    getDimension(seriesIndex: number): number {
         const optDim = this.option.dimension;
+        if (isArray(optDim) && seriesIndex != null) {
+            const idx = this.getTargetSeriesIndices().indexOf(seriesIndex);
+            return optDim[idx] ?? optDim[optDim.length - 1];
+        }
+        return [].concat(optDim)[0];
+    }
+
+    getDataDimensionIndex(data: SeriesData, seriesIndex: number): DimensionIndex {
+        const optDim = this.getDimension(seriesIndex);
 
         if (optDim != null) {
             return data.getDimensionIndex(optDim);

src/component/visualMap/visualEncoding.ts

@@ -42,7 +42,7 @@ export const visualMapEncodingHandlers: StageHandler[] = [
                     visualMapModel.stateList,
                     visualMapModel.targetVisuals,
                     zrUtil.bind(visualMapModel.getValueState, visualMapModel),
-                    visualMapModel.getDataDimensionIndex(seriesModel.getData())
+                    visualMapModel.getDataDimensionIndex(seriesModel.getData(), seriesModel.seriesIndex)
                 ));
             });
 
@@ -65,7 +65,7 @@ export const visualMapEncodingHandlers: StageHandler[] = [
                         outerColors: []
                     } as VisualMeta;
 
-                    const dimIdx = visualMapModel.getDataDimensionIndex(data);
+                    const dimIdx = visualMapModel.getDataDimensionIndex(data, seriesModel.seriesIndex);
                     if (dimIdx >= 0) {
                         // visualMeta.dimension should be dimension index, but not concrete dimension.
                         visualMeta.dimension = dimIdx;