diff --git a/examples/211-mapper-projection-spread.sx b/examples/211-mapper-projection-spread.sx new file mode 100644 index 0000000..9121340 --- /dev/null +++ b/examples/211-mapper-projection-spread.sx @@ -0,0 +1,18 @@ +// Phase 6 — `mapper(..sources.value)`: project a method over a pack and spread +// the results into a closure call. The mapper lambda's params are contextually +// typed from the `Closure(...)` parameter even though `apply` is a pack-fn. + +#import "modules/std.sx"; + +VL :: protocol(T: Type) { get :: () -> T; } +IntCell :: struct { v: s64; } +impl VL(s64) for IntCell { get :: (self: *IntCell) -> s64 => self.v; } + +apply :: (mapper: Closure(s64, s64) -> s64, ..sources: VL) -> s64 { + return mapper(..sources.get); // (a, b) => a + b applied to (s0.get(), s1.get()) +} + +main :: () -> s32 { + print("{}\n", apply((a, b) => a + b, IntCell.{ v = 40 }, IntCell.{ v = 2 })); // 42 + 0; +} diff --git a/src/ir/lower.zig b/src/ir/lower.zig index cd2ac80..167d435 100644 --- a/src/ir/lower.zig +++ b/src/ir/lower.zig @@ -9195,7 +9195,13 @@ pub const Lowering = struct { if (isPackParam(p)) break; if (ri >= call_node.args.len) break; if (!p.is_comptime) { + // Contextually type the arg from the param (so a lambda arg + // `(x) => …` takes its param types from a `Closure(...)` param). + const saved_tt = self.target_type; + const pty = self.resolveParamType(&p); + if (pty != .unresolved) self.target_type = pty; args.append(self.alloc, self.lowerExpr(call_node.args[ri])) catch return self.builder.constInt(0, .void); + self.target_type = saved_tt; } ri += 1; } diff --git a/tests/expected/211-mapper-projection-spread.exit b/tests/expected/211-mapper-projection-spread.exit new file mode 100644 index 0000000..573541a --- /dev/null +++ b/tests/expected/211-mapper-projection-spread.exit @@ -0,0 +1 @@ +0 diff --git a/tests/expected/211-mapper-projection-spread.txt b/tests/expected/211-mapper-projection-spread.txt new file mode 100644 index 0000000..d81cc07 --- /dev/null +++ b/tests/expected/211-mapper-projection-spread.txt @@ -0,0 +1 @@ +42