impl OperatorNorm

Author: termoshtt

src/impl2/mod.rs

@@ -0,0 +1,2 @@
+
+pub mod opnorm;

src/impl2/opnorm.rs

@@ -0,0 +1,47 @@
+//! Implement Operator norms for matrices
+
+use lapack::c;
+
+use types::*;
+use layout::*;
+
+#[repr(u8)]
+pub enum NormType {
+    One = b'o',
+    Infinity = b'i',
+    Frobenius = b'f',
+}
+
+pub trait OperatorNorm: Sized {
+    type Output;
+
+    fn opnorm(NormType, Layout, &[Self]) -> Self::Output;
+
+    fn opnorm_one(l: Layout, a: &[Self]) -> Self::Output {
+        Self::opnorm(NormType::One, l, a)
+    }
+
+    fn opnorm_inf(l: Layout, a: &[Self]) -> Self::Output {
+        Self::opnorm(NormType::Infinity, l, a)
+    }
+
+    fn opnorm_fro(l: Layout, a: &[Self]) -> Self::Output {
+        Self::opnorm(NormType::Frobenius, l, a)
+    }
+}
+
+macro_rules! impl_opnorm {
+    ($scalar:ty, $output:ty, $lange:path) => {
+impl OperatorNorm for $scalar {
+    type Output = $output;
+    fn opnorm(t: NormType, l: Layout, a: &[Self]) -> Self::Output {
+        let (m, n) = l.ffi_size();
+        $lange(l.ffi_layout(), t as u8, m, n, a, m)
+    }
+}
+}} // impl_opnorm!
+
+impl_opnorm!(f64, f64, c::dlange);
+impl_opnorm!(f32, f32, c::slange);
+impl_opnorm!(c64, f64, c::zlange);
+impl_opnorm!(c32, f32, c::clange);

src/layout.rs

@@ -2,6 +2,7 @@
 use ndarray::*;
 
 use super::error::*;
+use lapack::c::Layout as FFI_Layout;
 
 pub enum Layout {
     C((usize, usize)),
@@ -15,6 +16,18 @@ impl Layout {
             &Layout::F(s) => s,
         }
     }
+
+    pub fn ffi_size(&self) -> (i32, i32) {
+        let (n, m) = self.size();
+        (m as i32, n as i32)
+    }
+
+    pub fn ffi_layout(&self) -> FFI_Layout {
+        match self {
+            &Layout::C(_) => FFI_Layout::RowMajor,
+            &Layout::F(_) => FFI_Layout::ColumnMajor,
+        }
+    }
 }
 
 pub trait AllocatedArray2D {

src/lib.rs

@@ -43,9 +43,10 @@ extern crate enum_error_derive;
 extern crate derive_new;
 
 pub mod types;
-pub mod impls;
 pub mod error;
 pub mod layout;
+pub mod impls;
+pub mod impl2;
 
 pub mod vector;
 pub mod matrix;