Add Image Moments functions

Author: 9prady9

src/defines.rs

@@ -333,3 +333,14 @@ pub enum MarkerType {
     PLUS     = 6,
     STAR     = 7
 }
+
+/// Image moment types
+#[repr(C)]
+#[derive(Clone, Copy, Debug, PartialEq)]
+pub enum MomentType {
+    M00 = 1,    // 1<<0
+    M01 = 2,    // 1<<1
+    M10 = 4,    // 1<<2
+    M11 = 8,    // 1<<3
+    FIRST_ORDER = 1<<0 | 1<<1 | 1<<2 | 1<<3
+}

src/image/mod.rs

@@ -1,7 +1,7 @@
 extern crate libc;
 
 use array::Array;
-use defines::{AfError, BorderType, ColorSpace, Connectivity, InterpType, YCCStd};
+use defines::{AfError, BorderType, ColorSpace, Connectivity, InterpType, YCCStd, MomentType};
 use error::HANDLE_ERROR;
 use util::HasAfEnum;
 use self::libc::{uint8_t, c_uint, c_int, c_float, c_double};
@@ -93,6 +93,9 @@ extern {
     fn af_rgb2ycbcr(out: MutAfArray, input: AfArray, stnd: c_int) -> c_int;
     fn af_is_image_io_available(out: *mut c_int) -> c_int;
     fn af_transform_coordinates(out: MutAfArray, tf: AfArray, d0: c_float, d1: c_float) -> c_int;
+
+    fn af_moments(out: MutAfArray, input: AfArray, moment: c_int) ->c_int;
+    fn af_moments_all(out: *mut c_double, input: AfArray, moment: c_int) ->c_int;
 }
 
 /// Calculate the gradients
@@ -1136,3 +1139,47 @@ pub fn transform_coords(tf: &Array, d0: f32, d1: f32) -> Array {
         Array::from(temp)
     }
 }
+
+/// Find Image moments
+///
+/// # Parameters
+///
+/// - `input` is the input image
+/// - `moment` is the type of moment to be computed, takes a value of
+/// [enum](./enum.MomentType.html)
+///
+/// # Return Values
+///
+/// Moments Array
+pub fn moments(input: &Array, moment: MomentType) -> Array {
+    unsafe {
+        let mut temp: i64 = 0;
+        let err_val = af_moments(&mut temp as MutAfArray,
+                                 input.get() as AfArray,
+                                 moment as c_int);
+        HANDLE_ERROR(AfError::from(err_val));
+        Array::from(temp)
+    }
+}
+
+/// Find Image moment for whole image
+///
+/// # Parameters
+///
+/// - `input` is the input image
+/// - `moment` is the type of moment to be computed, takes a value of
+/// [enum](./enum.MomentType.html)
+///
+/// # Return Values
+///
+/// Moment value of the whole image
+pub fn moments_all(input: &Array, moment: MomentType) -> f64 {
+    unsafe {
+        let mut temp: f64 = 0.0;
+        let err_val = af_moments_all(&mut temp as *mut c_double,
+                                     input.get() as AfArray,
+                                     moment as c_int);
+        HANDLE_ERROR(AfError::from(err_val));
+        temp
+    }
+}

src/lib.rs

@@ -47,7 +47,7 @@ mod device;
 pub use defines::{DType, AfError, Backend, ColorMap, YCCStd, HomographyType};
 pub use defines::{InterpType, BorderType, MatchType, NormType};
 pub use defines::{Connectivity, ConvMode, ConvDomain, ColorSpace, MatProp};
-pub use defines::{MarkerType};
+pub use defines::{MarkerType, MomentType};
 mod defines;
 
 pub use dim4::Dim4;
@@ -74,6 +74,7 @@ pub use image::{gradient, histogram, hist_equal, regions};
 pub use image::{gray2rgb, rgb2gray, hsv2rgb, rgb2hsv, color_space};
 pub use image::{bilateral, mean_shift, medfilt, sobel};
 pub use image::{unwrap, wrap, sat, rgb2ycbcr, ycbcr2rgb, is_imageio_available, transform_coords};
+pub use image::{moments, moments_all};
 mod image;
 
 pub use lapack::{svd, lu, qr, cholesky, solve, solve_lu, inverse, det, rank, norm};