numo-gsl 0.1.0

data/ext/numo/gsl/gen/erbpp_gsl.rb

@@ -0,0 +1,166 @@
+require_relative "func_parser"
+
+module ErbppGsl
+
+  module_function
+
+  def FM(*args,**opts)
+    FuncMatch.new(*args,**opts)
+  end
+
+  def dbl;   "double"   end
+  def dblp;  "double *" end
+  def str;   "char *"   end
+  def szt;   "size_t"   end
+  def sztp;  "size_t *" end
+  def int;   "int"      end
+  def intp;  "int *"    end
+  def long;  "long"     end
+  def uint;  "unsigned int" end
+  def ulong; /^unsigned long/ end
+  def void;  "void"     end
+  def tp; get(:struct)+" *" end
+
+  def read_eval(prefix)
+    fmt = prefix + "_%s.rb"
+    fn = fmt % `gsl-config --version`.chomp
+    if !File.exist?(fn)
+      fn = fmt % "def"
+    end
+    File.exist?(fn) ? eval(open(fn).read) : []
+  end
+
+  def read_func(prefix="func")
+    read_eval(prefix).each do |h|
+      h[:desc].gsub!(/\/\*/,"//")
+      h[:desc].gsub!(/\*\//,"")
+    end
+  end
+
+  def read_const
+    read_eval("const")
+  end
+
+  def read_enum
+    read_eval("enum")
+  end
+
+  def read_type
+    read_eval("type")
+  end
+
+  def read_func_pattern(*a)
+    read_func.each do |h|
+      no_match = true
+      a.each do |re,list|
+        if re === h[:func_name]
+          list << h
+          no_match = false
+          break
+        end
+      end
+      if no_match
+        $stderr.puts "skip "+h[:func_name]
+      end
+    end
+  end
+
+end
+
+class DefSubclassNew < DefMethod
+  def initialize(parent,tmpl,var,subtp=nil,**h)
+    super(parent, tmpl, name:"new", **h)
+    if !subtp
+      subtp = var.sub(/gsl_#{parent.name}_/,"")
+    end
+    set subtype_var: var
+    set subtype_name: subtp
+    set subtype_class: subtp.split('_').map{|x| x.capitalize}.join("")
+    set c_superclass_new: "#{parent.name}_s_new"
+  end
+
+  def c_func(narg=nil)
+    super(narg)
+    "#{@parent.name}_#{get(:subtype_name)}_s_new"
+  end
+
+  def init_def
+    if n_arg != :nodef
+      "{ VALUE c#{subtype_class} = rb_define_class_under(#{_mod_var}, \"#{subtype_class}\", #{_mod_var});
+      rb_define_singleton_method(c#{subtype_class}, \"new\", #{c_func}, #{n_arg}); }"
+    end
+  end
+end
+
+
+class DefGslClass < DefClass
+  include ErbppGsl
+
+  def lookup(h)
+    # implement in Subclass
+    raise NotImplementedError
+  end
+
+  def check_func(h)
+    fn = h[:func_name]
+    if t = lookup(h)
+      m = to_method_name(fn)
+      define_method(t, name:m, **h)
+    elsif /_free$/ =~ fn
+      set free_func: fn
+    else
+      $stderr.puts "skip #{fn}"
+    end
+  end
+
+  def to_method_name(s)
+    s.sub(/^gsl_#{name}_/,"")
+  end
+
+  def define_method(t,**h)
+    DefGslMethod.new(self, t, **h)
+  end
+end
+
+class DefGslModule < DefModule
+  include ErbppGsl
+
+  def lookup(h)
+    # implement in Subclass
+  end
+
+  def check_func(h)
+    if t = lookup(h)
+      m = to_method_name(h[:func_name])
+      define_method(t, name:m, **h)
+    else
+      $stderr.puts "skip #{h[:func_name]}"
+    end
+  end
+
+  def to_method_name(s)
+    s.sub(/^gsl_#{name}_/,"")
+  end
+
+  def define_method(t,**h)
+    DefGslModuleFunction.new(self, t, **h)
+  end
+end
+
+class DefGslModuleFunction < DefModuleFunction
+  include FuncParser
+
+  def initialize(parent,tmpl,**h)
+    super(parent,tmpl,**h)
+    parse_args(h)
+  end
+end
+
+class DefGslMethod < DefMethod
+  include FuncParser
+
+  def initialize(parent,tmpl,**h)
+    super(parent,tmpl,**h)
+    parse_args(h)
+  end
+end

data/ext/numo/gsl/gen/func_parser.rb

@@ -0,0 +1,520 @@
+require_relative "erbpp2"
+
+class Argument
+
+  def description
+    @func.param_desc[@name] ||
+      if @prop[:param]
+        "@param [#{rb_class}] #{@name}"
+      elsif @prop[:narray]
+        "@param [#{na_class}] #{@name}"
+      end
+  end
+
+  NUM2DATA =
+    {
+     "double"=>"%2 = NUM2DBL(%1)",
+     "int"=>"%2 = NUM2INT(%1)",
+     "unsigned int"=>"%2 = NUM2UINT(%1)",
+     "size_t"=>"%2 = NUM2SIZET(%1)",
+     "gsl_mode_t"=>"%2 = NUM2INT(%1)",
+     "gsl_sf_legendre_t"=>"%2 = NUM2INT(%1)",
+     "gsl_sf_mathieu_workspace *"=>'
+    if (rb_obj_is_kind_of(%1,cWorkspace)!=Qtrue) {
+        rb_raise(rb_eTypeError,"last argument must be "
+                 "Numo::GSL::Sf::MathieuWorkspace class");
+    }
+    TypedData_Get_Struct(%1, gsl_sf_mathieu_workspace, &mathieuws_type, %2)'
+    }
+
+  DATA2NUM =
+    {
+     "double"=>"DBL2NUM(%1)",
+     "int"=>"INT2NUM(%1)",
+     "unsigned int"=>"UINT2NUM(%1)",
+     "size_t"=>"SIZET2NUM(%1)",
+     "gsl_sf_result *"=>["DBL2NUM(%1.val)","DBL2NUM(%1.err)"]
+    }
+
+  TYPEMAP =
+    {
+     "double"=>"cDF",
+     "gsl_complex"=>"cDC",
+     "unsigned int"=>"cUI",
+     "int"=>"cI",
+     "size_t"=>"cSZ",
+    }
+
+  NACLASSMAP =
+    {
+     "double"=>"Numo::DFloat",
+     "gsl_complex"=>"Numo::DComplex",
+     "unsigned int"=>"Numo::UInt",
+     "int"=>"Numo::Int",
+     "size_t"=>"Numo::UInt64",
+     "gsl_sf_result *"=>["Numo::DFloat"]*2,
+     "gsl_sf_result_e10 *"=>["Numo::DFloat"]*2+["Numo::Int"],
+     "gsl_sf_mathieu_workspace *"=>"Numo::GSL::Sf::MathieuWorkspace"
+    }
+
+  RBCLASSMAP =
+    {
+     "double"=>"Float",
+     "gsl_complex"=>"Complex",
+     "unsigned int"=>"Integer",
+     "int"=>"Integer",
+     "size_t"=>"Integer",
+     "gsl_mode_t"=>"Integer",
+     "gsl_sf_result *"=>["Float"]*2
+    }
+
+  CTYPEMAP =
+    {
+     "double"=>"double",
+     #"unsigned int"=>"uint32_t",
+     #"int"=>"int32_t",
+     "unsigned int"=>"unsigned int",
+     "int"=>"int",
+     "size_t"=>"uint64_t",
+     "gsl_complex"=>"gsl_complex",
+    }
+
+  RETNAMEMAP =
+    {
+     "double"=>"Numo::DFloat",
+     "gsl_complex"=>"Numo::DComplex",
+     #"unsigned int"=>"Numo::UInt32",
+     #"int"=>"Numo::Int32",
+     "unsigned int"=>"Numo::UInt",
+     "int"=>"Numo::Int",
+     "size_t"=>"Numo::UInt64",
+     "gsl_sf_result *"=>["Numo::DFloat"]*2,
+     "gsl_sf_result_e10 *"=>["Numo::DFloat"]*2+["Numo::Int"]
+    }
+
+  def initialize(func,idx,type,name,prop)
+    @func = func
+    @idx  = idx
+    @type = type
+    @name = name
+    @prop = prop
+    @pass = @prop[:pass]
+
+    if /(.+)\*$/ =~ @type
+      @type2 = $1.strip
+    end
+
+    if @prop[:output]
+      @n_out =
+        case @type
+        when "gsl_sf_result_e10 *";  3
+        when "gsl_sf_result *";      2
+        else                         1
+        end
+    end
+
+    @ctype = CTYPEMAP[@type] || CTYPEMAP[@type2]
+  end
+
+  attr_reader :type, :name
+  attr_reader :pass, :n_out
+
+  [:input,:output,:param,:narray].each do |sym|
+    define_method(sym){@prop[sym]}
+  end
+
+  def c_var
+    "c#{@idx}"
+  end
+
+  def c_arg
+    (@pass == :pointer) ? "&"+c_var : c_var
+  end
+
+  def v_var
+    if @func.n_arg < 0
+      "v[#{@idx}]"
+    else
+      "v#{@idx}"
+    end
+  end
+
+  def ain_def
+    "{#{na_type},0}"
+  end
+
+  def aout_def
+    case @type
+    when "gsl_sf_result_e10 *"
+      return "{cDF,0},{cDF,0},{cI,0}"
+    when "gsl_sf_result *"
+      return "{cDF,0},{cDF,0}"
+    end
+    case @pass
+    when :array
+      "{#{na_type},1,shape}"
+    when :pointer,:return
+      "{#{na_type},0}"
+    else
+      raise "no aout_def for #{self.inspect}"
+    end
+  end
+
+  def na_type
+    TYPEMAP[@type] || TYPEMAP[@type2]
+  end
+
+  def na_class
+    NACLASSMAP[@type] || NACLASSMAP[@type2]
+  end
+
+  def rb_class
+    RBCLASSMAP[@type] || RBCLASSMAP[@type2]
+  end
+
+  def ret_val
+    case @type
+    when "gsl_sf_result *"
+      [@name+".val",@name+".err"]
+    when "gsl_sf_result_e10 *"
+      [@name+".val",@name+".err",@name+".e10"]
+    else
+      @name
+    end
+  end
+
+  def def_var
+    if @prop[:param]
+      "#{@type} #{c_var};"
+    elsif @prop[:input]
+      "#{@type} #{c_var};"
+    elsif @pass == :array
+      "#{@type} *#{c_var};"
+    elsif @pass == :pointer
+      "#{@type2} #{c_var};"
+    elsif @pass == :return
+      "#{@type} #{c_var};"
+    else
+      raise "no variable definition"
+    end
+  end
+
+  def def_value
+    "VALUE #{v_var};"
+  end
+
+  def set_value
+    NUM2DATA[@type].gsub(/%1/,v_var).sub(/%2/,c_var)+";"
+  end
+
+  def get_value
+    case a = DATA2NUM[@type]
+    when String
+      a.gsub(/%1/,c_var)
+    when Array
+      a.map{|s| s.gsub(/%1/,c_var)}
+    end
+  end
+
+  def store_to_array(va)
+    a = get_value
+    a = [a] unless Array === a
+    a.map{|x|"rb_ary_push(#{va},#{x});"}.join("\n"+" "*8)
+  end
+
+  def set_param(opt)
+    n = @func.n_input
+    if @type == "gsl_mode_t" && n == @idx+1
+      return <<EOL
+if (argc==#{@idx}) {
+        #{c_var} = GSL_MODE_DEFAULT;
+    } else if (argc==#{@idx+1}) {
+        #{set_value}
+    } else {
+        rb_raise(rb_eArgError,"invalid number of argument: %d for #{n-1} or #{n}",argc);
+    }
+    #{opt} = &#{c_var}; //#{@name}
+EOL
+    else
+      return "#{set_value} #{opt} = &#{c_var}; //#{@name}"
+    end
+  end
+
+  def get_param(opt)
+    "#{c_var} = *(#{@type}*)(#{opt}); //#{@name}"
+  end
+
+  def get_data(lp)
+    "#{c_var} = *(#{@ctype}*)GET_PTR(#{lp},#{cnt}); //#{@name}"
+  end
+
+  def get_ptr(lp)
+    if @pass == :array
+      "#{c_var} = (#{@ctype}*)GET_PTR(#{lp},#{cnt}); //#{@name}"
+    end
+  end
+
+  def set_data(lp)
+    if @pass != :array
+      if "gsl_sf_result_e10 *" == @type
+        "*(double*)GET_PTR(#{lp},#{cnt}) = #{c_var}.val; "+
+        "*(double*)GET_PTR(#{lp},#{cnt}) = #{c_var}.err; "+
+        "*(int*)GET_PTR(#{lp},#{cnt}) = #{c_var}.e10; //#{@name}"
+      elsif "gsl_sf_result *" == @type
+        "*(double*)GET_PTR(#{lp},#{cnt}) = #{c_var}.val; "+
+        "*(double*)GET_PTR(#{lp},#{cnt}) = #{c_var}.err; //#{@name}"
+      else
+        "*(#{@ctype}*)GET_PTR(#{lp},#{cnt}) = #{c_var}; //#{@name}"
+      end
+    end
+  end
+
+  def cnt
+    @func.counter.inc
+  end
+end
+
+#----------------------------------------------------------
+
+class FuncMatch
+
+  def initialize(*args,type:nil,name:nil)
+    @func_type = type
+    @func_name = name
+    @args = args
+  end
+
+  def ===(h)
+    if /These functions are now deprecated/m =~ h[:desc]
+      $stderr.puts "depricated: #{h[:func_name]}"
+      return false
+    end
+    if /This function is now deprecated/m =~ h[:desc]
+      $stderr.puts "depricated: #{h[:func_name]}"
+      return false
+    end
+    if @func_type
+      return false unless t = h[:func_type]
+      t.sub!(/^const\s+/,"")
+      #$stderr.puts [t,@func_type].inspect
+      return false unless @func_type === t
+    end
+    if @func_name
+      return false unless @func_name === h[:func_name]
+    end
+    if @args
+      return false unless h_args = h[:args]
+      #$stderr.puts [@args,h_args].inspect
+      return false unless @args.size == 0 || @args.size == h_args.size
+      @args.each_with_index do |a,i|
+        #$stderr.puts [a,h_args[i]].inspect
+        return false unless h_a = h_args[i]
+        case a
+        when Array
+          if a[0] # C type
+            return false unless t = h_a[0]
+            t.sub!(/^const\s+/,"")
+            return false unless a[0] === t
+          end
+          if a[1] # var name
+            return false unless a[1] === h_a[1]
+          end
+        else # String or Regexp
+          #$stderr.puts [a,h_a[0]].inspect
+          return false unless t = h_a[0]
+          t.sub!(/^const\s+/,"")
+          #$stderr.puts [a,t].inspect
+          return false unless a === t
+        end
+      end
+    end
+    #$stderr.puts self.inspect
+    true
+  end
+
+end
+
+class Counter
+  def initialize
+    @i = 0
+  end
+  def inc
+    i = @i
+    @i += 1
+    i
+  end
+end
+
+#----------------------------------------------------------
+
+module FuncParser
+
+  def self.lookup(h)
+    false
+  end
+
+  PARAM_DESC = {}
+  PARAM_NAMES = {}
+
+  def param_desc
+    self.class::PARAM_DESC
+  end
+
+  def param_names
+    self.class::PARAM_NAMES
+  end
+
+  def parse_args(h)
+    i = 0
+    @parsed_args = []
+    h[:args].each do |type,name|
+      type = type.sub(/^const\s+/,"").strip
+      if /_array$/ =~ name && /(.+)\*$/ =~ type
+        type = $1.strip
+        name = name+"[]"
+      end
+      prop = argument_property(type,name)
+      @parsed_args << Argument.new(self,i,type,name,prop)
+      i += 1
+    end
+    t = h[:func_type]
+    if t != "void"
+      prop = argument_property(t,"return")
+      @parsed_args << Argument.new(self,i,t,"return",prop)
+    end
+    #
+    @args_param = @parsed_args.select{|a|a.param}
+    @args_input = @parsed_args.select{|a|a.input}
+    @args_in = @parsed_args.select{|a|a.narray && a.input}
+    @args_out = @parsed_args.select{|a|a.narray && a.output}
+    @parsed_args.each do |a|
+      if a.pass == :array
+        @generate_array = true
+        break
+      end
+    end
+    @counter = Counter.new
+    if @args_param.any?{|a| a.type=="gsl_mode_t"}
+      set n_arg: -1
+    else
+      set n_arg: @args_param.size+@args_in.size
+    end
+  end
+
+  #attr_reader :name
+  attr_reader :args_out, :args_in, :args_param
+  attr_reader :generate_array
+  attr_reader :counter
+
+  def argument_property(type,name)
+    if name == "return"
+      {output:true, narray:true, pass: :return}
+    elsif /\[\]$/ =~ name
+      {output:true, narray:true, pass: :array}
+    elsif is_param(type,name)
+      {input:true, param:true}
+    elsif /\*$/ =~ type
+      {output:true, narray:true, pass: :pointer}
+    else
+      {input:true, narray:true}
+    end
+  end
+
+  def is_param(tp,nm)
+    #$stderr.puts "type='#{tp}' name='#{nm}'"
+    a = self.class::PARAM_NAMES[tp]
+    case a
+    when Array
+      a.include?(nm)
+    else
+      a
+    end
+  end
+
+  def n_param
+    @args_param.size
+  end
+
+  def n_in
+    @args_in.size
+  end
+
+  def n_input
+    @args_input.size
+  end
+
+  def n_out
+    n = 0
+    @args_out.each{|a| n+=a.n_out}
+    n
+  end
+
+  def def_ain
+    @args_in.map{|a| a.ain_def}.join(",")
+  end
+
+  def def_aout
+    @args_out.map{|a| a.aout_def}.join(",")
+  end
+
+  def find_name(name)
+    @parsed_args.find{|a| a.name==name}
+  end
+
+  def c_args
+    @parsed_args.select{|a| a.pass != :return}.
+      map{|a| a.c_arg}.join(",")
+  end
+
+  def recv
+    a = args_out.last
+    if a && a.pass == :return
+      a.c_var + " = "
+    else
+      ""
+    end
+  end
+
+  def method_args
+    s = @args_input.map{|a| a.name}.join(",")
+    s.sub(/,mode$/,",[mode]")
+  end
+
+  def cdef_args
+    if n_arg == -1
+      "int argc, VALUE *v, VALUE mod"
+    else
+      "VALUE mod," +
+        @args_input.map{|a|"VALUE #{a.v_var}"}.join(",")
+    end
+  end
+
+  def ndl_args
+    @args_in.map{|a| a.v_var}.join(",")
+  end
+
+  def desc_param
+    #t = (["Numo::DFloat"]*n_out).join(",")
+    t = @args_out.map{|a| a.na_class}.flatten.join(", ")
+    v = @args_out.map{|a| a.ret_val}.flatten.join(", ")
+    if n_out > 1
+      r = "@return [[#{t}]]  array of [#{v}]"
+    else
+      r = "@return [#{t}]  #{v}"
+    end
+    @args_input.map{|a| a.description}+[r]
+  end
+
+  def desc_param_scalar
+    #t = (["Numo::DFloat"]*n_out).join(",")
+    t = @args_out.map{|a| a.rb_class}.flatten.join(", ")
+    v = @args_out.map{|a| a.ret_val}.flatten.join(", ")
+    if n_out > 1
+      r = "@return [[#{t}]]  array of [#{v}]"
+    else
+      r = "@return [#{t}]  #{v}"
+    end
+    @args_input.map{|a| a.description}+[r]
+  end
+
+end