numo-gsl-alt 0.1.3

data/ext/numo/gsl/wavelet/parse_wavelet.rb

@@ -0,0 +1,47 @@
+require_relative "../gen/erbpp_gsl"
+
+class DefWavelet < DefGslClass
+
+  WAVELET_TYPES = ErbppGsl.read_type.select{|s| /gsl_wavelet_/ =~ s}
+
+  def lookup(h)
+    t = get(:struct)
+    ws = t + "_workspace *"
+    dr = t + "_direction"
+    case h
+    when FM(name:/_free$/);                     false
+    when FM(name:"gsl_wavelet_workspace_alloc"); "c_new_sizet"
+    when FM(/_type /, szt, name:/_alloc$/);     "wavelet_new"
+    when FM(tp, type:str);                      "c_str_f_void"
+    when FM(tp, type:uint);                     "c_uint_f_void"
+
+    when FM(tp,dblp,szt,szt,dr,ws);             "wavelet_transform"
+    when FM(tp,dblp,szt,szt,ws);                "wavelet_transform2"
+    when FM(tp,dblp,*[szt]*3,dr,ws);            "wavelet2d_transform"
+    when FM(tp,dblp,*[szt]*3,ws);               "wavelet_transform2"
+
+    when FM(tp);                                "c_self_f_void"
+    end
+  end
+
+  def check_func(h)
+    if t = lookup(h)
+      m = h[:func_name].sub(/^gsl_[^_]+_/,"")
+      DefMethod.new(self, t, name:m, **h)
+      def_type_new(h)
+      return true
+    end
+    $stderr.puts "skip #{h[:func_name]}"
+    false
+  end
+
+  def def_type_new(h)
+    case h[:func_name]
+    when "gsl_wavelet_alloc"
+      t = "wavelet_type_new"
+      WAVELET_TYPES.each do |st|
+        DefSubclassNew.new(self, t, st, **h)
+      end
+    end
+  end
+end

data/ext/numo/gsl/wavelet/tmpl/wavelet2d_transform.c

