nmatrix-fftw 0.2.1

data/ext/nmatrix/math/trsm.h

@@ -0,0 +1,332 @@
+/////////////////////////////////////////////////////////////////////
+// = NMatrix
+//
+// A linear algebra library for scientific computation in Ruby.
+// NMatrix is part of SciRuby.
+//
+// NMatrix was originally inspired by and derived from NArray, by
+// Masahiro Tanaka: http://narray.rubyforge.org
+//
+// == Copyright Information
+//
+// SciRuby is Copyright (c) 2010 - 2014, Ruby Science Foundation
+// NMatrix is Copyright (c) 2012 - 2014, John Woods and the Ruby Science Foundation
+//
+// Please see LICENSE.txt for additional copyright notices.
+//
+// == Contributing
+//
+// By contributing source code to SciRuby, you agree to be bound by
+// our Contributor Agreement:
+//
+// * https://github.com/SciRuby/sciruby/wiki/Contributor-Agreement
+//
+// == trsm.h
+//
+// trsm function in native C++.
+//
+/*
+ *             Automatically Tuned Linear Algebra Software v3.8.4
+ *                    (C) Copyright 1999 R. Clint Whaley
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *   1. Redistributions of source code must retain the above copyright
+ *      notice, this list of conditions and the following disclaimer.
+ *   2. Redistributions in binary form must reproduce the above copyright
+ *      notice, this list of conditions, and the following disclaimer in the
+ *      documentation and/or other materials provided with the distribution.
+ *   3. The name of the ATLAS group or the names of its contributers may
+ *      not be used to endorse or promote products derived from this
+ *      software without specific written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
+ * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE ATLAS GROUP OR ITS CONTRIBUTORS
+ * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+#ifndef TRSM_H
+#define TRSM_H
+
+
+namespace nm { namespace math {
+
+
+/*
+ * This version of trsm doesn't do any error checks and only works on column-major matrices.
+ *
+ * For row major, call trsm<DType> instead. That will handle necessary changes-of-variables
+ * and parameter checks.
+ *
+ * Note that some of the boundary conditions here may be incorrect. Very little has been tested!
+ * This was converted directly from dtrsm.f using f2c, and then rewritten more cleanly.
+ */
+template <typename DType>
+inline void trsm_nothrow(const enum CBLAS_SIDE side, const enum CBLAS_UPLO uplo,
+                         const enum CBLAS_TRANSPOSE trans_a, const enum CBLAS_DIAG diag,
+                         const int m, const int n, const DType alpha, const DType* a,
+                         const int lda, DType* b, const int ldb)
+{
+
+  // (row-major) trsm: left upper trans nonunit m=3 n=1 1/1 a 3 b 3
+
+  if (m == 0 || n == 0) return; /* Quick return if possible. */
+
+  if (alpha == 0) { // Handle alpha == 0
+    for (int j = 0; j < n; ++j) {
+      for (int i = 0; i < m; ++i) {
+        b[i + j * ldb] = 0;
+      }
+    }
+    return;
+  }
+
+  if (side == CblasLeft) {
+    if (trans_a == CblasNoTrans) {
+
+      /* Form  B := alpha*inv( A )*B. */
+      if (uplo == CblasUpper) {
+        for (int j = 0; j < n; ++j) {
+          if (alpha != 1) {
+            for (int i = 0; i < m; ++i) {
+              b[i + j * ldb] = alpha * b[i + j * ldb];
+            }
+          }
+          for (int k = m-1; k >= 0; --k) {
+            if (b[k + j * ldb] != 0) {
+              if (diag == CblasNonUnit) {
+                b[k + j * ldb] /= a[k + k * lda];
+              }
+
+              for (int i = 0; i < k-1; ++i) {
+                b[i + j * ldb] -= b[k + j * ldb] * a[i + k * lda];
+              }
+            }
+          }
+        }
+      } else {
+        for (int j = 0; j < n; ++j) {
+          if (alpha != 1) {
+            for (int i = 0; i < m; ++i) {
+              b[i + j * ldb] = alpha * b[i + j * ldb];
+            }
+          }
+          for (int k = 0; k < m; ++k) {
+            if (b[k + j * ldb] != 0.) {
+              if (diag == CblasNonUnit) {
+                b[k + j * ldb] /= a[k + k * lda];
+              }
+              for (int i = k+1; i < m; ++i) {
+                b[i + j * ldb] -= b[k + j * ldb] * a[i + k * lda];
+              }
+            }
+          }
+        }
+      }
+    } else { // CblasTrans
+
+      /*           Form  B := alpha*inv( A**T )*B. */
+      if (uplo == CblasUpper) {
+        for (int j = 0; j < n; ++j) {
+          for (int i = 0; i < m; ++i) {
+            DType temp = alpha * b[i + j * ldb];
+            for (int k = 0; k < i; ++k) { // limit was i-1. Lots of similar bugs in this code, probably.
+              temp -= a[k + i * lda] * b[k + j * ldb];
+            }
+            if (diag == CblasNonUnit) {
+              temp /= a[i + i * lda];
+            }
+            b[i + j * ldb] = temp;
+          }
+        }
+      } else {
+        for (int j = 0; j < n; ++j) {
+          for (int i = m-1; i >= 0; --i) {
+            DType temp= alpha * b[i + j * ldb];
+            for (int k = i+1; k < m; ++k) {
+              temp -= a[k + i * lda] * b[k + j * ldb];
+            }
+            if (diag == CblasNonUnit) {
+              temp /= a[i + i * lda];
+            }
+            b[i + j * ldb] = temp;
+          }
+        }
+      }
+    }
+  } else { // right side
+
+    if (trans_a == CblasNoTrans) {
+
+      /*           Form  B := alpha*B*inv( A ). */
+
+      if (uplo == CblasUpper) {
+        for (int j = 0; j < n; ++j) {
+          if (alpha != 1) {
+            for (int i = 0; i < m; ++i) {
+              b[i + j * ldb] = alpha * b[i + j * ldb];
+            }
+          }
+          for (int k = 0; k < j-1; ++k) {
+            if (a[k + j * lda] != 0) {
+              for (int i = 0; i < m; ++i) {
+                b[i + j * ldb] -= a[k + j * lda] * b[i + k * ldb];
+              }
+            }
+          }
+          if (diag == CblasNonUnit) {
+            DType temp = 1 / a[j + j * lda];
+            for (int i = 0; i < m; ++i) {
+              b[i + j * ldb] = temp * b[i + j * ldb];
+            }
+          }
+        }
+      } else {
+        for (int j = n-1; j >= 0; --j) {
+          if (alpha != 1) {
+            for (int i = 0; i < m; ++i) {
+              b[i + j * ldb] = alpha * b[i + j * ldb];
+            }
+          }
+
+          for (int k = j+1; k < n; ++k) {
+            if (a[k + j * lda] != 0.) {
+              for (int i = 0; i < m; ++i) {
+                b[i + j * ldb] -= a[k + j * lda] * b[i + k * ldb];
+              }
+            }
+          }
+          if (diag == CblasNonUnit) {
+            DType temp = 1 / a[j + j * lda];
+
+            for (int i = 0; i < m; ++i) {
+              b[i + j * ldb] = temp * b[i + j * ldb];
+            }
+          }
+        }
+      }
+    } else { // CblasTrans
+
+      /*           Form  B := alpha*B*inv( A**T ). */
+
+      if (uplo == CblasUpper) {
+        for (int k = n-1; k >= 0; --k) {
+          if (diag == CblasNonUnit) {
+            DType temp= 1 / a[k + k * lda];
+            for (int i = 0; i < m; ++i) {
+              b[i + k * ldb] = temp * b[i + k * ldb];
+            }
+          }
+          for (int j = 0; j < k-1; ++j) {
+            if (a[j + k * lda] != 0.) {
+              DType temp= a[j + k * lda];
+              for (int i = 0; i < m; ++i) {
+                b[i + j * ldb] -= temp * b[i + k *  ldb];
+              }
+            }
+          }
+          if (alpha != 1) {
+            for (int i = 0; i < m; ++i) {
+              b[i + k * ldb] = alpha * b[i + k * ldb];
+            }
+          }
+        }
+      } else {
+        for (int k = 0; k < n; ++k) {
+          if (diag == CblasNonUnit) {
+            DType temp = 1 / a[k + k * lda];
+            for (int i = 0; i < m; ++i) {
+              b[i + k * ldb] = temp * b[i + k * ldb];
+            }
+          }
+          for (int j = k+1; j < n; ++j) {
+            if (a[j + k * lda] != 0.) {
+              DType temp = a[j + k * lda];
+              for (int i = 0; i < m; ++i) {
+                b[i + j * ldb] -= temp * b[i + k * ldb];
+              }
+            }
+          }
+          if (alpha != 1) {
+            for (int i = 0; i < m; ++i) {
+              b[i + k * ldb] = alpha * b[i + k * ldb];
+            }
+          }
+        }
+      }
+    }
+  }
+}
+
+/*
+ * BLAS' DTRSM function, generalized.
+ */
+template <typename DType, typename = typename std::enable_if<!std::is_integral<DType>::value>::type>
+inline void trsm(const enum CBLAS_ORDER order,
+                 const enum CBLAS_SIDE side, const enum CBLAS_UPLO uplo,
+                 const enum CBLAS_TRANSPOSE trans_a, const enum CBLAS_DIAG diag,
+                 const int m, const int n, const DType alpha, const DType* a,
+                 const int lda, DType* b, const int ldb)
+{
+  /*using std::cerr;
+  using std::endl;*/
+
+  int                     num_rows_a = n;
+  if (side == CblasLeft)  num_rows_a = m;
+
+  if (lda < std::max(1,num_rows_a)) {
+    fprintf(stderr, "TRSM: num_rows_a = %d; got lda=%d\n", num_rows_a, lda);
+    rb_raise(rb_eArgError, "TRSM: Expected lda >= max(1, num_rows_a)");
+  }
+
+  // Test the input parameters.
+  if (order == CblasRowMajor) {
+    if (ldb < std::max(1,n)) {
+      fprintf(stderr, "TRSM: M=%d; got ldb=%d\n", m, ldb);
+      rb_raise(rb_eArgError, "TRSM: Expected ldb >= max(1,N)");
+    }
+
+    // For row major, need to switch side and uplo
+    enum CBLAS_SIDE side_ = side == CblasLeft  ? CblasRight : CblasLeft;
+    enum CBLAS_UPLO uplo_ = uplo == CblasUpper ? CblasLower : CblasUpper;
+
+/*
+    cerr << "(row-major) trsm: " << (side_ == CblasLeft ? "left " : "right ")
+         << (uplo_ == CblasUpper ? "upper " : "lower ")
+         << (trans_a == CblasTrans ? "trans " : "notrans ")
+         << (diag == CblasNonUnit ? "nonunit " : "unit ")
+         << n << " " << m << " " << alpha << " a " << lda << " b " << ldb << endl;
+*/
+    trsm_nothrow<DType>(side_, uplo_, trans_a, diag, n, m, alpha, a, lda, b, ldb);
+
+  } else { // CblasColMajor
+
+    if (ldb < std::max(1,m)) {
+      fprintf(stderr, "TRSM: M=%d; got ldb=%d\n", m, ldb);
+      rb_raise(rb_eArgError, "TRSM: Expected ldb >= max(1,M)");
+    }
+/*
+    cerr << "(col-major) trsm: " << (side == CblasLeft ? "left " : "right ")
+         << (uplo == CblasUpper ? "upper " : "lower ")
+         << (trans_a == CblasTrans ? "trans " : "notrans ")
+         << (diag == CblasNonUnit ? "nonunit " : "unit ")
+         << m << " " << n << " " << alpha << " a " << lda << " b " << ldb << endl;
+*/
+    trsm_nothrow<DType>(side, uplo, trans_a, diag, m, n, alpha, a, lda, b, ldb);
+
+  }
+
+}
+
+} }  // namespace nm::math
+#endif // TRSM_H

