updated viewshed gpu example

Author: brendancol

examples/datasets.yml

@@ -48,4 +48,7 @@ data:
     files: LC80030172015001LGN00_B11.TIF
   - url: https://xarrayspatial.blob.core.windows.net/examples-data/LC80030172015001LGN00_BQA.tiff
     title: 'qa'
-    files: LC80030172015001LGN00_BQA.TIF
+    files: LC80030172015001LGN00_BQA.TIF
+  - url: https://makepath-gpu-dataset.s3.us-east-1.amazonaws.com/colorado_merge_3arc_resamp.tif
+    title: 'Colorado Elevation Data'
+    files: colorado_merge_3arc_resamp.tif

examples/viewshed_gpu.ipynb

@@ -1,19 +1,5 @@
 {
  "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "!pip install rasterio==1.2.10\n",
-    "!pip install xarray-spatial==0.3.5\n",
-    "!pip install rtxpy\n",
-    "!pip install numba==0.55.1\n",
-    "!pip install geopandas\n",
-    "!pip install spatialpandas"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {
@@ -49,9 +35,12 @@
     "import xarray as xr\n",
     "import numpy as np\n",
     "import cupy\n",
+    "import rioxarray\n",
     "\n",
     "from xrspatial import hillshade\n",
-    "from xrspatial import viewshed"
+    "from xrspatial import viewshed\n",
+    "\n",
+    "import rioxarray"
    ]
   },
   {
@@ -64,7 +53,9 @@
     "\n",
     "#### Raster data\n",
     "\n",
-    "The input ***elevation raster*** data covers the state of Colorado, USA. We'll load it from a local input file and calculate its bounding box to define the study area."
+    "The input ***elevation raster*** data covers the state of Colorado, USA. We'll load it from a local input file and calculate its bounding box to define the study area.\n",
+    "\n",
+    "Example data available here: https://makepath-gpu-dataset.s3.us-east-1.amazonaws.com/colorado_merge_3arc_resamp.tif"
    ]
   },
   {
@@ -73,8 +64,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "file_name = 'colorado_merge_3arc_resamp.tif'\n",
-    "raster = xr.open_rasterio(file_name).sel(band=1).drop('band', dim=None)\n",
+    "file_name = './data/colorado_merge_3arc_resamp.tif'\n",
+    "\n",
+    "raster = rioxarray.open_rasterio(file_name).sel(band=1).drop_vars('band')\n",
     "raster.name = 'Colorado Elevation Raster'\n",
     "\n",
     "xmin, xmax = raster.x.data.min(), raster.x.data.max()\n",
@@ -236,10 +228,12 @@
     "for i, peak in top_peaks.iterrows():\n",
     "    viewpoint_x, viewpoint_y = peak['geometry'].x, peak['geometry'].y\n",
     "    viewshed_grid = viewshed(raster, x=viewpoint_x, y=viewpoint_y)\n",
+    "    print(type(viewshed_grid.data))\n",
     "    # assign name for the output grid\n",
     "    viewshed_grid.name = peak['NAME']\n",
     "    # set visible cells to VISIBLE\n",
-    "    viewshed_grid = viewshed_grid.where(viewshed_grid == INVISIBLE, other=VISIBLE)\n",
+    "    mask_invisible = (viewshed_grid == INVISIBLE)\n",
+    "    viewshed_grid = mask_invisible * INVISIBLE + (~mask_invisible) * VISIBLE\n",
     "    visibility_grids.append(viewshed_grid)"
    ]
   },
@@ -321,10 +315,24 @@
   }
  ],
  "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
   "language_info": {
-   "name": "python"
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.14.2"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 0
+ "nbformat_minor": 4
 }