#!/usr/bin/env ruby

$LOAD_PATH << File.expand_path(File.join(File.dirname( File.symlink?(__FILE__) ? File.readlink(__FILE__) : __FILE__ ), "../lib"))

require 'optparse'
require 'ffi_yajl/benchmark'

opts = {}
optparse = OptionParser.new do |o|
  o.banner = "Usage: ffi-yajl-bench"

  opts[:profile] = false
  o.on( '-p', '--profile', 'Run perftools.rb profiling' ) do
    opts[:profile] = true
  end

  o.on( '-F', '--ffi', 'Force using FFI' ) do
    opts[:ffi] = true
  end

  o.on( '-E', '--ext', 'Force using C ext' ) do
    opts[:ext] = true
  end
end

optparse.parse!

ENV['FORCE_FFI_YAJL'] = 'ffi' if opts[:ffi]
ENV['FORCE_FFI_YAJL'] = 'ext' if opts[:ext]

if opts[:profile]
  FFI_Yajl::Benchmark::ParseProfileRubyProf.new.run
else
  FFI_Yajl::Benchmark::Parse.new.run
  FFI_Yajl::Benchmark::Encode.new.run
end
