Contents

require 'facets/functor'

module Enumerable

  EWISE = {}

  # Returns an elementwise Functor designed to make R-like
  # elementwise operations possible. This is very much like
  # the #every method, but it treats array argument specially.
  #
  #   [1,2].ewise + 3          #=> [4,5]
  #   [1,2].ewise + [4,5]      #=> [5,7]
  #   [1,2].ewise + [[4,5],3]  #=> [[5,7],[4,5]
  #
  # Special thanks to Martin DeMello for helping to develop this.

  def ewise(count=1)
    EWISE[[self,count]] ||= Functor.new do |op,*args|
      if args.empty?
        r = self
        count.times do
          r = r.collect{ |a| a.send(op) }
        end
        r
      else
        r = args.collect do |arg|
          if Array === arg #arg.kind_of?(Enumerable)
            x = self
            count.times do
              ln = (arg.length > length ? length : arg.length )
              x = x.slice(0...ln).zip(arg[0...ln]).collect{ |a,b| a.send(op,b) }
              #slice(0...ln).zip(arg[0...1n]).collect{ |a,b| b ? a.send(op,b) : nil }
            end
            x
          else
            x = self
            count.times do
              x = x.collect{ |a| a.send(op,arg) }
            end
            x
          end
        end
        r.flatten! if args.length == 1
        r
      end
    end
  end

  # Long-term for #ewise.
  #
  #   a = [1,2]
  #   a.elementwise + 3          #=> [4,5]
  #   a.elementwise + [4,5]      #=> [5,7]
  #   a.elementwise + [[4,5],3]  #=> [[5,7],[4,5]
  #
  alias_method :elementwise, :ewise

  # Operator equivalent of #elementwise.
  #
  # This was deprecated along with it's file "op_tilde.rb".
  #
  #def ~@
  #  ewise
  #end

end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
facets-2.8.4	lib/core/facets/enumerable/ewise.rb
facets-2.8.3	lib/core/facets/enumerable/ewise.rb
facets-2.8.2	lib/core/facets/enumerable/ewise.rb