koi-vm-ruby 0.0.4

data/lib/koi-vm-ruby.rb

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

data/lib/koi-vm-ruby/accessors.rb

@@ -0,0 +1,63 @@
+module KoiVMRuby
+  class VM
+  
+    def data_stack
+      return @state[:fiber][:data_stack]
+    end
+    
+    def data_stack=(stack)
+      @state[:fiber][:data_stack] = stack
+    end
+    
+    def return_stack
+      return @state[:fiber][:return_stack]
+    end
+    
+    def return_stack=(stack)
+      @state[:fiber][:return_stack] = stack
+    end
+    
+    def instruction_pointer
+      return @state[:fiber][:instruction_pointer]
+    end
+    
+    def instruction_pointer=(ip)
+      @state[:fiber][:instruction_pointer] = ip
+    end
+    
+    def opcodes
+      return @state[:opcodes]
+    end
+    
+    def locals
+      return @state[:fiber][:locals][@state[:fiber][:level]]
+    end
+    
+    def globals
+      return @state[:globals]
+    end
+    
+    def increment_scope
+      @state[:fiber][:level] += 1
+      @state[:fiber][:locals][@state[:fiber][:level]] = @state[:fiber][:locals][@state[:fiber][:level] - 1].dup
+    end
+    
+    def decrement_scope
+      @state[:fiber][:locals].delete_at(@state[:fiber][:level])
+      @state[:fiber][:level] -= 1
+    end
+    
+    def reload_scope
+      @state[:fiber][:locals][@state[:fiber][:level]] = {}
+    end
+    
+    def level
+      return @state[:fiber][:level]
+    end
+    
+    def level=(level)
+      @state[:fiber][:level] = level
+    end
+  
+  end
+end

data/lib/koi-vm-ruby/core.rb

@@ -0,0 +1,48 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction = []
+    
+    attr_accessor :state, :files
+    
+    def initialize(state = {}, opcodes = [])
+      default_state = {
+        :opcodes => opcodes,
+        :globals => {},
+        :fiber => {
+          :data_stack => [],
+          :return_stack => [],
+          :locals => [{}],
+          :instruction_pointer => 0,
+          :level => 0
+        }
+      }
+      @state = default_state.merge(state)
+      @files = {} 
+      raise ArgumentError, "state[:fibers][x] must be a hash" unless(@state[:fiber].is_a?(Hash))
+      raise ArgumentError, "state[:fibers][x][:stack] must be an Array" unless(@state[:fiber][:data_stack].is_a?(Array))
+      raise ArgumentError, "state[:fibers][x][:stack] must be an Array" unless(@state[:fiber][:return_stack].is_a?(Array))
+      raise ArgumentError, "state[:fibers][x][:instruction_pointer] must be an Integer" unless(@state[:fiber][:instruction_pointer].is_a?(Integer))
+      raise ArgumentError, "state[:fibers][x][:level] must be an Integer" unless(@state[:fiber][:level].is_a?(Integer))
+      raise ArgumentError, "state[:globals] must be a hash" unless(@state[:globals].is_a?(Hash))
+      raise ArgumentError, "state[:opcodes] must be an array" unless(@state[:opcodes].is_a?(Array))
+    end
+  
+    def run(opcodes = [])
+      @state[:opcodes].concat(opcodes)
+      opcode_size = @state[:opcodes].size
+      while (@state[:fiber][:instruction_pointer] < opcode_size)
+        begin
+          break if(@@instruction[@state[:opcodes][@state[:fiber][:instruction_pointer]]].call(self) == true)
+        rescue NoMethodError
+          @state.delete(:opcodes)
+          puts "\n\n" + @state.inspect + "\n\n" unless(defined?($test) && $test == true)
+          raise 
+        end
+      end
+      
+      return @state
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/exceptions.rb

@@ -0,0 +1,8 @@
+class StackError < RuntimeError
+end
+
+class ReturnStackError < RuntimeError
+end
+
+class OperandError < RuntimeError
+end

data/lib/koi-vm-ruby/helpers.rb

@@ -0,0 +1,5 @@
+module KoiVMRuby
+  class VM
+  
+  end
+end

data/lib/koi-vm-ruby/opcodes/_opcode_constants.rb

@@ -0,0 +1,68 @@
+module KoiVMRuby
+  # 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/lib/koi-vm-ruby/opcodes/_value_constants.rb

@@ -0,0 +1,11 @@
+module KoiVMRuby
+  NIL_       = 0
+  BOOL_      = 1
+  INTEGER_   = 2
+  FLOAT_     = 3
+  STRING_    = 4
+  HASH_      = 5
+  
+  
+  FUNCTION_  = 16
+end

data/lib/koi-vm-ruby/opcodes/comparative_operations/equal.rb

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

data/lib/koi-vm-ruby/opcodes/comparative_operations/greater_than.rb

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

data/lib/koi-vm-ruby/opcodes/comparative_operations/invert.rb

@@ -0,0 +1,12 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[INVERT] = Proc.new() do |vm|
+      raise StackError, "Expecting at least one item on the data stack" unless(vm.data_stack.length > 0)
+      raise StackError, "Expecting boolean value" unless(vm.data_stack[-1][0] == BOOL_)
+      vm.data_stack[-1][1] = !vm.data_stack[-1][1]
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/opcodes/comparative_operations/less_than.rb

@@ -0,0 +1,14 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[LESS_THAN] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      result = [BOOL_, vm.data_stack[-2][1] < vm.data_stack[-1][1]]
+      vm.data_stack.pop
+      vm.data_stack.pop
+      vm.data_stack.push(result)
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/opcodes/control_operations/exit.rb

@@ -0,0 +1,10 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[EXIT] = Proc.new() do |vm|
+      vm.instruction_pointer = vm.instruction_pointer + 1
+      true
+    end
+    
+  end
+end

data/lib/koi-vm-ruby/opcodes/control_operations/no_op.rb

@@ -0,0 +1,9 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[NO_OP] = Proc.new() do |vm|
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end

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

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

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

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

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

@@ -0,0 +1,19 @@
+module KoiVMRuby
+  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) + vm.instruction_pointer
+        if(vm.opcodes.slice(vm.instruction_pointer, 4) == [END_FUNCTION, 0, END_FUNCTION, function_id])
+          break
+        else
+          vm.instruction_pointer = vm.instruction_pointer + 1
+        end
+      end
+      vm.instruction_pointer = vm.instruction_pointer + 4
+    end
+    
+  end
+end

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

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

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

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

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

@@ -0,0 +1,18 @@
+module KoiVMRuby
+  class VM
+    
+    @@instruction[HAS_KEY] = Proc.new() do |vm|
+      raise StackError, "Expecting at least two items on the data stack" unless(vm.data_stack.length > 1)
+      raise StackError, "Expecting a hash" unless(vm.data_stack[-2][0] == HASH_)
+      key = vm.data_stack.pop
+      hash = vm.data_stack.pop[1]
+      if(hash.has_key?(key))
+        vm.data_stack.push([BOOL_, true])
+      else
+        vm.data_stack.push([BOOL_, false])
+      end
+      vm.instruction_pointer = vm.instruction_pointer + 1
+    end
+    
+  end
+end