seafoam 0.6 → 0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 69f3387c345f401ca95791c9f5b69d57b1ab123e60abcb31b9160b8734b5507e
-  data.tar.gz: 154f61a73ddb2ff1cf271c8d57fbf9422ddbb46fa3e0dc6d1256d1f4c2c0addc
+  metadata.gz: f8665fe768d4900820ca676269cd25028306e8253c2109c71ff4cb31dd7c6e44
+  data.tar.gz: fd8c52e5cf3daafa9b2673265067b743cc644755c3772c02904c2b815b3a93d2
 SHA512:
-  metadata.gz: 29067885733970836583c0060a933ad13ec78ba35a01c6462ea0af4ac7df2137f8d0202dcdb7f57078aab0aba30c2da08973c9f31415191c9229f0a7adaf816b
-  data.tar.gz: 69b2872db8937ed1bd73eefee880139055ee9a40b3a3cd50c041f3c623307780ff5e72d3ad2eaf9882d4234b58b9668021270f768c767229e930d12613483f8d
+  metadata.gz: 815e7031027718cea7a933b7fc5ad822f754f05e3ae5bdc2066775183a9ca98c841b54e682a148c4437a29b1f95c4ede5833a2a1f2995d2c55296a800e71b269
+  data.tar.gz: 462aaad5134c8b6cce229e8456742151354333c5e7c02283e2e90917fa7ad238f3f4f71770519357117f42b1579928ddb0bba3f04b31ddef68d7e41e92f8c016

data/bin/bgv2isabelle

@@ -5,12 +5,8 @@ require 'seafoam'
 # This is the 'bgv2isabelle' command line entry point.
 
 begin
-  # Load configuraiton.
-  config = Seafoam::Config.new
-  config.load_config
-
   # Run the command line.
-  commands = Seafoam::Commands.new($stdout, config)
+  commands = Seafoam::Commands.new($stdout)
   commands.bgv2isabelle(*ARGV)
 rescue StandardError => e
   if $DEBUG

data/bin/bgv2json

@@ -5,12 +5,8 @@ require 'seafoam'
 # This is the 'bgv2json' command line entry point.
 
 begin
-  # Load configuraiton.
-  config = Seafoam::Config.new
-  config.load_config
-
   # Run the command line.
-  commands = Seafoam::Commands.new($stdout, config)
+  commands = Seafoam::Commands.new($stdout)
   commands.bgv2json(*ARGV)
 rescue StandardError => e
   if $DEBUG

data/bin/cfg2asm

@@ -5,12 +5,8 @@ require 'seafoam'
 # This is the 'cfg2asm' command line entry point.
 
 begin
-  # Load configuraiton.
-  config = Seafoam::Config.new
-  config.load_config
-
   # Run the command line.
-  commands = Seafoam::Commands.new($stdout, config)
+  commands = Seafoam::Commands.new($stdout)
   commands.cfg2asm(*ARGV)
 rescue StandardError => e
   if $DEBUG

data/bin/seafoam

@@ -5,12 +5,8 @@ require 'seafoam'
 # This is the 'seafoam' command line entry point.
 
 begin
-  # Load configuraiton.
-  config = Seafoam::Config.new
-  config.load_config
-
   # Run the command line.
-  commands = Seafoam::Commands.new($stdout, config)
+  commands = Seafoam::Commands.new($stdout)
   commands.seafoam(*ARGV)
 rescue StandardError => e
   if $DEBUG

data/lib/seafoam/bgv/bgv_parser.rb

@@ -127,13 +127,21 @@ module Seafoam
             end
           end
         end
-        skip_blocks
+
+        # Read block information.
+        @reader.read_sint32.times do
+          block_id = @reader.read_sint32
+          block_nodes = @reader.read_sint32.times.map { @reader.read_sint32 }
+          # Followers aren't used but could be.
+          @reader.read_sint32.times.map { @reader.read_sint32 }
+          graph.create_block block_id, block_nodes
+        end
         graph
       end
 
       # Skip over a graph, having read or skipped its headers.
       def skip_graph
