#!/usr/bin/env ruby

require 'zipf'

class KbestItem
  attr_accessor :rank, :model, :rr, :gold, :f
  def initialize s
    a = s.split "\t"
    @rank = a[0].to_i
    @gold = a[1].to_f
    @model = a[2].to_f
    @rr    = -1.0
    @f = SparseVector.from_kv a[3], "=", " "
  end

  def to_s
    return "#{@model}\t#{@gold}"
  end
end

w = SparseVector.from_kv ReadFile.new(ARGV[0]).read, /\s/, "\n"

def o kl
  scores = []
  scores << kl.first.gold
  kl.sort! { |i,j| j.model <=> i.model }
  scores << kl.first.gold
  kl.sort! { |i,j| j.rr <=> i.rr }
  scores << kl.first.gold

  puts scores.join "\t"
end

STDERR.write "reranking..\n"
cur = []
k_sum = 0
j = 0
while line = STDIN.gets
  item = KbestItem.new line.strip
  item.rr = w.dot(item.f)
  if item.rank == 0 && cur.size > 0
    o cur
    cur = []
    j += 1
  end
  cur << item
end
o cur