@@ -0,0 +1,56 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   n1, n2;
+    double  *p;
+    void   **opts;
+    <%=struct%> *w;
+    gsl_wavelet_workspace *ws;
+    int      dir;
+
+    opts = (void **)(lp->opt_ptr);
+    w    = (<%=struct%>*)(opts[0]);
+    ws   = (gsl_wavelet_workspace*)(opts[1]);
+    dir  = *(int*)(opts[2]);
+
+    p = (double*)(lp->args[0].ptr + lp->args[0].iter[0].pos);
+    n1 = lp->args[0].shape[0];
+    n2 = lp->args[0].shape[1];
+    <%=func_name%>(w, p, n2, n1, n2, dir, ws);
+}
+
+/*
+  @overload <%=name%>(<%=args[1][1]%>,<%=args[5][1]%>)
+  @param  [DFloat]    <%=args[1][1]%>
+  @param  [Integer]   <%=args[5][1]%>
+  @return [DFloat]    result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE self, VALUE v1, VALUE v2)
+{
+    ndfunc_arg_in_t ain[1] = {{OVERWRITE,2}};
+    ndfunc_t ndf = {iter_<%=c_func%>, NO_LOOP, 1,0, ain,0};
+    <%=struct%> *w;
+    gsl_wavelet_workspace *a;
+    size_t    n;
+    int       dir;
+    void     *opts[3];
+    VALUE     vws;
+
+    TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
+    opts[0] = w;
+
+    v1 = wavelet_array_check(v1, 2, &n);
+
+    vws = wavelet_workspace_s_new(cWaveletWorkspace, SIZET2NUM(n));
+    TypedData_Get_Struct(vws, gsl_wavelet_workspace, &wavelet_workspace_data_type, a);
+    opts[1] = a;
+    dir = NUM2INT(v2);
+    opts[2] = &dir;
+
+    na_ndloop3(&ndf, opts, 1, v1);
+    RB_GC_GUARD(vws);
+    return v1;
+}

data/ext/numo/gsl/wavelet/tmpl/wavelet_array_check.c

@@ -0,0 +1,56 @@
+static int
+power_of_two(size_t n)
+{
+    size_t k = 1;
+
+    while (1) {
+        if (n == k) return 1;
+        k *= 2;
+        if (n < k || k==0) break;
+    }
+    rb_raise(nary_eShapeError,"array size is not 2**n");
+    return 0;
+}
+
+static VALUE
+wavelet_array_check(VALUE v1, int nd, size_t *size)
+{
+    size_t    n1, n2;
+    VALUE     tmp;
+    narray_t *na;
+
+    GetNArray(v1,na);
+    if (na->ndim < nd) {
+        rb_raise(nary_eDimensionError,"ndim(=%d) should >= %d", na->ndim, nd);
+    }
+    switch(nd) {
+    case 2:
+        n1 = na->shape[na->ndim-2];
+        n2 = na->shape[na->ndim-1];
+        if (n1 != n2) {
+            rb_raise(nary_eShapeError,"not square 2d array");
+        }
+        power_of_two(n1);
+        power_of_two(n2);
+        *size = n2;
+        break;
+    case 1:
+        n2 = na->shape[na->ndim-1];
+        power_of_two(n2);
+        *size = n2;
+        break;
+    default:
+        rb_bug("invalid value for nd");
+    }
+    if (TEST_INPLACE(v1)) {
+        v1 = rb_funcall(cDF, rb_intern("cast"), 1, v1);
+        if (!RTEST(na_check_contiguous(v1))) {
+            v1 = na_copy(v1);
+        }
+    } else {
+        tmp = rb_narray_new(cDF, na->ndim, na->shape);
+        rb_funcall(tmp, rb_intern("store"), 1, v1);
+        v1 = tmp;
+    }
+    return v1;
+}

data/ext/numo/gsl/wavelet/tmpl/wavelet_macro.c

@@ -0,0 +1 @@
+#define gsl_wavelet2d_free gsl_wavelet_free

data/ext/numo/gsl/wavelet/tmpl/wavelet_new.c

@@ -0,0 +1,20 @@
+<%
+set name: "new"
+set singleton: true
+%>
+/*
+  :nodoc:
+ */
+static VALUE
+<%=c_func(:nodef)%>(VALUE self, const gsl_wavelet_type *t, VALUE v1)
+{
+    <%=struct%> *w;
+    size_t k;
+
+    k = NUM2SIZET(v1);
+    w = <%=func_name%>(t, k);
+    if (!w) {
+        rb_raise(rb_eNoMemError,"fail to allocate struct");
+    }
+    return TypedData_Wrap_Struct(<%=class_var%>, &<%=data_type_var%>, (void*)w);
+}

data/ext/numo/gsl/wavelet/tmpl/wavelet_transform.c

