{
    "cells": [
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "<a href=\"https://githubtocolab.com/gee-community/geemap/blob/master/examples/notebooks/49_colorbar.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open in Colab\"/></a>\n",
                "\n",
                "Uncomment the following line to install [geemap](https://geemap.org) if needed."
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "# !pip install geemap"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "# How to add a colorbar to the map\n",
                "\n",
                "## For ipyleaflet maps\n",
                "\n",
                "### Continuous colorbar"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "import ee\n",
                "import geemap"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "# geemap.update_package()"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "Map = geemap.Map()\n",
                "\n",
                "dem = ee.Image('USGS/SRTMGL1_003')\n",
                "vis_params = {\n",
                "    'min': 0,\n",
                "    'max': 4000,\n",
                "    'palette': ['006633', 'E5FFCC', '662A00', 'D8D8D8', 'F5F5F5'],\n",
                "}\n",
                "\n",
                "Map.addLayer(dem, vis_params, 'SRTM DEM')\n",
                "\n",
                "colors = vis_params['palette']\n",
                "vmin = vis_params['min']\n",
                "vmax = vis_params['max']\n",
                "\n",
                "Map.add_colorbar_branca(colors=colors, vmin=vmin, vmax=vmax, layer_name=\"SRTM DEM\")\n",
                "\n",
                "# nlcd_2016 = ee.Image('USGS/NLCD/NLCD2016').select('landcover')\n",
                "# Map.addLayer(nlcd_2016, {}, \"NLCD\")\n",
                "# Map.add_legend(legend_title=\"NLCD\", builtin_legend=\"NLCD\", layer_name=\"NLCD\")\n",
                "\n",
                "Map"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Categorical colorbar"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "Map = geemap.Map()\n",
                "\n",
                "dem = ee.Image('USGS/SRTMGL1_003')\n",
                "vis_params = {\n",
                "    'min': 0,\n",
                "    'max': 4000,\n",
                "    'palette': ['006633', 'E5FFCC', '662A00', 'D8D8D8', 'F5F5F5'],\n",
                "}\n",
                "\n",
                "Map.addLayer(dem, vis_params, 'SRTM DEM')\n",
                "\n",
                "colors = vis_params['palette']\n",
                "vmin = vis_params['min']\n",
                "vmax = vis_params['max']\n",
                "\n",
                "Map.add_colorbar_branca(\n",
                "    colors=colors, vmin=vmin, vmax=vmax, categorical=True, step=4, layer_name=\"SRTM DEM\"\n",
                ")\n",
                "\n",
                "Map"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## For folium maps\n",
                "\n",
                "### Continuous colorbar"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "import ee\n",
                "import geemap.foliumap as geemap"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "Map = geemap.Map()\n",
                "\n",
                "dem = ee.Image('USGS/SRTMGL1_003')\n",
                "vis_params = {\n",
                "    'min': 0,\n",
                "    'max': 4000,\n",
                "    'palette': ['006633', 'E5FFCC', '662A00', 'D8D8D8', 'F5F5F5'],\n",
                "}\n",
                "\n",
                "Map.addLayer(dem, vis_params, 'SRTM DEM')\n",
                "\n",
                "colors = vis_params['palette']\n",
                "vmin = vis_params['min']\n",
                "vmax = vis_params['max']\n",
                "\n",
                "Map.add_colorbar(vis_params, label='Elevation (m)')\n",
                "\n",
                "Map.addLayerControl()\n",
                "Map"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Categorical colorbar"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "Map = geemap.Map()\n",
                "\n",
                "dem = ee.Image('USGS/SRTMGL1_003')\n",
                "vis_params = {\n",
                "    'min': 0,\n",
                "    'max': 4000,\n",
                "    'palette': ['006633', 'E5FFCC', '662A00', 'D8D8D8', 'F5F5F5'],\n",
                "}\n",
                "\n",
                "Map.addLayer(dem, vis_params, 'SRTM DEM')\n",
                "\n",
                "colors = vis_params['palette']\n",
                "vmin = vis_params['min']\n",
                "vmax = vis_params['max']\n",
                "\n",
                "Map.add_colorbar(vis_params, categorical=True, step=4)\n",
                "\n",
                "Map.addLayerControl()\n",
                "Map"
            ]
        },
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "### Draggable legend"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": null,
            "metadata": {},
            "outputs": [],
            "source": [
                "Map = geemap.Map()\n",
                "\n",
                "legend_dict = {\n",
                "    '11 Open Water': '466b9f',\n",
                "    '12 Perennial Ice/Snow': 'd1def8',\n",
                "    '21 Developed, Open Space': 'dec5c5',\n",
                "    '22 Developed, Low Intensity': 'd99282',\n",
                "    '23 Developed, Medium Intensity': 'eb0000',\n",
                "    '24 Developed High Intensity': 'ab0000',\n",
                "    '31 Barren Land (Rock/Sand/Clay)': 'b3ac9f',\n",
                "    '41 Deciduous Forest': '68ab5f',\n",
                "    '42 Evergreen Forest': '1c5f2c',\n",
                "    '43 Mixed Forest': 'b5c58f',\n",
                "    '51 Dwarf Scrub': 'af963c',\n",
                "    '52 Shrub/Scrub': 'ccb879',\n",
                "    '71 Grassland/Herbaceous': 'dfdfc2',\n",
                "    '72 Sedge/Herbaceous': 'd1d182',\n",
                "    '73 Lichens': 'a3cc51',\n",
                "    '74 Moss': '82ba9e',\n",
                "    '81 Pasture/Hay': 'dcd939',\n",
                "    '82 Cultivated Crops': 'ab6c28',\n",
                "    '90 Woody Wetlands': 'b8d9eb',\n",
                "    '95 Emergent Herbaceous Wetlands': '6c9fb8',\n",
                "}\n",
                "\n",
                "landcover = ee.Image('USGS/NLCD/NLCD2016').select('landcover')\n",
                "Map.addLayer(landcover, {}, 'NLCD Land Cover')\n",
                "\n",
                "Map.add_legend(title=\"NLCD Land Cover Classification\", legend_dict=legend_dict)\n",
                "\n",
                "Map.addLayerControl()\n",
                "\n",
                "Map"
            ]
        }
    ],
    "metadata": {
        "kernelspec": {
            "display_name": "Python 3",
            "language": "python",
            "name": "python3"
        }
    },
    "nbformat": 4,
    "nbformat_minor": 5
}