mt-lang 0.3.5 → 0.3.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1ba5797b3cf3ff504321adf78e281c017a4ca3aa30088f5d472c01fe157f7b4f
-  data.tar.gz: f1b7c0dd477f0e0868e6b3da0fa7f640ace5ef9d53703a7aa8ab89b8907999c9
+  metadata.gz: 2d5caebfa149a9d43d576f3cbf1b36b175ce70b28bfe474bc942fc2d11f0f1e8
+  data.tar.gz: bf9f4a30b1cabadbed2b2d276dd29d567e118d008000791eefb2ec0e81f80c4f
 SHA512:
-  metadata.gz: 9db5f6898f03726a69d907b0a5dec9d19a52ac3141e8cfdda443797476d9fc8f7b0e426812288babdbba3f48d270499e49998224a4509705286992bfd117e810
-  data.tar.gz: 5de95e68ce290fa2e3f80e5d595e29dba7d19093f91dcc169c6773a027fe80f3866f38a213713709911aeda767ccd28b30eb3b4765b5d58319997aaec587f414
+  metadata.gz: 681c1614c3449f16bb71fbb0f9ee7e9ec10420e302602b131c4ed813b087742e7474ff71e9111f5d35cb1e4315eb440c3c85e29d66e8c733159063cbe946ea01
+  data.tar.gz: 6ea9caabdb7407f64d634f6576135df74e614878f81680777062c2e5f64f75278bb56f71bb48406153e40745e2d2b458e99c46193c9c85837c2b28b3b7aca4b4

data/lib/milk_tea/base.rb

@@ -3,7 +3,7 @@
 require "pathname"
 
 module MilkTea
-  VERSION = "0.3.5"
+  VERSION = "0.3.6"
 
   def self.root
     @root ||= Pathname.new(File.expand_path("../..", __dir__))

data/lib/milk_tea/core/c_backend/expressions.rb

@@ -33,7 +33,12 @@ module MilkTea
             when IR::NullableSpanIndex
               "#{nullable_span_index_helper_name(expression.receiver_type)}(#{emit_expression(expression.receiver)}, #{emit_expression(expression.index)})"
             when IR::Call
-              raise CBackendError, "array-return call must be materialized before C emission" if array_type?(expression.type)
+              if array_type?(expression.type)
+                location = [@source_path, @current_line && "line #{@current_line}"].compact.join(" ")
+                msg = "array-return call must be materialized before C emission"
+                msg += " (#{location})" unless location.empty?
+                raise CBackendError.new(msg, line: @current_line, path: @source_path)
+              end
 
               emit_call_expression(expression)
             when IR::Unary

data/lib/milk_tea/core/c_backend/statements.rb

@@ -23,6 +23,10 @@ module MilkTea
             indent = INDENT * level
             aliases = checked_index_aliases_for_statement(statement)
             alias_lines = emit_checked_index_alias_declarations(aliases, indent)
+            if statement.respond_to?(:line) && statement.line
+              @current_line = statement.line
+              @source_path ||= statement.source_path if statement.respond_to?(:source_path) && statement.source_path
+            end
             line_directive = if @emit_line_directives && statement.respond_to?(:line) && statement.line
                                sp = (statement.respond_to?(:source_path) && statement.source_path) || @source_path
                                sp ? ["#line #{statement.line} #{sp.inspect}"] : []

data/lib/milk_tea/core/c_backend.rb

@@ -55,6 +55,8 @@ module MilkTea
       @checked_index_alias_id = 0
       @loop_guard_id = 0
       @emitted_span_type_names = Set.new
+      @source_path = program.source_path
+      @current_line = nil
     end
 
     def emit
@@ -218,10 +220,16 @@ module MilkTea
       end
 
       constants.each do |constant|
+        @current_line = constant.line if constant.line
+        @source_path ||= constant.source_path if constant.source_path
         lines << "#{constant_storage(constant.type)} #{c_declaration(constant.type, constant.linkage_name)} = #{emit_initializer(constant.value)};"
       end
       @program.globals.each do |global|
         next unless emitted_globals.include?(global)
+        if global.respond_to?(:line) && global.line
+          @current_line = global.line
+          @source_path ||= global.source_path if global.respond_to?(:source_path) && global.source_path
+        end
         lines << "#{global_storage(global.type)} #{c_declaration(global.type, global.linkage_name)} = #{emit_initializer(global.value)};"
       end
       lines << "" unless constants.empty? && @program.globals.empty?

data/lib/milk_tea/core/compile_time.rb

@@ -114,7 +114,12 @@ module MilkTea
                        else
                          @checker.instance_variable_get(:@ctx).types
                        end
