first version of OpenCL support for operations on rasters (and soon f…

datatypes/src/raster/typed_raster.rs

@@ -91,6 +91,146 @@ where
  }
  None
  }
+
+ pub fn get_u8_ref(&self) -> Option<&BaseRaster<D, u8, Vec<u8>>> {
+ if let TypedRasterNDim::U8(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u16_ref(&self) -> Option<&BaseRaster<D, u16, Vec<u16>>> {
+ if let TypedRasterNDim::U16(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u32_ref(&self) -> Option<&BaseRaster<D, u32, Vec<u32>>> {
+ if let TypedRasterNDim::U32(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u64_ref(&self) -> Option<&BaseRaster<D, u64, Vec<u64>>> {
+ if let TypedRasterNDim::U64(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i8_ref(&self) -> Option<&BaseRaster<D, i8, Vec<i8>>> {
+ if let TypedRasterNDim::I8(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i16_ref(&self) -> Option<&BaseRaster<D, i16, Vec<i16>>> {
+ if let TypedRasterNDim::I16(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i32_ref(&self) -> Option<&BaseRaster<D, i32, Vec<i32>>> {
+ if let TypedRasterNDim::I32(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i64_ref(&self) -> Option<&BaseRaster<D, i64, Vec<i64>>> {
+ if let TypedRasterNDim::I64(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_f32_ref(&self) -> Option<&BaseRaster<D, f32, Vec<f32>>> {
+ if let TypedRasterNDim::F32(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_f64_ref(&self) -> Option<&BaseRaster<D, f64, Vec<f64>>> {
+ if let TypedRasterNDim::F64(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u8_ref_mut(&mut self) -> Option<&mut BaseRaster<D, u8, Vec<u8>>> {
+ if let TypedRasterNDim::U8(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u16_ref_mut(&mut self) -> Option<&mut BaseRaster<D, u16, Vec<u16>>> {
+ if let TypedRasterNDim::U16(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u32_ref_mut(&mut self) -> Option<&mut BaseRaster<D, u32, Vec<u32>>> {
+ if let TypedRasterNDim::U32(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_u64_ref_mut(&mut self) -> Option<&mut BaseRaster<D, u64, Vec<u64>>> {
+ if let TypedRasterNDim::U64(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i8_ref_mut(&mut self) -> Option<&mut BaseRaster<D, i8, Vec<i8>>> {
+ if let TypedRasterNDim::I8(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i16_ref_mut(&mut self) -> Option<&mut BaseRaster<D, i16, Vec<i16>>> {
+ if let TypedRasterNDim::I16(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i32_ref_mut(&mut self) -> Option<&mut BaseRaster<D, i32, Vec<i32>>> {
+ if let TypedRasterNDim::I32(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_i64_ref_mut(&mut self) -> Option<&mut BaseRaster<D, i64, Vec<i64>>> {
+ if let TypedRasterNDim::I64(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_f32_ref_mut(&mut self) -> Option<&mut BaseRaster<D, f32, Vec<f32>>> {
+ if let TypedRasterNDim::F32(r) = self {
+ return Some(r);
+ }
+ None
+ }
+
+ pub fn get_f64_ref_mut(&mut self) -> Option<&mut BaseRaster<D, f64, Vec<f64>>> {
+ if let TypedRasterNDim::F64(r) = self {
+ return Some(r);
+ }
+ None
+ }
 }
 
 // TODO: use a macro?

operators/src/error.rs

@@ -58,6 +58,14 @@ pub enum Error {
  SerdeJson {
  source: serde_json::Error,
  },
+ OCL {
+ #[snafu(source(from(ocl::error::Error, failure::Fail::compat)))]
+ source: failure::Compat<ocl::error::Error>,
+ },
+ CLProgramInvalidRasterIndex,
+ CLProgramInvalidRasterDataType,
+ CLProgramUnspecifiedRaster,
+ CLInvalidInputsForIterationType,
 }
 
 impl From<geoengine_datatypes::error::Error> for Error {

operators/src/lib.rs

@@ -22,5 +22,6 @@
 pub mod engine;
 pub mod error;
 pub mod mock;
+pub mod opencl;
 pub mod source;
 pub mod util;