Skip to content

Commit 455b810

Browse files
committed
refactor(sedona-gdal): prefer imported raster helpers
1 parent 22874bc commit 455b810

3 files changed

Lines changed: 14 additions & 10 deletions

File tree

c/sedona-gdal/src/raster/polygonize.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,7 @@ mod tests {
138138
use crate::global::with_global_gdal_api;
139139
use crate::raster::Buffer;
140140
use crate::vector::FieldDefn;
141+
use crate::vsi::unlink_mem_file;
141142

142143
with_global_gdal_api(|api| {
143144
let mem_driver = DriverManager::get_driver_by_name(api, "MEM").unwrap();
@@ -209,7 +210,7 @@ mod tests {
209210
.count();
210211
assert_eq!(ones_8, 1);
211212

212-
crate::vsi::unlink_mem_file(api, gpkg_path).unwrap();
213+
unlink_mem_file(api, gpkg_path).unwrap();
213214
})
214215
.unwrap();
215216
}
@@ -222,6 +223,7 @@ mod tests {
222223
use crate::global::with_global_gdal_api;
223224
use crate::raster::Buffer;
224225
use crate::vector::FieldDefn;
226+
use crate::vsi::unlink_mem_file;
225227

226228
with_global_gdal_api(|api| {
227229
let mem_driver = DriverManager::get_driver_by_name(api, "MEM").unwrap();
@@ -267,7 +269,7 @@ mod tests {
267269
let only_val = layer.features().next().unwrap().field_as_integer(0);
268270
assert_eq!(only_val, Some(7));
269271

270-
crate::vsi::unlink_mem_file(api, gpkg_path).unwrap();
272+
unlink_mem_file(api, gpkg_path).unwrap();
271273
})
272274
.unwrap();
273275
}

c/sedona-gdal/src/raster/rasterize.rs

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ use std::ptr;
2323

2424
use crate::cpl::CslStringList;
2525
use crate::dataset::Dataset;
26-
use crate::errors::Result;
26+
use crate::errors::{GdalError, Result};
2727
use crate::gdal_api::{call_gdal_api, GdalApi};
2828
use crate::gdal_dyn_bindgen::*;
2929
use crate::vector::Geometry;
@@ -147,12 +147,12 @@ pub fn rasterize(
147147
options: Option<RasterizeOptions>,
148148
) -> Result<()> {
149149
if band_list.is_empty() {
150-
return Err(crate::errors::GdalError::BadArgument(
150+
return Err(GdalError::BadArgument(
151151
"`band_list` must not be empty".to_string(),
152152
));
153153
}
154154
if burn_values.len() != geometries.len() {
155-
return Err(crate::errors::GdalError::BadArgument(format!(
155+
return Err(GdalError::BadArgument(format!(
156156
"burn_values length ({}) must match geometries length ({})",
157157
burn_values.len(),
158158
geometries.len()
@@ -162,7 +162,7 @@ pub fn rasterize(
162162
for &band in band_list {
163163
let is_good = band > 0 && (band as usize) <= raster_count;
164164
if !is_good {
165-
return Err(crate::errors::GdalError::BadArgument(format!(
165+
return Err(GdalError::BadArgument(format!(
166166
"Band index {} is out of bounds",
167167
band
168168
)));
@@ -211,6 +211,8 @@ pub fn rasterize(
211211
#[cfg(test)]
212212
mod tests {
213213
use super::*;
214+
use crate::driver::DriverManager;
215+
use crate::global::with_global_gdal_api;
214216

215217
#[test]
216218
fn test_rasterizeoptions_as_ptr() {
@@ -237,11 +239,11 @@ mod tests {
237239
#[cfg(feature = "gdal-sys")]
238240
#[test]
239241
fn test_rasterize() {
240-
crate::global::with_global_gdal_api(|api| {
242+
with_global_gdal_api(|api| {
241243
let wkt = "POLYGON ((2 2, 2 4.25, 4.25 4.25, 4.25 2, 2 2))";
242244
let poly = Geometry::from_wkt(api, wkt).unwrap();
243245

244-
let driver = crate::driver::DriverManager::get_driver_by_name(api, "MEM").unwrap();
246+
let driver = DriverManager::get_driver_by_name(api, "MEM").unwrap();
245247
let dataset = driver.create("", 5, 5, 1).unwrap();
246248

247249
let bands = [1];

c/sedona-gdal/src/raster/rasterize_affine.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,8 @@ mod tests {
178178

179179
use crate::driver::{Driver, DriverManager};
180180
use crate::global::with_global_gdal_api;
181-
use crate::raster::rasterize;
182181
use crate::raster::types::Buffer;
182+
use crate::raster::{rasterize, RasterizeOptions};
183183

184184
fn mem_driver(api: &'static GdalApi) -> Driver {
185185
DriverManager::get_driver_by_name(api, "MEM").unwrap()
@@ -284,7 +284,7 @@ mod tests {
284284
&[1],
285285
&geom_refs,
286286
&[1.0],
287-
Some(crate::raster::RasterizeOptions {
287+
Some(RasterizeOptions {
288288
all_touched: true,
289289
..Default::default()
290290
}),

0 commit comments

Comments
 (0)