resolve type of comparison op to @vector(N, bool) if lhs is a vector (#…

…1982) Co-authored-by: Techatrix <techatrix@mailbox.org>
zigtools · Aug 5, 2024 · f2f5339 · f2f5339
1 parent dfe958e
commit f2f5339
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/src/analysis.zig b/src/analysis.zig
@@ -2071,6 +2071,28 @@ fn resolveTypeOfNodeUncached(analyser: *Analyser, node_handle: NodeWithHandle) e
         .greater_than,
         .less_or_equal,
         .greater_or_equal,
+        => {
+            const ty = try analyser.resolveTypeOfNodeInternal(
+                .{ .node = datas[node].lhs, .handle = handle },
+            ) orelse return try Type.typeValFromIP(analyser, .bool_type);
+            const typeof = ty.typeOf(analyser);
+
+            if (typeof.data == .ip_index) {
+                const key = analyser.ip.indexToKey(typeof.data.ip_index.index);
+                if (key == .vector_type) {
+                    const vector_ty_ip_index = try analyser.ip.get(analyser.gpa, .{
+                        .vector_type = .{
+                            .len = key.vector_type.len,
+                            .child = .bool_type,
+                        },
+                    });
+
+                    return try Type.typeValFromIP(analyser, vector_ty_ip_index);
+                }
+            }
+            return try Type.typeValFromIP(analyser, .bool_type);
+        },
+
         .bool_and,
         .bool_or,
         .bool_not,

diff --git a/tests/lsp_features/inlay_hints.zig b/tests/lsp_features/inlay_hints.zig
@@ -222,6 +222,10 @@ test "inlay destructuring" {
     , .{ .kind = .Type });
 }
 test "var decl" {
+    try testInlayHints(
+        \\const a<@Vector(2,u8)> = @Vector(2, u8){1,2};
+        \\const foo<@Vector(2,bool)> = a == a;
+    , .{ .kind = .Type });
     try testInlayHints(
         \\const foo<comptime_int> = 5;
     , .{ .kind = .Type });