pages/04_helene.py

import os
import leafmap
import solara
import ipywidgets as widgets
import pandas as pd
import geopandas as gpd
import tempfile
from shapely.geometry import Point

event = "HurricaneHelene-Oct24"
url = "https://raw.githubusercontent.com/opengeos/maxar-open-data/master"
repo = "https://github.com/opengeos/maxar-open-data/blob/master/datasets"


def get_datasets():
    datasets = f"{url}/datasets.csv"
    df = pd.read_csv(datasets)
    return df


def get_catalogs(name):
    dataset = f"{url}/datasets/{name}.tsv"
    basename = os.path.basename(dataset)
    tempdir = tempfile.gettempdir()
    tmp_dataset = os.path.join(tempdir, basename)
    if os.path.exists(tmp_dataset):
        dataset_df = pd.read_csv(tmp_dataset, sep="\t")
    else:
        dataset_df = pd.read_csv(dataset, sep="\t")
        dataset_df.to_csv(tmp_dataset, sep="\t", index=False)
    catalog_ids = dataset_df["catalog_id"].unique().tolist()
    catalog_ids.sort()
    return catalog_ids


def get_image_date(catalog_id, m):
    gdf = m.footprint
    image_date = pd.Timestamp(
        gdf[gdf["catalog_id"] == catalog_id]["datetime"].values[0]
    ).strftime("%Y-%m-%d %H:%M:%S")
    return image_date


def add_widgets(m):
    datasets = get_datasets()["dataset"].tolist()
    setattr(m, "zoom_to_layer", True)
    style = {"description_width": "initial"}
    padding = "0px 0px 0px 5px"
    dataset = widgets.Dropdown(
        options=datasets,
        description="Event:",
        value=event,
        style=style,
        layout=widgets.Layout(width="270px", padding=padding),
    )

    catalog_ids = get_catalogs(dataset.value)
    setattr(m, "catalog_ids", catalog_ids)

    date_picker = widgets.DatePicker(
        description="Start date:",
        value=pd.to_datetime("2021-01-01").date(),
        style=style,
        layout=widgets.Layout(width="270px", padding=padding),
    )

    image = widgets.Dropdown(
        value=None,
        options=m.catalog_ids,
        description="Image:",
        style=style,
        layout=widgets.Layout(width="270px", padding=padding),
    )

    checkbox = widgets.Checkbox(
        value=True,
        description="Footprints",
        style=style,
        layout=widgets.Layout(width="90px", padding="0px"),
    )

    split = widgets.Checkbox(
        value=False,
        description="Split map",
        style=style,
        layout=widgets.Layout(width="92px", padding=padding),
    )

    reset = widgets.Checkbox(
        value=False,
        description="Reset",
        style=style,
        layout=widgets.Layout(width="75px", padding="0px"),
    )

    output = widgets.Output()

    def reset_map(change):
        if change.new:
            image.value = None
            image.options = m.catalog_ids
            m.layers = m.layers[:3]
            m.zoom_to_layer = True
            reset.value = False
            date_picker.value = pd.to_datetime("2021-01-01").date()
            m.remove_layer(m.find_layer("Footprint"))

            m.add_gdf(
                m.footprint,
                layer_name="Footprint",
                zoom_to_layer=False,
                info_mode="on_hover",
            )
            satellite_layer = m.find_layer("Google Satellite")
            satellite_layer.visible = False
            output.outputs = ()

    reset.observe(reset_map, names="value")

    def change_dataset(change):
        default_geojson = f"{url}/datasets/{change.new}_union.geojson"
        m.layers = m.layers[:2]

        basename = os.path.basename(default_geojson)
        tempdir = tempfile.gettempdir()
        tmp_geojson = os.path.join(tempdir, basename)
        if os.path.exists(tmp_geojson):
            default_geojson = tmp_geojson
        else:
            leafmap.download_file(default_geojson, tmp_geojson, quiet=True)
        m.add_geojson(default_geojson, layer_name="Footprint", zoom_to_layer=True)
        setattr(m, "gdf", gpd.read_file(default_geojson))

        image.options = get_catalogs(change.new)

    dataset.observe(change_dataset, names="value")

    def change_date(change):
        if change.new:
            start_date = change.new.strftime("%Y-%m-%d")
            sub_gdf = m.gdf[m.gdf["datetime"] >= start_date]
            sub_catalog_ids = sub_gdf["catalog_id"].values.tolist()
            image.options = sub_catalog_ids
            m.remove_layer(m.find_layer("Footprint"))

            m.add_gdf(
                sub_gdf,
                layer_name="Footprint",
                zoom_to_layer=False,
                info_mode="on_hover",
            )
            m.gdf = sub_gdf

    date_picker.observe(change_date, names="value")

    def change_image(change):
        if change.new:
            if change.new not in m.get_layer_names():
                mosaic = f"{url}/datasets/{dataset.value}/{image.value}.json"
                m.add_stac_layer(mosaic, name=image.value, fit_bounds=m.zoom_to_layer)
                image_date = get_image_date(image.value, m)
                output.outputs = ()
                output.append_stdout(f"Image date: {image_date}\n")

    image.observe(change_image, names="value")

    def change_footprint(change):
        geojson_layer = m.find_layer("Footprint")
        if change.new:
            geojson_layer.visible = True
        else:
            geojson_layer.visible = False

    checkbox.observe(change_footprint, names="value")

    def change_split(change):
        if change.new:
            if image.value is not None:
                left_layer = m.find_layer(image.value)
                right_layer = m.find_layer("Google Satellite")
                right_layer.visible = True
                footprint_layer = m.find_layer("Footprint")
                footprint_layer.visible = False
                checkbox.value = False
                m.split_map(
                    left_layer=left_layer,
                    right_layer=right_layer,
                    add_close_button=True,
                    left_label=image.value,
                    right_label="Google Satellite",
                )
                split.value = False
            else:
                left_layer = None

    split.observe(change_split, names="value")

    def handle_click(**kwargs):
        if kwargs.get("type") == "click":
            latlon = kwargs.get("coordinates")
            geometry = Point(latlon[::-1])
            selected = m.gdf[m.gdf.intersects(geometry)]
            setattr(m, "zoom_to_layer", False)
            if len(selected) > 0:
                catalog_ids = selected["catalog_id"].values.tolist()

                if len(catalog_ids) > 1:
                    image.options = catalog_ids
                image.value = catalog_ids[0]
            else:
                image.value = None

    m.on_interaction(handle_click)

    box = widgets.VBox(
        [dataset, date_picker, image, widgets.HBox([checkbox, split, reset]), output]
    )
    m.add_widget(box, position="topright", add_header=False)


