pnmatrix 1.2.4

data/lib/nmatrix/mkmf.rb

@@ -0,0 +1,100 @@
+require "mkmf"
+
+if RUBY_VERSION < '1.9'
+  raise NotImplementedError, "Sorry, you need at least Ruby 1.9!"
+end
+
+# Function derived from NArray's extconf.rb.
+def create_conf_h(file) #:nodoc:
+  print "creating #{file}\n"
+  File.open(file, 'w') do |hfile|
+    header_guard = file.upcase.sub(/\s|\./, '_')
+
+    hfile.puts "#ifndef #{header_guard}"
+    hfile.puts "#define #{header_guard}"
+    hfile.puts
+
+    # FIXME: Find a better way to do this:
+    hfile.puts "#define RUBY_2 1" if RUBY_VERSION >= '2.0'
+
+    for line in $defs
+      line =~ /^-D(.*)/
+      hfile.printf "#define %s 1\n", $1
+    end
+
+    hfile.puts
+    hfile.puts "#endif"
+  end
+end
+
+def find_newer_gplusplus #:nodoc:
+  print "checking for apparent GNU g++ binary with C++0x/C++11 support... "
+  [9,8,7,6,5,4,3].each do |minor|
+    ver = "4.#{minor}"
+    gpp = "g++-#{ver}"
+    result = `which #{gpp}`
+    next if result.empty?
+    CONFIG['CXX'] = gpp
+    puts ver
+    return CONFIG['CXX']
+  end
+  false
+end
+
+def gplusplus_version
+  cxxvar = proc { |n| `#{CONFIG['CXX']} -E -dM - <#{File::NULL} | grep #{n}`.chomp.split(' ')[2] }
+  major = cxxvar.call('__GNUC__')
+  minor = cxxvar.call('__GNUC_MINOR__')
+  patch = cxxvar.call('__GNUC_PATCHLEVEL__')
+
+  raise("unable to determine g++ version (match to get version was nil)") if major.nil? || minor.nil? || patch.nil?
+
+  "#{major}.#{minor}.#{patch}"
+end
+
+
+if /cygwin|mingw/ =~ RUBY_PLATFORM
+  CONFIG["DLDFLAGS"] << " --output-lib libnmatrix.a"
+end
+
+# Fix compiler pairing
+if CONFIG['CC'] == 'clang' && CONFIG['CXX'] != 'clang++'
+  puts "WARNING: CONFIG['CXX'] is not 'clang++' even though CONFIG['CC'] is 'clang'.",
+       "WARNING: Force to use clang++ together with clang."
+
+  CONFIG['CXX'] = 'clang++'
+end
+
+if CONFIG['CXX'] == 'clang++'
+  $CXX_STANDARD = 'c++11'
+else
+  version = gplusplus_version
+  if version < '4.3.0' && CONFIG['CXX'] == 'g++'  # see if we can find a newer G++, unless it's been overridden by user
+    if !find_newer_gplusplus
+      raise("You need a version of g++ which supports -std=c++0x or -std=c++11. If you're on a Mac and using Homebrew, we recommend using mac-brew-gcc.sh to install a more recent g++.")
+    end
+    version = gplusplus_version
+  end
+
+  if version < '4.7.0'
+    $CXX_STANDARD = 'c++0x'
+  else
+    $CXX_STANDARD = 'c++11'
+  end
+  puts "using C++ standard... #{$CXX_STANDARD}"
+  puts "g++ reports version... " + `#{CONFIG['CXX']} --version|head -n 1|cut -f 3 -d " "`
+end
+
+# For release, these next two should both be changed to -O3.
+$CFLAGS += " -O3 "
+#$CFLAGS += " -static -O0 -g "
+$CXXFLAGS += " -O3 -std=#{$CXX_STANDARD} " #-fmax-errors=10 -save-temps
+#$CXXFLAGS += " -static -O0 -g -std=#{$CXX_STANDARD} "
+
+if CONFIG.has_key?('warnflags')
+  CONFIG['warnflags'].gsub!('-Wshorten-64-to-32', '') # doesn't work except in Mac-patched gcc (4.2)
+  CONFIG['warnflags'].gsub!('-Wdeclaration-after-statement', '')
+  CONFIG['warnflags'].gsub!('-Wimplicit-function-declaration', '')
+end
+  
+have_func("rb_array_const_ptr", "ruby.h")

