# encoding: utf-8

require "rubygems"
require "pdf/writer"
require "pdf/simpletable"
gem "pdf-wrapper", ">=0.1.2"
require "pdf/wrapper"
require "fastercsv"
require "benchmark"

$LOAD_PATH << File.join(File.dirname(__FILE__), '..', 'lib')
$LOAD_PATH << File.join(File.dirname(__FILE__), '..', 'examples')
require "prawn"
require "ruport_formatter"

csv_data = FasterCSV.read("#{Prawn::BASEDIR}/examples/currency.csv") * 
  (ARGV.first || 1).to_i

####################################
# PDF::Writer Table Rendering Prep #
####################################
pdf = PDF::Writer.new
pdf.select_font("Helvetica")

table = PDF::SimpleTable.new do |tab|
  tab.column_order.push(*%w(date rate))

  tab.columns["date"] = PDF::SimpleTable::Column.new("date") { |col|
    col.heading = "Date"
  }
  tab.columns["rate"] = PDF::SimpleTable::Column.new("rate") { |col|
    col.heading = "Rate"
  }

  tab.orientation   = :center

  data = csv_data.map do |e| 
    { "date" => e[0], "rate" => e[1] }
  end

  tab.data.replace data
end

##############################
# Prawn Table Rendering Prep #
##############################
doc = Prawn::Document.new

#####################
# Ruport/Prawn Prep #
#####################

ruport_table = Table(%w[Date Rate], :data => csv_data)
ruport_doc   = Prawn::Document.new

##############################
# PDF::Wrapper Table Rendering Prep
##############################
wrapper_doc = PDF::Wrapper.new
wrapper_table = PDF::Wrapper::Table.new do |t|
  t.data = csv_data
  t.table_options :font_size => 6
end

#######################
# Benchmarking code   #
#######################

puts "Processing #{csv_data.length} records"

Benchmark.bmbm do |x|
  x.report("PDF Wrapper") do
    wrapper_doc.table( wrapper_table, :width => 100 ) unless wrapper_doc.finished?
    wrapper_doc.render_to_file('currency_pdf_wrapper.pdf')
  end
  x.report("Prawn") do
    doc.table(csv_data, :font_size          => 10, 
                        :vertical_padding   => 2,
                        :horizontal_padding => 5, 
                        :position           => :center, 
                        :row_colors         => :pdf_writer,
                        :headers            => ["Date","Rate"])
    doc.render_file('currency_prawn.pdf')
  end
  x.report("Ruport/Prawn") do
    ruport_table.save_as('currency_ruport.pdf', :document => ruport_doc)
  end
  x.report("PDF Writer") do
    table.render_on(pdf) 
    pdf.save_as('currency_pdf_writer.pdf')
  end
end