here we go

app.py

@@ -14,19 +14,133 @@
 # +
 import leafmap.foliumap as leafmap
 import streamlit as st
+import altair as alt
+import ibis
+from ibis import _
+import ibis.selectors as s
+
 st.set_page_config(layout="wide", page_title="Protected Areas Explorer", page_icon=":globe:")
 
-low = st.slider("Red when richness is less than", 0, 30, 2)
-high = st.slider("Green for richness larger than", 0, 30, 3)
-"missing data in grey. in-between values in blue"
+'''
+# US Protected Area Database Explorer
+
+The map below shows all continental US protected areas overlaid on [NatureServe's Map of Biodiversity Importance](https://www.natureserve.org/map-biodiversity-importance)
+which shows species richness for some 2,216 imperiled species in (990 m grid cells).
+Use the later tool in the top right to toggle layers on and off for visibility.
+Zoom into the map for more detailed rendering -- e.g. try exploring the many tiny public mixed used lands inside Los Angeles.
+Mouse over any polygon to see details about name, ownership, mean species richness, and mean rarity-weighted-richness.
+
+By default, areas are divided into the following categories: 
+- Publicly managed lands in Gap 1 & 2 (e.g. National Parks) ("public use", blue)
+- Publicly managed lands  in Gap 3 & 4 (most national forests, BLM lands, etc)
+- Privately managed lands
+- Tribally managed lands
+
+Alternative color rules can be applied, for instance, to distinguish between easements (rented) and fee (owned) protected areas. Below the interactive map, 
+charts summarize the total protected area and mean richness and mean RSR within each group.  
+
+'''
+# -
 
-red = "#850101"
-purple = "#BF40BF"
-green = "#005a00"
-blue = "#3388ff"
+private_color = "#850101" # red
+tribal_color = "#BF40BF" # purple
+mixed_color = "#005a00" # green
+public_color = "#3388ff" # blue
 
 
 # +
