Nixyri/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2025-01-06 10:20:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

codegen - lower str_lit to vector

Browse Source
This commit is contained in:
Travis Staloch 2022-12-13 13:55:59 -08:00 committed by Andrew Kelley
parent 8a0a6b7387
commit 1ebb761244
4 changed files with 53 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/codegen.zig
View File

@ -854,6 +854,14 @@ pub fn generateSymbol(
}
return Result{ .appended = {} };
},
.str_lit => {
const str_lit = typed_value.val.castTag(.str_lit).?.data;
const mod = bin_file.options.module.?;
const bytes = mod.string_literal_bytes.items[str_lit.index..][0..str_lit.len];
try code.ensureUnusedCapacity(str_lit.len);
code.appendSliceAssumeCapacity(bytes);
return Result{ .appended = {} };
},
else => unreachable,
},
else => |t| {

26
src/codegen/llvm.zig
View File

@ -3882,6 +3882,32 @@ pub const DeclGen = struct {
@intCast(c_uint, llvm_elems.len),
);
},
.str_lit => {
// Note, sentinel is not stored
const str_lit = tv.val.castTag(.str_lit).?.data;
const bytes = dg.module.string_literal_bytes.items[str_lit.index..][0..str_lit.len];
const vector_len = @intCast(usize, tv.ty.arrayLen());
assert(vector_len == bytes.len);
const elem_ty = tv.ty.elemType();
const llvm_elems = try dg.gpa.alloc(*llvm.Value, vector_len);
defer dg.gpa.free(llvm_elems);
for (llvm_elems) |*elem, i| {
var byte_payload: Value.Payload.U64 = .{
.base = .{ .tag = .int_u64 },
.data = bytes[i],
};
elem.* = try dg.lowerValue(.{
.ty = elem_ty,
.val = Value.initPayload(&byte_payload.base),
});
}
return llvm.constVector(
llvm_elems.ptr,
@intCast(c_uint, llvm_elems.len),
);
},
else => unreachable,
},

1
test/behavior.zig
View File

@ -161,6 +161,7 @@ test {
_ = @import("behavior/int_div.zig");
_ = @import("behavior/inttoptr.zig");
_ = @import("behavior/ir_block_deps.zig");
_ = @import("behavior/lower_strlit_to_vector.zig");
_ = @import("behavior/math.zig");
_ = @import("behavior/maximum_minimum.zig");
_ = @import("behavior/member_func.zig");

18
test/behavior/lower_strlit_to_vector.zig Normal file
View File

@ -0,0 +1,18 @@
const std = @import("std");
const builtin = @import("builtin");
test "strlit to vector" {
if (builtin.zig_backend == .stage2_x86_64) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_arm) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_sparc64) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_wasm) return error.SkipZigTest; // TODO
if (builtin.zig_backend == .stage2_c) return error.SkipZigTest; // TODO
const strlit = "0123456789abcdef0123456789ABCDEF";
const vec_from_strlit: @Vector(32, u8) = strlit.*;
const arr_from_vec = @as([32]u8, vec_from_strlit);
for (strlit) |c, i|
try std.testing.expect(c == arr_from_vec[i]);
try std.testing.expectEqualSlices(u8, strlit, &arr_from_vec);
}