numo-linalg-alt 0.4.1 → 0.6.0

data/ext/numo/linalg/lapack/geev.c

@@ -6,22 +6,6 @@ struct _geev_option {
   char jobvr;
 };
 
-char _get_jobvl(VALUE val) {
-  const char jobvl = NUM2CHR(val);
-  if (jobvl != 'N' && jobvl != 'V') {
-    rb_raise(rb_eArgError, "jobvl must be 'N' or 'V'");
-  }
-  return jobvl;
-}
-
-char _get_jobvr(VALUE val) {
-  const char jobvr = NUM2CHR(val);
-  if (jobvr != 'N' && jobvr != 'V') {
-    rb_raise(rb_eArgError, "jobvr must be 'N' or 'V'");
-  }
-  return jobvr;
-}
-
 #define DEF_LINALG_FUNC(tDType, tNAryClass, fLapackFunc)                                       \
   static void _iter_##fLapackFunc(na_loop_t* const lp) {                                       \
     tDType* a = (tDType*)NDL_PTR(lp, 0);                                                       \
@@ -35,8 +19,8 @@ char _get_jobvr(VALUE val) {
       (lapack_int)(opt->matrix_layout == LAPACK_ROW_MAJOR ? NDL_SHAPE(lp, 0)[0]                \
                                                           : NDL_SHAPE(lp, 0)[1]);              \
     const lapack_int lda = n;                                                                  \
-    const lapack_int ldvl = (opt->jobvl == 'N') ? 1 : n;                                       \
-    const lapack_int ldvr = (opt->jobvr == 'N') ? 1 : n;                                       \
+    const lapack_int ldvl = n;                                                                 \
+    const lapack_int ldvr = n;                                                                 \
     lapack_int i = LAPACKE_##fLapackFunc(                                                      \
       opt->matrix_layout, opt->jobvl, opt->jobvr, n, a, lda, wr, wi, vl, ldvl, vr, ldvr        \
     );                                                                                         \
