Improved to_html function gee-community#361

AbrihaDavid · Mar 14, 2021 · 339f8be · 339f8be
1 parent aacc6ca
commit 339f8be
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/docs/notebooks/21_export_map_to_html_png.ipynb b/docs/notebooks/21_export_map_to_html_png.ipynb
@@ -97,7 +97,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "Map = geemap.Map(toolbar_ctrl=True)\n",
+    "Map = geemap.Map(toolbar_ctrl=True, layer_ctrl=True)\n",
     "Map"
    ]
   },
@@ -122,7 +122,7 @@
     "# Add Earth Eninge layers to Map\n",
     "Map.addLayer(dem, vis_params, 'SRTM DEM', True, 0.5)\n",
     "Map.addLayer(landcover, {}, 'Land cover')\n",
-    "Map.addLayer(landsat7, {'bands': ['B4', 'B3', 'B2'], 'min': 20, 'max': 200}, 'Landsat 7')\n",
+    "Map.addLayer(landsat7, {'bands': ['B4', 'B3', 'B2'], 'min': 20, 'max': 200, 'gamma':1.5}, 'Landsat 7')\n",
     "Map.addLayer(states, {}, \"US States\")"
    ]
   },

diff --git a/examples/notebooks/21_export_map_to_html_png.ipynb b/examples/notebooks/21_export_map_to_html_png.ipynb
@@ -97,7 +97,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "Map = geemap.Map(toolbar_ctrl=True)\n",
+    "Map = geemap.Map(toolbar_ctrl=True, layer_ctrl=True)\n",
     "Map"
    ]
   },
@@ -122,7 +122,7 @@
     "# Add Earth Eninge layers to Map\n",
     "Map.addLayer(dem, vis_params, 'SRTM DEM', True, 0.5)\n",
     "Map.addLayer(landcover, {}, 'Land cover')\n",
-    "Map.addLayer(landsat7, {'bands': ['B4', 'B3', 'B2'], 'min': 20, 'max': 200}, 'Landsat 7')\n",
+    "Map.addLayer(landsat7, {'bands': ['B4', 'B3', 'B2'], 'min': 20, 'max': 200, 'gamma':1.5}, 'Landsat 7')\n",
     "Map.addLayer(states, {}, \"US States\")"
    ]
   },

diff --git a/geemap/geemap.py b/geemap/geemap.py
@@ -2960,14 +2960,23 @@ def add_landsat_ts_gif(
         except Exception as e:
             raise Exception(e)
 
-    def to_html(self, outfile, title="My Map", width="100%", height="880px"):
+    def to_html(
+        self,
+        outfile,
+        title="My Map",
+        width="100%",
+        height="880px",
+        add_layer_control=True,
+    ):
         """Saves the map as a HTML file.
 
         Args:
             outfile (str): The output file path to the HTML file.
             title (str, optional): The title of the HTML file. Defaults to 'My Map'.
             width (str, optional): The width of the map in pixels or percentage. Defaults to '100%'.
             height (str, optional): The height of the map in pixels. Defaults to '880px'.
+            add_layer_control (bool, optional): Whether to add the LayersControl. Defaults to True.
+
         """
         try:
 
@@ -2979,6 +2988,11 @@ def to_html(self, outfile, title="My Map", width="100%", height="880px"):
             if not os.path.exists(out_dir):
                 os.makedirs(out_dir)
 
+            if add_layer_control and self.layer_control is None:
+                layer_control = ipyleaflet.LayersControl(position="topright")
+                self.layer_control = layer_control
+                self.add_control(layer_control)
+
             before_width = self.layout.width
             before_height = self.layout.height