Extracts a block of subtree from a callgraph of a given function (matched by a substring).

truffle_calltree_find.rb

=begin

Extracts a block of subtree from a callgraph of a given function (matched by a substring).

Use:

jt ruby --cpusampler=calltree -e 'a=[];a.push(1);p(a.size)' | ruby truffle_calltree_find.rb "IO#puts"

=end

pattern = ARGV[0]

in_match = false
tab_size = nil

STDIN.each_line.with_index do |line, idx|
	if !in_match
		if line.index(pattern) != nil
			in_match = true
			tab_size = line.match(/^ */)[0].size
			puts(line)
		end
	else
		current_tab_size = line.match(/^ */)[0].size
		if current_tab_size == tab_size
			in_match = false
			puts("\n\n---\n\n")
		else
			puts(line)
		end
	end
end