Merge branch 'master' into possible-speedups

korny · korny · commit ffdd9ddb5cb5 · 2016-12-28T12:08:14.000+01:00
diff --git a/bench/bench.rb b/bench/bench.rb
@@ -15,22 +15,26 @@
 format = ARGV.fetch(1, 'html').downcase
 encoder = CodeRay.encoder(format)
 
-size = ARGV.fetch(2, 1000).to_i * 1000
+size = ARGV.fetch(2, 2000).to_i * 1000
 unless size.zero?
   data += data until data.size >= size
   data = data[0, size]
 end
 size = data.size
 puts "encoding %d kB of #{lang} code to #{format}..." % [(size / 1000.0).round]
 
-n = ARGV.fetch(3, 5).to_s[/\d+/].to_i
+n = ARGV.fetch(3, 10).to_s[/\d+/].to_i
 require 'profile' if ARGV.include? '-p'
+times = []
 n.times do |i|
   time = Benchmark.realtime { encoder.encode(data, lang) }
   puts "run %d: %5.2f s, %4.0f kB/s" % [i + 1, time, size / time / 1000.0]
+  times << time
 end
 
-STDIN.gets if ARGV.include? '-w'
+times_sum = times.inject(0) { |time, sum| sum + time }
+puts 'Average time: %5.2f s, %4.0f kB/s' % [times_sum / times.size, (size * n) / times_sum / 1000.0]
+puts 'Best time:    %5.2f s, %4.0f kB/s' % [times.min, size / times.min / 1000.0]
 
 __END__
 Usage:
diff --git a/rake_tasks/benchmark.rake b/rake_tasks/benchmark.rake
@@ -1,6 +1,6 @@
 desc 'Do a benchmark'
 task :benchmark do
-  ruby 'bench/bench.rb ruby html 3000'
+  ruby 'bench/bench.rb ruby html'
 end
 
 task :bench => :benchmark
