#!/usr/bin/env ruby

require 'csv'

# current as of July 2013
fare_types = {
  "pub_aircon" => {
    "fare" => 12.00,
    "firstkm" => 5,
    "perkm" => 2.2,
    "limit" => 60
  },
  "pub_ordinary" => {
    "fare" => 10.00,
    "firstkm" => 5,
    "perkm" => 1.85,
    "limit" => 60
  },
  "puj" => {
    "fare" => 8.00,
    "firstkm" => 4,
    "perkm" => 1.4,
    "limit" => 50,
    "override" => {
      10 => { "discounted" => '13.25' },
      12 => { "discounted" => '15.50' },
      14 => { "discounted" => '17.75' },
      18 => { "discounted" => '22.25' },
      20 => { "discounted" => '24.50' },
      22 => { "discounted" => '26.75' },
      33 => { "discounted" => '38.75' },
      35 => { "discounted" => '41.25' },
      37 => { "discounted" => '43.50' },
      39 => { "discounted" => '45.75' },
      45 => { "discounted" => '52.50' },
      47 => { "discounted" => '54.75' }
    }
  }
}

def round(original)
  "%.2f" % ((original*4).round/4.0)
end

def if_override(original, type, dist, name)
  value = original
  value = type['override'][dist][name] if type['override'] && type['override'][dist] && type['override'][dist][name]
  value
end

fare_types.each do |key, type|
  CSV.open(key+'.csv', 'wb') do |csv|
    csv << ['distance', 'regular', 'discounted']
    type['limit'].times do |x|
      dist = x+1
      fare = type['fare']
      if dist > type['firstkm']
        fare += type['perkm'] * (dist - type['firstkm'])
      end
      discounted = round(fare*0.8)
      fare = round(fare)
      discounted = if_override(discounted, type, dist, 'discounted')
      fare = if_override(fare, type, dist, 'fare')
      csv << [dist, fare, discounted]
    end
  end
end