Update numpy_to_raster

banesullivan · Jun 14, 2023 · 70e1a22 · 70e1a22
1 parent ff70753
commit 70e1a22
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/localtileserver/__init__.py b/localtileserver/__init__.py
@@ -1,6 +1,7 @@
 # flake8: noqa: F401
 from localtileserver._version import __version__
 from localtileserver.client import RemoteTileClient, TileClient, get_or_create_tile_client
+from localtileserver.helpers import hillshade, parse_shapely, polygon_to_geojson, save_new_raster
 from localtileserver.report import Report
 from localtileserver.tiler import get_cache_dir, make_vsi, purge_cache
 from localtileserver.validate import validate_cog

diff --git a/localtileserver/helpers.py b/localtileserver/helpers.py
@@ -32,11 +32,16 @@ def numpy_to_raster(ras_meta, data, out_path: str = None):
  use a temporary file
 
  """
+ if data.ndim == 2:
+ data = data[np.newaxis, ...]
+
  ras_meta = ras_meta.copy()
  ras_meta.update({"count": data.shape[0]})
  ras_meta.update({"dtype": str(data.dtype)})
  ras_meta.update({"height": data.shape[1]})
  ras_meta.update({"width": data.shape[2]})
+ ras_meta.update({"compress": "lzw"})
+ ras_meta.update({"driver": "GTiff"})
 
  if not out_path:
  ext = get_extensions_from_driver(ras_meta["driver"])[0]