koi-vm 0.0.1

data/lib/koi-vm/opcodes/flow_control_operations/jump.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[JUMP] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting integer value but got #{operand.inspect}" unless(operand.is_a?(Integer))
+      vm.instruction_pointer = vm.instruction_pointer + operand
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/flow_control_operations/jump_if.rb

@@ -0,0 +1,16 @@
+module KoiVM
+  class VM
+    
+    @@instruction[JUMP_IF] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting integer value but got #{operand.inspect}" unless(operand.is_a?(Integer))
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      if(vm.data_stack.pop[1])
+        vm.instruction_pointer = vm.instruction_pointer + operand
+      else
+        vm.instruction_pointer = vm.instruction_pointer + 2
+      end
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/flow_control_operations/jump_unless.rb

@@ -0,0 +1,16 @@
+module KoiVM
+  class VM
+    
+    @@instruction[JUMP_UNLESS] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting integer value but got #{operand.inspect}" unless(operand.is_a?(Integer))
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      unless(vm.data_stack.pop[1])
+        vm.instruction_pointer = vm.instruction_pointer + operand
+      else
+        vm.instruction_pointer = vm.instruction_pointer + 2
+      end
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/function_operations/call.rb

@@ -0,0 +1,13 @@
+module KoiVM
+  class VM
+    
+    @@instruction[CALL] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting function" unless(vm.data_stack[-1][0] == FUNCTION_)
+      vm.return_stack.push( vm.instruction_pointer + 1 )
+      vm.instruction_pointer = vm.data_stack.pop[1]
+      vm.increment_scope
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/function_operations/push_function.rb

@@ -0,0 +1,15 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PUSH_FUNCTION] = Proc.new() do |vm|
+      vm.data_stack.push([FUNCTION_, vm.instruction_pointer + 2])
+      function_id = vm.opcodes[vm.instruction_pointer + 1]
+      while(true)
+        vm.instruction_pointer = vm.opcodes.slice(vm.instruction_pointer, vm.opcodes.length).index(END_FUNCTION)
+        break if(vm.opcodes.slice(vm.instruction_pointer, 4) == [END_FUNCTION, 0, END_FUNCTION, function_id])
+      end
+      vm.instruction_pointer = vm.instruction_pointer + 4
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/function_operations/return.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[RETURN] = Proc.new() do |vm|
+      raise ReturnStackError, "Expecting at least one item on the return stack" unless(vm.return_stack.length > 0)
+      vm.instruction_pointer = vm.return_stack.pop
+      vm.decrement_scope
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/function_operations/tailcall.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[TAILCALL] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting function" unless(vm.data_stack[-1][0] == FUNCTION_)
+      vm.instruction_pointer = vm.data_stack.pop[1]
+      vm.reload_scope
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/io_operations/gets.rb

@@ -0,0 +1,10 @@
+module KoiVM
+  class VM
+    
+    @@instruction[GETS] = Proc.new() do |vm|
+      vm.data_stack.push([STRING_, $stdin.gets("\n")])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/io_operations/print.rb

@@ -0,0 +1,13 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PRINT] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      value = vm.data_stack.pop
+      raise StackError, "Expecting topmost stack item to be a string but was #{value.class}" unless(value[0] == STRING_)
+      print value[1]
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/math_operations/add.rb

@@ -0,0 +1,14 @@
+module KoiVM
+  class VM
+    
+    @@instruction[ADD] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      vm.data_stack[-2][1] += vm.data_stack[-1][1]
+      vm.data_stack.pop
+      vm.data_stack[-1][0] = INTEGER_ if(vm.data_stack[-1][1].is_a?(Integer))
+      vm.data_stack[-1][0] = FLOAT_ if(vm.data_stack[-1][1].is_a?(Float))
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/math_operations/divide.rb

@@ -0,0 +1,14 @@
+module KoiVM
+  class VM
+    
+    @@instruction[DIVIDE] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      vm.data_stack[-2][1] = vm.data_stack[-2][1] / vm.data_stack[-1][1]
+      vm.data_stack.pop
+      vm.data_stack[-1][0] = INTEGER_ if(vm.data_stack[-1][1].is_a?(Integer))
+      vm.data_stack[-1][0] = FLOAT_ if(vm.data_stack[-1][1].is_a?(Float))
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/math_operations/multiply.rb

@@ -0,0 +1,14 @@
+module KoiVM
+  class VM
+    
+    @@instruction[MULTIPLY] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      vm.data_stack[-2][1] = vm.data_stack[-2][1] * vm.data_stack[-1][1]
+      vm.data_stack.pop
+      vm.data_stack[-1][0] = INTEGER_ if(vm.data_stack[-1][1].is_a?(Integer))
+      vm.data_stack[-1][0] = FLOAT_ if(vm.data_stack[-1][1].is_a?(Float))
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/math_operations/subtract.rb

