Feature/graph analysis ipyleaflet visualisation

notebooks/exploratory/rdnfn-13-geographviewer-test.ipynb

@@ -0,0 +1,89 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "polyphonic-gateway",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%load_ext autoreload\n",
+    "%autoreload 2\n",
+    "%config IPCompleter.greedy=True"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "educational-texture",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from src.models import geograph\n",
+    "from src.data_loading import test_data\n",
+    "from src.constants import PREFERRED_CRS\n",
+    "\n",
+    "test_gdf = test_data.get_polygon_gdf(\"chernobyl_10_squares_touching\")\n",
+    "test_gdf['class_label'] = list(range(10))\n",
+    "graph = geograph.GeoGraph(test_gdf, crs=PREFERRED_CRS)\n",
+    "graph.add_habitat(name=\"eagle_habitat\", valid_classes=[0], max_travel_distance=10000)\n",
+    "\n",
+    "from src.visualisation import geoviewer, control_widgets"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "distinct-kenya",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import ipyleaflet\n",
+    "\n",
+    "viewer = geoviewer.GeoGraphViewer()\n",
+    "viewer.add_graph(graph, name='main_graph')\n",
+    "viewer.add_layer(ipyleaflet.basemaps.Esri.WorldImagery)\n",
+    "viewer.enable_graph_controls()\n",
+    "#viewer.log_handler.show_logs()\n",
+    "viewer"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "unsigned-butterfly",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "touched-beaver",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.8.8"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

notebooks/exploratory/rdnfn-6-geographviewer-ipyleaflet-prototype.ipynb

