renames Typed to Vm

lib/vm/tree/assignment.rb

@@ -0,0 +1,13 @@
+module Vm
+  module Tree
+    class Assignment < Statement
+      attr_accessor :name , :value
+      def initialize(n = nil , v = nil )
+        @name , @value = n , v
+      end
+      def to_s
+        "#{name} = #{value}\n"
+      end
+    end
+  end
+end

lib/vm/tree/basic_values.rb

@@ -0,0 +1,55 @@
+module Vm
+  module Tree
+    class IntegerExpression < Expression
+      include ValuePrinter
+      attr_accessor :value
+      def initialize(value)
+        @value = value
+      end
+    end
+    class FloatExpression < Expression
+      include ValuePrinter
+      attr_accessor :value
+      def initialize(value)
+        @value = value
+      end
+    end
+    class TrueExpression < Expression
+      def to_s
+        "true"
+      end
+    end
+    class FalseExpression < Expression
+      def to_s
+        "false"
+      end
+    end
+    class NilExpression < Expression
+      def to_s
+        "nil"
+      end
+    end
+    class StringExpression < Expression
+      include ValuePrinter
+      attr_accessor :value
+      def initialize(value)
+        @value = value
+      end
+    end
+    class NameExpression < Expression
+      include ValuePrinter
+      attr_accessor :value
+      alias :name :value
+      def initialize(value)
+        @value = value
+      end
+    end
+    class ClassExpression < Expression
+      include ValuePrinter
+      attr_accessor :value
+      def initialize(value)
+        @value = value
+      end
+    end
+  end
+end

lib/vm/tree/call_site.rb

@@ -0,0 +1,12 @@
+module Vm
+  module Tree
+    class CallSite < Expression
+      attr_accessor :name , :receiver , :arguments
+
+      def to_s
+        str = receiver ? "#{receiver}.#{name}" : name.to_s
+        str + arguments.collect{|a| a.to_s }.join(",")
+      end
+    end
+  end
+end

lib/vm/tree/field_access.rb

@@ -0,0 +1,10 @@
+module Vm
+  module Tree
+    class FieldAccess < Expression
+      attr_accessor :receiver , :field
+      def to_s
+        "#{receiver}.#{field}"
+      end
+    end
+  end
+end

lib/vm/tree/if_statement.rb

@@ -0,0 +1,12 @@
+module Vm
+  module Tree
+    class IfStatement < Statement
+      attr_accessor :branch_type , :condition , :if_true , :if_false
+      def to_s
+        str = "if_#{branch_type}(#{condition}) \n  #{if_true}\n"
+        str += "else\n  #{if_false}\n" if if_false
+        str + "end\n"
+      end
+    end
+  end
+end

lib/vm/tree/operator_expression.rb

@@ -0,0 +1,10 @@
+module Vm
+  module Tree
+    class OperatorExpression < Expression
+      attr_accessor :operator , :left_expression , :right_expression
+      def to_s
+        "#{left_expression} #{operator} #{right_expression}"
+      end
+    end
+  end
+end

lib/vm/tree/return_statement.rb

@@ -0,0 +1,11 @@
+module Vm
+  module Tree
+    class ReturnStatement < Statement
+      attr_accessor :return_value
+
+      def to_s
+        "return #{return_value}"
+      end
+    end
+  end
+end

lib/vm/tree/statements.rb

@@ -0,0 +1,9 @@
+module Vm
+  class Statements < Statement
+    attr_accessor :statements
+    def to_s
+      return "" unless statements
+      statements.collect() { |s| s.to_s }.join
+    end
+  end
+end

lib/vm/tree/to_code.rb

@@ -0,0 +1,132 @@
+module Vm
+
+  def self.ast_to_code statement
+    compiler = ToCode.new
+    compiler.process statement
+  end
+
+  class ToCode < AST::Processor
+
+    def handler_missing node
+      raise "No handler  on_#{node.type}(node)"
+    end
+
+    def on_parameters statement
+      params = {}
+      statement.children.each do |param , type , name|
+        type , name = *param
+        params[name] = type
+      end
+      params
+    end
+
+    def on_while_statement statement
+      branch_type , condition , statements = *statement
+      w = Tree::WhileStatement.new()
+      w.branch_type = branch_type
+      w.condition = process(condition)
+      w.statements = process(statements)
+      w
+    end
+
+    def on_if_statement statement
+      branch_type , condition , if_true , if_false = *statement
+      w = Tree::IfStatement.new()
+      w.branch_type = branch_type
+      w.condition = process(condition)
+      w.if_true = process(if_true)
+      w.if_false = process(if_false)
+      w
+    end
+
+    def process_first code
+      raise "Too many children #{code.inspect}" if code.children.length != 1
+      process code.children.first
+    end
+    alias  :on_conditional :process_first
+    alias  :on_condition :process_first
+    alias  :on_field :process_first
+
+    def on_statements statement
+      w = Statements.new()
+      return w unless statement.children
+      return w unless statement.children.first
+      w.statements = process_all(statement.children)
+      w
+    end
+    alias :on_true_statements :on_statements
+    alias :on_false_statements :on_statements
+
+    def on_return statement
+      w = Tree::ReturnStatement.new()
+      w.return_value = process(statement.children.first)
+      w
+    end
+
+    def on_operator_value statement
+      operator , left_e , right_e = *statement
+      w = Tree::OperatorExpression.new()
+      w.operator = operator
+      w.left_expression = process(left_e)
+      w.right_expression = process(right_e)
+      w
+    end
+
+    def on_field_access statement
+      receiver_ast , field_ast = *statement
+      w = Tree::FieldAccess.new()
+      w.receiver = process(receiver_ast)
+      w.field = process(field_ast)
+      w
+    end
+
+    def on_receiver expression
+      process expression.children.first
+    end
+
+    def on_call statement
+      name_s , arguments , receiver = *statement
+      w = Tree::CallSite.new()
+      w.name = name_s.children.first
+      w.arguments = process_all(arguments)
+      w.receiver = process(receiver)
+      w
+    end
+
+    def on_int expression
+      Tree::IntegerExpression.new(expression.children.first)
+    end
+
+    def on_true _expression
+      Tree::TrueExpression.new
+    end
+
+    def on_false _expression
+      Tree::FalseExpression.new
+    end
+
+    def on_nil _expression
+      Tree::NilExpression.new
+    end
+
+    def on_name statement
+      Tree::NameExpression.new(statement.children.first)
+    end
+    def on_string expression
+      Tree::StringExpression.new(expression.children.first)
+    end
+
+    def on_class_name expression
+      Tree::ClassExpression.new(expression.children.first)
+    end
+
+    def on_assignment statement
+      name , value = *statement
+      w = Vm::Tree::Assignment.new()
+      w.name = process name
+      w.value = process(value)
+      w
+    end
+
+  end
+end

lib/vm/tree/while_statement.rb

@@ -0,0 +1,11 @@
+module Vm
+  module Tree
+    class WhileStatement < Statement
+      attr_accessor :branch_type , :condition , :statements
+      def to_s
+        str = "while_#{branch_type}(#{condition}) do\n"
+        str + statements.to_s + "\nend\n"
+      end
+    end
+  end
+end