+with st.sidebar:
+
+    "## Map controls"
+    
+    style_choice = st.radio(
+    "Color areas by",
+    ["management type",
+     "Fee/Easement",
+     "species richness (thresholds)", 
+     "species richness (continuous)",
+     "custom"]
+    )
+
+    "### Management Types Color selector"
+
+    public_color = st.color_picker('Color for Public Protected (Gap 1 & 2) lands', public_color)
+    mixed_color = st.color_picker('Color for Public Mixed Use lands', mixed_color)
+    tribal_color = st.color_picker('Color for Tribal managed lands', tribal_color)
+    private_color = st.color_picker('Color for privately managed land', private_color)
+    
+    "### Threshold selector"
+    
+    low = st.slider("Red when richness is less than", 0, 30, 2)
+    high = st.slider("Green for richness larger than", 0, 30, 3)
+    "missing data in grey. in-between values in blue"
+
+
+    "### Custom styles"
+    custom = st.text_area("Define a custom mapbox fill-color rule",
+                          value = '''{
+    "fill-color": 
+    ['case',
+    ['==', ['get', 'bucket'], 'public'],
+    public_color,
+    ['==', ['get', 'bucket'], 'private'],
+    private_color,
+    ['==', ['get', 'bucket'], 'tribal'],
+    tribal_color,
+    mixed_color # default (bucket == mixed)
+    ],
+     "fill-opacity": 0.5
+    }
+''',
+                          height = 300
+)
+
+
+
+# +
+
+bucket = {
+    "fill-color": 
+                      ['case',
+                      ['==', ['get', 'bucket'], 'public'],
+                       public_color,
+                      ['==', ['get', 'bucket'], 'private'],
+                       private_color,
+                      ['==', ['get', 'bucket'], 'tribal'],
+                       tribal_color,
+                       mixed_color # default (bucket == mixed)
+                       ],
+     "fill-opacity": 0.5
+    }
+
+
+easement = {
+    "fill-color": 
+                      ['case',
+                      ['==', ['get', ''], 'Fee'],
+                       public_color,
+                      ['==', ['get', 'FeatClass'], 'Easement'],
+                       private_color,
+                      ['==', ['get', 'FeatClass'], 'Proclamation'],
+                       tribal_color,
+                       mixed_color # default (bucket == mixed)
+                       ],
+     "fill-opacity": 0.5
+    }
+
+
+thresholds = {
+    "fill-color": 
+                      ['case',
+                      ['<', ['get', 'richness'], low],
+                       private_color,
+                      ['>=', ['get', 'richness'], high],
+                       mixed_color,
+                       public_color # default
+                       ],
+     "fill-opacity": 0.5
+    }
+
 
 richness = {
     "fill-color": [
@@ -37,12 +151,41 @@ richness = {
                   # ["min", ["number", ["get", "richness"]]],
                   "rgb(252, 226, 220)",
                   #["max", ["get", "richness"]],
-                  8,
+                  6,
                   "rgb(102, 16, 66)"
                 ],
     "fill-opacity": 0.9
 }
 
+rsr = {
+    "fill-color": [
+                  "interpolate",
+                  ["linear"],
+                  ["get", "rsr"],
+                  0,
+                  # ["min", ["number", ["get", "richness"]]],
+                  "rgb(252, 226, 220)",
+                  #["max", ["get", "richness"]],
+                  6,
+                  "rgb(102, 16, 66)"
+                ],
+    "fill-opacity": 0.9
+}
+
+
+# +
+if style_choice == "management type":
+    paint_fill = bucket
+if style_choice == "Fee/Easement":
+    paint_fill = easement
+if style_choice == "species richness (thresholds)":
+    paint_fill = thresholds
+if style_choice == "species richness (continuous)":
+    paint_fill = richness
+if style_choice == "custom":
+    paint_fill = custom
+
+
 r = {
     "fill-color": [
         "rgb",
@@ -59,31 +202,7 @@ r = {
 
 # +
 
-thresholds = {
-    "fill-color": 
-                      ['case',
-                      ['<', ['get', 'richness'], low],
-                       red,
-                      ['>=', ['get', 'richness'], high],
-                       green,
-                       blue # default
-                       ],
-     "fill-opacity": 0.5
-    }
 
-bucket = {
-    "fill-color": 
-                      ['case',
-                      ['==', ['get', 'bucket'], 'public'],
-                       blue,
-                      ['==', ['get', 'bucket'], 'private'],
-                       red,
-                      ['==', ['get', 'bucket'], 'tribal'],
-                       purple,
-                       green # default (bucket == mixed)
-                       ],
-     "fill-opacity": 0.5
-    }
 
 # -
 
@@ -106,22 +225,116 @@ style = {
             "source": "pad",
             "source-layer": "pad-mobi",
             "type": "fill",
-            "paint": r
+            "paint": paint_fill
         }]}
-m = leafmap.Map(center=[35, -100], zoom=3)
+m = leafmap.Map(center=[35, -100], 
+                zoom=4, 
+                layers_control=True,
+                draw_control=False,
+                measure_control=False,
+                fullscreen_control=True)
 #m.add_basemap("CartoDB.DarkMatter")
-m.add_pmtiles(url, name="Public", style=style, overlay=True, show=True, zoom_to_layer=False)
-#m.to_streamlit(height=900)
-m
+
+m.add_basemap("Esri.WorldShadedRelief")
+
+m.add_tile_layer(
+    url="https://minio.carlboettiger.info/shared-data/mobi-tiles/red/{z}/{x}/{y}.png",
+    name="MOBI",
+    attribution="NatureServe",
+    opacity=0.9
+)
+
+
+m.add_pmtiles(url, name="Protected Areas (PAD-US-3)", style=style, overlay=True, show=True, zoom_to_layer=False)
+
+
+m.to_streamlit(height=900)
+
+# -
+
+
 
 # +
-import ibis
-from ibis import _
+#parquet = "https://data.source.coop/cboettig/pad-us-3/pad-mobi.parquet"
+parquet = "https://minio.carlboettiger.info/public-biodiversity/pad-us-3/pad-mobi.parquet"
+
+t = ibis.read_parquet(parquet)
+
+
+# +
+us_lower_48_area_m2 = 7.8e+12
+
+@st.cache_data()
+def summary_table():
+    x = ibis.memtable({"bucket": ["public", "tribal", "mixed", "private"], 
+                       "color": [public_color, tribal_color, mixed_color, private_color]})
+    df = (t.
+        group_by(_.bucket).
+        aggregate(percent_protected =  100 * _.area.sum() / us_lower_48_area_m2,
+                mean_richness = (_.richness * _.area).sum() / _.area.sum(),
+                mean_rsr = (_.rsr * _.area).sum() / _.area.sum()
+               ).
+        mutate(percent_protected = _.percent_protected.round())
+        ).inner_join(x, "bucket")
+    return df.to_pandas()
+
+df = summary_table()
+# st.table(richness_table)
+
+
+# +
+#summary_table.to_pandas()
+
+# +
+
+base = alt.Chart(df).encode(
+    alt.Theta("percent_protected:Q").stack(True),
+    alt.Color("color:N").scale(None).legend(None)
+)
+
 
-t = ibis.read_parquet("https://data.source.coop/cboettig/pad-us-3/pad-mobi.parquet")
+area_chart = (
+  base.mark_arc(innerRadius=50, outerRadius=120) +
+  base.mark_text(radius=170, size=20).encode(text="bucket") +
+  base.mark_text(radius=135, size=20).encode(text="percent_protected:N")
+)
 
+# area_chart
+
+# +
+
+richness_chart = alt.Chart(df).mark_bar().encode(
+    x='bucket',
+    y='mean_richness',
+    color=alt.Color('color').scale(None)
+)
+#richness_chart
+
+
+# +
+
+rsr_chart = alt.Chart(df).mark_bar().encode(
+    x='bucket',
+    y='mean_rsr',
+    color=alt.Color('color').scale(None)
+)
 # -
 
-t.group_by(_.bucket).aggregate(mean = (_.richness * _.area).mean() / _.area.sum()).to_pandas()
+"## Summary Statistics"
+
+# +
+col1, col2, col3 = st.columns(3)
+
+with col1: 
+    st.altair_chart(area_chart, use_container_width=True)
+    "Percent of Continental US Area in each group"
+# -
+
+with col2: 
+    st.altair_chart(richness_chart, use_container_width=True)
+
+
+with col3: 
+    st.altair_chart(rsr_chart, use_container_width=True)
+
 
-t.group_by(_.bucket).aggregate(mean = (_.rsr * _.area).mean() / _.area.sum()).to_pandas()