data/ext/nmatrix/math/util.h

@@ -0,0 +1,148 @@
+/////////////////////////////////////////////////////////////////////
+// = NMatrix
+//
+// A linear algebra library for scientific computation in Ruby.
+// NMatrix is part of SciRuby.
+//
+// NMatrix was originally inspired by and derived from NArray, by
+// Masahiro Tanaka: http://narray.rubyforge.org
+//
+// == Copyright Information
+//
+// SciRuby is Copyright (c) 2010 - 2014, Ruby Science Foundation
+// NMatrix is Copyright (c) 2012 - 2014, John Woods and the Ruby Science Foundation
+//
+// Please see LICENSE.txt for additional copyright notices.
+//
+// == Contributing
+//
+// By contributing source code to SciRuby, you agree to be bound by
+// our Contributor Agreement:
+//
+// * https://github.com/SciRuby/sciruby/wiki/Contributor-Agreement
+//
+// == util.h
+//
+// Collect a few utility functions which convert ruby symbols into arguments
+// that CBLAS or LAPACK can understand: either enum's for CBLAS or char's
+// for LAPACK.
+//
+
+#ifndef UTIL_H
+#define UTIL_H
+
+/* Interprets cblas argument which could be any of false/:no_transpose, :transpose, or :complex_conjugate,
+ * into an enum recognized by cblas.
+ *
+ * Called by nm_cblas_gemm -- basically inline.
+ *
+ */
+static inline enum CBLAS_TRANSPOSE blas_transpose_sym(VALUE op) {
+  if (op == Qfalse || rb_to_id(op) == nm_rb_no_transpose) return CblasNoTrans;
+  else if (rb_to_id(op) == nm_rb_transpose) return CblasTrans;
+  else if (rb_to_id(op) == nm_rb_complex_conjugate) return CblasConjTrans;
+  else rb_raise(rb_eArgError, "Expected false, :transpose, or :complex_conjugate");
+  return CblasNoTrans;
+}
+
+/* Interprets transpose argument which could be any of false/:no_transpose, :transpose, or :complex_conjugate,
+ * into an character recognized by LAPACKE. LAPACKE uses a different system than CBLAS for this.
+ *
+ */
+static inline char lapacke_transpose_sym(VALUE op) {
+  if (op == Qfalse || rb_to_id(op) == nm_rb_no_transpose) return 'N';
+  else if (rb_to_id(op) == nm_rb_transpose) return 'T';
+  else if (rb_to_id(op) == nm_rb_complex_conjugate) return 'C';
+  else rb_raise(rb_eArgError, "Expected false, :transpose, or :complex_conjugate");
+  return 'N';
+}
+
+/*
+ * Interprets cblas argument which could be :left or :right
+ *
+ * Called by nm_cblas_trsm -- basically inline
+ */
+static inline enum CBLAS_SIDE blas_side_sym(VALUE op) {
+  ID op_id = rb_to_id(op);
+  if (op_id == nm_rb_left)  return CblasLeft;
+  if (op_id == nm_rb_right) return CblasRight;
+  rb_raise(rb_eArgError, "Expected :left or :right for side argument");
+  return CblasLeft;
+}
+
+/*
+ * Interprets cblas argument which could be :upper or :lower
+ *
+ * Called by nm_cblas_trsm -- basically inline
+ */
+static inline enum CBLAS_UPLO blas_uplo_sym(VALUE op) {
+  ID op_id = rb_to_id(op);
+  if (op_id == nm_rb_upper) return CblasUpper;
+  if (op_id == nm_rb_lower) return CblasLower;
+  rb_raise(rb_eArgError, "Expected :upper or :lower for uplo argument");
+  return CblasUpper;
+}
+
+/*
+ * Interprets argument which could be :upper or :lower for LAPACKE
+ *
+ * Called by nm_cblas_trsm -- basically inline
+ */
+static inline char lapacke_uplo_sym(VALUE op) {
+  ID op_id = rb_to_id(op);
+  if (op_id == nm_rb_upper) return 'U';
+  if (op_id == nm_rb_lower) return 'L';
+  rb_raise(rb_eArgError, "Expected :upper or :lower for uplo argument");
+  return 'U';
+}
+
+/*
+ * Interprets cblas argument which could be :unit (true) or :nonunit (false or anything other than true/:unit)
+ *
+ * Called by nm_cblas_trsm -- basically inline
+ */
+static inline enum CBLAS_DIAG blas_diag_sym(VALUE op) {
+  if (rb_to_id(op) == nm_rb_unit || op == Qtrue) return CblasUnit;
+  return CblasNonUnit;
+}
+
+/*
+ * Interprets cblas argument which could be :row or :col
+ * 
+ * This function, unlike the other ones, works for LAPACKE as well as for CBLAS/CLAPACK.
+ * Although LAPACKE calls this an int instead of a enum, the magic values are the same
+ * (101 for row-major, 102 for column-major).
+ */
+static inline enum CBLAS_ORDER blas_order_sym(VALUE op) {
+  if (rb_to_id(op) == rb_intern("row") || rb_to_id(op) == rb_intern("row_major")) return CblasRowMajor;
+  else if (rb_to_id(op) == rb_intern("col") || rb_to_id(op) == rb_intern("col_major") ||
+           rb_to_id(op) == rb_intern("column") || rb_to_id(op) == rb_intern("column_major")) return CblasColMajor;
+  rb_raise(rb_eArgError, "Expected :row or :col for order argument");
+  return CblasRowMajor;
+}
+
+/*
+ * Interprets lapack jobu and jobvt arguments, for which LAPACK needs character values A, S, O, or N.
+ *
+ * Called by lapack_gesvd -- basically inline. svd stands for singular value decomposition.
+ */
+static inline char lapack_svd_job_sym(VALUE op) {
+  if (rb_to_id(op) == rb_intern("all") || rb_to_id(op) == rb_intern("a")) return 'A';
+  else if (rb_to_id(op) == rb_intern("return") || rb_to_id(op) == rb_intern("s")) return 'S';
+  else if (rb_to_id(op) == rb_intern("overwrite") || rb_to_id(op) == rb_intern("o")) return 'O';
+  else if (rb_to_id(op) == rb_intern("none") || rb_to_id(op) == rb_intern("n")) return 'N';
+  else rb_raise(rb_eArgError, "Expected :all, :return, :overwrite, :none (or :a, :s, :o, :n, respectively)");
+  return 'a';
+}
+
+/*
+ * Interprets lapack jobvl and jobvr arguments, for which LAPACK needs character values N or V.
+ *
+ * Called by lapack_geev -- basically inline. evd stands for eigenvalue decomposition.
+ */
+static inline char lapack_evd_job_sym(VALUE op) {
+  if (op == Qfalse || op == Qnil || rb_to_id(op) == rb_intern("n")) return 'N';
+  else return 'V';
+}
+
+#endif