From 89d23c51eed9b74b7aad0d4bb5ada7e9e3d27982 Mon Sep 17 00:00:00 2001
From: netthier <lp@senselabs.de>
Date: Thu, 18 Apr 2024 17:37:13 +0200
Subject: [PATCH] Add `Rasterband::fill`

Signed-off-by: netthier <lp@senselabs.de>
---
 CHANGES.md               |  3 +++
 src/raster/rasterband.rs | 10 ++++++++++
 src/raster/tests.rs      | 10 ++++++++++
 3 files changed, 23 insertions(+)

diff --git a/CHANGES.md b/CHANGES.md
index ad974f38..eace9875 100644
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -2,6 +2,9 @@
 
 ## Unreleased
 
+- Added `Rasterband::fill`
+  - <https://github.com/georust/gdal/pull/528>
+
 - Added `Dataset::rasterbands`.
   - <https://github.com/georust/gdal/pull/523>
 
diff --git a/src/raster/rasterband.rs b/src/raster/rasterband.rs
index 43ccdaa5..5ca872e2 100644
--- a/src/raster/rasterband.rs
+++ b/src/raster/rasterband.rs
@@ -815,6 +815,16 @@ impl<'a> RasterBand<'a> {
             Ok(())
         }
     }
+
+    /// Fill this band with a constant value.
+    pub fn fill(&mut self, real_value: f64, imaginary_value: Option<f64>) -> Result<()> {
+        let rv = unsafe { gdal_sys::GDALFillRaster(self.c_rasterband, real_value, imaginary_value.unwrap_or(0.0)) };
+        if rv != CPLErr::CE_None {
+            return Err(_last_cpl_err(rv));
+        }
+        Ok(())
+    }
+
     /// Returns the color interpretation of this band.
     pub fn color_interpretation(&self) -> ColorInterpretation {
         let interp_index = unsafe { gdal_sys::GDALGetRasterColorInterpretation(self.c_rasterband) };
diff --git a/src/raster/tests.rs b/src/raster/tests.rs
index 0f5898a4..026b16ef 100644
--- a/src/raster/tests.rs
+++ b/src/raster/tests.rs
@@ -483,6 +483,16 @@ fn test_set_no_data_value() {
     assert_eq!(rasterband.no_data_value(), None);
 }
 
+#[test]
+fn test_fill() {
+    let driver = DriverManager::get_driver_by_name("MEM").unwrap();
+    let dataset = driver.create("", 5, 5, 1).unwrap();
+    let mut rasterband = dataset.rasterband(1).unwrap();
+    assert!(rasterband.fill(5.4, None).is_ok());
+    let contents = rasterband.read_as::<u8>((0, 0), (5, 5), (5, 5), None).unwrap();
+    assert!(contents.data().iter().all(|&v| v == 5));
+}
+
 #[test]
 fn test_get_scale() {
     let dataset = Dataset::open(fixture("offset_scaled_tinymarble.tif")).unwrap();