-              if (type = types[call_expr.callee.name]) && type.is_a?(Types::Struct)
+              callee_name = if call_expr.callee.is_a?(AST::Specialization) && call_expr.callee.callee.respond_to?(:name)
+                              call_expr.callee.callee.name
+                            elsif call_expr.callee.respond_to?(:name)
+                              call_expr.callee.name
+                            end
+              if callee_name && (type = types[callee_name]) && type.is_a?(Types::Struct)
                 fields = {}
                 call_expr.arguments.each do |argument|
                   val = evaluate_expression(argument.value, scopes:)
@@ -123,6 +128,19 @@ module MilkTea
                 end
                 next fields
               end
+
+              if call_expr.callee.is_a?(AST::Specialization) && @checker.respond_to?(:resolve_type_expression)
+                resolved = @checker.send(:resolve_type_expression, call_expr.callee)
+                if resolved && @checker.respond_to?(:array_type?) && @checker.send(:array_type?, resolved)
+                  values = []
+                  call_expr.arguments.each do |argument|
+                    val = evaluate_expression(argument.value, scopes:)
+                    return nil unless val
+                    values << val
+                  end
+                  next values
+                end
+              end
               
               @checker.send(:evaluate_compile_time_const_value, call_expr, scopes:)
             },

data/lib/milk_tea/core/ir.rb

@@ -6,7 +6,9 @@ module MilkTea
       def initialize(module_name:, includes:, constants:, globals:, opaques:, structs:, unions:, enums:, variants:, static_asserts:, functions:, source_path: nil) = super
     end
     Include = Data.define(:header)
-    Constant = Data.define(:name, :linkage_name, :type, :value)
+    Constant = Data.define(:name, :linkage_name, :type, :value, :line, :source_path) do
+      def initialize(name:, linkage_name:, type:, value:, line: nil, source_path: nil) = super
+    end
     Global = Data.define(:name, :linkage_name, :type, :value)
     OpaqueDecl = Data.define(:name, :linkage_name, :forward_declarable, :source_module) do
       def initialize(name:, linkage_name:, forward_declarable:, source_module: nil) = super

data/lib/milk_tea/core/lowering/block.rb

@@ -353,7 +353,8 @@ module MilkTea
         value = lower_static_storage_initializer(decl.value, env:, expected_type: decl.type ? resolve_type_ref(decl.type) : nil)
         type = decl.type ? resolve_type_ref(decl.type) : infer_expression_type(decl.value, env:)
         linkage_name = value_c_name(decl.name)
-        constant = IR::Constant.new(name: decl.name, linkage_name:, type:, value:)
+        constant = IR::Constant.new(name: decl.name, linkage_name:, type:, value:,
+                                    line: decl.line, source_path: @ctx.current_analysis_path)
         @artifacts.emitted_declarations << constant
         []
       end
@@ -511,7 +512,7 @@ module MilkTea
         env: local_env,
         expected_type: target.type,
         allow_root_statement_foreign: true,
-        materialize_array_calls: !array_type?(target.type),
+        materialize_array_calls: true,
       )
       lowered.concat(prepared_setup)
       if (foreign_call = foreign_call_info(prepared_value, local_env))
@@ -700,7 +701,7 @@ module MilkTea
           env: local_env,
           expected_type: storage_type,
           allow_root_statement_foreign: true,
-          materialize_array_calls: !array_type?(storage_type),
+          materialize_array_calls: true,
         )
         lowered.concat(prepared_setup)
       end

data/lib/milk_tea/core/lowering/declarations.rb

@@ -41,7 +41,8 @@ module MilkTea
             value = lower_static_storage_initializer(decl.value, env: empty_env, expected_type: type)
           end
 
-          IR::Constant.new(name: decl.name, linkage_name: value_c_name(decl.name), type:, value:)
+          IR::Constant.new(name: decl.name, linkage_name: value_c_name(decl.name), type:, value:,
+                           line: decl.line, source_path: @ctx.current_analysis_path)
         end
       end
 

data/lib/milk_tea/core/lowering/resolve.rb

@@ -666,11 +666,11 @@ module MilkTea
         end
 
         if (callable_resolution = resolve_specialized_callable_binding(callee, env:))
