Contents

require 'narray'

class ObjectTable::Column < NArray

  def self.make(value)
    value = case value
    when self
      value
    when NArray
      if value.rank <= 0
        self.new(value.typecode, 0)
      else
        cast(value)
      end
    when Range
      to_na(value.to_a)
    when Array
      to_na(value)
    else
      raise ArgumentError.new("Expected NArray or Array, got #{value.class}")
    end
    value
  end

  def slice(*)
    self.class.make super
  end

  def [](*a)
    result = super
    result.is_a?(NArray) ? self.class.make(result) : result
  end

  def []=(*args)
    if (args[-1].is_a?(Array) or args[-1].is_a?(NArray)) and args[-1].empty? and self.empty?
      return args[-1]
    end

    super
  end

  def to_object
    to_type('object')
  end

  def to_bool
    map{|i| i ? 1 : 0}.to_type('byte')
  end

  def uniq
    self.class.make to_a.uniq
  end

  def coerce_rev(other, operator)
    other.send(operator, NArray.refer(self))
  end

  def method_missing(*args)
    collect{|x| x.send(*args)}
  end

#   def collect(*)
#     self.class.make super, name
#   end

  def _refer(value)
    value.is_a?(NArray) ? NArray.refer(value) : value
  end

  %w{ + - * / }.each do |op|
    define_method(op) do |other|
      #self.class.make super(_refer(other)), name
      super(_refer(other))
    end
  end

  %w{ xor or and <= >= le ge < > gt lt % ** ne eq & | ^ to_type }.each do |op|
    define_method(op) do |other|
      self.class.make super(other)
    end
  end

#   %w{ not abs -@ ~ }.each do |op|
#     define_method(op) do
#       self.class.make super()
# #     end
#   end

  def stack(*others)
    columns = [self] + others
    new_rows = columns.map{|x| x.shape[-1]}.reduce(:+)
    new_col = self.class.new(typecode, *shape[0...-1], new_rows)

    padding = [nil] * (rank - 1)

    row = 0
    columns.each do |col|
      new_col[*padding, row ... (row + col.shape[-1])] = col
      row += col.shape[-1]
    end

    new_col
  end

end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
object_table-0.2.8	lib/object_table/column.rb
object_table-0.2.4	lib/object_table/column.rb
object_table-0.2.3	lib/object_table/column.rb