-        # Already read BEGIN_GRAPH, id, format, args, and props
+        # Already read BEGIN_GRAPH, id, format, args, and props.
         @reader.read_sint32.times do
           @reader.skip_int32
           node_class = read_pool_object

data/lib/seafoam/commands.rb

@@ -1,11 +1,11 @@
 require 'json'
+require 'set'
 
 module Seafoam
   # Implementations of the command-line commands that you can run in Seafoam.
   class Commands
-    def initialize(out, config)
+    def initialize(out)
       @out = out
-      @config = config
     end
 
     # Run the general seafoam command.
@@ -15,24 +15,7 @@ module Seafoam
       when nil, 'help', '-h', '--help', '-help'
         raise ArgumentError, "unexpected arguments #{args.join(' ')}" unless args.empty?
 
-        @out.puts 'seafoam file.bgv info'
-        @out.puts '        file.bgv list'
-        @out.puts '        file.bgv[:graph][:node[-edge]] search term...'
-        @out.puts '        file.bgv[:graph][:node[-edge]] edges'
-        @out.puts '        file.bgv[:graph][:node[-edge]] props'
-        @out.puts '        file.bgv:graph:node source'
-        @out.puts '        file.bgv:graph render'
-        @out.puts '              --spotlight n,n,n...'
-        @out.puts '              --out graph.pdf'
-        @out.puts '                    graph.svg'
-        @out.puts '                    graph.png'
-        @out.puts '                    graph.dot'
-        @out.puts '               --show-frame-state'
-        @out.puts '               --hide-floating'
-        @out.puts '               --no-reduce-edges'
-        @out.puts '               --option key value'
-        @out.puts '        --help'
-        @out.puts '        --version'
+        help(*args)
       when 'version', '-v', '-version', '--version'
         version(*args)
       else
@@ -57,6 +40,8 @@ module Seafoam
           render name, *args
         when 'debug'
           debug name, *args
+        when 'describe'
+          describe name, *args
         else
           raise ArgumentError, "unknown command #{command}"
         end
@@ -262,7 +247,7 @@ module Seafoam
     end
 
     def search_object(tag, object, terms)
-      full_text = JSON.generate(object)
+      full_text = JSON.generate(JSONWriter.prepare_json(object))
       full_text_down = full_text.downcase
       start = 0
       terms.each do |t|
@@ -298,7 +283,7 @@ module Seafoam
         parser.read_graph_header
         graph = parser.read_graph
         if node_id
-          Annotators.apply graph
+          Passes.apply graph
           node = graph.nodes[node_id]
           raise ArgumentError, 'node not found' unless node
 
@@ -391,14 +376,66 @@ module Seafoam
       end
     end
 
+    # seafoam file.bgv:n describe
+    def describe(name, *args)
+      file, graph_index, *rest = parse_name(name)
+
+      if graph_index.nil? || !rest.all?(&:nil?)
+        raise ArgumentError, 'describe only works with a graph'
+      end
+      raise ArgumentError, 'describe does not take arguments' unless args.empty?
+
+      parser = BGV::BGVParser.new(file)
+      parser.read_file_header
+      parser.skip_document_props
+
+      loop do
+        index, = parser.read_graph_preheader
+        break unless index
+
+        parser.skip_graph_header
+
+        if index != graph_index
+          parser.skip_graph
+          next
+        end
+
+        graph = parser.read_graph
+        notes = Set.new
+
+        graph.nodes.each_value do |node|
+          node_class = node.props.dig(:node_class, :node_class)
+          case node_class
+          when 'org.graalvm.compiler.nodes.IfNode'
+            notes.add 'branches'
+          when 'org.graalvm.compiler.nodes.LoopBeginNode'
+            notes.add 'loops'
+          when 'org.graalvm.compiler.nodes.InvokeNode', 'org.graalvm.compiler.nodes.InvokeWithExceptionNode'
+            notes.add 'calls'
+          end
+        end
+
+        notes.add 'deopts' if graph.nodes[0].outputs.map(&:to)
+                                   .all? { |t| t.props.dig(:node_class, :node_class) == 'org.graalvm.compiler.nodes.DeoptimizeNode' }
+
+        notes.add 'linear' unless notes.include?('branches') || notes.include?('loops')
+
+        @out.puts ["#{graph.nodes.size} nodes", *notes].join(', ')
+
+        break
+      end
+    end
+
     # seafoam file.bgv:n render options...
     def render(name, *args)
       file, graph_index, *rest = parse_name(name)
       raise ArgumentError, 'render needs at least a graph' unless graph_index
       raise ArgumentError, 'render only works with a graph' unless rest == [nil, nil]
 
