ssb-sgis 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

sgis/raster/cubebase.py

@@ -1,39 +1,25 @@
-def _method_as_func(self, method, **kwargs):
-    return getattr(self, method)(**kwargs)
-
-
-def _raster_from_path(path, raster_type, res, **kwargs):
-    return raster_type.from_path(path, res=res, **kwargs)
-
-
-def _from_gdf_func(gdf, raster_type, **kwargs):
-    return raster_type.from_gdf(gdf, **kwargs)
-
-
-def _write_func(raster, path, **kwargs):
-    raster.write(path, **kwargs)
-    return raster
-
-
-def _add(raster, scalar):
-    return raster + scalar
+from collections.abc import Callable
+from pathlib import Path
 
+from geopandas import GeoDataFrame
 
-def _mul(raster, scalar):
-    return raster * scalar
+from .raster import Raster
 
 
-def _sub(raster, scalar):
-    return raster - scalar
+def _from_gdf_func(gdf: GeoDataFrame, **kwargs) -> Raster:
+    return Raster.from_gdf(gdf, **kwargs)
 
 
-def _truediv(raster, scalar):
-    return raster / scalar
+def _raster_from_path(path: str, **kwargs) -> Raster:
+    return Raster.from_path(path, **kwargs)
 
 
-def _floordiv(raster, scalar):
-    return raster // scalar
+def _method_as_func(self: Raster, method: str, **kwargs) -> Callable:
+    return getattr(self, method)(**kwargs)
 
 
-def _pow(raster, scalar):
-    return raster**scalar
+def _write_func(raster: Raster, folder: str, **kwargs):
+    path = str(Path(folder) / Path(raster.name).stem) + ".tif"
+    raster.write(path, **kwargs)
+    raster.path = path
+    return raster