mdarray 0.4.0-java

data/lib/mdarray/ruby_generic_functions.rb

@@ -0,0 +1,37 @@
+
+##########################################################################################
+#
+##########################################################################################
+
+module GenericFunctions
+  extend FunctionCreation
+  extend RubyFunctions
+
+  @null = Proc.new { }
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  make_unary_op("dim_set", :set_block, 
+                ["ruby_dim_set", "RubyFunctions", @null, "*", "*", "void"])
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  make_binary_op("fill", :fill, 
+                 ["ruby_fill", "RubyFunctions", @null, "*", "*", "void"])
+
+
+end # GenericFunctions
+
+##########################################################################################
+#
+##########################################################################################
+
+class MDArray
+
+  include GenericFunctions
+
+end # NumericalMDArray

data/lib/mdarray/ruby_math.rb

@@ -0,0 +1,57 @@
+##########################################################################################
+#
+##########################################################################################
+
+module RubyMath
+  extend FunctionCreation
+  extend RubyFunctions
+  
+  @acos = Proc.new { |val| Math.acos(val) }
+  @acosh = Proc.new { |val| Math.acosh(val) }
+  @asin = Proc.new { |val| Math.asin(val) }
+  @asinh = Proc.new { |val| Math.asinh(val) }
+  @atan = Proc.new { |val| Math.atan(val) }
+  @atan2 = Proc.new { |val| Math.atan2(val) }
+  @atanh = Proc.new { |val| Math.atanh(val) }
+  @cbrt = Proc.new { |val| Math.cbrt(val) }
+  @cos = Proc.new { |val| Math.cos(val) }
+  @cosh = Proc.new { |val| Math.cosh(val) }
+  @erf = Proc.new { |val| Math.erf(val) }
+  @erfc = Proc.new { |val| Math.erfc(val) }
+  @exp = Proc.new { |val| Math.exp(val) }
+  @gamma = Proc.new { |val| Math.gamma(val) }
+  @hypot = Proc.new { |val| Math.hypotn(val) }
+  @ldexp = Proc.new { |val| Math.ldexp(val) }
+  @log = Proc.new { |val| Math.log(val) }
+  @log10 = Proc.new { |val| Math.log10(val) }
+  @log2 = Proc.new { |val| Math.log2(val) }
+  @sin = Proc.new { |val| Math.sin(val) }
+  @sinh = Proc.new { |val| Math.sinh(val) }
+  @sqrt = Proc.new { |val| Math.sqrt(val) }
+  @tan = Proc.new { |val| Math.tan(val) }
+  @tanh = Proc.new { |val| Math.tanh(val) }
+  @neg = Proc.new { |val| -1 * val }
+
+  @unary_methods = [:acos, :acosh, :asin, :asinh, :atan, :atan2, 
+                    :atanh, :cbrt, :cos, :erf, :exp, :gamma, :hypot, :ldexp, 
+                    :log, :log10, :log2, :sin, :sinh, :sqrt, :tan, :tanh, :neg]
+
+  @unary_methods.each do |method|
+    make_unary_operators(method.to_s,
+                         ruby_unary_function("#{method.to_s}_ruby", 
+                                             instance_variable_get("@#{method.to_s}")))
+  end
+
+  alias :-@ :neg
+
+end # RubyMath
+
+##########################################################################################
+#
+##########################################################################################
+
+class NumericalMDArray
+
+  include RubyMath
+  
+end # NumericalMDArray

data/lib/mdarray/ruby_numeric_functions.rb

