mirrors/zig
1
0
Fork 0
mirror of https://codeberg.org/ziglang/zig.git synced 2025-12-06 05:44:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

std.compress.lzma2: optimize appendLz

Browse source 
make the hot loop be a for loop without any failures or allocation.
change a O(N) addition into O(1)
This commit is contained in:
Andrew Kelley 2025-09-02 12:00:43 -07:00
parent 00b0beb682
commit 90ac62cc75
1 changed files with 10 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
lib/std/compress/lzma2.zig
View file

@ -79,23 +79,18 @@ pub const AccumBuffer = struct {
_ = writer;
const buf_len = self.buf.items.len;
if (dist > buf_len) {
return error.CorruptInput;
}
if (dist > buf_len) return error.CorruptInput;
// ensure we have enough capacity for all new bytes.
// This prevents the buffer's memory from being reallocated (and freed)
// while we are still reading from it.
try self.buf.ensureTotalCapacity(allocator, buf_len + len);
try self.buf.ensureUnusedCapacity(allocator, len);
const buffer = self.buf.allocatedSlice();
const src = buffer[buf_len - dist ..][0..len];
const dst = buffer[buf_len..][0..len];
var offset = buf_len - dist;
var i: usize = 0;
while (i < len) : (i += 1) {
const x = self.buf.items[offset];
// Since capacity is guaranteed, it's safe to use appendAssumeCapacity.
self.buf.appendAssumeCapacity(x);
offset += 1;
}
// This is not a @memmove; it intentionally repeats patterns caused by
// iterating one byte at a time.
for (dst, src) |*d, s| d.* = s;
self.buf.items.len = buf_len + len;
self.len += len;
}