@@ -0,0 +1,55 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   n;
+    double  *p;
+    void   **opts;
+    <%=struct%> *w;
+    gsl_wavelet_workspace *ws;
+    int      dir;
+
+    opts = (void **)(lp->opt_ptr);
+    w    = (<%=struct%>*)(opts[0]);
+    ws   = (gsl_wavelet_workspace*)(opts[1]);
+    dir  = *(int*)(opts[2]);
+
+    p = (double*)(lp->args[0].ptr + lp->args[0].iter[0].pos);
+    n = lp->args[0].shape[0];
+    <%=func_name%>(w, p, 1, n, dir, ws);
+}
+
+/*
+  @overload <%=name%>(<%=args[1][1]%>,<%=args[4][1]%>)
+  @param  [DFloat]    <%=args[1][1]%>
+  @param  [Integer]   <%=args[4][1]%>
+  @return [DFloat]    result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE self, VALUE v1, VALUE v2)
+{
+    ndfunc_arg_in_t ain[1] = {{OVERWRITE,1}};
+    ndfunc_t ndf = {iter_<%=c_func%>, NO_LOOP, 1,0, ain,0};
+    <%=struct%> *w;
+    gsl_wavelet_workspace *ws;
+    size_t    n;
+    int       dir;
+    void     *opts[3];
+    VALUE     vws;
+
+    TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
+    opts[0] = w;
+
+    v1 = wavelet_array_check(v1, 1, &n);
+
+    vws = wavelet_workspace_s_new(cWaveletWorkspace, SIZET2NUM(n));
+    TypedData_Get_Struct(vws, gsl_wavelet_workspace, &wavelet_workspace_data_type, ws);
+    opts[1] = ws;
+    dir = NUM2INT(v2);
+    opts[2] = &dir;
+
+    na_ndloop3(&ndf, opts, 1, v1);
+    RB_GC_GUARD(vws);
+    return v1;
+}

data/ext/numo/gsl/wavelet/tmpl/wavelet_transform2.c

@@ -0,0 +1,22 @@
+/*
+  @overload <%=name%>(<%=args[1][1]%>)
+  @param  [DFloat]    <%=args[1][1]%>
+  @return [DFloat]    result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(1)%>(VALUE self, VALUE v1)
+{
+    int dir;
+    VALUE v2;
+
+    <% if /_forward$/ =~ name %>
+    dir = gsl_wavelet_forward;
+    <% elsif /_inverse$/ =~ name %>
+    dir = gsl_wavelet_backward;
+    <% end %>
+    v2 = INT2FIX(dir);
+
+    return <%=c_func.sub(/_[^_]+$/,"")%>(self, v1, v2);
+}

data/ext/numo/gsl/wavelet/tmpl/wavelet_type_new.c

@@ -0,0 +1,15 @@
+/*
+  @overload <%=name%>(<%=args[1][1]%>)
+  @param  [DFloat]  <%=args[1][1]%>
+
+  Generate an instance of <%=full_class_name%>::<%=subtype_class%> class,
+  a subclass of <%=full_class_name%> class with <%=subtype_var%> type.
+
+  <%= desc %>
+
+  */
+static VALUE
+<%=c_func(1)%>(VALUE self, VALUE v1)
+{
+    return <%=c_superclass_new%>(self, <%=subtype_var%>, v1);
+}

data/ext/numo/gsl/wavelet/type_def.rb

@@ -0,0 +1,6 @@
+["gsl_wavelet_daubechies",
+ "gsl_wavelet_daubechies_centered",
+ "gsl_wavelet_haar",
+ "gsl_wavelet_haar_centered",
+ "gsl_wavelet_bspline",
+ "gsl_wavelet_bspline_centered"]

data/lib/numo/gsl.rb

@@ -0,0 +1,17 @@
+require 'numo/narray'
+require 'numo/gsl/err'
+require 'numo/gsl/sys'
+require 'numo/gsl/const'
+require 'numo/gsl/poly'
+require 'numo/gsl/sf'
+require 'numo/gsl/stats'
+require 'numo/gsl/rstat'
+require 'numo/gsl/rng'
+require 'numo/gsl/ran'
+require 'numo/gsl/pdf'
+require 'numo/gsl/cdf'
+require 'numo/gsl/histogram'
+require 'numo/gsl/interp'
+require 'numo/gsl/wavelet'
+require 'numo/gsl/fit'
+require 'numo/gsl/spmatrix'

data/numo-gsl-alt.gemspec

@@ -0,0 +1,33 @@
+
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+open("ext/numo/gsl/version.h") do |f|
+  f.each_line do |l|
+    if /NUMO_GSL_VERSION "([\d.]+)"/ =~ l
+      VERSION = $1
+      break
+    end
+  end
+end
+
+Gem::Specification.new do |spec|
+  spec.name          = "numo-gsl-alt"
+  spec.version       = VERSION
+  spec.authors       = ["Masahiro TANAKA", "Matijs van Zuijlen"]
+  spec.email         = ["masa16.tanaka@gmail.com", "matijs@matijs.net"]
+  spec.description   = %q{Numo::GSL Alternative is an alternative to Numo::GSL that depends on Numo::NArray Alternative (numo-narray-alt)}
+  spec.summary       = %q{Numo::GSL bindings, alternative version}
+  spec.homepage      = "https://github.com/mvz/numo-gsl-alt"
+  spec.license       = "GPL-3"
+
+  spec.files         = `git ls-files Gemfile README.md Rakefile ext lib numo-gsl-alt.gemspec spec`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+  spec.extensions    = spec.files.grep(%r{^ext/numo/gsl/.*/extconf\.rb$})
+
+  spec.add_development_dependency "rake", "~> 13.0"
+  spec.add_runtime_dependency "numo-narray-alt", ">= 0.9.0.8"
+end