Adapt Heatmap layer for an easier use with a ColormapControl.

Author: HaudinFlorence

ipyleaflet/leaflet.py

@@ -5,11 +5,10 @@
 import copy
 import asyncio
 import json
-from folium import LinearColormap
 import xyzservices
 from datetime import date, timedelta
 from math import isnan
-from branca.colormap import linear, ColorMap
+from branca.colormap import linear, LinearColormap, ColorMap
 from IPython.display import display
 import warnings
 
@@ -800,8 +799,16 @@ class Heatmap(RasterLayer):
         Radius of the data points.
     blur: float, default 15.
         Blurring intensity.
-    gradient: dict, default {0.4: 'blue', 0.6: 'cyan', 0.7: 'lime', 0.8: 'yellow', 1.0: 'red'}
+    values : list, default [0.4, 0.6, 0.7, 0.8, 1.0]
+        list of values to define the gradient
+    colors : list, default ['blue', 'cyan', 'lime', 'yellow', 'red']
+        list of colors to define the gradient
+    dict: dict, default built from values (keys) and colors lists (values)
+        dictionnary mapping value to colors.
+    gradient : Dict, default built from dict
         Colors used for the color-mapping from low to high heatmap intensity.
+    colormap: branca.colormap.LinearColorMap instance
+        The colormap used for displaying the HeatMap data, defined with the same min and max values and colors than the gradient.
     """
 
     _view_name = Unicode('LeafletHeatmapView').tag(sync=True)
@@ -823,8 +830,7 @@ class Heatmap(RasterLayer):
     vmin = values[0]
     vmax = values[len(values)-1]
     gradient = Dict(dict).tag(sync=True, o=True)
-    colormap = LinearColormap([colors], vmin=vmin, vmax=vmax)
-
+    colormap = LinearColormap(colors, vmin=vmin, vmax=vmax)
 
 
 class VectorTileLayer(Layer):