class BetterBytes

# File lib/better_bytes.rb, line 27
def self.dehumanize(str)
  m = str.match(%r{(\d*(\.\d+)?)\s*(\w{0,3})}) or raise "Unable to dehumanize: #{str}"

  v = m[2] ? m[1].to_f : m[1].to_i

  abbrev, mult = m[3][1] == ?i ? BINARY : DECIMAL

  unless exp = abbrev.index(m[3])
    if m[3] == nil || m[3].empty?
      exp = 0
    else
      raise "Unknown unit: #{m[3]} (extracted from #{str})"
    end
  end

  return v * (mult ** exp)
end

# File lib/better_bytes.rb, line 5
def self.humanize(val, precision: 0.1, units: :binary)
  val.is_a? String and val = dehumanize(val)

  if units.is_a? Symbol
    abbrev, mult = units == :binary ? BINARY : DECIMAL
    t = [1,'B']
    abbrev.each_with_index do |a,i|
      m = mult ** i
      val < m and break
      t = [m,a]
    end
  elsif i = BINARY[0].index(units)
    t = [BINARY[1] ** i, units]
  elsif i = DECIMAL[0].index(units)
    t = [DECIMAL[1] ** i, units]
  else
    raise "Unknown units: #{units}"
  end

  return "%#{precision}f %s" % [val.to_f / t[0], t[1]]
end