diff --git a/localtileserver/__init__.py b/localtileserver/__init__.py index 81a834e5..f21f1091 100644 --- 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 index 5beaf958..9a7934f3 100644 --- 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]