@@ -0,0 +1,14 @@
+module KoiVM
+  class VM
+    
+    @@instruction[SUBTRACT] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      vm.data_stack[-2][1] -= vm.data_stack[-1][1]
+      vm.data_stack.pop
+      vm.data_stack[-1][0] = INTEGER_ if(vm.data_stack[-1][1].is_a?(Integer))
+      vm.data_stack[-1][0] = FLOAT_ if(vm.data_stack[-1][1].is_a?(Float))
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/push_operations/push_bool.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PUSH_BOOL] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting boolean value but got #{operand.inspect}" unless(operand.is_a?(TrueClass) or operand.is_a?(FalseClass))
+      vm.data_stack.push([BOOL_, operand])
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/push_operations/push_float.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PUSH_FLOAT] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting float value but got #{operand.inspect}" unless(operand.is_a?(Float))
+      vm.data_stack.push([FLOAT_, operand])
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/push_operations/push_int.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PUSH_INT] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting integer value but got #{operand.inspect}" unless(operand.is_a?(Integer))
+      vm.data_stack.push([INTEGER_, operand])
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/push_operations/push_nil.rb

@@ -0,0 +1,10 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PUSH_NIL] = Proc.new() do |vm|
+      vm.data_stack.push([NIL_, nil])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/push_operations/push_string.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[PUSH_STRING] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting string value but got #{operand.inspect}" unless(operand.is_a?(String))
+      vm.data_stack.push([STRING_, operand])
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/stack_operations/dup.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[DUP] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 0)
+      vm.data_stack.push(vm.data_stack[-1])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/stack_operations/pop.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[POP] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      vm.data_stack.pop
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/stack_operations/stksize.rb

@@ -0,0 +1,10 @@
+module KoiVM
+  class VM
+    
+    @@instruction[STKSIZE] = Proc.new() do |vm|
+      vm.data_stack.push([INTEGER_, vm.data_stack.length])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/stack_operations/swap.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[SWAP] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      vm.data_stack.insert(-2, vm.data_stack.pop)
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/stack_operations/top.rb

@@ -0,0 +1,13 @@
+module KoiVM
+  class VM
+    
+    @@instruction[TOP] = Proc.new() do |vm|
+      operand = vm.opcodes[vm.instruction_pointer + 1]
+      raise OperandError, "Expecting operand to be a integer but got #{operand.class}" unless(operand.is_a?(Integer))
+      raise StackError, "Expecting at least #{operand} items on the data stack" unless(vm.data_stack.length > (operand - 1))
+      vm.data_stack.concat(vm.data_stack.slice(-1 - (operand - 1), vm.data_stack.length))
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/stack_operations/typeof.rb

@@ -0,0 +1,16 @@
+module KoiVM
+  class VM
+    
+    @@instruction[TYPEOF] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      value = vm.data_stack.pop
+      vm.data_stack.push([STRING_, "nil"]) if(value[0] == NIL_)
+      vm.data_stack.push([STRING_, "bool"]) if(value[0] == BOOL_)
+      vm.data_stack.push([STRING_, "integer"]) if(value[0] == INTEGER_)
+      vm.data_stack.push([STRING_, "float"]) if(value[0] == FLOAT_)
+      vm.data_stack.push([STRING_, "string"]) if(value[0] == STRING_)
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/string_operations/concat.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[CONCAT] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      raise StackError, "Expecting at least two strings on the stack" unless((vm.data_stack[-1][0] == STRING_) and (vm.data_stack[-2][0] == STRING_))
+      vm.data_stack[-2][1].concat(vm.data_stack.pop[1])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/string_operations/strlen.rb

@@ -0,0 +1,12 @@
+module KoiVM
+  class VM
+    
+    @@instruction[STRLEN] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one items on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting topmost item on stack to be a string" unless((vm.data_stack[-1][0] == STRING_))
+      vm.data_stack.push([INTEGER_, vm.data_stack.pop[1].length])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/string_operations/to_string.rb

@@ -0,0 +1,16 @@
+module KoiVM
+  class VM
+    
+    @@instruction[TO_STRING] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      value = vm.data_stack.pop
+      if(value[0] == NIL_)
+        vm.data_stack.push([STRING_, "nil"])
+      else
+        vm.data_stack.push([STRING_, value[1].to_s])
+      end
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/variable_operations/get_global.rb

@@ -0,0 +1,10 @@
+module KoiVM
+  class VM
+    
+    @@instruction[GET_GLOBAL] = Proc.new() do |vm|
+      vm.data_stack.push( vm.globals[vm.opcodes[vm.instruction_pointer + 1]] )
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/variable_operations/get_local.rb

