Added Error & Display traits for AfError type

Author: 9prady9

examples/helloworld.rs

@@ -10,7 +10,10 @@ fn main() {
     let dims = Dim4::new(&[5, 3, 1, 1]);
 
     println!("Create a 5-by-3 matrix of random floats on the GPU");
-    let a = randu(dims, Aftype::F32).unwrap();
+    let a = match randu(dims, Aftype::F32) {
+        Ok(value) => value,
+        Err(error) => panic!("{}", error),
+    };
     print(&a);
 
     println!("Element-wise arithmetic");
@@ -38,8 +41,7 @@ fn main() {
     // af_print(B);
 
     println!("Fourier transform the result");
-    let c = &fft(&b, 1.0, 0).unwrap();
-    print(&c);
+    fft(&b, 1.0, 0).map(|x| print(&x));
 
     // printf("Grab last row\n");
     // array c = C.row(end);
@@ -57,9 +59,11 @@ fn main() {
 
     // // Sort A
     println!("Sort A and print sorted array and corresponding indices");
-    let (vals, inds) = sort_index(&a, 0, true).unwrap();
-    print(&vals);
-    print(&inds);
+    sort_index(&a, 0, true)
+        .map(| x | {
+            print(&x.0);
+            print(&x.1);
+        });
 
     println!("u8 constant array");
     let u8_cnst = &constant(1 as u8, dims).unwrap();

examples/histogram.rs

@@ -15,20 +15,20 @@ fn main() {
 
     let img_wnd = match Window::new(480, 640, String::from("Input Image")) {
         Ok(v) => { v.set_position(100, 100).unwrap(); v },
-        Err(e)=> panic!("Window creation failed, exiting: {:?}", e),
+        Err(e)=> panic!("Window creation failed, exiting: {}", e),
     };
 
     let hst_wnd = match Window::new(512, 512, String::from("Input Image Histogram")) {
         Ok(v) => { v.set_position(600, 100).unwrap(); v },
-        Err(e)=> panic!("Window creation failed, exiting: {:?}", e),
+        Err(e)=> panic!("Window creation failed, exiting: {}", e),
     };
 
     let (man, hst) = match load_image(format!("{}/man.jpg", assets_dir.display()), false) {
         Ok(v) => match histogram(&v, 256, 0.0, 255.0) {
             Ok(h) => (v, h),
-            Err(e)=> panic!("Histogram computation failed, exiting: {:?}", e),
+            Err(e)=> panic!("Histogram computation failed, exiting: {}", e),
         },
-        Err(e)=> panic!("Image loading failed, exiting: {:?}", e),
+        Err(e)=> panic!("Image loading failed, exiting: {}", e),
     };
 
     let disp_img = man.dims()

src/defines.rs

@@ -1,3 +1,7 @@
+use std::error::Error;
+use std::fmt::{Display, Formatter};
+use std::fmt::Error as FmtError;
+
 #[repr(C)]
 #[derive(Clone, Copy, Debug)]
 pub enum AfError {
@@ -74,6 +78,35 @@ pub enum AfError {
     ERR_UNKNOWN        = 999
 }
 
+impl Display for AfError {
+    fn fmt(&self, f: &mut Formatter) -> Result<(), FmtError> {
+        write!(f, "{}", self.description())
+    }
+}
+
+impl Error for AfError {
+    fn description(&self) -> &str {
+        match *self {
+            AfError::SUCCESS => "Function returned successfully",
+            AfError::ERR_NO_MEM => "The system or device ran out of memory",
+            AfError::ERR_DRIVER => "Device driver error",
+            AfError::ERR_RUNTIME => "Error in runtime environment",
+            AfError::ERR_INVALID_ARRAY => "Input is not a valid Array Object",
+            AfError::ERR_ARG => "One of the function arguments is incorrect",
+            AfError::ERR_SIZE => "The size is incorrect",
+            AfError::ERR_TYPE => "The type is not supported by this function",
+            AfError::ERR_DIFF_TYPE => "The type of input arrays are not compatible",
+            AfError::ERR_BATCH => "Function does not support GFOR / batch mode",
+            AfError::ERR_NOT_SUPPORTED => "The option is not supported",
+            AfError::ERR_NOT_CONFIGURED => "This build of ArrayFire does not support this feature",
+            AfError::ERR_NO_DBL => "This device does not support double",
+            AfError::ERR_NO_GFX => "This build of ArrayFire was not built with graphics or this device does not support graphics",
+            AfError::ERR_INTERNAL => "There was an internal error in either ArrayFire or upstream project",
+            AfError::ERR_UNKNOWN => "Unkown Error",
+        }
+    }
+}
+
 #[derive(Copy, Clone)]
 pub enum Aftype {
     F32 = 0,