numo-gsl 0.1.0

data/ext/numo/gsl/gen/tmpl/c_void_f_double_x2.c

@@ -0,0 +1,18 @@
+/*
+  @overload <%=name%>(<%=args[1..2].map{|a| a[1]}.join(",")%>)
+  @param  [Float]  <%=args[1][1]%>
+  @param  [Float]  <%=args[2][1]%>
+  @return [Qnil]
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE self, VALUE v1, VALUE v2)
+{
+    <%=struct%> *w;
+
+    TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
+
+    <%=func_name%>(w, NUM2DBL(v1), NUM2DBL(v2));
+    return Qnil;
+}

data/ext/numo/gsl/gen/tmpl/c_void_f_double_x4.c

@@ -0,0 +1,20 @@
+/*
+  @overload <%=name%>(<%=args[1..4].map{|a| a[1]}.join(",")%>)
+  @param  [Float]  <%=args[1][1]%>
+  @param  [Float]  <%=args[2][1]%>
+  @param  [Float]  <%=args[3][1]%>
+  @param  [Float]  <%=args[4][1]%>
+  @return [Qnil]
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(4)%>(VALUE self, VALUE v1, VALUE v2, VALUE v3, VALUE v4)
+{
+    <%=struct%> *w;
+
+    TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
+
+    <%=func_name%>(w, NUM2DBL(v1),NUM2DBL(v2),NUM2DBL(v3),NUM2DBL(v4));
+    return Qnil;
+}

data/ext/numo/gsl/gen/tmpl/c_void_f_ulong.c

@@ -0,0 +1,17 @@
+/*
+  @overload <%=name%>(<%=args[1][1]%>)
+  @param  [Integer]  <%=args[1][1]%>
+  @return [Qnil]
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(1)%>(VALUE self, VALUE v1)
+{
+    <%=struct%> *w;
+
+    TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
+
+    <%=func_name%>(w, NUM2ULONG(v1));
+    return Qnil;
+}

data/ext/numo/gsl/gen/tmpl/c_void_f_void.c

@@ -0,0 +1,14 @@
+/*
+  @overload <%=name%>
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(0)%>(VALUE self)
+{
+    <%=struct%> *w;
+
+    TypedData_Get_Struct(self, <%=struct%>, &<%=data_type_var%>, w);
+    <%=func_name%>(w);
+    return Qnil;
+}

data/ext/numo/gsl/gen/tmpl/cast_1d_contiguous.c

@@ -0,0 +1,15 @@
+static VALUE
+cast_1d_contiguous(VALUE v1, VALUE type)
+{
+    narray_t *na;
+
+    v1 = rb_funcall(type, rb_intern("cast"), 1, v1);
+    GetNArray(v1,na);
+    if (NA_NDIM(na) != 1) {
+        rb_raise(nary_eShapeError, "argument should be 1-dimensional array");
+    }
+    if (!RTEST(na_check_contiguous(v1))) {
+        v1 = na_copy(v1);
+    }
+    return v1;
+}

data/ext/numo/gsl/gen/tmpl/cast_2d_contiguous.c

@@ -0,0 +1,15 @@
+static VALUE
+cast_2d_contiguous(VALUE v1, VALUE type)
+{
+    narray_t *na;
+
+    v1 = rb_funcall(type, rb_intern("cast"), 1, v1);
+    GetNArray(v1,na);
+    if (NA_NDIM(na) != 2) {
+        rb_raise(nary_eShapeError, "argument should be 2-dimensional array");
+    }
+    if (!RTEST(na_check_contiguous(v1))) {
+        v1 = na_copy(v1);
+    }
+    return v1;
+}

data/ext/numo/gsl/gen/tmpl/class.c

@@ -0,0 +1,26 @@
+/*
+  class definition: <%= full_class_name %>
+*/
+
+static VALUE <%=class_var%>;
+
+static void
+<%=name%>_free(void *ptr)
+{
+    <%=free_func%>(ptr);
+}
+
+static size_t
+<%=name%>_memsize(const void *ptr)
+{
+    return sizeof(<%=struct%>);
+}
+
+static const rb_data_type_t <%=name%>_data_type = {
+    "<%=full_class_name%>",
+    {NULL, <%=name%>_free, <%=name%>_memsize,},
+    0, 0, RUBY_TYPED_FREE_IMMEDIATELY|RUBY_TYPED_WB_PROTECTED
+};
+<% set data_type_var:name+"_data_type" %>
+
+<%= method_code %>

