Contents

module Daru
  module Accessors
    # Internal class for wrapping ruby array
    class ArrayWrapper
      include Enumerable
      extend Forwardable

      def_delegators :@data, :slice!

      def each(&block)
        @data.each(&block)
        self
      end

      def map!(&block)
        @data.map!(&block)
        self
      end

      attr_accessor :size
      attr_reader   :data

      def initialize vector, context
        @data = vector.to_a
        @context = context

        set_size
      end

      def [] *index
        @data[*index]
      end

      def []= index, value
        @data[index] = value
        set_size
      end

      def == other
        @data == other
      end

      def delete_at index
        @data.delete_at index
        set_size
      end

      def index *args, &block
        @data.index(*args, &block)
      end

      def << element
        @data << element
        set_size
      end

      def uniq
        @data.uniq
      end

      def to_a
        @data
      end

      def dup
        ArrayWrapper.new @data.dup, @context
      end

      def compact
        @data - @context.missing_values
      end

      def mean
        values_to_sum = compact
        return nil if values_to_sum.empty?
        sum = values_to_sum.inject :+
        sum.quo(values_to_sum.size).to_f
      end

      def product
        compact.inject :*
      end

      def max
        compact.max
      end

      def min
        compact.min
      end

      def sum
        compact.inject :+
      end

      private

      def set_size
        @size = @data.size
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
daru-0.1.3.1	lib/daru/accessors/array_wrapper.rb
daru-0.1.3	lib/daru/accessors/array_wrapper.rb