arturoherrero/powercore

class Array
  # Drops n elements from the array.
  old_drop = instance_method(:drop)
  define_method(:drop) do |n|
    n >= 0 ? old_drop.bind(self).call(n) : self[0..n-1]
  end

  # Returns the array without the indexes specified.
  def except(*indexes)
    self.reject.with_index { |_, index| indexes.include?(index) }
  end

  # Extracts the nested value specified by the sequence of indexes.
  # If the key can’t be found, there are two options:
  # with no other argument, it will raise an IndexError exception;
  # if the optional code block is specified,
  # it will be executed and its result will be returned.
  def fetch_dig(*indexes, &block)
    indexes.inject(self) { |array, index| array.fetch(index) }
  rescue IndexError
    block&.call || raise
  end

  # Returns the head of the array.
  def head
    self.first
  end

  # Builds the histogram in a hash.
  def histogram
    self.each_with_object(Hash.new(0)) { |n, h| h[n] += 1 }
  end

  # The initial part of the array without its last element.
  def init
    self[0..-2]
  end

  # Calculates the mean of the elements.
  def mean
    self.empty? ? nil : self.sum.to_f / self.size
  end
  alias_method :average, :mean

  # Calculates the median of the elements.
  def median
    return nil if self.empty?
    sorted = self.sort
    size = self.size

    if size % 2 == 1
      sorted[size / 2]
    else
      (sorted[(size - 1) / 2] + sorted[size / 2]) / 2.0
    end
  end

  # Finds the mode value/s.
  def mode
    max = histogram.values.max
    histogram.map { |key, value| key if value == max }.compact
  end

  # Returns the percentile value for a given percentage.
  def percentile(percentage)
    size = self.size

    if size > 1
      index = size * percentage / 100.0
      self.sort[index]
    else
      self.first
    end
  end

  # Returns n elements from the array.
  old_take = instance_method(:take)
  define_method(:take) do |n|
    n >= 0 ? old_take.bind(self).call(n) : self.last(n.abs)
  end

  # The rest of the array without its first element.
  def tail
    self[1..-1]
  end

  # Assumes that self is an array of arrays or ranges
  # and transposes the rows and columns.
  old_transpose = instance_method(:transpose)
  define_method(:transpose) do
    old_transpose.bind(self.map(&:to_a)).call
  end
end