From 50cdb654ac4ff1bde2f432d205932a6887f6f92f Mon Sep 17 00:00:00 2001
From: Jeremy Kemper <jeremy@bitsweat.net>
Date: Sun, 4 Apr 2010 11:56:49 -0700
Subject: [PATCH] Use call stack printer if available

---
 tools/profile | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/tools/profile b/tools/profile
index 927467bc4e..0ccef6c26c 100755
--- a/tools/profile
+++ b/tools/profile
@@ -60,8 +60,14 @@ after = GC.allocated_size
 usage = (after - before) / 1024.0
 
 if mode
-  File.open("#{File.basename(path, '.rb')}.#{mode}.callgrind", 'w') do |out|
-    RubyProf::CallTreePrinter.new(results).print(out)
+  if RubyProf.const_defined?(:CallStackPrinter)
+    File.open("#{File.basename(path, '.rb')}.#{mode}.html", 'w') do |out|
+      RubyProf::CallStackPrinter.new(results).print(out)
+    end
+  else
+    File.open("#{File.basename(path, '.rb')}.#{mode}.callgrind", 'w') do |out|
+      RubyProf::CallTreePrinter.new(results).print(out)
+    end
   end
 end