Contents

#
# TODO: Reimplement Matrix so it's not so awful, and give it a new name (like Grid).
#       (Goal: To be as nice as numpy)
#
# Features:
# * Better constructors
# * Pretty-printer support
# * .rows, .cols, .each_{row,col}, .neighbourhood
# * Vec class, Point class
#

require 'matrix'

#
# Matrix extensions
#
class Matrix

  alias_class_method :zeros, :zero
  alias_class_method :zeroes, :zero

  #
  # Create a matrix of the specified size full of random numbers.
  #
  def self.random(*dims)
    build(*dims) { rand }
  end


  #
  # The size of the matrix, returned as `[rows, columns]`.
  #
  def size
    [row_size, column_size]
  end

  #
  # Print the matrix to the STDOUT.
  #
  def print(header=true, separator=" ")
    max_width = map {|num| num.to_s.size }.max

    case first
    when Integer
      justify = :rjust
    when Float
      justify = :ljust
    else
      raise "Unknown matrix element type: #{first.class}"
    end

    # print it!
    puts "#{size.join("x")} matrix:" if header

    rows.each do |row|
      puts "  " + row.map { |n| n.to_s.send(justify, max_width) }.join(separator)
    end

    puts
  end

  alias_method :draw, :print

  #
  # Allow mathematical operations (*, /, +, -) with a scalar (integer or float) on the right side.
  #
  # eg: Matrix.zero(3) + 5
  #
  %i[* / + -].each do |op|
    class_eval %{
      def #{op}(other)
        case other
        when Numeric
          map { |e| e #{op} other }
        else
          super(other)
        end
      end
    }
  end

  public :[]=

end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
epitools-0.5.116	lib/epitools/core_ext/matrix.rb
epitools-0.5.115	lib/epitools/core_ext/matrix.rb
epitools-0.5.114	lib/epitools/core_ext/matrix.rb
epitools-0.5.113	lib/epitools/core_ext/matrix.rb
epitools-0.5.112	lib/epitools/core_ext/matrix.rb