furnace 0.3.1 → 0.4.0.beta.1

data/lib/furnace/ssa/value.rb

@@ -0,0 +1,62 @@
+module Furnace
+  class SSA::Value
+    def initialize
+      @uses = Set.new
+    end
+
+    def initialize_copy(original)
+      @uses = Set.new
+    end
+
+    def type
+      SSA.void
+    end
+
+    def constant?
+      false
+    end
+
+    def add_use(use)
+      @uses.add use
+    end
+
+    def remove_use(use)
+      @uses.delete use
+    end
+
+    def each_use(&block)
+      @uses.each(&block)
+    end
+
+    def use_count
+      @uses.count
+    end
+
+    def used?
+      @uses.any?
+    end
+
+    def replace_all_uses_with(value)
+      each_use do |user|
+        user.replace_uses_of self, value
+      end
+    end
+
+    def to_value
+      self
+    end
+
+    def ==(other)
+      other.respond_to?(:to_value) &&
+          equal?(other.to_value)
+    end
+
+    def pretty_print(p=SSA::PrettyPrinter.new)
+      inspect_as_value(p)
+    end
+
+    def inspect_as_value(p=SSA::PrettyPrinter.new)
+      p.text inspect
+    end
+  end
+end

data/lib/furnace/ssa.rb

@@ -0,0 +1,66 @@
+module Furnace
+  module SSA
+    def self.void
+      SSA::VoidType.instance
+    end
+
+    def self.void_value
+      SSA::VoidType.value
+    end
+
+    def self.class_name_to_opcode(klass)
+      klass.to_s.split('::').last.gsub(/([a-z]|^)([A-Z])/) do
+        if $1.empty?
+          $2.downcase
+        else
+          "#{$1}_#{$2.downcase}"
+        end
+      end.gsub(/_insn$/, '')
+    end
+
+    def self.opcode_to_class_name(opcode)
+      opcode.to_s.gsub(/(?:([a-z])_|^)([a-z])/) do
+        if $1.nil?
+          $2.upcase
+        else
+          "#{$1}#{$2.upcase}"
+        end
+      end + 'Insn'
+    end
+
+    def self.inspect_type(type)
+      type ?  type.inspect : '<?>'
+    end
+  end
+
+  require_relative 'ssa/pretty_printer'
+
+  require_relative 'ssa/type'
+  require_relative 'ssa/generic_type'
+
+  require_relative 'ssa/types/void'
+  require_relative 'ssa/types/basic_block'
+  require_relative 'ssa/types/function'
+
+  require_relative 'ssa/value'
+  require_relative 'ssa/constant'
+  require_relative 'ssa/named_value'
+  require_relative 'ssa/argument'
+  require_relative 'ssa/user'
+
+  require_relative 'ssa/instruction'
+  require_relative 'ssa/instruction_syntax'
+  require_relative 'ssa/generic_instruction'
+  require_relative 'ssa/terminator_instruction'
+
+  require_relative 'ssa/basic_block'
+  require_relative 'ssa/function'
+
+  require_relative 'ssa/instructions/phi'
+  require_relative 'ssa/instructions/branch'
+  require_relative 'ssa/instructions/return'
+
+  require_relative 'ssa/module'
+
+  require_relative 'ssa/builder'
+end

data/lib/furnace/transform/iterative.rb

@@ -0,0 +1,27 @@
+module Furnace
+  module Transform
+    class Iterative
+      def initialize(stages)
+        @stages = stages
+      end
+
+      def run(context)
+        self_changed = false
+
+        loop do
+          changed = false
+
+          @stages.each do |stage|
+            return self_changed if stage.nil?
+
+            if stage.run(context)
+              self_changed = changed = true
+            end
+          end
+
+          return self_changed unless changed
+        end
+      end
+    end
+  end
+end

data/lib/furnace/transform/pipeline.rb

@@ -5,14 +5,14 @@ module Furnace
         @stages = stages
       end
 
-      def run(*sequence)
+      def run(context)
         @stages.each do |stage|
           break if stage.nil?
 
-          sequence = stage.transform *sequence
+          stage.run context
         end
 
-        sequence
+        true
       end
     end
   end

data/lib/furnace/version.rb

@@ -1,3 +1,3 @@
 module Furnace
-  VERSION = "0.3.1"
+  VERSION = "0.4.0.beta.1"
 end

data/lib/furnace.rb

@@ -6,7 +6,7 @@
 # data structure being used:
 #
 #  * Abstract syntax trees: {AST}
-#  * Control flow graphs: {CFG}
+#  * Static single assignment representation: {SSA}
 #  * Transformations: {Transform}
 #
 # Additionally, a simple {Graphviz} adapter is provided.
@@ -14,10 +14,10 @@ module Furnace
   require "furnace/version"
 
   require "furnace/ast"
-  require "furnace/cfg"
+  require "furnace/ssa"
 
   require "furnace/transform/pipeline"
-  require "furnace/transform/iterative_process"
+  require "furnace/transform/iterative"
 
   require "furnace/graphviz"
 end

data/test/ast_test.rb

@@ -1,6 +1,8 @@
 require_relative 'test_helper'
 
 describe AST::Node do
+  extend Furnace::AST::Sexp
+
   class MetaNode < AST::Node
     attr_reader :meta
   end
@@ -60,6 +62,17 @@ describe AST::Node do
     AST::Node.new(:a, [ :sym,
       AST::Node.new(:b, [ @node, @node ])
     ]).to_sexp.should.equal "(a :sym\n  (b\n    (node 0 1)\n    (node 0 1)))"
+    AST::Node.new(:a, [
+      { :foo => :bar, :baz => 100, :val => AST::Node.new(:val, [ @node ]) }, @node
+    ]).to_sexp.should == <<-END.rstrip
+(a {
+    :foo => :bar
+    :baz => 100
+    :val => (val
+      (node 0 1))
+  }
+  (node 0 1))
+    END
   end
 
   it 'should return self in to_ast' do
@@ -86,9 +99,19 @@ describe AST::Node do
     end
     @node.should.equal mock_node
   end
+
+  it 'should allow to decompose nodes with a, b = *node' do
+    node = s(:gasgn, :$foo, s(:integer, 1))
+
+    var_name, value = *node
+    var_name.should.equal :$foo
+    value.should.equal s(:integer, 1)
+  end
 end
 
 describe AST::Processor do
+  extend Furnace::AST::Sexp
+
   def have_sexp(text)
     text = text.lines.map { |line| line.sub /^ +\|(.+)/, '\1' }.join.rstrip
     lambda { |ast| ast.to_sexp == text }
@@ -162,8 +185,6 @@ describe AST::Processor do
     SEXP
   end
 
-  extend Furnace::AST::Sexp
-
   it 'should build sexps' do
     s(:add,
       s(:integer, 1),
@@ -182,4 +203,12 @@ describe AST::Processor do
     -> { @processor.process(nil) }.should.raise NoMethodError, %r|to_ast|
     -> { @processor.process([]) }.should.raise NoMethodError, %r|to_ast|
   end
-end
+
+  it 'should allow to visit nodes with process_all(node)' do
+    @processor.process_all s(:foo, s(:bar), s(:integer, 1))
+    @processor.counts.should.equal({
+      bar:     1,
+      integer: 1,
+    })
+  end
+end