Benford Counter

benford_counter.rb

class BenfordCounter
  def initialize
    @num_count = {}

    # Set all values to zero
    (1..9).each {|num| @num_count[num] = 0 }

    @total_records = 0
    @min_value = 9999999999
    @max_value = 0
  end

  def count(string_value)
    # Turn it into a nice string if it isn't already
    string_value = string_value.to_s
    string_value.gsub!(',', '')
    string_value.strip!

    # Create an integer for nice comparisons
    int_value = string_value.to_i

    # Make sure we have a number and it's positive
    if string_value.length > 0 && int_value > 0
      # Get the first number from the string
      first_num = string_value[0].chr.to_i

      if first_num > 0
        @min_value = int_value if int_value < @min_value
        @max_value = int_value if int_value > @max_value

        @num_count[first_num] += 1
        @total_records += 1
      end
    end
  end

  def results
    @num_count.each do |first_num, num_records|
      percentage = num_records.quo(@total_records) * 100
      puts first_num.to_s + ": " + num_records.to_s + ": " + percentage.to_f.to_s + "%"
    end

    puts "Total records: #{@total_records.to_s}"
    puts "Min value: #{@min_value.to_s}"
    puts "Max value: #{@max_value.to_s}"
  end
end