From 7daf1e10d5d6309d9b13340067f697f499488d3c Mon Sep 17 00:00:00 2001
From: atmorling <atmorling@gmail.com>
Date: Wed, 18 Dec 2024 00:02:05 +0200
Subject: [PATCH] legend ignores transparent colors (#360)

---
 ecoscope/mapping/map.py | 34 ++++++++++++++++------------------
 tests/test_ecomap.py    |  7 +++----
 2 files changed, 19 insertions(+), 22 deletions(-)

diff --git a/ecoscope/mapping/map.py b/ecoscope/mapping/map.py
index 020150fb..12bd21cf 100644
--- a/ecoscope/mapping/map.py
+++ b/ecoscope/mapping/map.py
@@ -7,7 +7,7 @@
 import numpy as np
 import pandas as pd
 import rasterio as rio
-from ecoscope.base.utils import color_tuple_to_css
+from ecoscope.base.utils import color_tuple_to_css, hex_to_rgba
 from io import BytesIO
 from typing import Dict, IO, List, Optional, TextIO, Union
 from pathlib import Path
@@ -254,29 +254,27 @@ def add_legend(self, labels: list | pd.Series, colors: list | pd.Series, **kwarg
         style: dict
             Additional style params
         """
-        nans = None
         if isinstance(labels, pd.Series):
-            if pd.api.types.is_numeric_dtype(labels):
-                nans = np.argwhere(np.isnan(labels))
-            labels = labels.unique().tolist()
+            labels = labels.tolist()
         if isinstance(colors, pd.Series):
-            colors = colors.unique().tolist()
+            colors = colors.tolist()
 
+        labels = list(dict.fromkeys(labels))
+        colors = list(dict.fromkeys(colors))
         if len(labels) != len(colors):
             raise ValueError("Unique label and color values must be of equal number")
 
-        if nans is not None:
-            filtered_labels = []
-            filtered_colors = []
-            for index, value in enumerate(labels):
-                if index not in nans:
-                    filtered_labels.append(labels[index])
-                    filtered_colors.append(colors[index])
-            labels = filtered_labels
-            colors = filtered_colors
-
-        widget_labels = [str(label) for label in labels]
-        widget_colors = [color_tuple_to_css(color) if isinstance(color, tuple) else color for color in colors]
+        filtered_labels = []
+        filtered_colors = []
+        for index, value in enumerate(colors):
+            if isinstance(value, str):
+                value = hex_to_rgba(value)
+            if len(value) == 4 and value[3] > 0:
+                filtered_labels.append(labels[index])
+                filtered_colors.append(colors[index])
+
+        widget_labels = [str(label) for label in filtered_labels]
+        widget_colors = [color_tuple_to_css(color) if isinstance(color, tuple) else color for color in filtered_colors]
 
         self.add_widget(LegendWidget(labels=widget_labels, colors=widget_colors, **kwargs))
 
diff --git a/tests/test_ecomap.py b/tests/test_ecomap.py
index e6f4d553..6b8886ec 100644
--- a/tests/test_ecomap.py
+++ b/tests/test_ecomap.py
@@ -2,7 +2,6 @@
 import random
 import ee
 import geopandas as gpd
-import numpy as np
 import pandas as pd
 import pytest
 import ecoscope
@@ -89,16 +88,16 @@ def test_add_legend_series():
     assert legend.colors == ["rgba(0, 0, 0, 1.0)", "rgba(255, 255, 255, 1.0)"]
 
 
-def test_add_legend_series_with_nan():
+def test_add_legend_series_with_transparent():
     m = EcoMap(default_widgets=False)
     m.add_legend(
-        labels=pd.Series([0, 1, np.nan, 5, np.nan]),
+        labels=pd.Series([1, 3, 0, 5, 0]),
         colors=pd.Series([(0, 0, 0, 255), (255, 255, 255, 255), (0, 0, 0, 0), (100, 100, 100, 255), (0, 0, 0, 0)]),
     )
     assert len(m.deck_widgets) == 1
     legend = m.deck_widgets[0]
     assert isinstance(legend, LegendWidget)
-    assert legend.labels == ["0.0", "1.0", "5.0"]
+    assert legend.labels == ["1", "3", "5"]
     assert legend.colors == ["rgba(0, 0, 0, 1.0)", "rgba(255, 255, 255, 1.0)", "rgba(100, 100, 100, 1.0)"]