@@ -0,0 +1,292 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "mental-garden",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%load_ext autoreload\n",
+    "%autoreload 2\n",
+    "%config IPCompleter.greedy=True"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "turned-dispatch",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from src.models import geograph, visualisation\n",
+    "from src.data_loading import test_data\n",
+    "\n",
+    "test_gdf = test_data.get_polygon_gdf(\"chernobyl_squares_touching\")\n",
+    "test_gdf['class_label'] = 0\n",
+    "test_gdf = test_gdf[['geometry', 'class_label']]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "diagnostic-motor",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "test_gdf2 = test_data.get_polygon_gdf(\"chernobyl_squares_apart\")\n",
+    "test_gdf2['class_label'] = 0\n",
+    "test_gdf2 = test_gdf2[['geometry', 'class_label']]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "thirty-pulse",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "graph = geograph.GeoGraph(test_gdf)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "identified-right",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "graph2 = geograph.GeoGraph(test_gdf2)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "hungarian-clear",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#widgets.link((m, 'layers'), (viewer, 'layers'))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fourth-injury",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from traitlets import HasTraits, Integer, observe, Dict, Tuple\n",
+    "import ipywidgets as widgets\n",
+    "from ipywidgets import HTML\n",
+    "import ipyleaflet\n",
+    "from ipyleaflet import TileLayer, GeoData, basemaps\n",
+    "import copy\n",
+    "import markdown \n",
+    "import pandas as pd\n",
+    "\n",
+    "from src.constants import CHERNOBYL_COORDS_WGS84, WGS84\n",
+    "\n",
+    "class GeoGraphViewer(ipyleaflet.Map):\n",
+    "    #layers = Tuple()\n",
+    "    log_tab = widgets.Output(layout={'border': '1px solid black'})\n",
+    "    \n",
+    "    def __init__(self):\n",
+    "        super().__init__(center=CHERNOBYL_COORDS_WGS84, zoom = 7, scroll_wheel_zoom=True)\n",
+    "        self.map = ipyleaflet.Map(center=CHERNOBYL_COORDS_WGS84, zoom = 7, scroll_wheel_zoom=True)\n",
+    "        self.layer_dict = dict(Basemap=dict(layer=TileLayer(base=True, max_zoom=19, min_zoom=4),active=True, layer_type='map'))\n",
+    "        #self.create_traitlets_links()\n",
+    "        self.custom_style = dict(style={'color': 'black', 'fillColor': '#3366cc'},\n",
+    "                          hover_style={'fillColor': 'red' , 'fillOpacity': 0.2},\n",
+    "                          point_style={'radius': 10, 'color': 'red', 'fillOpacity': 0.8, 'fillColor': 'blue', 'weight': 3})\n",
+    "    \n",
+    "    @log_tab.capture()     \n",
+    "    def add_graph(self,graph, name = 'Graph'):\n",
+    "        nodes, edges = visualisation.create_node_edge_geometries(graph.graph)\n",
+    "        graph_geometries = pd.concat([nodes,edges]).reset_index()\n",
+    "        graph_geo_data = ipyleaflet.GeoData(geo_dataframe = graph_geometries.to_crs(WGS84),\n",
+    "                          style ={'color': 'black', 'fillColor': '#3366cc'},\n",
+    "                          hover_style={'fillColor': 'red' , 'fillOpacity': 0.2},\n",
+    "                          point_style={'radius': 10, 'color': 'red', 'fillOpacity': 0.8, 'fillColor': 'blue', 'weight': 3},\n",
+    "                          name = name)\n",
+    "        self.layer_dict[name] = dict(layer=graph_geo_data, active=True, layer_type='graph')\n",
+    "        self.layer_update()\n",
+    "    \n",
+    "    @log_tab.capture() \n",
+    "    def layer_update(self):\n",
+    "        self.layers = tuple([entry['layer'] for entry in self.layer_dict.values() if entry['active']])\n",
+    "        \n",
+    "    @log_tab.capture()    \n",
+    "    def set_graph_style(self, radius):\n",
+    "        for name, entry in self.layer_dict.items():\n",
+    "            if entry['layer_type'] == \"graph\":\n",
+    "                layer = entry['layer']\n",
+    "                #layer.point_style['radius'] = radius   #doesn't work because traitlet change not observed\n",
+    "                self.custom_style['point_style']['radius'] = radius\n",
+    "                layer = ipyleaflet.GeoData(geo_dataframe = layer.geo_dataframe, name = layer.name, **self.custom_style)\n",
+    "                self.layer_dict[name]['layer'] = layer\n",
+    "        self.layer_update()\n",
+    "            \n",
+    "    \n",
+    "    @log_tab.capture()    \n",
+    "    def remove_graphs(self, button):\n",
+    "        for idx, layer in enumerate(self.layers):\n",
+    "            if layer.name[0:5] == \"Graph\":\n",
+    "                self.remove_layer(layer)\n",
+    "                \n",
+    "    @log_tab.capture()    \n",
+    "    def checkbox_layer_switch(self, change_dict):\n",
+    "        if change_dict['name'] == 'value':\n",
+    "            layer_name = change_dict['owner'].description\n",
+    "            self.layer_dict[layer_name]['active'] = change_dict['new']\n",
+    "            \n",
+    "        self.layer_update()\n",
+    "    \n",
+    "    \n",
+    "    @log_tab.capture()    \n",
+    "    def create_habitat_tab(self):\n",
+    "        checkboxes = []\n",
+    "        for name in self.layer_dict.keys():\n",
+    "            checkbox = widgets.Checkbox(\n",
+    "                    value=True,\n",
+    "                    description=name,\n",
+    "                    disabled=False,\n",
+    "                    indent=False\n",
+    "                )\n",
+    "            checkbox.observe(self.checkbox_layer_switch)\n",
+    "            checkboxes.append(checkbox)\n",
+    "            \n",
+    "        self.add_habitat_button = widgets.Button(\n",
+    "            description='Remove graph',\n",
+    "            disabled=False,\n",
+    "            button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
+    "            tooltip='Click me',\n",
+    "            icon='plus' # (FontAwesome names without the `fa-` prefix)\n",
+    "        )\n",
+    "        self.add_habitat_button.on_click(self.remove_graphs)\n",
+    "        \n",
+    "        habitat_accordion = widgets.Accordion(children=[self.add_habitat_button])\n",
+    "        habitat_accordion.set_title(0, 'Habitats')\n",
+    "        \n",
+    "        checkboxes.append(habitat_accordion)\n",
+    "        \n",
+    "        habitat_tab = widgets.VBox(checkboxes)\n",
+    "            \n",
+    "        return habitat_tab\n",
+    "    \n",
+    "    @log_tab.capture()    \n",
+    "    def create_diff_tab(self):\n",
+    "        time_slider1 = widgets.IntSlider(min=1960, max=2021, step=1, value=1990, description=\"Start time:\")\n",
+    "        time_slider2 = widgets.IntSlider(min=1960, max=2021, step=1, value=2010, description=\"End time:\")\n",
+    "        \n",
+    "        compute_node_button = widgets.Button(\n",
+    "            description='Compute node diff',\n",
+    "            disabled=False,\n",
+    "            button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
+    "            tooltip='This computes the differences between of the nodes in the graph at start time and the graph at end time.',\n",
+    "            icon='' # (FontAwesome names without the `fa-` prefix)\n",
+    "        )\n",
+    "        \n",
+    "        compute_pgon_button = widgets.Button(\n",
+    "            description='Compute polygon diff',\n",
+    "            disabled=False,\n",
+    "            button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
+    "            tooltip='This computes the differences between of the polygons in the graph at start time and the graph at end time.',\n",
+    "            icon='' # (FontAwesome names without the `fa-` prefix)\n",
+    "        )\n",
+    "        \n",
+    "        diff_tab = widgets.VBox([time_slider1, time_slider2, compute_node_button, compute_pgon_button])\n",
+    "        \n",
+    "        return diff_tab\n",
+    "    \n",
+    "    @log_tab.capture()    \n",
+    "    def create_settings_tab(self):\n",
+    "        radius_slider = widgets.FloatSlider(min=0.01, max=100.0, step=0.005, value=5.0, description=\"Node radius:\")\n",
+    "        w = widgets.interactive(self.set_graph_style, radius=radius_slider)\n",
+    "        \n",
+    "        zoom_slider = widgets.FloatSlider(description='Zoom level:', min=0, max=15, value=7)\n",
+    "        widgets.jslink((zoom_slider, 'value'), (self, 'zoom'))\n",
+    "        \n",
+    "        settings_tab = widgets.VBox([zoom_slider, radius_slider, radius_slider, radius_slider, radius_slider, radius_slider])\n",
+    "        \n",
+    "        return settings_tab\n",
+    "        \n",
+    "    @log_tab.capture()    \n",
+    "    def add_widgets(self):\n",
+    "        self.add_settings_widget()\n",
+    "        self.add_control(ipyleaflet.FullScreenControl())\n",
+    "            \n",
+    "    @log_tab.capture()\n",
+    "    def add_settings_widget(self):\n",
+    "        habitats_tab = self.create_habitat_tab()\n",
+    "        diff_tab = self.create_diff_tab()\n",
+    "        settings_tab = self.create_settings_tab()\n",
+    "        \n",
+    "        tab_nest = widgets.Tab()\n",
+    "        tab_nest.children = [habitats_tab, settings_tab, diff_tab, settings_tab, self.log_tab]\n",
+    "        for i, title in enumerate([\"Habitats\",\"Metrics\",\"Diff\",\"Settings\",\"Log\"]):\n",
+    "            tab_nest.set_title(i, title)\n",
+    "            \n",
+    "        header = widgets.HTML(markdown.markdown(\"\"\"  GeoGraph  \"\"\"))\n",
+    "        widget_control2 = ipyleaflet.WidgetControl(widget=header, position='bottomright')\n",
+    "        self.add_control(widget_control2)\n",
+    "        \n",
+    "        self.control = ipyleaflet.LayersControl(position='topleft')\n",
+    "        self.add_control(self.control)\n",
+    "        \n",
+    "        widget_panel = widgets.VBox([tab_nest])\n",
+    "        widget_control = ipyleaflet.WidgetControl(widget=widget_panel, position='topright')\n",
+    "        self.add_control(widget_control)\n",
+    "        \n",
+    "\n",
+    "\n",
+    "        \n",
+    "viewer = GeoGraphViewer()\n",
+    "viewer.add_graph(graph)\n",
+    "viewer.add_graph(graph2, name = 'Graph 2')\n",
+    "viewer.create_habitat_tab()\n",
+    "viewer.add_widgets()\n",
+    "viewer"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "copyrighted-denmark",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "viewer.log_tab"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "extraordinary-excess",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.8.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}