data/lib/nmatrix/monkeys.rb

@@ -0,0 +1,137 @@
+#--
+# = 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
+#
+# == monkeys.rb
+#
+# Ruby core extensions for NMatrix.
+#++
+
+#######################
+# Classes and Modules #
+#######################
+
+class Array
+  # Convert a Ruby Array to an NMatrix.
+  #
+  # You must provide a shape for the matrix as the first argument.
+  #
+  # == Arguments:
+  # <tt>shape</tt> :: Array describing matrix dimensions (or Integer for square).
+  #   If not provided, will be intuited through #shape.
+  # <tt>dtype</tt> :: Override data type (e.g., to store a Float as :float32
+  #   instead of :float64) -- optional.
+  # <tt>stype</tt> :: Optional storage type (defaults to :dense)
+  def to_nm(shape = nil, dtype = nil, stype = :dense)
+    elements = self.dup
+
+    guess_dtype = ->(type) {
+      case type
+      when Integer  then :int64
+      when Float    then :float64
+      when Complex  then :complex128
+      end
+    }
+
+    guess_shape = lambda { |shapey; shape|
+      # Get the size of the current dimension
+      shape = [shapey.size]
+      shape << shapey.map {|s|
+        if s.respond_to?(:size) && s.respond_to?(:map)
+          guess_shape.call(s)
+        else
+          nil
+        end
+      }
+      if shape.last.any? {|s| (s != shape.last.first) || s.nil?}
+        shape.pop
+      end
+      if (shape.first != shape.last) && shape.last.all? {|s| s == shape.last.first}
+        shape[-1] = shape.last.first
+      end
+      shape.flatten
+    }
+
+    unless shape
+      shape = guess_shape.call(elements)
+      elements.flatten!(shape.size - 1)
+      if elements.flatten != elements
+        dtype = :object
+      else
+        dtype ||= guess_dtype[elements[0]]
+      end
+    end
+
+    dtype ||= guess_dtype[self[0]]
+
+    matrix = NMatrix.new(:dense, shape, elements, dtype)
+
+    if stype != :dense then matrix.cast(stype, dtype) else matrix end
+  end
+end
+
+class Object #:nodoc:
+  def returning(value)
+    yield(value)
+    value
+  end
+end
+
+
+module Math #:nodoc:
+  class << self
+    NMatrix::NMMath::METHODS_ARITY_2.each do |meth|
+      define_method "nm_#{meth}" do |arg0, arg1|
+        if arg0.is_a? NMatrix then
+          arg0.send(meth, arg1)
+        elsif arg1.is_a? NMatrix then
+          arg1.send(meth, arg0, true)
+        else
+          self.send("old_#{meth}".to_sym, arg0, arg1)
+        end
+      end
+      alias_method "old_#{meth}".to_sym, meth
+      alias_method meth, "nm_#{meth}".to_sym
+    end
+  end
+end
+
+class String
+  def underscore
+    self.gsub(/::/, '/').
+    gsub(/([A-Z]+)([A-Z][a-z])/,'\1_\2').
+    gsub(/([a-z\d])([A-Z])/,'\1_\2').
+    tr("-", "_").
+    downcase
+  end
+end
+
+# Since `autoload` will most likely be deprecated (due to multi-threading concerns),
+# we'll use `const_missing`. See: https://www.ruby-forum.com/topic/3036681 for more info.
+module AutoloadPatch #:nodoc
+  def const_missing(name)
+    file = name.to_s.underscore
+    require "nmatrix/io/#{file}"
+    klass = const_get(name)
+    return klass if klass
+  end
+end