class RubyProf::CallTreePrinter

# File lib/ruby-prof/printers/call_tree_printer.rb, line 90
def self.needs_dir?
  true
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 94
def base_name
  @options[:profile] || "profile"
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 69
def convert(value)
  (value * @value_scale).round
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 16
def determine_event_specification_and_value_scale
  @event_specification = "events: "
  case RubyProf.measure_mode
    when RubyProf::PROCESS_TIME
      @value_scale = RubyProf::CLOCKS_PER_SEC
      @event_specification << 'process_time'
    when RubyProf::WALL_TIME
      @value_scale = 1_000_000
      @event_specification << 'wall_time'
    when RubyProf.const_defined?(:CPU_TIME) && RubyProf::CPU_TIME
      @value_scale = RubyProf.cpu_frequency
      @event_specification << 'cpu_time'
    when RubyProf.const_defined?(:ALLOCATIONS) && RubyProf::ALLOCATIONS
      @value_scale = 1
      @event_specification << 'allocations'
    when RubyProf.const_defined?(:MEMORY) && RubyProf::MEMORY
      @value_scale = 1
      @event_specification << 'memory'
    when RubyProf.const_defined?(:GC_RUNS) && RubyProf::GC_RUNS
      @value_scale = 1
      @event_specification << 'gc_runs'
    when RubyProf.const_defined?(:GC_TIME) && RubyProf::GC_TIME
      @value_scale = 1000000
      @event_specification << 'gc_time'
    else
      raise "Unknown measure mode: #{RubyProf.measure_mode}"
  end
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 73
def file(method)
  File.expand_path(method.source_file)
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 104
def file_name_for_thread(thread)
  if thread.fiber_id == Fiber.current.object_id
    [base_name, "callgrind.out", $$].join(".")
  else
    [base_name, "callgrind.out", $$, thread.fiber_id].join(".")
  end
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 112
def file_path_for_thread(thread)
  File.join(path, file_name_for_thread(thread))
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 86
def path
  @options[:path] || "."
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 45
def print(options = {})
  validate_print_params(options)
  setup_options(options)
  determine_event_specification_and_value_scale
  print_threads
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 116
def print_headers(output, thread)
  output << "#{@event_specification}\n\n"
  # this doesn't work. kcachegrind does not fully support the spec.
  # output << "thread: #{thread.id}\n\n"
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 122
def print_method(output, method)
  # Print out the file and method name
  output << "fl=#{file(method)}\n"
  output << "fn=#{method.calltree_name}\n"

  # Now print out the function line number and its self time
  output << "#{method.line} #{convert(method.self_time)}\n"

  # Now print out all the children methods
  method.children.each do |callee|
    output << "cfl=#{file(callee.target)}\n"
    output << "cfn=#{callee.target.calltree_name}\n"
    output << "calls=#{callee.called} #{callee.line}\n"

    # Print out total times here!
    output << "#{callee.line} #{convert(callee.total_time)}\n"
  end
  output << "\n"
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 77
def print_thread(thread)
  File.open(file_path_for_thread(thread), "w") do |f|
    print_headers(f, thread)
    thread.methods.reverse_each do |method|
      print_method(f, method)
    end
  end
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 60
def print_threads
  remove_subsidiary_files_from_previous_profile_runs
  # TODO: merge fibers of a given thread here, instead of relying
  # on the profiler to merge fibers.
  @result.threads.each do |thread|
    print_thread(thread)
  end
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 98
def remove_subsidiary_files_from_previous_profile_runs
  pattern = [base_name, "callgrind.out", $$, "*"].join(".")
  files = Dir.glob(File.join(path, pattern))
  FileUtils.rm_f(files)
end

# File lib/ruby-prof/printers/call_tree_printer.rb, line 52
def validate_print_params(options)
  if options.is_a?(IO)
    raise ArgumentError, "#{self.class.name}#print cannot print to IO objects"
  elsif !options.is_a?(Hash)
    raise ArgumentError, "#{self.class.name}#print requires an options hash"
  end
end