BUG: Handle additional data types in rio.reproject (#619)

corteva · Dec 19, 2022 · c2f066c · c2f066c
1 parent 9b9a5ae
commit c2f066c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/docs/history.rst b/docs/history.rst
@@ -3,6 +3,7 @@ History
 
 Latest
 ------
+- BUG: Handle data type error in `rio.reproject` (issue #618)
 
 0.13.2
 ------

diff --git a/rioxarray/raster_array.py b/rioxarray/raster_array.py
@@ -65,8 +65,8 @@
 
 # DTYPE TO NODATA MAP
 # Based on: https://github.com/OSGeo/gdal/blob/
-# cde27dc7641964a872efdc6bbcf5e3d3f7ab9cfd/gdal/
-# swig/python/gdal-utils/osgeo_utils/gdal_calc.py#L62
+# dee861e7c91c2da7ef8ff849947713e4d9bd115c/
+# swig/python/gdal-utils/osgeo_utils/gdal_calc.py#L61
 _NODATA_DTYPE_MAP = {
     1: 255,  # GDT_Byte
     2: 65535,  # GDT_UInt16
@@ -75,10 +75,13 @@
     5: -2147483647,  # GDT_Int32
     6: 3.402823466e38,  # GDT_Float32
     7: 1.7976931348623158e308,  # GDT_Float64
-    8: -32768,  # GDT_CInt16
-    9: -2147483647,  # GDT_CInt32
+    8: None,  # GDT_CInt16
+    9: None,  # GDT_CInt32
     10: 3.402823466e38,  # GDT_CFloat32
     11: 1.7976931348623158e308,  # GDT_CFloat64
+    12: None,  # GDT_Int64
+    13: None,  # GDT_UInt64
+    14: None,  # GDT_Int8
 }
 
 
@@ -510,7 +513,7 @@ def reproject(
 
     def _get_dst_nodata(self, nodata: Optional[float]) -> Optional[float]:
         default_nodata = (
-            _NODATA_DTYPE_MAP[dtype_rev[self._obj.dtype.name]]
+            _NODATA_DTYPE_MAP.get(dtype_rev[self._obj.dtype.name])
             if self.nodata is None
             else self.nodata
         )
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,6 +3,7 @@ History @@
     Latest
     ------
+    - BUG: Handle data type error in `rio.reproject` (issue #618)
 .13.2
     ------
@@ Expand Down @@