Added country analysis

pages/01_gswis.py

@@ -42,6 +42,8 @@ def add_analysis_gui(m=None, position='topright', opened=True):
     """
 
     fc = ee.FeatureCollection('users/giswqs/public/countries')
+    countries = fc.aggregate_array('NAME').getInfo()
+    countries.sort()
     gswe = ee.ImageCollection("users/h2i_lab/gswe/gswe_datasets")
     image = gswe.mosaic()
     # esa = gswe.select("esa").mosaic()
@@ -76,17 +78,22 @@ def add_analysis_gui(m=None, position='topright', opened=True):
         style={"description_width": "initial"},
     )
 
+    country = widgets.Dropdown(
+        options=countries,
+        value=None,
+        layout=widgets.Layout(width=widget_width, padding=padding),
+    )
+
     buttons = widgets.ToggleButtons(
         value=None,
         options=["Apply", "Reset", "Close"],
         tooltips=["Apply", "Reset", "Close"],
         button_style="primary",
+        layout=widgets.Layout(padding="0px 2px 4px 2px"),
     )
     buttons.style.button_width = "88px"
     label = widgets.Label("Draw an area on the map first.")
 
-    output = widgets.Output(layout=widgets.Layout(width=widget_width, padding=padding))
-
     toolbar_widget = widgets.VBox()
     toolbar_widget.children = [toolbar_button]
     toolbar_header = widgets.HBox()
@@ -95,9 +102,30 @@ def add_analysis_gui(m=None, position='topright', opened=True):
     toolbar_footer.children = [
         radio,
         buttons,
-        output,
     ]
 
+    def change_radio(change):
+        if change["new"] == "Select a country":
+            toolbar_footer.children = [radio, country, buttons]
+        else:
+            toolbar_footer.children = [radio, buttons]
+
+    radio.observe(change_radio, "value")
+
+    m.selected_country = None
+
+    def change_country(change):
+        if change["new"]:
+            country_name = country.value
+            country_fc = fc.filter(ee.Filter.eq('NAME', country_name))
+            vec_style = {'color': '000000ff', 'width': 3, 'fillColor': '00000000'}
+            m.addLayer(country_fc.style(**vec_style), {}, 'Selected Country')
+            m.centerObject(country_fc)
+            m.selected_country = country_fc
+            toolbar_footer.children = [radio, country, buttons]
+
+    country.observe(change_country, "value")
+
     def toolbar_btn_click(change):
         if change["new"]:
             close_button.value = False
@@ -121,10 +149,16 @@ def add_analysis_gui(m=None, position='topright', opened=True):
 
     def button_clicked(change):
         if change["new"] == "Apply":
+            output = widgets.Output(
+                layout=widgets.Layout(width=widget_width, padding=padding)
+            )
+            if radio.value == "Select a country":
+                toolbar_footer.children = [radio, country, buttons, output]
+            else:
+                toolbar_footer.children = [radio, buttons, output]
             with output:
                 output.clear_output()
-                # com_label = widgets.Label("PLEASE WAIT ...")
-                # display(com_label)
+
                 buttons.value = None
 
                 if radio.value == "Draw an area":
@@ -134,9 +168,18 @@ def add_analysis_gui(m=None, position='topright', opened=True):
                     else:
                         chart = zonal_stats_chart(image, m.user_roi, scale=100)
                         display(chart)
+                elif radio.value == "Select a country":
+                    if m.selected_country is not None:
+                        chart = zonal_stats_chart(
+                            image, m.selected_country.geometry(), scale=100
+                        )
+                        display(chart)
 
         elif change["new"] == "Reset":
-            output.clear_output()
+            country.value = None
+            radio.value = "Draw an area"
+            toolbar_footer.children = [radio, buttons]
+
         elif change["new"] == "Close":
             if m is not None:
                 if m.tool_control is not None and m.tool_control in m.controls: