nmatrix-gemv 0.0.3

data/ext/nmatrix_gemv/ruby_constants.cpp

@@ -0,0 +1,153 @@
+/////////////////////////////////////////////////////////////////////
+// = 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
+//
+// == ruby_symbols.cpp
+//
+// Ruby symbols used throught the NMatrix project.
+
+/*
+ * Standard Includes
+ */
+
+#include <ruby.h>
+
+/*
+ * Global Variables
+ */
+
+ID	nm_rb_dtype,
+    nm_rb_stype,
+
+    nm_rb_capacity,
+    nm_rb_default,
+
+    nm_rb_real,
+		nm_rb_imag,
+
+		nm_rb_numer,
+		nm_rb_denom,
+
+		nm_rb_complex_conjugate,
+		nm_rb_transpose,
+		nm_rb_no_transpose,
+    nm_rb_left,
+    nm_rb_right,
+    nm_rb_upper,
+    nm_rb_lower,
+    nm_rb_unit,
+    nm_rb_nonunit,
+
+		nm_rb_dense,
+		nm_rb_list,
+		nm_rb_yale,
+
+		nm_rb_row,
+		nm_rb_column,
+		nm_rb_add,
+		nm_rb_sub,
+		nm_rb_mul,
+		nm_rb_div,
+    nm_rb_both,
+    nm_rb_none,
+
+		nm_rb_negate,
+
+		nm_rb_percent,
+		nm_rb_gt,
+		nm_rb_lt,
+		nm_rb_eql,
+		nm_rb_neql,
+		nm_rb_gte,
+		nm_rb_lte,
+
+		nm_rb_hash;
+
+VALUE cNMatrix,
+      cNMatrix_IO,
+      cNMatrix_IO_Matlab,
+			cNVector,
+			cNMatrix_YaleFunctions,
+			cNMatrix_BLAS,
+			cNMatrix_LAPACK,
+
+      cNMatrix_GC_holder,
+			
+			nm_eDataTypeError,
+      nm_eConvergenceError,
+			nm_eStorageTypeError;
+
+/*
+ * Functions
+ */
+
+void nm_init_ruby_constants(void) {
+  nm_rb_dtype             = rb_intern("dtype");
+  nm_rb_stype             = rb_intern("stype");
+
+  nm_rb_capacity          = rb_intern("capacity");
+  nm_rb_default           = rb_intern("default");
+
+	nm_rb_real							= rb_intern("real");
+	nm_rb_imag							= rb_intern("imag");
+
+	nm_rb_numer							= rb_intern("numerator");
+	nm_rb_denom							= rb_intern("denominator");
+
+	nm_rb_complex_conjugate	= rb_intern("complex_conjugate");
+	nm_rb_transpose					= rb_intern("transpose");
+	nm_rb_no_transpose			= rb_intern("no_transpose");
+
+	nm_rb_dense 						= rb_intern("dense");
+	nm_rb_list							= rb_intern("list");
+	nm_rb_yale							= rb_intern("yale");
+
+	nm_rb_add								= rb_intern("+");
+	nm_rb_sub								= rb_intern("-");
+	nm_rb_mul								= rb_intern("*");
+	nm_rb_div								= rb_intern("/");
+
+	nm_rb_negate            = rb_intern("-@");
+
+	nm_rb_percent						= rb_intern("%");
+	nm_rb_gt								= rb_intern(">");
+	nm_rb_lt								= rb_intern("<");
+	nm_rb_eql								= rb_intern("==");
+	nm_rb_neql							= rb_intern("!=");
+	nm_rb_gte								= rb_intern(">=");
+	nm_rb_lte								= rb_intern("<=");
+
+	nm_rb_left              = rb_intern("left");
+	nm_rb_right             = rb_intern("right");
+	nm_rb_upper             = rb_intern("upper");
+	nm_rb_lower             = rb_intern("lower");
+	nm_rb_unit              = rb_intern("unit");
+	nm_rb_nonunit           = rb_intern("nonunit");
+	nm_rb_hash              = rb_intern("hash");
+
+	nm_rb_column            = rb_intern("column");
+	nm_rb_row               = rb_intern("row");
+
+  //Added by Ryan
+  nm_rb_both              = rb_intern("both");
+  nm_rb_none              = rb_intern("none");
+}

data/ext/nmatrix_gemv/ruby_constants.h

@@ -0,0 +1,107 @@
+/////////////////////////////////////////////////////////////////////
+// = 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
+//
+// == data.h
+//
+// Header file for dealing with data types.
+
+#ifndef RUBY_CONSTANTS_H
+#define RUBY_CONSTANTS_H
+
+/*
+ * Standard Includes
+ */
+
+#include <ruby.h>
+
+/*
+ * Data
+ */
+
+extern ID nm_rb_dtype,
+          nm_rb_stype,
+
+          nm_rb_capacity,
+          nm_rb_default,
+
+          nm_rb_real,
+					nm_rb_imag,
+		
+					nm_rb_numer,
+					nm_rb_denom,
+		
+					nm_rb_complex_conjugate,
+					nm_rb_transpose,
+					nm_rb_no_transpose,
+					nm_rb_left,
+					nm_rb_right,
+					nm_rb_upper,
+					nm_rb_lower,
+					nm_rb_unit,
+					nm_rb_nonunit,
+		
+					nm_rb_dense,
+					nm_rb_list,
+					nm_rb_yale,
+
+          nm_rb_row,
+          nm_rb_column,
+		
+					nm_rb_add,
+					nm_rb_sub,
+					nm_rb_mul,
+					nm_rb_div,
+
+					nm_rb_negate,
+		
+					nm_rb_percent,
+					nm_rb_gt,
+					nm_rb_lt,
+					nm_rb_eql,
+					nm_rb_neql,
+					nm_rb_gte,
+					nm_rb_lte,
+
+					nm_rb_hash;
+
+extern VALUE	cNMatrix,
+              cNMatrix_IO,
+              cNMatrix_IO_Matlab,
+							cNVector,
+							cNMatrix_YaleFunctions,
+							cNMatrix_BLAS,
+							cNMatrix_LAPACK,
+
+							cNMatrix_GC_holder,
+			
+							nm_eDataTypeError,
+              nm_eConvergenceError,
+							nm_eStorageTypeError;
+
+/*
+ * Functions
+ */
+
+void nm_init_ruby_constants(void);
+
+#endif // RUBY_CONSTANTS_H

data/ext/nmatrix_gemv/ruby_nmatrix.c

@@ -0,0 +1,84 @@
+/////////////////////////////////////////////////////////////////////
+// = 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
+//
+// == ruby_nmatrix.c
+//
+// Ruby-facing NMatrix C functions. Not compiled directly -- included
+// into nmatrix.cpp.
+//
+
+/*
+#ifdef BENCHMARK
+static double get_time(void);
+#endif
+*/
+
+///////////////////
+// Ruby Bindings //
+///////////////////
+
+void Init_nmatrix_gemv() {
+
+
+	///////////////////////
+	// Class Definitions //
+	///////////////////////
+
+	cNMatrix = rb_define_class("NMatrix", rb_cObject);
+
+	// Special exceptions
+
+	/*
+	 * Exception raised when there's a problem with data.
+	 */
+	nm_eDataTypeError    = rb_define_class("DataTypeError", rb_eStandardError);
+
+	/*
+	 * Exception raised when something goes wrong with the storage of a matrix.
+	 */
+	nm_eStorageTypeError = rb_define_class("StorageTypeError", rb_eStandardError);
+
+  /*
+   * Class that holds values in use by the C code.
+   */
+  cNMatrix_GC_holder = rb_define_class("NMGCHolder", rb_cObject);
+
+	///////////////////////
+	// Symbol Generation //
+	///////////////////////
+
+	nm_init_ruby_constants();
+
+	/////////////////
+	// BLAS module //
+	/////////////////
+
+	nm_math_init_blas();
+
+	/////////////////////////////////////////////////
+	// Force compilation of necessary constructors //
+	/////////////////////////////////////////////////
+	nm_init_data();
+}
+
+

data/ext/nmatrix_gemv/ttable_helper.rb

@@ -0,0 +1,122 @@
+#!/usr/bin/ruby
+
+# A helper file for generating and maintaining template tables.
+
+
+DTYPES = [
+          :uint8_t,
+          :int8_t,
+          :int16_t,
+          :int32_t,
+          :int64_t,
+          :float32_t,
+          :float64_t,
+          :'nm::Complex64',
+          :'nm::Complex128',
+          :'nm::Rational32',
+          :'nm::Rational64',
+          :'nm::Rational128',
+          :'nm::RubyObject'
+         ]
+
+def nullify(disabled = []) #:nodoc:
+  DTYPES.map { |t| if disabled.include?(t) then :NULL else t end }
+end
+
+ITYPES = [
+          :uint8_t,
+          :uint16_t,
+          :uint32_t,
+          :uint64_t
+         ]
+
+EWOPS = [
+         :'nm::EW_ADD',
+         :'nm::EW_SUB',
+         :'nm::EW_MUL',
+         :'nm::EW_DIV',
+         :'nm::EW_POW',
+         :'nm::EW_MOD',
+         :'nm::EW_EQEQ',
+         :'nm::EW_NEQ',
+         :'nm::EW_LT',
+         :'nm::EW_GT',
+         :'nm::EW_LEQ',
+         :'nm::EW_GEQ'
+        ]
+
+LR_ALLOWED = {
+  :uint8_t	 		=> DTYPES,
+  :int8_t				=> DTYPES,
+  :int16_t			=> DTYPES,
+  :int32_t			=> DTYPES,
+  :int64_t			=> DTYPES,
+  :float32_t		=> DTYPES,
+  :float64_t		=> DTYPES,
+  :'nm::Complex64'		=> DTYPES,
+  :'nm::Complex128'		=> DTYPES,
+  :'nm::Rational32'		=> nullify([:float32_t, :float64_t, :'nm::Complex64', :'nm::Complex128']),
+  :'nm::Rational64'		=> nullify([:float32_t, :float64_t, :'nm::Complex64', :'nm::Complex128']),
+  :'nm::Rational128'	=> nullify([:float32_t, :float64_t, :'nm::Complex64', :'nm::Complex128']),
+  :'nm::RubyObject'		=> DTYPES
+}
+
+lines =
+  case ARGV[0]
+  when 'OPLR'
+    '{' +
+      EWOPS.map do |op|
+
+    '{' +
+      DTYPES.map do |l_dtype|
+
+      '{' +
+        LR_ALLOWED[l_dtype].map do |r_dtype|
+        if r_dtype == :NULL
+          'NULL'
+        else
+          "fun<#{op}, #{l_dtype}, #{r_dtype}>"
+        end
+      end.join(', ') +
+        '}'
+
+    end.join(",\n") +
+      '}'
+
+  end.join(",\n") +
+      '}'
+
+  when 'OPID'
+    '{' +
+      EWOPS.map do |op|
+    '{' +
+      ITYPES.map do |itype|
+      '{' +
+        DTYPES.map do |dtype|
+
+        if dtype == :NULL
+          'NULL'
+        else
+          "fun<#{op}, #{itype}, #{dtype}>"
+        end
+
+      end.join(",") +
+        '}'
+    end.join(",\\\n") +
+      '}'
+  end.join(",\\\n") +
+      '}'
+
+  when 'LR'
+    '{' + DTYPES.map do |l_dtype|
+      '{' + LR_ALLOWED[l_dtype].map do |r_dtype|
+        if r_dtype == :NULL
+          'NULL'
+        else
+          "fun<#{l_dtype}, #{r_dtype}>"
+        end
+      end.join(', ') + '}'
+    end.join(",\n") + '}'
+  end
+
+puts lines