koi-vm-ruby 0.0.4

data/lib/koi-vm-ruby/opcodes/hash_operations/length.rb

@@ -0,0 +1,19 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[LENGTH] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting a hash" unless(vm.data_stack[-1][0] == HASH_)
+      hash = vm.data_stack.pop[1]
+      integer_keys = hash.keys.reject{|x| x[0] != INTEGER_ }.map{|x| x[1]}
+      highest_key = integer_keys.sort.last
+      if(highest_key.nil?)
+        vm.data_stack.push([INTEGER_, 0])
+      else
+        vm.data_stack.push([INTEGER_, highest_key + 1])
+      end
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/opcodes/hash_operations/num_pairs.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[NUM_PAIRS] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting a hash" unless(vm.data_stack[-1][0] == HASH_)
+      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-ruby/opcodes/hash_operations/pairs.rb

@@ -0,0 +1,22 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[PAIRS] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting a hash" unless(vm.data_stack[-1][0] == HASH_)
+      hash = vm.data_stack.pop[1]
+      hashed_hash = {}
+      index = 0
+      hash.each_pair do |k,v|
+        hashed_hash[[INTEGER_, index]] = [HASH_, {
+          [INTEGER_, 0] => k,
+          [INTEGER_, 1] => v
+        }]
+        index += 1
+      end
+      vm.data_stack.push([HASH_, hashed_hash])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/opcodes/hash_operations/push_hash.rb

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[PUSH_HASH] = Proc.new() do |vm|
+      vm.data_stack.push([HASH_, {}])
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/opcodes/hash_operations/set_key.rb

@@ -0,0 +1,14 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[SET_KEY] = Proc.new() do |vm|
+      raise StackError, "Expecting at least three items on the data stack" unless(vm.data_stack.length > 2)
+      raise StackError, "Expecting a hash" unless(vm.data_stack[-3][0] == HASH_)
+      value = vm.data_stack.pop
+      key = vm.data_stack.pop
+      vm.data_stack[-1][1][key] = value
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

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

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  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-ruby/opcodes/io_operations/print.rb

@@ -0,0 +1,13 @@
+module KoiVMRuby
+  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-ruby/opcodes/math_operations/add.rb

@@ -0,0 +1,14 @@
+module KoiVMRuby
+  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-ruby/opcodes/math_operations/divide.rb

@@ -0,0 +1,14 @@
+module KoiVMRuby
+  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-ruby/opcodes/math_operations/multiply.rb

@@ -0,0 +1,14 @@
+module KoiVMRuby
+  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-ruby/opcodes/math_operations/subtract.rb

@@ -0,0 +1,14 @@
+module KoiVMRuby
+  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-ruby/opcodes/push_operations/push_bool.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  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-ruby/opcodes/push_operations/push_float.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  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-ruby/opcodes/push_operations/push_int.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  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-ruby/opcodes/push_operations/push_nil.rb

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  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-ruby/opcodes/push_operations/push_string.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  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-ruby/opcodes/stack_operations/dup.rb

@@ -0,0 +1,11 @@
+module KoiVMRuby
+  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-ruby/opcodes/stack_operations/pop.rb

@@ -0,0 +1,11 @@
+module KoiVMRuby
+  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-ruby/opcodes/stack_operations/stksize.rb

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  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-ruby/opcodes/stack_operations/swap.rb

@@ -0,0 +1,11 @@
+module KoiVMRuby
+  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-ruby/opcodes/stack_operations/top.rb

@@ -0,0 +1,13 @@
+module KoiVMRuby
+  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-ruby/opcodes/stack_operations/typeof.rb

@@ -0,0 +1,18 @@
+module KoiVMRuby
+  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.data_stack.push([STRING_, "hash"]) if(value[0] == HASH_)
+      vm.data_stack.push([STRING_, "function"]) if(value[0] == FUNCTION_)
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

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

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  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-ruby/opcodes/string_operations/strlen.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  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-ruby/opcodes/string_operations/to_string.rb

@@ -0,0 +1,16 @@
+module KoiVMRuby
+  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-ruby/opcodes/variable_operations/get_global.rb

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  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-ruby/opcodes/variable_operations/get_local.rb

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  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-ruby/opcodes/variable_operations/set_global.rb

@@ -0,0 +1,11 @@
+module KoiVMRuby
+  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-ruby/opcodes/variable_operations/set_local.rb

@@ -0,0 +1,11 @@
+module KoiVMRuby
+  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/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 KoiVMRuby
+  
+  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