-          callable_kind, function_binding, receiver = callable_resolution
+          callable_kind, function_binding, receiver, method_entry_receiver_type = callable_resolution
           if callable_kind == :method
             return [
               :method,
-              function_binding_c_name(function_binding, module_name: function_binding.owner.module_name, receiver_type: function_binding.type.receiver_type),
+              function_binding_c_name(function_binding, module_name: function_binding.owner.module_name, receiver_type: method_entry_receiver_type),
               receiver,
               function_binding.type,
               function_binding,
@@ -1277,7 +1277,7 @@ module MilkTea
         return nil unless binding
 
         type_arguments = resolve_specialization_type_arguments(expression)
-        [callable_kind, instantiate_function_binding_with_receiver(binding, type_arguments, receiver_type:), receiver]
+        [callable_kind, instantiate_function_binding_with_receiver(binding, type_arguments, receiver_type:), receiver, method_entry_receiver_type]
       end
 
       def resolve_default_specialization(expression, env:)

data/lib/milk_tea/core/semantic_analyzer/top_level.rb

@@ -376,6 +376,29 @@ module MilkTea
           else
             callee_name = expression.callee.callee.is_a?(AST::Identifier) ? expression.callee.callee.name : nil
             if callee_name
+              resolved_type = resolve_type_expression(expression.callee)
+              if resolved_type && resolved_type.is_a?(Types::Struct)
+                fields = {}
+                expression.arguments.each do |argument|
+                  val = CompileTime.evaluate(argument.value, resolve_identifier: lambda { |id|
+                    if scopes
+                      binding = lookup_value(id.name, scopes)
+                      return binding.const_value unless binding&.const_value.nil?
+                    end
+                    resolve_current_module_const_value(id.name)
+                  }, resolve_member_access: lambda { |ma|
+                    if (receiver_type = resolve_type_expression(ma.receiver))
+                      next resolve_enum_member_const_value(receiver_type, ma.member)
+                    end
+                    nil
+                  }, resolve_call: lambda { |inner_call|
+                    evaluate_compile_time_call(inner_call, scopes:)
+                  })
+                  return nil unless val
+                  fields[argument.name] = val
+                end
+                return fields
+              end
               func = @ctx.top_level_functions[callee_name]
               if func&.ast&.respond_to?(:const) && func.ast.const
                 evaluate_const_function_body(func, expression.arguments, scopes:)

data/lib/milk_tea/tooling/error_formatter.rb

@@ -16,19 +16,11 @@ module MilkTea
       severity = error.respond_to?(:severity) ? error.severity : :error
       code = error.respond_to?(:code) ? error.code : nil
 
-      source ||= read_source(path)
-      return "#{severity_label(severity)}: #{error_message} at #{path}:#{line}:#{column}" unless source
-
-      source_lines = source.split("\n", -1)
-      source_line = source_lines[line - 1] || ""
-      stripped = source_line.gsub(/\t/, " ")
+      # Strip location suffix from message (errors embed "at path:line:col" in message text)
+      location_suffix = " at #{path}:#{line}:#{column}"
+      error_message = error_message.delete_suffix(location_suffix).strip
 
-      indent = " " * [column - 1, 0].max
-      highlight = if length && length > 1
-                    "^" + "~" * (length - 1)
-                  else
-                    "^"
-                  end
+      source ||= read_source(path)
 
       bold  = color ? "\e[1m"  : ""
       red   = color ? "\e[31m" : ""
@@ -46,9 +38,21 @@ module MilkTea
 
       code_text = code ? "#{cyan}[#{code}]#{reset}" : ""
 
+      return "#{sev_color}#{sev_text}#{code_text}#{reset}: #{error_message} -- #{path}:#{line}:#{column}" unless source
+
+      source_lines = source.split("\n", -1)
+      source_line = source_lines[line - 1] || ""
+      stripped = source_line.gsub(/\t/, " ")
+
+      indent = " " * [column - 1, 0].max
+      highlight = if length && length > 1
+                    "^" + "~" * (length - 1)
+                  else
+                    "^"
+                  end
+
       [
-        "#{bold}#{sev_color}#{sev_text}#{code_text}#{reset}: #{error_message}",
-        "  #{bold}-->#{reset} #{path}:#{line}:#{column}",
+        "#{sev_color}#{sev_text}#{code_text}#{reset}: #{error_message} -- #{path}:#{line}:#{column}",
         "   #{bold}|#{reset}",
         "#{line.to_s.rjust(5)} #{bold}|#{reset} #{stripped}",
         "      #{bold}|#{reset} #{sev_color}#{indent}#{highlight}#{reset}",

data/std/mem/ptr.mt

@@ -10,3 +10,12 @@ extending ptr[T]:
 
     public function store(value: T) -> void:
         unsafe: read(this) = value
+
+    public function load_at(offset: ptr_uint) -> T:
+        return unsafe: read(this + offset)
+
+    public function store_at(offset: ptr_uint, value: T) -> void:
+        unsafe: read(this + offset) = value
+
+    public function cast[U]() -> ptr[U]:
+        return unsafe: ptr[U]<-this

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mt-lang
 version: !ruby/object:Gem::Version
-  version: 0.3.5
+  version: 0.3.6
 platform: ruby
 authors:
 - Long (Teefan) Tran
@@ -602,7 +602,7 @@ metadata:
   homepage_uri: https://teefan.github.io/mt-lang/
   source_code_uri: https://github.com/teefan/mt-lang
 post_install_message: |
-  Milk Tea 0.3.5 installed!
+  Milk Tea 0.3.6 installed!
 
   System requirements:
     - A C compiler (gcc or clang) must be available on PATH