Fixes scale and offset missing GeoDataset (#205)

* Fixes scale and offset missing GeoDataset * Fixes CHANGELOG entry.
DOI-USGS · Apr 2, 2024 · f0b0bba · f0b0bba
1 parent 28ce2cf
commit f0b0bba
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -34,6 +34,8 @@ release.
 -->
 
 ## [Unreleased]
+### Fixed
+- Fixed a bug where scale and offset were not being read or applied to GeoDataset objects.
 
 ## [1.5.5]()
 ### Fixed

diff --git a/plio/io/io_gdal.py b/plio/io/io_gdal.py
@@ -506,14 +506,18 @@ def read_array(self, band=1, pixels=None, dtype=None):
 
         """
         band = self.dataset.GetRasterBand(band)
-
+        offset = band.GetOffset()
+        offset = 0 if offset is None else offset
+        scale = band.GetScale()
+        scale = 1 if scale is None else scale
+
         if dtype is None:
             dtype = GDAL2NP_CONVERSION[band.DataType]
 
         dtype = getattr(np, dtype)
 
         if not pixels:
-            array = band.ReadAsArray().astype(dtype)
+            array = (band.ReadAsArray().astype(dtype) + offset) * scale
             #if self.north_up == False:
             #    array = np.flipud(array)
         else:
@@ -534,7 +538,7 @@ def read_array(self, band=1, pixels=None, dtype=None):
 
             if ystart + ycount > ymax:
                 ycount = ymax - ystart
-            array = band.ReadAsArray(xstart, ystart, xcount, ycount).astype(dtype)
+            array = (band.ReadAsArray(xstart, ystart, xcount, ycount).astype(dtype) + offset) * scale
 
         return array