snuxoll.com

# Simple matrix class for applications that need 2-dimensonal arrays
class Matrix
  # Creates a new matrix, and may initialize it with an array
  def initialize(array=[])
    @matrix = array
  end
  # Retrieves a row from the matrix, will also set the value of a row
  # if a block is passed
  def row(_row, &block)
    if block_given?
      @matrix[_row] = block.call
    else
      @matrix[_row]
    end
  end
  # Retrieves a specific cell from the matrix
  def [](_row, _column)
    @matrix[_row][_column]
  end
  # Sets the value for a cell in the matrix
  def []=(_row, _column, value)
    @matrix[_row][_column] = value
  end
  # Appends a row to the matrix
  def <<(_row)
    @matrix << _row
  end
end
 
# Helper class to create matrixes
class M
  def self.[](*rows)
    array = []
    rows.each do |row|
      array << row
    end
    Matrix.new(array)
  end
end

matrix = 
M[
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
matrix[0, 0] => 1
matrix[4, 0] = 10