class JSONToChoropleth

# File lib/jsontochoropleth.rb, line 4
def initialize(input, location, number)
  @input = JSON.parse(input)
  @location = location
  @number = number
end

# File lib/jsontochoropleth.rb, line 10
def genfill(type)
  countryhash = Hash.new

  rangehash = Hash.new
  fullcount = 0
  @input.each do |r|
    fullcount = fullcount + r[@number].to_f
  end

  average = fullcount.to_f/@input.length.to_f

  if type == "percent"
    rangehash["#FFFF99"] = average/3
    rangehash["#FFFF00"] = average/2
    rangehash["#FFA800"] = average
    rangehash["#FF6800"] = average/0.5
    rangehash["#FF0000"] = average/0.25
  elsif type == "full"
    rangehash["#FFFF99"] = average.to_f/3.0
    rangehash["#FFFF00"] = average.to_f/2.0
    rangehash["#FFA800"] = average
    rangehash["#FF6800"] = average*2
    rangehash["#FF0000"] = average*4
  end

  @input.each do |t|
    numusers = t[@number]
    prevv = 0
    count = 0

    rangehash.each do |k,v|
      if (numusers > prevv) && (numusers <= v)
        countryhash[t[@location]] = k
      elsif (count == 4) && (numusers >= v)
        countryhash[t[@location]] = k
      end
      prevv = v
      count = count + 1
    end
  end

  return JSON.pretty_generate(countryhash)
end