zoom = solara.reactive(2)
center = solara.reactive((20, 0))


class Map(leafmap.Map):
    def __init__(self, **kwargs):
        kwargs["toolbar_control"] = False
        super().__init__(**kwargs)
        basemap = {
            "url": "https://mt1.google.com/vt/lyrs=s&x={x}&y={y}&z={z}",
            "attribution": "Google",
            "name": "Google Satellite",
        }
        self.add_tile_layer(**basemap, shown=False)
        self.add_layer_manager(opened=False)
        add_widgets(self)
        default_geojson = f"{url}/datasets/{event}_union.geojson"
        basename = os.path.basename(default_geojson)
        tempdir = tempfile.gettempdir()
        tmp_geojson = os.path.join(tempdir, basename)
        if os.path.exists(tmp_geojson):
            default_geojson = tmp_geojson
        else:
            leafmap.download_file(default_geojson, tmp_geojson, quiet=True)
        self.add_geojson(
            default_geojson,
            layer_name="Footprint",
            zoom_to_layer=True,
            info_mode="on_hover",
        )
        gdf = gpd.read_file(default_geojson)
        setattr(self, "gdf", gdf)
        setattr(self, "footprint", gdf)


@solara.component
def Page():
    with solara.Column(style={"min-width": "500px"}):
        # solara components support reactive variables
        # solara.SliderInt(label="Zoom level", value=zoom, min=1, max=20)
        # using 3rd party widget library require wiring up the events manually
        # using zoom.value and zoom.set
        Map.element(  # type: ignore
            zoom=zoom.value,
            on_zoom=zoom.set,
            center=center.value,
            on_center=center.set,
            scroll_wheel_zoom=True,
            toolbar_ctrl=False,
            data_ctrl=False,
            height="780px",
        )
        solara.Text(f"Center: {center.value}")
        solara.Text(f"Zoom: {zoom.value}")