feat(world-map): support color by metric or country column

superset-frontend/packages/superset-ui-chart-controls/src/shared-controls/components/RadioButtonControl.tsx

@@ -53,8 +53,12 @@ export default function RadioButtonControl({
         '.btn:focus': {
           outline: 'none',
         },
+        '.control-label': {
+          color: theme.colors.grayscale.base,
+          marginBottom: theme.gridUnit,
+        },
         '.control-label + .btn-group': {
-          marginTop: 1,
+          marginTop: theme.gridUnit * 0.25,
         },
         '.btn-group .btn-default': {
           color: theme.colors.grayscale.dark1,

superset-frontend/plugins/legacy-plugin-chart-world-map/src/WorldMap.js

@@ -23,8 +23,10 @@ import { extent as d3Extent } from 'd3-array';
 import {
   getNumberFormatter,
   getSequentialSchemeRegistry,
+  CategoricalColorNamespace,
 } from '@superset-ui/core';
 import Datamap from 'datamaps/dist/datamaps.world.min';
+import { ColorBy } from './utils';
 
 const propTypes = {
   data: PropTypes.arrayOf(
@@ -55,6 +57,9 @@ function WorldMap(element, props) {
     showBubbles,
     linearColorScheme,
     color,
+    colorBy,
+    colorScheme,
+    sliceId,
   } = props;
   const div = d3.select(element);
   div.classed('superset-legacy-chart-world-map', true);
@@ -69,15 +74,27 @@ function WorldMap(element, props) {
     .domain([extRadius[0], extRadius[1]])
     .range([1, maxBubbleSize]);
 
-  const colorScale = getSequentialSchemeRegistry()
-    .get(linearColorScheme)
-    .createLinearScale(d3Extent(filteredData, d => d.m1));
+  let processedData;
+  let colorScale;
+  if (colorBy === ColorBy.country) {
+    colorScale = CategoricalColorNamespace.getScale(colorScheme);
 
-  const processedData = filteredData.map(d => ({
-    ...d,
-    radius: radiusScale(Math.sqrt(d.m2)),
-    fillColor: colorScale(d.m1),
-  }));
+    processedData = filteredData.map(d => ({
+      ...d,
+      radius: radiusScale(Math.sqrt(d.m2)),
+      fillColor: colorScale(d.name, sliceId),
+    }));
+  } else {
+    colorScale = getSequentialSchemeRegistry()
+      .get(linearColorScheme)
+      .createLinearScale(d3Extent(filteredData, d => d.m1));
+
+    processedData = filteredData.map(d => ({
+      ...d,
+      radius: radiusScale(Math.sqrt(d.m2)),
+      fillColor: colorScale(d.m1),
+    }));
+  }
 
   const mapData = {};
   processedData.forEach(d => {

superset-frontend/plugins/legacy-plugin-chart-world-map/src/controlPanel.ts

@@ -22,6 +22,7 @@ import {
   formatSelectOptions,
   sections,
 } from '@superset-ui/chart-controls';
+import { ColorBy } from './utils';
 
 const config: ControlPanelConfig = {
   controlPanelSections: [
@@ -106,7 +107,25 @@ const config: ControlPanelConfig = {
           },
         ],
         ['color_picker'],
+        [
+          {
+            name: 'color_by',
+            config: {
+              type: 'RadioButtonControl',
+              label: t('Color by'),
+              default: ColorBy.metric,
+              options: [
+                [ColorBy.metric, t('metric')],
+                [ColorBy.country, t('country')],
+              ],
+              description: t(
+                'Whether to define a color by metric or country column',
+              ),
+            },
+          },
+        ],
         ['linear_color_scheme'],
+        ['color_scheme'],
       ],
     },
   ],
@@ -115,10 +134,6 @@ const config: ControlPanelConfig = {
       label: t('Country Column'),
       description: t('3 letter code of the country'),
     },
-    metric: {
-      label: t('Metric for Color'),
-      description: t('Metric that defines the color of the country'),
-    },
     secondary_metric: {
       label: t('Bubble Size'),
       description: t('Metric that defines the size of the bubble'),
@@ -128,6 +143,13 @@ const config: ControlPanelConfig = {
     },
     linear_color_scheme: {
       label: t('Country Color Scheme'),
+      visibility: ({ controls }) =>
+        Boolean(controls?.color_by.value === ColorBy.metric),
+    },
+    color_scheme: {
+      label: t('Country Color Scheme'),
+      visibility: ({ controls }) =>
+        Boolean(controls?.color_by.value === ColorBy.country),
     },
   },
 };

superset-frontend/plugins/legacy-plugin-chart-world-map/src/transformProps.js

@@ -20,8 +20,15 @@ import { rgb } from 'd3-color';
 
 export default function transformProps(chartProps) {
   const { width, height, formData, queriesData } = chartProps;
-  const { maxBubbleSize, showBubbles, linearColorScheme, colorPicker } =
-    formData;
+  const {
+    maxBubbleSize,
+    showBubbles,
+    linearColorScheme,
+    colorPicker,
+    colorBy,
+    colorScheme,
+    sliceId,
+  } = formData;
   const { r, g, b } = colorPicker;
 
   return {
@@ -32,5 +39,8 @@ export default function transformProps(chartProps) {
     showBubbles,
     linearColorScheme,
     color: rgb(r, g, b).hex(),
+    colorBy,
+    colorScheme,
+    sliceId,
   };
 }

superset-frontend/plugins/legacy-plugin-chart-world-map/src/utils.ts

@@ -0,0 +1,23 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+export enum ColorBy {
+  metric = 'metric',
+  country = 'country',
+}