@@ -0,0 +1,187 @@
+
+##########################################################################################
+#
+##########################################################################################
+
+module NumericFunctions
+  include_package "ucar.ma2.MAMath"
+
+  extend FunctionCreation
+  extend RubyFunctions
+
+  @add = Proc.new { |val1, val2| val1 + val2 }
+  @sub = Proc.new { |val1, val2| val1 - val2 }
+  @mul = Proc.new { |val1, val2| val1 * val2 }
+  @div = Proc.new { |val1, val2| val1 / val2 }
+  @power = Proc.new { |val1, val2| val1 ** val2 }
+  @abs = Proc.new { |val| val.abs }
+  @ceil = Proc.new { |val| val.ceil }
+  @floor = Proc.new { |val| val.floor }
+  @truncate = Proc.new { |val| val.truncate }
+  @is_zero = Proc.new { |val| val.zero }
+  @square = Proc.new { |val| val ** 2 }
+  @cube = Proc.new { |val| val ** 3 }
+  @fourth = Proc.new { |val| val ** 4 }
+  
+  @min = Proc.new { |val1, val2| val1 < val2 ? val1 : val2 }
+  @max = Proc.new { |val1, val2| val1 > val2 ? val1 : val2 }
+  
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def zero?(val)
+    @is_zero.call(val)
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def fast_add(other_val)
+    arr = Java::UcarMa2::MAMath.add(@nc_array, other_val.nc_array)
+  end
+  
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+
+  @binary_methods = [:add, :sub, :mul, :div, :power, :min, :max]
+  
+  @unary_methods = [:abs, :ceil, :floor, :truncate, :is_zero, :square, 
+                    :cube, :fourth]
+  
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  @binary_methods.each do |method|
+    make_binary_operators(method.to_s,
+                          ruby_binary_function("#{method.to_s}_ruby", 
+                                               instance_variable_get("@#{method.to_s}")))
+  end
+  
+  @unary_methods.each do |method|
+    make_unary_operators(method.to_s,
+                         ruby_unary_function("#{method.to_s}_ruby", 
+                                             instance_variable_get("@#{method.to_s}")))
+  end
+
+  #======================================================================================
+  # Arithmetic
+  #======================================================================================
+
+  alias :+ :add
+  alias :- :sub
+  alias :* :mul
+  alias :/ :div
+  alias :** :power
+  alias :addAssign :add!
+  alias :subAssign :sub!
+  alias :mulAssign :mul!
+  alias :powAssign :power!
+
+end # NumericFunctions
+
+##########################################################################################
+#
+##########################################################################################
+
+module ComparisonOperators
+  extend FunctionCreation
+  extend RubyFunctions
+
+  @ge = Proc.new { |val1, val2| val1 >= val2 ? true : false}
+  @gt = Proc.new { |val1, val2| val1 > val2 ? true : false}
+  @le = Proc.new { |val1, val2| val1 <= val2 ? true : false}
+  @lt = Proc.new { |val1, val2| val1 < val2 ? true : false}
+  @eq = Proc.new { |val1, val2| val1 == val2 ? true : false}
+
+  @binary_methods = [:ge, :gt, :le, :lt, :eq]
+
+  @binary_methods.each do |method|
+    make_comparison_op(method.to_s,
+                       ruby_binary_function("#{method.to_s}_ruby", 
+                                            instance_variable_get("@#{method.to_s}")))
+  end
+
+  alias :>= :ge
+  alias :> :gt
+  alias :<= :le
+  alias :< :lt
+  alias :== :eq
+
+end # ComparisonOperators
+
+##########################################################################################
+#
+##########################################################################################
+
+module BitwiseOperators
+  extend FunctionCreation
+  extend RubyFunctions
+
+  @binary_and = Proc.new { |val1, val2| val1 & val2 }
+  @binary_or = Proc.new { |val1, val2| val1 | val2 }
+  @binary_xor = Proc.new { |val1, val2| val1 ^ val2 }
+  @binary_left_shift = Proc.new { |val1, val2| val1 << val2 }
+  @binary_right_shift = Proc.new { |val1, val2| val1 >> val2 }
+
+  @binary_ones_complement = Proc.new { |val| ~val }
+  
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+
+  @binary_methods = [:binary_and, :binary_or, :binary_xor, :binary_left_shift,
+                     :binary_right_shift]
+  
+  @unary_methods = [:binary_ones_complement]
+  
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  @binary_methods.each do |method|
+    make_binary_operators(method.to_s,
+                          ruby_binary_function("#{method.to_s}_ruby", 
+                                               instance_variable_get("@#{method.to_s}")))
+  end
+  
+  @unary_methods.each do |method|
+    make_unary_operators(method.to_s,
+                         ruby_unary_function("#{method.to_s}_ruby", 
+                                             instance_variable_get("@#{method.to_s}")))
+  end
+
+  alias :& :binary_and
+  alias :| :binary_or
+  alias :^ :binary_xor
+  alias :~ :binary_ones_complement
+  alias :<< :binary_left_shift
+  alias :>> :binary_right_shift
+
+end # BitwiseOperators
+
+##########################################################################################
+#
+##########################################################################################
+
+class NumericalMDArray
+
+  include NumericFunctions
+  include ComparisonOperators
+
+end # NumericalMDArray
+
+##########################################################################################
+#
+##########################################################################################
+
+class LongMDArray
+
+  include BitwiseOperators
+
+end # LongMDArray

data/lib/mdarray/ruby_operators.rb

