Support an extra profile printer arg

This commit is contained in:
Jeremy Kemper
2010-06-24 01:08:12 -07:00
parent 7b730a2f1b
commit 4883082ff1

View File

@@ -63,7 +63,9 @@ after = GC.allocated_size
usage = (after - before) / 1024.0
if mode
if RubyProf.const_defined?(:CallStackPrinter)
if printer = ARGV.shift
RubyProf.const_get("#{printer.to_s.classify}Printer").new(results).print($stdout)
elsif RubyProf.const_defined?(:CallStackPrinter)
File.open("#{File.basename(path, '.rb')}.#{mode}.html", 'w') do |out|
RubyProf::CallStackPrinter.new(results).print(out)
end