require "./lib/contracts"
require "benchmark"
require "rubygems"
require "method_profiler"
require "ruby-prof"

include Contracts

def add opts
  opts[:a] + opts[:b]
end

Contract ({ :a => Num, :b => Num}) => Num
def contracts_add opts
  opts[:a] + opts[:b]
end

def explicit_add opts
  a = opts[:a]
  b = opts[:b]
  fail unless a.is_a?(Numeric)
  fail unless b.is_a?(Numeric)
  c = a + b
  fail unless c.is_a?(Numeric)
  c
end

def benchmark
  Benchmark.bm 30 do |x|
    x.report "testing add" do
      1_000_000.times do |_|
        add(:a => rand(1000), :b => rand(1000))
      end
    end
    x.report "testing contracts add" do
      1_000_000.times do |_|
        contracts_add(:a => rand(1000), :b => rand(1000))
      end
    end
  end
end

def profile
  profilers = []
  profilers << MethodProfiler.observe(Contract)
  profilers << MethodProfiler.observe(Object)
  profilers << MethodProfiler.observe(Contracts::MethodDecorators)
  profilers << MethodProfiler.observe(Contracts::Decorator)
  profilers << MethodProfiler.observe(Contracts::Support)
  profilers << MethodProfiler.observe(UnboundMethod)
  10_000.times do |_|
    contracts_add(:a => rand(1000), :b => rand(1000))
  end
  profilers.each { |p| puts p.report }
end

def ruby_prof
  RubyProf.start
  100_000.times do |_|
    contracts_add(:a => rand(1000), :b => rand(1000))
  end
  result = RubyProf.stop
  printer = RubyProf::FlatPrinter.new(result)
  printer.print(STDOUT)
end

benchmark
profile
ruby_prof if ENV["FULL_BENCH"] # takes some time