@@ -52,8 +36,8 @@ char _get_jobvr(VALUE val) {
     rb_get_kwargs(kw_args, kw_table, 0, 3, kw_values);                                         \
     const int matrix_layout =                                                                  \
       kw_values[0] != Qundef ? get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;             \
-    const char jobvl = kw_values[1] != Qundef ? _get_jobvl(kw_values[1]) : 'V';                \
-    const char jobvr = kw_values[2] != Qundef ? _get_jobvr(kw_values[2]) : 'V';                \
+    const char jobvl = kw_values[1] != Qundef ? get_job(kw_values[1], "jobvl") : 'V';          \
+    const char jobvr = kw_values[2] != Qundef ? get_job(kw_values[2], "jobvr") : 'V';          \
                                                                                                \
     if (CLASS_OF(a_vnary) != tNAryClass) {                                                     \
       a_vnary = rb_funcall(tNAryClass, rb_intern("cast"), 1, a_vnary);                         \
@@ -101,8 +85,8 @@ char _get_jobvr(VALUE val) {
       (lapack_int)(opt->matrix_layout == LAPACK_ROW_MAJOR ? NDL_SHAPE(lp, 0)[0]                \
                                                           : NDL_SHAPE(lp, 0)[1]);              \
     const lapack_int lda = n;                                                                  \
-    const lapack_int ldvl = (opt->jobvl == 'N') ? 1 : n;                                       \
-    const lapack_int ldvr = (opt->jobvr == 'N') ? 1 : n;                                       \
+    const lapack_int ldvl = n;                                                                 \
+    const lapack_int ldvr = n;                                                                 \
     lapack_int i = LAPACKE_##fLapackFunc(                                                      \
       opt->matrix_layout, opt->jobvl, opt->jobvr, n, a, lda, w, vl, ldvl, vr, ldvr             \
     );                                                                                         \
@@ -118,8 +102,8 @@ char _get_jobvr(VALUE val) {
     rb_get_kwargs(kw_args, kw_table, 0, 3, kw_values);                                         \
     const int matrix_layout =                                                                  \
       kw_values[0] != Qundef ? get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;             \
-    const char jobvl = kw_values[1] != Qundef ? _get_jobvl(kw_values[1]) : 'V';                \
-    const char jobvr = kw_values[2] != Qundef ? _get_jobvr(kw_values[2]) : 'V';                \
+    const char jobvl = kw_values[1] != Qundef ? get_job(kw_values[1], "jobvl") : 'V';          \
+    const char jobvr = kw_values[2] != Qundef ? get_job(kw_values[2], "jobvr") : 'V';          \
                                                                                                \
     if (CLASS_OF(a_vnary) != tNAryClass) {                                                     \
       a_vnary = rb_funcall(tNAryClass, rb_intern("cast"), 1, a_vnary);                         \

data/ext/numo/linalg/lapack/geev.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_geev(VALUE mLapack);
 

data/ext/numo/linalg/lapack/gehrd.c

@@ -0,0 +1,77 @@
+#include "gehrd.h"
+
+struct _gehrd_option {
+  int matrix_layout;
+  int ilo;
+  int ihi;
+};
+
+#define DEF_LINALG_FUNC(tDType, tNAryClass, fLapackFunc)                                       \
+  static void _iter_##fLapackFunc(na_loop_t* const lp) {                                       \
+    tDType* a = (tDType*)NDL_PTR(lp, 0);                                                       \
+    tDType* tau = (tDType*)NDL_PTR(lp, 1);                                                     \
+    int* info = (int*)NDL_PTR(lp, 2);                                                          \
+    struct _gehrd_option* opt = (struct _gehrd_option*)(lp->opt_ptr);                          \
+    const lapack_int ilo = opt->ilo;                                                           \
+    const lapack_int ihi = opt->ihi;                                                           \
+    const lapack_int n =                                                                       \
+      (lapack_int)(opt->matrix_layout == LAPACK_ROW_MAJOR ? NDL_SHAPE(lp, 0)[0]                \
+                                                          : NDL_SHAPE(lp, 0)[1]);              \
+    const lapack_int lda = n;                                                                  \
+    lapack_int i = LAPACKE_##fLapackFunc(opt->matrix_layout, n, ilo, ihi, a, lda, tau);        \
+    *info = (int)i;                                                                            \
+  }                                                                                            \
+                                                                                               \
+  static VALUE _linalg_lapack_##fLapackFunc(int argc, VALUE* argv, VALUE self) {               \
+    VALUE a_vnary = Qnil;                                                                      \
+    VALUE kw_args = Qnil;                                                                      \
+    rb_scan_args(argc, argv, "1:", &a_vnary, &kw_args);                                        \
+    ID kw_table[3] = { rb_intern("ilo"), rb_intern("ihi"), rb_intern("order") };               \
+    VALUE kw_values[3] = { Qundef, Qundef, Qundef };                                           \
+    rb_get_kwargs(kw_args, kw_table, 2, 1, kw_values);                                         \
+    const int ilo = NUM2INT(kw_values[0]);                                                     \
+    const int ihi = NUM2INT(kw_values[1]);                                                     \
+    const int matrix_layout =                                                                  \
+      kw_values[2] != Qundef ? get_matrix_layout(kw_values[2]) : LAPACK_ROW_MAJOR;             \
+                                                                                               \
+    if (CLASS_OF(a_vnary) != tNAryClass) {                                                     \
+      a_vnary = rb_funcall(tNAryClass, rb_intern("cast"), 1, a_vnary);                         \
+    }                                                                                          \
+    if (!RTEST(nary_check_contiguous(a_vnary))) {                                              \
+      a_vnary = nary_dup(a_vnary);                                                             \
+    }                                                                                          \
+                                                                                               \
+    narray_t* a_nary = NULL;                                                                   \
+    GetNArray(a_vnary, a_nary);                                                                \
+    const int n_dims = NA_NDIM(a_nary);                                                        \
+    if (n_dims != 2) {                                                                         \
+      rb_raise(rb_eArgError, "input array a must be 2-dimensional");                           \
+      return Qnil;                                                                             \
+    }                                                                                          \
+                                                                                               \
+    size_t n = matrix_layout == LAPACK_ROW_MAJOR ? NA_SHAPE(a_nary)[0] : NA_SHAPE(a_nary)[1];  \
+    size_t shape_tau[1] = { n - 1 };                                                           \
+    ndfunc_arg_in_t ain[1] = { { OVERWRITE, 2 } };                                             \
+    ndfunc_arg_out_t aout[2] = { { tNAryClass, 1, shape_tau }, { numo_cInt32, 0 } };           \
+    ndfunc_t ndf = { _iter_##fLapackFunc, NO_LOOP | NDF_EXTRACT, 1, 2, ain, aout };            \
+    struct _gehrd_option opt = { matrix_layout, ilo, ihi };                                    \
+    VALUE res = na_ndloop3(&ndf, &opt, 1, a_vnary);                                            \
+    VALUE ret = rb_ary_concat(rb_ary_new3(1, a_vnary), res);                                   \
+                                                                                               \
+    RB_GC_GUARD(a_vnary);                                                                      \
+    return ret;                                                                                \
+  }
+
+DEF_LINALG_FUNC(double, numo_cDFloat, dgehrd)
+DEF_LINALG_FUNC(float, numo_cSFloat, sgehrd)
+DEF_LINALG_FUNC(lapack_complex_double, numo_cDComplex, zgehrd)
+DEF_LINALG_FUNC(lapack_complex_float, numo_cSComplex, cgehrd)
+
+#undef DEF_LINALG_FUNC
+
+void define_linalg_lapack_gehrd(VALUE mLapack) {
+  rb_define_module_function(mLapack, "dgehrd", _linalg_lapack_dgehrd, -1);
+  rb_define_module_function(mLapack, "sgehrd", _linalg_lapack_sgehrd, -1);
+  rb_define_module_function(mLapack, "zgehrd", _linalg_lapack_zgehrd, -1);
+  rb_define_module_function(mLapack, "cgehrd", _linalg_lapack_cgehrd, -1);
+}

data/ext/numo/linalg/lapack/gehrd.h

@@ -0,0 +1,15 @@
+#ifndef NUMO_LINALG_ALT_LAPACK_GEHRD_H
+#define NUMO_LINALG_ALT_LAPACK_GEHRD_H 1
+
+#include <lapacke.h>
+
+#include <ruby.h>
+
+#include <numo/narray.h>
+#include <numo/template.h>
+
+#include "lapack_util.h"
+
+void define_linalg_lapack_gehrd(VALUE mLapack);
+
+#endif // NUMO_LINALG_ALT_LAPACK_GEHRD_H

data/ext/numo/linalg/lapack/gelsd.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_gelsd(VALUE mLapack);
 

data/ext/numo/linalg/lapack/geqrf.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_geqrf(VALUE mLapack);
 

data/ext/numo/linalg/lapack/gerqf.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_gerqf(VALUE mLapack);
 

data/ext/numo/linalg/lapack/gesdd.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_gesdd(VALUE mLapack);
 

data/ext/numo/linalg/lapack/gesv.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_gesv(VALUE mLapack);
 

data/ext/numo/linalg/lapack/gesvd.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_gesvd(VALUE mLapack);
 

data/ext/numo/linalg/lapack/getrf.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_getrf(VALUE mLapack);
 

data/ext/numo/linalg/lapack/getri.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_getri(VALUE mLapack);
 

data/ext/numo/linalg/lapack/getrs.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_getrs(VALUE mLapack);
 

data/ext/numo/linalg/lapack/gges.c

@@ -49,8 +49,8 @@
     rb_get_kwargs(kw_args, kw_table, 0, 4, kw_values);                                         \
     const int matrix_layout =                                                                  \
       kw_values[0] != Qundef ? get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;             \
-    const char jobvsl = kw_values[1] != Qundef ? get_jobvs(kw_values[1]) : 'V';                \
-    const char jobvsr = kw_values[2] != Qundef ? get_jobvs(kw_values[2]) : 'V';                \
+    const char jobvsl = kw_values[1] != Qundef ? get_job(kw_values[1], "jobvsl") : 'V';        \
+    const char jobvsr = kw_values[2] != Qundef ? get_job(kw_values[2], "jobvsr") : 'V';        \
                                                                                                \
     if (CLASS_OF(a_vnary) != tNAryClass) {                                                     \
       a_vnary = rb_funcall(tNAryClass, rb_intern("cast"), 1, a_vnary);                         \
@@ -141,8 +141,8 @@
     rb_get_kwargs(kw_args, kw_table, 0, 4, kw_values);                                         \
     const int matrix_layout =                                                                  \
       kw_values[0] != Qundef ? get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;             \
-    const char jobvsl = kw_values[1] != Qundef ? get_jobvs(kw_values[1]) : 'V';                \
-    const char jobvsr = kw_values[2] != Qundef ? get_jobvs(kw_values[1]) : 'V';                \
+    const char jobvsl = kw_values[1] != Qundef ? get_job(kw_values[1], "jobvsl") : 'V';        \
+    const char jobvsr = kw_values[2] != Qundef ? get_job(kw_values[2], "jobvsr") : 'V';        \
                                                                                                \
     if (CLASS_OF(a_vnary) != tNAryClass) {                                                     \
       a_vnary = rb_funcall(tNAryClass, rb_intern("cast"), 1, a_vnary);                         \

data/ext/numo/linalg/lapack/gges.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_gges(VALUE mLapack);
 

data/ext/numo/linalg/lapack/heev.c

@@ -26,7 +26,7 @@ struct _heev_option {
     ID kw_table[3] = { rb_intern("jobz"), rb_intern("uplo"), rb_intern("order") };             \
     VALUE kw_values[3] = { Qundef, Qundef, Qundef };                                           \
     rb_get_kwargs(kw_args, kw_table, 0, 3, kw_values);                                         \
-    const char jobz = kw_values[0] != Qundef ? get_jobz(kw_values[0]) : 'V';                   \
+    const char jobz = kw_values[0] != Qundef ? get_job(kw_values[0], "jobz") : 'V';            \
     const char uplo = kw_values[1] != Qundef ? get_uplo(kw_values[1]) : 'U';                   \
     const int matrix_layout =                                                                  \
       kw_values[2] != Qundef ? get_matrix_layout(kw_values[2]) : LAPACK_ROW_MAJOR;             \

data/ext/numo/linalg/lapack/heev.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_heev(VALUE mLapack);
 

data/ext/numo/linalg/lapack/heevd.c

@@ -26,7 +26,7 @@ struct _heevd_option {
     ID kw_table[3] = { rb_intern("jobz"), rb_intern("uplo"), rb_intern("order") };             \
     VALUE kw_values[3] = { Qundef, Qundef, Qundef };                                           \
     rb_get_kwargs(kw_args, kw_table, 0, 3, kw_values);                                         \
-    const char jobz = kw_values[0] != Qundef ? get_jobz(kw_values[0]) : 'V';                   \
+    const char jobz = kw_values[0] != Qundef ? get_job(kw_values[0], "jobz") : 'V';            \
     const char uplo = kw_values[1] != Qundef ? get_uplo(kw_values[1]) : 'U';                   \
     const int matrix_layout =                                                                  \
       kw_values[2] != Qundef ? get_matrix_layout(kw_values[2]) : LAPACK_ROW_MAJOR;             \

data/ext/numo/linalg/lapack/heevd.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_heevd(VALUE mLapack);
 

data/ext/numo/linalg/lapack/heevr.c

@@ -40,7 +40,7 @@
                        rb_intern("iu"),   rb_intern("order") };                                \
     VALUE kw_values[8] = { Qundef, Qundef, Qundef, Qundef, Qundef, Qundef, Qundef, Qundef };   \
     rb_get_kwargs(kw_args, kw_table, 0, 8, kw_values);                                         \
-    const char jobz = kw_values[0] != Qundef ? get_jobz(kw_values[0]) : 'V';                   \
+    const char jobz = kw_values[0] != Qundef ? get_job(kw_values[0], "jobz") : 'V';            \
     const char range = kw_values[1] != Qundef ? get_range(kw_values[1]) : 'A';                 \
     const char uplo = kw_values[2] != Qundef ? get_uplo(kw_values[2]) : 'U';                   \
     const tRtDType vl = kw_values[3] != Qundef ? NUM2DBL(kw_values[3]) : 0.0;                  \

data/ext/numo/linalg/lapack/heevr.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_heevr(VALUE mLapack);
 

data/ext/numo/linalg/lapack/hegv.c

@@ -33,7 +33,7 @@ struct _hegv_option {
     VALUE kw_values[4] = { Qundef, Qundef, Qundef, Qundef };                                   \
     rb_get_kwargs(kw_args, kw_table, 0, 4, kw_values);                                         \
     const lapack_int itype = kw_values[0] != Qundef ? get_itype(kw_values[0]) : 1;             \
-    const char jobz = kw_values[1] != Qundef ? get_jobz(kw_values[1]) : 'V';                   \
+    const char jobz = kw_values[1] != Qundef ? get_job(kw_values[1], "jobz") : 'V';            \
     const char uplo = kw_values[2] != Qundef ? get_uplo(kw_values[2]) : 'U';                   \
     const int matrix_layout =                                                                  \
       kw_values[3] != Qundef ? get_matrix_layout(kw_values[3]) : LAPACK_ROW_MAJOR;             \

data/ext/numo/linalg/lapack/hegv.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_hegv(VALUE mLapack);
 

data/ext/numo/linalg/lapack/hegvd.c

@@ -33,7 +33,7 @@ struct _hegvd_option {
     VALUE kw_values[4] = { Qundef, Qundef, Qundef, Qundef };                                   \
     rb_get_kwargs(kw_args, kw_table, 0, 4, kw_values);                                         \
     const lapack_int itype = kw_values[0] != Qundef ? get_itype(kw_values[0]) : 1;             \
-    const char jobz = kw_values[1] != Qundef ? get_jobz(kw_values[1]) : 'V';                   \
+    const char jobz = kw_values[1] != Qundef ? get_job(kw_values[1], "jobz") : 'V';            \
     const char uplo = kw_values[2] != Qundef ? get_uplo(kw_values[2]) : 'U';                   \
     const int matrix_layout =                                                                  \
       kw_values[3] != Qundef ? get_matrix_layout(kw_values[3]) : LAPACK_ROW_MAJOR;             \

data/ext/numo/linalg/lapack/hegvd.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_hegvd(VALUE mLapack);
 

data/ext/numo/linalg/lapack/hegvx.c

@@ -46,7 +46,7 @@
                            Qundef, Qundef, Qundef, Qundef };                                   \
     rb_get_kwargs(kw_args, kw_table, 0, 9, kw_values);                                         \
     const lapack_int itype = kw_values[0] != Qundef ? get_itype(kw_values[0]) : 1;             \
-    const char jobz = kw_values[1] != Qundef ? get_jobz(kw_values[1]) : 'V';                   \
+    const char jobz = kw_values[1] != Qundef ? get_job(kw_values[1], "jobz") : 'V';            \
     const char range = kw_values[2] != Qundef ? get_range(kw_values[2]) : 'A';                 \
     const char uplo = kw_values[3] != Qundef ? get_uplo(kw_values[3]) : 'U';                   \
     const tRtDType vl = kw_values[4] != Qundef ? NUM2DBL(kw_values[4]) : 0.0;                  \

data/ext/numo/linalg/lapack/hegvx.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_hegvx(VALUE mLapack);
 

data/ext/numo/linalg/lapack/hetrf.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_hetrf(VALUE mLapack);
 

data/ext/numo/linalg/lapack/lange.c

@@ -5,10 +5,10 @@ struct _lange_option {
   char norm;
 };
 
-#define DEF_LINALG_FUNC(tDType, tNAryClass, fLapackFunc)                                       \
+#define DEF_LINALG_FUNC(tDType, tRtDType, tNAryClass, tRtNAryClass, fLapackFunc)               \
   static void _iter_##fLapackFunc(na_loop_t* const lp) {                                       \
     tDType* a = (tDType*)NDL_PTR(lp, 0);                                                       \
-    tDType* d = (tDType*)NDL_PTR(lp, 1);                                                       \
+    tRtDType* d = (tRtDType*)NDL_PTR(lp, 1);                                                   \
     struct _lange_option* opt = (struct _lange_option*)(lp->opt_ptr);                          \
     const lapack_int m = (lapack_int)NDL_SHAPE(lp, 0)[0];                                      \
     const lapack_int n = (lapack_int)NDL_SHAPE(lp, 0)[1];                                      \
@@ -42,8 +42,7 @@ struct _lange_option {
     }                                                                                          \
                                                                                                \
     ndfunc_arg_in_t ain[1] = { { tNAryClass, 2 } };                                            \
-    size_t shape_out[1] = { 1 };                                                               \
-    ndfunc_arg_out_t aout[1] = { { tNAryClass, 0, shape_out } };                               \
+    ndfunc_arg_out_t aout[1] = { { tRtNAryClass, 0 } };                                        \
     ndfunc_t ndf = { _iter_##fLapackFunc, NO_LOOP | NDF_EXTRACT, 1, 1, ain, aout };            \
     struct _lange_option opt = { matrix_layout, norm };                                        \
     VALUE ret = na_ndloop3(&ndf, &opt, 1, a_vnary);                                            \
@@ -52,10 +51,10 @@ struct _lange_option {
     return ret;                                                                                \
   }
 
-DEF_LINALG_FUNC(double, numo_cDFloat, dlange)
-DEF_LINALG_FUNC(float, numo_cSFloat, slange)
-DEF_LINALG_FUNC(lapack_complex_double, numo_cDComplex, zlange)
-DEF_LINALG_FUNC(lapack_complex_float, numo_cSComplex, clange)
+DEF_LINALG_FUNC(double, double, numo_cDFloat, numo_cDFloat, dlange)
+DEF_LINALG_FUNC(float, float, numo_cSFloat, numo_cSFloat, slange)
+DEF_LINALG_FUNC(lapack_complex_double, double, numo_cDComplex, numo_cDFloat, zlange)
+DEF_LINALG_FUNC(lapack_complex_float, float, numo_cSComplex, numo_cSFloat, clange)
 
 #undef DEF_LINALG_FUNC
 

data/ext/numo/linalg/lapack/lange.h

@@ -8,7 +8,7 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include "../util.h"
+#include "lapack_util.h"
 
 void define_linalg_lapack_lange(VALUE mLapack);
 

data/ext/numo/linalg/lapack/lapack_util.c

@@ -0,0 +1,57 @@
+#include "lapack_util.h"
+
+lapack_int get_itype(VALUE val) {
+  const lapack_int itype = NUM2INT(val);
+
+  if (itype != 1 && itype != 2 && itype != 3) {
+    rb_raise(rb_eArgError, "itype must be 1, 2 or 3");
+  }
+
+  return itype;
+}
+
+char get_job(VALUE val, const char* param_name) {
+  const char job = NUM2CHR(val);
+
+  if (job != 'N' && job != 'V') {
+    rb_raise(rb_eArgError, "%s must be 'N' or 'V'", param_name);
+  }
+
+  return job;
+}
+
+char get_range(VALUE val) {
+  const char range = NUM2CHR(val);
+
+  if (range != 'A' && range != 'V' && range != 'I') {
+    rb_raise(rb_eArgError, "range must be 'A', 'V' or 'I'");
+  }
+
+  return range;
+}
+
+char get_uplo(VALUE val) {
+  const char uplo = NUM2CHR(val);
+
+  if (uplo != 'U' && uplo != 'L') {
+    rb_raise(rb_eArgError, "uplo must be 'U' or 'L'");
+  }
+
+  return uplo;
+}
+
+int get_matrix_layout(VALUE val) {
+  const char option = NUM2CHR(val);
+
+  switch (option) {
+  case 'r':
+  case 'R':
+    break;
+  case 'c':
+  case 'C':
+    rb_warn("Numo::Linalg does not support column major.");
+    break;
+  }
+
+  return LAPACK_ROW_MAJOR;
+}

data/ext/numo/linalg/lapack/lapack_util.h

@@ -0,0 +1,27 @@
+#ifndef NUMO_LINALG_ALT_LAPACK_UTIL_H
+#define NUMO_LINALG_ALT_LAPACK_UTIL_H 1
+
+#include <ruby.h>
+
+#include <lapacke.h>
+
+#ifndef lapack_complex_float_real
+#define lapack_complex_float_real(z) ((z).real)
+#endif
+#ifndef lapack_complex_float_imag
+#define lapack_complex_float_imag(z) ((z).imag)
+#endif
+#ifndef lapack_complex_double_real
+#define lapack_complex_double_real(z) ((z).real)
+#endif
+#ifndef lapack_complex_double_imag
+#define lapack_complex_double_imag(z) ((z).imag)
+#endif
+
+lapack_int get_itype(VALUE val);
+char get_job(VALUE val, const char* param_name);
+char get_range(VALUE val);
+char get_uplo(VALUE val);
+int get_matrix_layout(VALUE val);
+
+#endif // NUMO_LINALG_ALT_LAPACK_UTIL_H