@@ -0,0 +1,10 @@
+module KoiVM
+  class VM
+    
+    @@instruction[GET_LOCAL] = Proc.new() do |vm|
+      vm.data_stack.push( vm.locals[vm.opcodes[vm.instruction_pointer + 1]] )
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/variable_operations/set_global.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[SET_GLOBAL] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      vm.globals[vm.opcodes[vm.instruction_pointer + 1]] = vm.data_stack.pop
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm/opcodes/variable_operations/set_local.rb

@@ -0,0 +1,11 @@
+module KoiVM
+  class VM
+    
+    @@instruction[SET_LOCAL] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      vm.locals[vm.opcodes[vm.instruction_pointer + 1]] = vm.data_stack.pop
+      vm.instruction_pointer = vm.instruction_pointer + 2
+    end
+    
+  end
+end

data/lib/koi-vm.rb

@@ -0,0 +1,10 @@
+require_files = []
+
+require_files << File.join(File.dirname(__FILE__), "koi-vm", "core.rb")
+require_files << File.join(File.dirname(__FILE__), "koi-vm", "helpers.rb")
+require_files << File.join(File.dirname(__FILE__), "koi-vm", "exceptions.rb")
+require_files.concat Dir[File.join(File.dirname(__FILE__), 'koi-vm', '**', '*.rb')]
+
+require_files.each do |file|
+  require File.expand_path(file)
+end

data/test/performance/.gitignore

@@ -0,0 +1 @@
+*.html

data/test/performance/simple_benchmark.rb

@@ -0,0 +1,87 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'test_helper.rb'))
+
+class SimpleBenchmark < Test::Unit::TestCase
+  
+  require "benchmark"
+  require "stringio"
+  require 'ruby-prof'
+  include KoiVM
+  
+  vm = VM.new
+  
+  test "simple benchmark" do
+    Benchmark.bmbm do |b|
+      b.report("Ruby") do
+        silence do
+          1000.times do
+            x = 10
+            while(x > -1) do
+              print "#{x}, "
+              x -= 1
+            end
+            print "Blast off!\n"
+          end
+        end
+      end
+      
+      b.report("Koi") do
+        silence do
+          1000.times do
+            vm.run [
+              PUSH_INT, 11,
+              PUSH_INT, 1,
+              SUBTRACT,
+              DUP,
+              TO_STRING,
+              PRINT,
+              PUSH_STRING, ", ",
+              PRINT,
+              DUP,
+              PUSH_INT, 0,
+              EQUAL,
+              JUMP_UNLESS, -13,
+              PUSH_STRING, "Blast Off!\n",
+              PRINT
+            ]
+          end
+        end
+      end
+    end
+  end
+  
+  test "simple profile" do
+    silence do
+      result = RubyProf.profile do
+        vm.run [
+          PUSH_INT, 11,
+          PUSH_INT, 1,
+          SUBTRACT,
+          DUP,
+          TO_STRING,
+          PRINT,
+          PUSH_STRING, ", ",
+          PRINT,
+          DUP,
+          PUSH_INT, 0,
+          EQUAL,
+          JUMP_UNLESS, -13,
+          PUSH_STRING, "Blast Off!\n",
+          PRINT
+        ]
+      end
+      file = File.new(File.join(File.dirname(__FILE__), "koi_profile.html"), "w+")
+      printer = RubyProf::GraphHtmlPrinter.new(result)
+      printer.print(file, :min_percent=>0)
+    end
+  end
+  
+  private
+    
+    def silence
+      old_stdout = $stdout
+      $stdout = StringIO.new
+      yield
+      $stdout = old_stdout
+    end
+  
+end

data/test/setup/test_unit_extensions.rb

@@ -0,0 +1,21 @@
+module Test::Unit
+  # Used to fix a minor minitest/unit incompatibility in flexmock
+  AssertionFailedError = Class.new(StandardError)
+  
+  class TestCase
+   
+    def self.test(name, &block)
+      test_name = "test_#{name.gsub(/\s+/,'_')}".to_sym
+      defined = instance_method(test_name) rescue false
+      raise "#{test_name} is already defined in #{self}" if defined
+      if block_given?
+        define_method(test_name, &block)
+      else
+        define_method(test_name) do
+          flunk "No implementation provided for #{name}"
+        end
+      end
+    end
+
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,13 @@
+$test = true
+
+require 'rubygems'
+require 'test/unit'
+require 'stringio'
+
+require_files = []
+require_files << File.join(File.dirname(__FILE__), '..', 'lib', 'koi-vm.rb')
+require_files.concat Dir[File.join(File.dirname(__FILE__), 'setup', '*.rb')]
+
+require_files.each do |file|
+  require File.expand_path(file)
+end