koi-reference-compiler 0.0.1

data/lib/koi-reference-compiler/node_extensions/hash_literals/hash_assignment.rb

@@ -0,0 +1,17 @@
+module KoiReferenceCompiler
+
+  class HashAssignment < SyntaxNode
+    def compile
+      expression = self.elements[3].compile
+    
+      bytecode = []
+      bytecode.concat( self.elements[0].compile )
+      bytecode.concat( self.elements[1].compile )
+      tmp = bytecode.slice!(-2,2)
+      bytecode.concat( expression )
+      bytecode.concat( tmp )
+      return bytecode
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/hash_literals/hash_literal.rb

@@ -0,0 +1,11 @@
+module KoiReferenceCompiler
+
+  class HashLiteral < SyntaxNode
+    def compile
+      bytecode = [PUSH_HASH]
+      bytecode.concat( self.elements.first.compile )
+      return bytecode
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/hash_literals/key_value.rb

@@ -0,0 +1,13 @@
+module KoiReferenceCompiler
+
+  class KeyValue < SyntaxNode
+    def compile
+      bytecode = []
+      bytecode.concat(self.elements[0].compile)
+      bytecode.concat(self.elements[1].compile)
+      bytecode << SET_KEY
+      return bytecode
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/hash_literals/key_value_list.rb

@@ -0,0 +1,13 @@
+module KoiReferenceCompiler
+
+  class KeyValueList < SyntaxNode
+    def compile
+      bytecode = []
+      self.elements.each do |element|
+        bytecode.concat(element.compile)
+      end
+      return bytecode
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/identifiers/identifier.rb

