From 4a1784d8e122a1638496eac8af96707a4f328c8e Mon Sep 17 00:00:00 2001
From: Bartosz Prusinowski <bartosz@interactivethings.com>
Date: Fri, 2 Feb 2024 13:58:46 +0100
Subject: [PATCH] fix: Retrieval of geo coordinates for shared dimensions

---
 app/charts/map/map-state-props.ts | 29 ++++++++++++++++-------------
 app/charts/shared/chart-state.ts  |  2 +-
 2 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/app/charts/map/map-state-props.ts b/app/charts/map/map-state-props.ts
index adc5ee7d4..aab00a700 100644
--- a/app/charts/map/map-state-props.ts
+++ b/app/charts/map/map-state-props.ts
@@ -54,11 +54,6 @@ export const useMapStateVariables = (
     );
   }
 
-  const { getValue: getArea } = useDimensionWithAbbreviations(
-    areaLayerDimension,
-    { observations, field: areaLayer }
-  );
-
   const symbolLayerDimension = dimensions.find(
     (d) => d.iri === symbolLayer?.componentIri
   );
@@ -70,17 +65,18 @@ export const useMapStateVariables = (
     );
   }
 
-  const { getValue: getSymbol } = useDimensionWithAbbreviations(
-    symbolLayerDimension,
-    { observations, field: symbolLayer }
-  );
+  const { getValue: getSymbol, getLabel: getSymbolLabel } =
+    useDimensionWithAbbreviations(symbolLayerDimension, {
+      observations,
+      field: symbolLayer,
+    });
 
   return {
     ...baseVariables,
     areaLayerDimension,
-    getArea,
     symbolLayerDimension,
     getSymbol,
+    getSymbolLabel,
   };
 };
 
@@ -96,7 +92,12 @@ export const useMapStateData = (
   variables: MapStateVariables
 ): MapStateData => {
   const { chartConfig, observations, shapes, coordinates } = chartProps;
-  const { areaLayerDimension, symbolLayerDimension, getSymbol } = variables;
+  const {
+    areaLayerDimension,
+    symbolLayerDimension,
+    getSymbol,
+    getSymbolLabel,
+  } = variables;
   const filters = useChartConfigFilters(chartConfig);
   // No need to sort the data for map.
   const plottableData = usePlottableData(observations, {});
@@ -134,8 +135,9 @@ export const useMapStateData = (
       const coordsByLabel = keyBy(coordinates, (d) => d.label);
 
       data.chartData.forEach((d) => {
-        const label = getSymbol(d);
-        const coords = coordsByLabel[label];
+        const value = getSymbol(d);
+        const label = getSymbolLabel(value);
+        const coords = coordsByLabel[value] ?? coordsByLabel[label];
 
         if (coords) {
           const { iri, label, latitude, longitude } = coords;
@@ -182,6 +184,7 @@ export const useMapStateData = (
   }, [
     symbolLayerDimension,
     getSymbol,
+    getSymbolLabel,
     data.chartData,
     shapes,
     coordinates,
diff --git a/app/charts/shared/chart-state.ts b/app/charts/shared/chart-state.ts
index a058c3087..c6cab2f51 100644
--- a/app/charts/shared/chart-state.ts
+++ b/app/charts/shared/chart-state.ts
@@ -330,7 +330,6 @@ export const useSegmentVariables = (
 
 export type AreaLayerVariables = {
   areaLayerDimension: GeoShapesDimension | undefined;
-  getArea: StringValueGetter;
 };
 
 export type SymbolLayerVariables = {
@@ -339,6 +338,7 @@ export type SymbolLayerVariables = {
     | GeoCoordinatesDimension
     | undefined;
   getSymbol: StringValueGetter;
+  getSymbolLabel: (d: string) => string;
 };
 
 export type ChartStateData = {