ruby-llvm 3.4.2 → 13.0.0

data/lib/llvm/execution_engine.rb

@@ -25,8 +25,8 @@ module LLVM
           @ptr = ptr.read_pointer
         else
           C.dispose_message(error)
-          error.autorelease=false
-          raise RuntimeError, "Error creating JIT compiler: #{message}"
+          error.autorelease = false
+          raise "Error creating JIT compiler: #{message}"
         end
       end
     end
@@ -49,6 +49,13 @@ module LLVM
       TargetDataLayout.from_ptr(C.get_execution_engine_target_data(self))
     end
 
+    # Get the associated target machine.
+    #
+    # @return [TargetMachine]
+    def target_machine
+      TargetMachine.from_ptr(C.get_execution_engine_target_machine(self))
+    end
+
     # Execute the given LLVM::Function with the supplied args (as
     # GenericValues).
     # Important: Call #dispose on the returned GenericValue to
@@ -56,7 +63,7 @@ module LLVM
     def run_function(fun, *args)
       FFI::MemoryPointer.new(FFI.type_size(:pointer) * args.size) do |args_ptr|
         new_values = []
-        args_ptr.write_array_of_pointer fun.params.zip(args).map { |p, a|
+        args_ptr.write_array_of_pointer(fun.params.zip(args).map do |p, a|
           if a.kind_of?(GenericValue)
             a
           else
@@ -64,7 +71,7 @@ module LLVM
             new_values << value
             value
           end
-        }
+        end)
         result = LLVM::GenericValue.from_ptr(
           C.run_function(self, fun, args.size, args_ptr))
         new_values.each(&:dispose)
@@ -77,6 +84,10 @@ module LLVM
       C.get_pointer_to_global(self, global)
     end
 
+    def function_address(name)
+      C.get_function_address(self, name)
+    end
+
     # Returns a ModuleCollection of all the Modules in the engine.
     # @return [ModuleCollection]
     def modules
@@ -116,7 +127,7 @@ module LLVM
           message = errorp.read_string unless errorp.null?
 
           C.dispose_message(error)
-          error.autorelease=false
+          error.autorelease = false
 
           raise "Error removing module: #{message}"
         end
@@ -159,35 +170,6 @@ module LLVM
     end
   end
 
-  class JITCompiler < ExecutionEngine
-    # Create a JIT execution engine.
-    #
-    # @note You should call `LLVM.init_jit` before creating an execution engine.
-    #
-    # @param [LLVM::Module] mod module
-    # @param [Hash{Symbol => Object}] options options
-    # @option options [Integer] :opt_level (3) Optimization level 
-    # @return [ExecutionEngine] Execution engine
-    def initialize(mod, options = {})
-      # Prior to ruby-llvm 3.4.0, signature is initialize(mod, opt_level = 3)
-      if options.kind_of?(Integer)
-        options = { :opt_level => options }
-      end
-
-      options = {
-        :opt_level => 3,
-      }.merge(options)
-
-      super
-    end
-
-    protected
-
-    def create_execution_engine_for_module(out_ee, mod, out_error, options)
-      C.create_jit_compiler_for_module(out_ee, mod, options[:opt_level], out_error)
-    end
-  end
-
   class MCJITCompiler < ExecutionEngine
     # Create a MCJIT execution engine.
     #
@@ -196,7 +178,7 @@ module LLVM
     #
     # @param [LLVM::Module] mod module
     # @param [Hash{Symbol => Object}] options options
-    # @option options [Integer] :opt_level (2) Optimization level 
+    # @option options [Integer] :opt_level (2) Optimization level
     # @option options [Integer] :code_model (0) Code model types
     # @option options [Boolean] :no_frame_pointer_elim (false) Disable frame pointer elimination optimization
     # @option options [Boolean] :enable_fast_i_sel (false) Enables fast-path instruction selection
@@ -204,7 +186,7 @@ module LLVM
     def initialize(mod, options = {})
       options = {
         :opt_level             => 2, # LLVMCodeGenLevelDefault
-        :code_model            => 0, # LLVMCodeModelDefault
+        :code_model            => 0,
         :no_frame_pointer_elim => false,
         :enable_fast_i_sel     => false,
         # TODO
@@ -214,6 +196,31 @@ module LLVM
       super
     end
 
+    def convert_type(type)
+      case type.kind
+      when :integer
+        if type.width <= 8
+          :int8
+        else
+          "int#{type.width}".to_sym
+        end
+      else
+        type.kind
+      end
+    end
+
+    def run_function(fun, *args)
+      args2 = fun.params.map {|e| convert_type(e.type)}
+      ptr = FFI::Pointer.new(function_address(fun.name))
+      raise "Couldn't find function" if ptr.null?
+
+      return_type = convert_type(fun.function_type.return_type)
+      f = FFI::Function.new(return_type, args2, ptr)
+      ret1 = f.call(*args)
+      ret2 = LLVM.make_generic_value(fun.function_type.return_type, ret1)
+      ret2
+    end
+
     protected
 
     def create_execution_engine_for_module(out_ee, mod, out_error, options)
@@ -230,6 +237,8 @@ module LLVM
     end
   end
 
+  JITCompiler = MCJITCompiler
+
   class GenericValue
     # @private
     def to_ptr