ruby-llvm 3.5.0 → 13.0.1

data/lib/llvm/execution_engine.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'llvm'
 require 'llvm/core'
 require 'llvm/target'
@@ -25,8 +27,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
@@ -63,7 +65,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
@@ -71,7 +73,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)
@@ -84,6 +86,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
@@ -123,7 +129,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
@@ -166,35 +172,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.
     #
@@ -211,7 +188,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
@@ -221,6 +198,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)
@@ -237,6 +239,8 @@ module LLVM
     end
   end
 
+  JITCompiler = MCJITCompiler
+
   class GenericValue
     # @private
     def to_ptr