@@ -0,0 +1,201 @@
+# -*- coding: utf-8 -*-
+
+##########################################################################################
+# Copyright © 2013 Rodrigo Botafogo. All Rights Reserved. Permission to use, copy, modify, 
+# and distribute this software and its documentation, without fee and without a signed 
+# licensing agreement, is hereby granted, provided that the above copyright notice, this 
+# paragraph and the following two paragraphs appear in all copies, modifications, and 
+# distributions.
+#
+# IN NO EVENT SHALL RODRIGO BOTAFOGO BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT, SPECIAL, 
+# INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF 
+# THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF RODRIGO BOTAFOGO HAS BEEN ADVISED OF THE 
+# POSSIBILITY OF SUCH DAMAGE.
+#
+# RODRIGO BOTAFOGO SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 
+# THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE 
+# SOFTWARE AND ACCOMPANYING DOCUMENTATION, IF ANY, PROVIDED HEREUNDER IS PROVIDED "AS IS". 
+# RODRIGO BOTAFOGO HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, 
+# OR MODIFICATIONS.
+##########################################################################################
+
+##########################################################################################
+#
+##########################################################################################
+
+class RubyBinaryOperator < BinaryOperator
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def initialize(name, exec_type, force_type = nil, pre_condition = nil, 
+                 post_condition = nil)
+    super(name, 2, exec_type, force_type, pre_condition, post_condition)
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def default(*args)
+
+    get_args(*args) do |op1_iterator, op2_iterator, shape, *other_args|
+      result = MDArray.build(@type, shape)
+      res_iterator = result.get_iterator_fast
+      if (@coerced)
+        while (res_iterator.has_next?)
+          res_iterator.set_next(@do_func.call(op2_iterator.get_next, 
+                                              op1_iterator.get_next))
+        end
+      else
+        while (res_iterator.has_next?)
+          res_iterator.set_next(@do_func.call(op1_iterator.get_next, 
+                                              op2_iterator.get_next))
+        end
+      end
+      return result
+    end
+
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def fill(*args)
+
+    get_args(*args) do |op1_iterator, op2_iterator, shape, *other_args|
+      while (op1_iterator.has_next?)
+        op1_iterator.set_next(op2_iterator.get_next)
+      end
+      return self
+    end
+
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+  
+  def in_place(*args)
+
+    get_args(*args) do |op1_iterator, op2_iterator, shape, *other_args|
+      while (op1_iterator.has_next?)
+        op1_iterator.set_current(@do_func.call(op1_iterator.get_next, 
+                                               op2_iterator.get_next))
+      end
+      return self
+    end
+
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def reduce(*args)
+
+    get_args(*args) do |op1_iterator, op2_iterator, shape, *other_args|
+      result = @pre_condition_result
+      while (op1_iterator.has_next?)
+        result = @do_func.call(result, op1_iterator.get_next, op2_iterator.get_next)
+      end
+      return result
+    end
+    
+  end
+
+end # RubyBinaryOperator
+
+##########################################################################################
+#
+##########################################################################################
+
+class RubyUnaryOperator < UnaryOperator
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def initialize(name, exec_type, force_type = nil, pre_condition = nil, 
+                 post_condition = nil)
+    super(name, 1, exec_type, force_type, pre_condition, post_condition)
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def default(*args)
+
+    get_args(*args) do |op_iterator, shape, *other_args|
+      result = MDArray.build(@type, shape)
+      res_iterator = result.get_iterator_fast
+      while (res_iterator.has_next?)
+        res_iterator.set_next(@do_func.call(op_iterator.get_next))
+      end
+      return result
+    end
+
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def set_block(*args)
+
+    get_args(*args) do |op_iterator, shape, *other_args|
+      block = other_args[0]
+      while (op_iterator.has_next?)
+        op_iterator.next
+        if (shape.size < 8)
+          op_iterator.set_current(block.call(*op_iterator.get_current_counter))
+        else
+          op_iterator.set_current(block.call(op_iterator.get_current_counter))
+        end
+      end if block
+    end
+
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def in_place(*args)
+
+    get_args(*args) do |op_iterator, *other_args|
+      while (op_iterator.has_next?)
+        op_iterator.set_current(@do_func.call(op_iterator.get_next))
+      end
+      return self
+    end
+
+  end
+
+  #---------------------------------------------------------------------------------------
+  #
+  #---------------------------------------------------------------------------------------
+
+  def reduce(*args)
+
+    get_args(*args) do |op_iterator, shape, *other_args|
+      result = @pre_condition_result
+      while (op_iterator.has_next?)
+        result = @do_func.call(result, op_iterator.get_next)
+      end
+      return result
+    end
+
+  end
+
+end # RubyUnaryOperator
+
+##########################################################################################
+#
+##########################################################################################
+
+MDArray.binary_operator = RubyBinaryOperator
+MDArray.unary_operator = RubyUnaryOperator