data/ext/numo/gsl/gen/tmpl/create_new_narray.c

@@ -0,0 +1,26 @@
+static VALUE
+create_new_narray(VALUE datatype, VALUE vshape)
+{
+    size_t size, *shape;
+    int j, ndim;
+
+    switch(TYPE(vshape)) {
+    case T_FIXNUM:
+    case T_BIGNUM:
+        size = NUM2SIZET(vshape);
+        ndim = 1;
+        shape = &size;
+        break;
+    case T_ARRAY:
+        ndim = RARRAY_LEN(vshape);
+        shape = ALLOCA_N(size_t,ndim);
+        size = 1;
+        for (j=0; j<ndim; j++) {
+            shape[j] = NUM2SIZET(RARRAY_AREF(vshape,j));
+        }
+        break;
+    default:
+        rb_raise(rb_eArgError,"invalid argument for shape");
+    }
+    return rb_narray_new(datatype,ndim,shape);
+}

data/ext/numo/gsl/gen/tmpl/init_class.c

@@ -0,0 +1,10 @@
+
+    /*
+      Document-class: <%= full_class_name %>
+      <%= description %>
+    */
+    {
+    <%=class_var%> = rb_define_class_under(<%=ns_var%>, "<%=class_name%>", <%=super_class%>);
+    <% @children.each do |m| %>
+    <%= m.init_def %><% end %>
+    }

data/ext/numo/gsl/gen/tmpl/init_module.c

@@ -0,0 +1,12 @@
+
+    /*
+      Document-module: <%= full_module_name %>
+      <%= description %>
+    */
+    {
+    <%  if module_var != ns_var %>
+    <%=module_var%> = rb_define_module_under(<%=ns_var%>, "<%=module_name%>");
+    <%  end %>
+    <% @children.each do |m| %>
+    <%= m.init_def %><% end %>
+    }

data/ext/numo/gsl/gen/tmpl/lib.c

@@ -0,0 +1,41 @@
+/*
+  <%= file_name %>
+  Ruby/Numo::GSL - GSL wrapper for Ruby/Numo::NArray
+
+  created on: 2017-03-11
+  Copyright (C) 2017 Masahiro Tanaka
+*/
+
+#include <ruby.h>
+#include <assert.h>
+#include "numo/narray.h"
+#include "numo/template.h"
+#include "../numo_gsl.h"
+<% include_files.each do |f| %>
+#include <<%=f%>>
+<% end %>
+
+static VALUE <%=ns_var%>;
+
+<% id_decl.each do |x| %>
+<%= x %><% end %>
+
+<% children.each do |c|%>
+<%= c.result %>
+
+<% end %>
+
+void
+Init_<%=lib_name%>(void)
+{
+    VALUE mN;
+    mN = rb_define_module("Numo");
+    <%=ns_var%> = rb_define_module_under(mN, "GSL");
+
+    <% id_assign.each do |x| %>
+    <%= x %><% end %>
+
+<% children.each do |c| %>
+<%= c.init_def %>
+<% end %>
+}

data/ext/numo/gsl/gen/tmpl/m_DFloat_Int_f_DFloat.c