@@ -0,0 +1,31 @@
+module KoiReferenceCompiler
+
+  class Identifier < SyntaxNode
+    def compile
+      identifier = self.text_value
+      parent = self.parent
+      if(identifier[0] == "$")
+        if(parent.is_a?(Expression) || parent.is_a?(MultitiveExpression) || parent.is_a?(AdditiveExpression) || parent.is_a?(ComparativeExpression) || parent.is_a?(HashAssignment) || parent.is_a?(HashAccess))
+          [ GET_GLOBAL, identifier.to_sym ]
+          
+        elsif(parent.is_a?(Assignment) || parent.is_a?(FunctionDefinition))
+          [ SET_GLOBAL, identifier.to_sym ]
+          
+        else
+          raise CompileError, "Unkown parent for Identfier: #{self.parent.class}"        
+        end
+      else
+        if(parent.is_a?(Expression) || parent.is_a?(MultitiveExpression) || parent.is_a?(AdditiveExpression) || parent.is_a?(ComparativeExpression) || parent.is_a?(HashAssignment) || parent.is_a?(HashAccess))
+          [ GET_LOCAL, identifier.to_sym ]
+          
+        elsif(parent.is_a?(Assignment) || parent.is_a?(FunctionDefinition))
+          [ SET_LOCAL, identifier.to_sym ]
+          
+        else
+          raise CompileError, "Unkown parent for Identfier: #{self.parent.class}"
+        end
+      end
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/literals/false_literal.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class FalseLiteral < SyntaxNode
+    def compile
+      [ PUSH_BOOL, false ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/literals/float_literal.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class FloatLiteral < SyntaxNode
+    def compile
+      [ PUSH_FLOAT, self.text_value.to_f ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/literals/integer_literal.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class IntegerLiteral < SyntaxNode
+    def compile
+      [ PUSH_INT, self.text_value.to_i ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/literals/nil_literal.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class NilLiteral < SyntaxNode
+    def compile
+      [ PUSH_NIL ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/literals/string_literal.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class StringLiteral < SyntaxNode
+    def compile
+      [ PUSH_STRING, self.text_value.slice(1, self.text_value.length - 2) ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/literals/true_literal.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class TrueLiteral < SyntaxNode
+    def compile
+      [ PUSH_BOOL, true ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/addition_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class AdditionOperator < SyntaxNode
+    def compile
+      [ ADD ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/assignment_operator.rb

@@ -0,0 +1,6 @@
+module KoiReferenceCompiler
+
+  class AssignmentOperator < SyntaxNode
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/division_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class DivisionOperator < SyntaxNode
+    def compile
+      [ DIVIDE ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/equality_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class EqualityOperator < SyntaxNode
+    def compile
+      [ EQUAL ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/great_than_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class GreaterThanOperator < SyntaxNode
+    def compile
+      [ GREATER_THAN ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/inequality_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class InequalityOperator < SyntaxNode
+    def compile
+      [ EQUAL, INVERT ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/less_than_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class LessThanOperator < SyntaxNode
+    def compile
+      [ LESS_THAN ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/multiplication_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class MultiplicationOperator < SyntaxNode
+    def compile
+      [ MULTIPLY ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/operators/subtraction_operator.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class SubtractionOperator < SyntaxNode
+    def compile
+      [ SUBTRACT ]
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/node_extensions/statements/statement.rb

@@ -0,0 +1,9 @@
+module KoiReferenceCompiler
+
+  class Statement < SyntaxNode
+    def compile
+      return self.elements.first.compile
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/syntax_node.rb

@@ -0,0 +1,18 @@
+module KoiReferenceCompiler
+  
+  class SyntaxNode
+    
+    attr_accessor :parent, :elements, :offset, :text_value
+    
+    def initialize(text_value, offset, elements = nil)
+      @parent = nil
+      @elements = elements
+      @offset = offset
+      @text_value = text_value
+      elements.each do |element|
+        element.parent = self
+      end unless(elements.nil?)
+    end
+  end
+  
+end

data/lib/koi-reference-compiler/vm_opcodes.rb

@@ -0,0 +1,70 @@
+module KoiReferenceCompiler
+
+  # 0
+  PUSH_NIL      = 0
+  PUSH_BOOL     = 1
+  PUSH_INT      = 2
+  PUSH_FLOAT    = 3
+  PUSH_STRING   = 4
+  
+  # 20
+  JUMP          = 20
+  JUMP_IF       = 21
+  JUMP_UNLESS   = 22
+  
+  # 40
+  POP           = 40
+  SWAP          = 41
+  DUP           = 42
+  STKSIZE       = 43
+  TYPEOF        = 44
+  TOP           = 45
+  
+  # 60
+  ADD           = 60
+  SUBTRACT      = 61
+  MULTIPLY      = 62
+  DIVIDE        = 63
+  
+  # 80
+  CONCAT        = 80
+  STRLEN        = 81
+  TO_STRING     = 82
+  
+  # 100
+  EQUAL         = 100
+  LESS_THAN     = 101
+  GREATER_THAN  = 102
+  INVERT        = 103
+  
+  # 120
+  PRINT         = 120
+  GETS          = 121
+  
+  # 140
+  SET_LOCAL     = 140
+  GET_LOCAL     = 141
+  SET_GLOBAL    = 142
+  GET_GLOBAL    = 143
+  
+  # 160
+  PUSH_FUNCTION = 160
+  END_FUNCTION  = 161
+  CALL          = 162
+  RETURN        = 163
+  TAILCALL      = 164
+  
+  # 180
+  PUSH_HASH     = 180
+  SET_KEY       = 181
+  GET_KEY       = 182
+  HAS_KEY       = 183
+  LENGTH        = 184
+  NUM_PAIRS     = 185
+  PAIRS         = 186
+  
+  # 240
+  NO_OP         = 240
+  EXIT          = 255
+
+end

data/test/compiler/unit/ast_hash_loader/ast_hash_loader_test.rb

@@ -0,0 +1,46 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', 'test_helper.rb'))
+
+class AstHashLoaderTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should turn AST hash into object" do
+    ast_hash = {
+      :parent => nil,
+      :elements => nil,
+      :offset => 0,
+      :name => "Identifier",
+      :text_value => "test"
+    }
+    tree = Compiler.load_ast_hash(ast_hash)
+    assert_kind_of Identifier, tree
+    assert_equal nil, tree.parent
+    assert_equal nil, tree.elements
+    assert_equal 0, tree.offset
+    assert_equal "test", tree.text_value
+  end
+  
+  test "should turn nested AST hash into nested objects" do
+    built_hash = {
+      :parent => nil,
+      :elements => nil,
+      :offset => 0,
+      :name => "Assignment",
+      :text_value => "test = 1"
+    }
+    built_hash[:elements] = [
+      {
+      :parent => built_hash,
+      :elements => nil,
+      :offset => 0,
+      :name => "Identifier",
+      :text_value => "test"
+      }
+    ]
+    tree = Compiler.load_ast_hash(built_hash)
+    assert_kind_of Assignment, tree
+    assert_kind_of Identifier, tree.elements.first
+    assert_equal tree, tree.elements.first.parent
+  end
+  
+end

data/test/compiler/unit/node_extensions/assignment/assignment_test.rb

@@ -0,0 +1,22 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class AssignmentTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile Assignment" do
+    tree = Assignment.new("test = 1", 0, [
+      Identifier.new("test", 0),
+      AssignmentOperator.new("=", 5),
+      Expression.new("1", 7, [
+        IntegerLiteral.new("1", 7)
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      PUSH_INT, 1, 
+      SET_LOCAL, :test
+    ], bytecode
+  end
+  
+end

data/test/compiler/unit/node_extensions/blocks/block_test.rb

@@ -0,0 +1,37 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'test_helper.rb'))
+
+class BlockTest < Test::Unit::TestCase
+
+  include KoiReferenceCompiler
+  
+  test "should compile Block" do
+    tree = Block.new("test = 1\ntest2 = 2", 0, [
+      Statement.new("test = 1\n", 0, [
+        Assignment.new("test = 1", 0, [
+          Identifier.new("test", 0),
+          AssignmentOperator.new("=", 5),
+          Expression.new("1", 7, [
+            IntegerLiteral.new("1", 7)
+          ])
+        ])
+      ]),
+      Statement.new("test2 = 2", 10, [
+        Assignment.new("test2 = 2", 10, [
+          Identifier.new("test2", 9),
+          AssignmentOperator.new("=", 15),
+          Expression.new("2", 17, [
+            IntegerLiteral.new("2", 17)
+          ])
+        ])
+      ])
+    ])
+    bytecode = tree.compile
+    assert_equal [
+      PUSH_INT, 1, 
+      SET_LOCAL, :test,
+      PUSH_INT, 2, 
+      SET_LOCAL, :test2
+    ], bytecode
+  end
+  
+end