-      annotator_options = {
+      pass_options = {
+        simplify_truffle_args: true,
         hide_frame_state: true,
+        hide_pi: true,
         hide_floating: false,
         reduce_edges: true
       }
@@ -406,6 +443,7 @@ module Seafoam
       args = args.dup
       out_file = nil
       explicit_out_file = false
+      draw_blocks = false
       until args.empty?
         arg = args.shift
         case arg
@@ -418,12 +456,18 @@ module Seafoam
           raise ArgumentError, 'no list for --spotlight' unless spotlight_arg
 
           spotlight_nodes = spotlight_arg.split(',').map { |n| Integer(n) }
+        when '--full-truffle-args'
+          pass_options[:simplify_truffle_args] = false
         when '--show-frame-state'
-          annotator_options[:hide_frame_state] = false
+          pass_options[:hide_frame_state] = false
+        when '--show-pi'
+          pass_options[:hide_pi] = false
         when '--hide-floating'
-          annotator_options[:hide_floating] = true
+          pass_options[:hide_floating] = true
         when '--no-reduce-edges'
-          annotator_options[:reduce_edges] = false
+          pass_options[:reduce_edges] = false
+        when '--draw-blocks'
+          draw_blocks = true
         when '--option'
           key = args.shift
           raise ArgumentError, 'no key for --option' unless key
@@ -432,7 +476,7 @@ module Seafoam
           raise ArgumentError, "no value for --option #{key}" unless out_file
 
           value = { 'true' => true, 'false' => 'false' }.fetch(key, value)
-          annotator_options[key.to_sym] = value
+          pass_options[key.to_sym] = value
         else
           raise ArgumentError, "unexpected option #{arg}"
         end
@@ -455,7 +499,7 @@ module Seafoam
       with_graph(file, graph_index) do |parser|
         parser.skip_graph_header
         graph = parser.read_graph
-        Annotators.apply graph, annotator_options
+        Passes.apply graph, pass_options
         if spotlight_nodes
           spotlight = Spotlight.new(graph)
           spotlight_nodes.each do |node_id|
@@ -469,14 +513,14 @@ module Seafoam
         if out_format == :dot
           File.open(out_file, 'w') do |stream|
             writer = GraphvizWriter.new(stream)
-            writer.write_graph graph
+            writer.write_graph graph, false, draw_blocks
           end
         else
           begin
             IO.popen(['dot', "-T#{out_format}", '-o', out_file], 'w') do |stream|
               writer = GraphvizWriter.new(stream)
               hidpi = out_format == :png
-              writer.write_graph graph, hidpi
+              writer.write_graph graph, hidpi, draw_blocks
             end
           rescue Errno::ENOENT
             raise 'Could not run Graphviz - is it installed?'
@@ -565,6 +609,32 @@ module Seafoam
       raise ArgumentError, 'graph not found' unless graph_found
     end
 
+    # Prints help.
+    def help(*_args)
+      @out.puts 'seafoam file.bgv info'
+      @out.puts '        file.bgv list'
+      @out.puts '        file.bgv[:graph][:node[-edge]] search term...'
+      @out.puts '        file.bgv[:graph][:node[-edge]] edges'
+      @out.puts '        file.bgv[:graph][:node[-edge]] props'
+      @out.puts '        file.bgv:graph:node source'
+      @out.puts '        file.bgv:graph describe'
+      @out.puts '        file.bgv:graph render'
+      @out.puts '              --spotlight n,n,n...'
+      @out.puts '              --out graph.pdf'
+      @out.puts '                    graph.svg'
+      @out.puts '                    graph.png'
+      @out.puts '                    graph.dot'
+      @out.puts '               --full-truffle-args'
+      @out.puts '               --show-frame-state'
+      @out.puts '               --show-pi'
+      @out.puts '               --hide-floating'
+      @out.puts '               --no-reduce-edges'
+      @out.puts '               --draw-blocks'
+      @out.puts '               --option key value'
+      @out.puts '        --help'
+      @out.puts '        --version'
+    end
+
     # Prints the version.
     def version(*args)
       raise ArgumentError, "unexpected arguments #{args.join(' ')}" unless args.empty?
@@ -598,10 +668,15 @@ module Seafoam
 
     # Open a file for the user if possible.
     def autoopen(file)
-      if RUBY_PLATFORM.include?('darwin') && @out.tty?
+      return unless @out.tty?
+
+      case RUBY_PLATFORM
+      when /darwin/
         system 'open', file
-        # Don't worry if it fails.
+      when /linux/
+        system 'xdg-open', file
       end
+      # Don't worry if it fails.
     end
   end
 end

data/lib/seafoam/graal/pi.rb

@@ -0,0 +1,18 @@
+module Seafoam
+  module Graal
+    # Routines for understanding pi nodes in Graal.
+    module Pi
+      # Find the actual value behind potentially a chain of pi nodes.
+      def self.follow_pi_object(node)
+        node = node.edges.find { |edge| edge.props[:name] == 'object' }.from while PI_NODES.include?(node.props.dig(:node_class, :node_class))
+        node
+      end
+
+      # Pi nodes add type information.
+      PI_NODES = [
+        'org.graalvm.compiler.nodes.PiNode',
+        'org.graalvm.compiler.nodes.PiArrayNode'
+      ]
+    end
+  end
+end

data/lib/seafoam/graph.rb

@@ -2,12 +2,14 @@ module Seafoam
   # A graph, with properties, nodes, and edges. We don't encapsulate the graph
   # too much - be careful.
   class Graph
-    attr_reader :props, :nodes, :edges
+    attr_reader :props, :nodes, :edges, :blocks, :new_id
 
     def initialize(props = nil)
       @props = props || {}
       @nodes = {}
       @edges = []
+      @blocks = []
+      @new_id = 0
     end
 
     # Create a node.
@@ -15,6 +17,7 @@ module Seafoam
       props ||= {}
       node = Node.new(id, props)
       @nodes[id] = node
+      @new_id = id + 1
       node
     end
 
@@ -27,6 +30,20 @@ module Seafoam
       to.inputs.push edge
       edge
     end
+
+    # Add a new basic block with given id and node id list.
+    def create_block(id, node_ids)
+      nodes = node_ids.select { |n| @nodes.key? n }.map { |n| @nodes[n] }
+      block = Block.new(id, nodes)
+      @blocks.push block
+      block
+    end
+
+    def remove_edge(edge)
+      edge.from.outputs.delete edge
+      edge.to.inputs.delete edge
+      edges.delete edge
+    end
   end
 
   # A node, with properties, input edges, and output edges.
@@ -88,4 +105,19 @@ module Seafoam
       "<Edge #{from.id} -> #{to.id}>"
     end
   end
+
+  # A control-flow basic block
+  class Block
+    attr_reader :id, :nodes
+
+    def initialize(id, nodes)
+      @id = id
+      @nodes = nodes
+    end
+
+    # Inspect.
+    def inspect
+      "<Block #{id}>"
+    end
+  end
 end

data/lib/seafoam/graphviz_writer.rb

@@ -7,15 +7,16 @@ module Seafoam
     end
 
     # Write a graph.
-    def write_graph(graph, hidpi = false)
+    def write_graph(graph, hidpi = false, draw_blocks = false)
       inline_attrs = {}
       attrs = {}
       attrs[:dpi] = 200 if hidpi
-      attrs[:bgcolor] = 'transparent'
+      attrs[:bgcolor] = 'white'
       @stream.puts 'digraph G {'
       @stream.puts "  graph #{write_attrs(attrs)};"
       write_nodes inline_attrs, graph
       write_edges inline_attrs, graph
+      write_blocks graph if draw_blocks
       @stream.puts '}'
     end
 
@@ -99,6 +100,9 @@ module Seafoam
         # from a shaded node.
         next if edge.to.props[:hidden] && edge.from.props[:spotlight] != 'shaded'
 
+        # Skip the edge if it's hidden itself
+        next if edge.props[:hidden]
+
         write_edge inline_attrs, edge
       end
     end
@@ -166,6 +170,24 @@ module Seafoam
       end
     end
 
+    # Write basic block outlines.
+    def write_blocks(graph)
+      graph.blocks.each do |block|
+        @stream.puts "  subgraph cluster_block#{block.id} {"
+        @stream.puts '    fontname = "Arial";'
+        @stream.puts "    label = \"B#{block.id}\";"
+        @stream.puts '    style=dotted;'
+
+        block.nodes.each do |node|
+          next if node.props[:hidden] || node.props[:inlined]
+
+          @stream.puts "    node#{node.id};"
+        end
+
+        @stream.puts '  }'
+      end
+    end
+
     # Return attributes for a node or edge modified to 'shade' them in terms
     # the spotlight functionality - so basically make them light grey.
     def shade(attrs)

data/lib/seafoam/json_writer.rb

@@ -36,9 +36,7 @@ module Seafoam
       @out.puts JSON.pretty_generate(prepare_json(object))
     end
 
-    private
-
-    def prepare_json(object)
+    def self.prepare_json(object)
       case object
       when Float
         if object.nan?

data/lib/seafoam/{annotators → passes}/fallback.rb

@@ -1,15 +1,15 @@
 module Seafoam
-  module Annotators
-    # The fallback annotator always applies, and adds some basic annotations.
+  module Passes
+    # The fallback pass always applies, and adds some basic properties.
     # Works for example with Truffle AST and call graphs, but also means anyone
     # can emit a graph with 'label' properties and we can do something useful
     # with it.
-    class FallbackAnnotator < Annotator
+    class FallbackPass < Pass
       def self.applies?(_graph)
         true
       end
 
-      def annotate(graph)
+      def apply(graph)
         graph.nodes.each_value do |node|
           if node.props[:label].nil? && node.props['label']
             node.props[:label] = node.props['label']

data/lib/seafoam/{annotators → passes}/graal.rb

@@ -1,18 +1,19 @@
 module Seafoam
-  module Annotators
-    # The Graal annotator applies if it looks like it was compiled by Graal or
+  module Passes
+    # The Graal pass applies if it looks like it was compiled by Graal or
     # Truffle.
-    class GraalAnnotator < Annotator
+    class GraalPass < Pass
       def self.applies?(graph)
         graph.props.values.any? do |v|
           TRIGGERS.any? { |t| v.to_s.include?(t) }
         end
       end
 
-      def annotate(graph)
-        annotate_nodes graph
-        annotate_edges graph
+      def apply(graph)
+        apply_nodes graph
+        apply_edges graph
         hide_frame_state graph if @options[:hide_frame_state]
+        hide_pi graph if @options[:hide_pi]
         hide_floating graph if @options[:hide_floating]
         reduce_edges graph if @options[:reduce_edges]
         hide_unused_nodes graph
@@ -21,8 +22,10 @@ module Seafoam
       private
 
       # Annotate nodes with their label and kind
-      def annotate_nodes(graph)
+      def apply_nodes(graph)
         graph.nodes.each_value do |node|
+          next if node.props[:label]
+
           # The Java class of the node.
           node_class = node.props.dig(:node_class, :node_class)
 
@@ -93,6 +96,11 @@ module Seafoam
             name_template = 'π'
           end
 
+          # Use a symbol for PiArrayNode.
+          if node_class == 'org.graalvm.compiler.nodes.PiArrayNode'
+            name_template = '[π]'
+          end
+
           # Use a symbol for PhiNode.
           if node_class == 'org.graalvm.compiler.nodes.ValuePhiNode'
             name_template = 'ϕ'
@@ -208,7 +216,7 @@ module Seafoam
       end
 
       # Annotate edges with their label and kind.
-      def annotate_edges(graph)
+      def apply_edges(graph)
         graph.edges.each do |edge|
           if edge.to.props.dig(:node_class, :node_class) == 'org.graalvm.compiler.nodes.ValuePhiNode' && edge.props[:name] == 'values'
             merge_node = edge.to.edges.find { |e| e.props[:name] == 'merge' }.from
@@ -254,8 +262,6 @@ module Seafoam
           # loopBegin edges point from LoopEndNode (continue) and LoopExitNode
           # (break) to the LoopBeginNode. Both are drawn reversed.
           when 'loopBegin'
-            edge.props[:hidden] = true
-
             case edge.to.props.dig(:node_class, :node_class)
             when 'org.graalvm.compiler.nodes.LoopEndNode'
               # If it's from the LoopEnd then it's the control edge to follow.
@@ -311,6 +317,24 @@ module Seafoam
         end
       end
 
+      # Hide pi nodes - they add information for optimisation, but not generally
+      # useful day-to-day for understanding the graph. Connect the user to the
+      # actual object, which may be several steps away.
+      def hide_pi(graph)
+        loop do
+          umodified = true
+          graph.edges.each do |edge|
+            next unless Graal::Pi::PI_NODES.include?(edge.from.props.dig(:node_class, :node_class))
+
+            object = Graal::Pi.follow_pi_object(edge.from)
+            graph.create_edge object, edge.to, edge.props.merge({ synthetic: true })
+            graph.remove_edge edge
+            umodified = false
+          end
+          break if umodified
+        end
+      end
+
       # Hide floating nodes. This highlights just the control flow backbone.
       def hide_floating(graph)
         graph.nodes.each_value do |node|
@@ -332,15 +356,19 @@ module Seafoam
         end
       end
 
-      # Hide nodes that have no non-hidden users and no control flow in. These
-      # would display as a node floating unconnected to the rest of the graph
-      # otherwise. An exception is made for node with an anchor edge coming in,
-      # as some guards are anchored like this.
+      # Hide nodes that have no non-hidden users or edges and no control flow
+      # in. These would display as a node floating unconnected to the rest of
+      # the graph otherwise. An exception is made for node with an anchor edge
+      # coming in, as some guards are anchored like this.
       def hide_unused_nodes(graph)
         loop do
           modified = false
           graph.nodes.each_value do |node|
-            next unless node.outputs.all? { |edge| edge.to.props[:hidden] } &&
+            # Call trees are like Graal graphs but don't have these edges -
+            # don't hide them.
+            next if node.props['truffleCallees']
+
+            next unless node.outputs.all? { |edge| edge.to.props[:hidden] || edge.props[:hidden] } &&
                         node.inputs.none? { |edge| edge.props[:kind] == 'control' } &&
                         node.inputs.none? { |edge| edge.props[:name] == 'anchor' }