@@ -0,0 +1,38 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   i;
+    char    *p1, *p2, *p3;
+    ssize_t  s1, s2, s3;
+    double   x, z;
+    int      y;
+
+    INIT_COUNTER(lp, i);
+    INIT_PTR(lp, 0, p1, s1);
+    INIT_PTR(lp, 1, p2, s2);
+    INIT_PTR(lp, 2, p3, s3);
+
+    for (; i--; ) {
+        GET_DATA_STRIDE(p1,s1,double,x);
+        z = <%=func_name%>(x,&y);
+        SET_DATA_STRIDE(p2,s2,double,z);
+        SET_DATA_STRIDE(p3,s3,int,y);
+    }
+}
+
+/*
+  @overload <%=name%>(<%=args[0][1]%>)
+  @param  [DFloat]   <%=args[0][1]%>
+  @return [DFloat]   array of [result, <%=args[1][1]%>]
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE mod, VALUE v0, VALUE v1)
+{
+    ndfunc_arg_in_t ain[1] = {{cDF,0}};
+    ndfunc_arg_out_t aout[2] = {{cDF,0},{cInt,0}};
+    ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 1,2, ain,aout};
+
+    return na_ndloop(&ndf, 1, v0);
+}

data/ext/numo/gsl/gen/tmpl/m_DFloat_f_DFloat.c

@@ -0,0 +1,35 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   i;
+    char    *p1, *p2;
+    ssize_t  s1, s2;
+    double   x, y;
+
+    INIT_COUNTER(lp, i);
+    INIT_PTR(lp, 0, p1, s1);
+    INIT_PTR(lp, 1, p2, s2);
+
+    for (; i--; ) {
+        GET_DATA_STRIDE(p1,s1,double,x);
+        y = <%=func_name%>(x);
+        SET_DATA_STRIDE(p2,s2,double,y);
+    }
+}
+
+/*
+  @overload <%=name%>(<%=args[0][1]%>)
+  @param  [DFloat]   <%=args[0][1]%>
+  @return [DFloat]   result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(1)%>(VALUE mod, VALUE v0)
+{
+    ndfunc_arg_in_t ain[1] = {{cDF,0}};
+    ndfunc_arg_out_t aout[1] = {{cDF,0}};
+    ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 1,1, ain,aout};
+
+    return na_ndloop(&ndf, 1, v0);
+}

data/ext/numo/gsl/gen/tmpl/m_DFloat_f_DFloat_Int.c

@@ -0,0 +1,39 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   i;
+    char    *p1, *p2, *p3;
+    ssize_t  s1, s2, s3;
+    double   x, z;
+    int      y;
+
+    INIT_COUNTER(lp, i);
+    INIT_PTR(lp, 0, p1, s1);
+    INIT_PTR(lp, 1, p2, s2);
+    INIT_PTR(lp, 2, p3, s3);
+
+    for (; i--; ) {
+        GET_DATA_STRIDE(p1,s1,double,x);
+        GET_DATA_STRIDE(p2,s2,int,y);
+        z = <%=func_name%>(x,y);
+        SET_DATA_STRIDE(p3,s3,double,z);
+    }
+}
+
+/*
+  @overload <%=name%>(<%=args[0][1]%>, <%=args[1][1]%>)
+  @param  [DFloat]   <%=args[0][1]%>
+  @param  [DFloat]   <%=args[1][1]%>
+  @return [DFloat]   result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE mod, VALUE v0, VALUE v1)
+{
+    ndfunc_arg_in_t ain[2] = {{cDF,0},{cInt,0}};
+    ndfunc_arg_out_t aout[1] = {{cDF,0}};
+    ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 2,1, ain,aout};
+
+    return na_ndloop(&ndf, 2, v0, v1);
+}

data/ext/numo/gsl/gen/tmpl/m_DFloat_f_DFloat_UInt.c

@@ -0,0 +1,39 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   i;
+    char    *p1, *p2, *p3;
+    ssize_t  s1, s2, s3;
+    double   x, z;
+    unsigned int y;
+
+    INIT_COUNTER(lp, i);
+    INIT_PTR(lp, 0, p1, s1);
+    INIT_PTR(lp, 1, p2, s2);
+    INIT_PTR(lp, 2, p3, s3);
+
+    for (; i--; ) {
+        GET_DATA_STRIDE(p1,s1,double,x);
+        GET_DATA_STRIDE(p2,s2,unsigned int,y);
+        z = <%=func_name%>(x,y);
+        SET_DATA_STRIDE(p3,s3,double,z);
+    }
+}
+
+/*
+  @overload <%=name%>(<%=args[0][1]%>, <%=args[1][1]%>)
+  @param  [DFloat]   <%=args[0][1]%>
+  @param  [DFloat]   <%=args[1][1]%>
+  @return [DFloat]   result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE mod, VALUE v0, VALUE v1)
+{
+    ndfunc_arg_in_t ain[2] = {{cDF,0},{cUInt,0}};
+    ndfunc_arg_out_t aout[1] = {{cDF,0}};
+    ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 2,1, ain,aout};
+
+    return na_ndloop(&ndf, 2, v0, v1);
+}

data/ext/numo/gsl/gen/tmpl/m_DFloat_f_DFloat_double.c

@@ -0,0 +1,42 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   i;
+    char    *p1, *p2;
+    ssize_t  s1, s2;
+    double   x, y;
+    double   c1;
+
+    INIT_COUNTER(lp, i);
+    INIT_PTR(lp, 0, p1, s1);
+    INIT_PTR(lp, 1, p2, s2);
+
+    c1 = *(double*)(lp->opt_ptr);
+
+    for (; i--;) {
+        GET_DATA_STRIDE(p1,s1,double,x);
+        y = <%=func_name%>(x, c1);
+        SET_DATA_STRIDE(p2,s2,double,y);
+    }
+}
+
+/*
+  @overload <%=name%>(<%=args.map{|a| a[1]}.join(",")%>)
+  @param  [DFloat]   <%=args[0][1]%>
+  @param  [Float]    <%=args[1][1]%> parameter
+  @return [DFloat]   result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(2)%>(VALUE mod, VALUE v0, VALUE v1)
+{
+    ndfunc_arg_in_t ain[1] = {{cDF,0}};
+    ndfunc_arg_out_t aout[1] = {{cDF,0}};
+    ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 1,1, ain,aout};
+    double c1;
+
+    c1 = NUM2DBL(v1);
+
+    return na_ndloop3(&ndf, &c1, 1, v0);
+}

data/ext/numo/gsl/gen/tmpl/m_DFloat_f_DFloat_double_x2.c

@@ -0,0 +1,47 @@
+static void
+iter_<%=c_func%>(na_loop_t *const lp)
+{
+    size_t   i;
+    char    *p1, *p2;
+    ssize_t  s1, s2;
+    double   x, y;
+    double  *opt;
+    double   c1, c2;
+
+    INIT_COUNTER(lp, i);
+    INIT_PTR(lp, 0, p1, s1);
+    INIT_PTR(lp, 1, p2, s2);
+
+    opt = (double*)(lp->opt_ptr);
+    c1 = opt[0];
+    c2 = opt[1];
+
+    for (; i--;) {
+        GET_DATA_STRIDE(p1,s1,double,x);
+        y = <%=func_name%>(x, c1, c2);
+        SET_DATA_STRIDE(p2,s2,double,y);
+    }
+}
+
+/*
+  @overload <%=name%>(<%=args.map{|a| a[1]}.join(",")%>)
+  @param  [DFloat]   <%=args[0][1]%>
+  @param  [Float]    <%=args[1][1]%> parameter
+  @param  [Float]    <%=args[2][1]%> parameter
+  @return [DFloat]   result
+
+  <%= description %>
+*/
+static VALUE
+<%=c_func(3)%>(VALUE mod, VALUE v0, VALUE v1, VALUE v2)
+{
+    ndfunc_arg_in_t ain[1] = {{cDF,0}};
+    ndfunc_arg_out_t aout[1] = {{cDF,0}};
+    ndfunc_t ndf = {iter_<%=c_func%>, STRIDE_LOOP|NDF_EXTRACT, 1,1, ain,aout};
+    double opt[2];
+
+    opt[0] = NUM2DBL(v1);
+    opt[1] = NUM2DBL(v2);
+
+    return na_ndloop3(&ndf, opt, 1, v0);
+}