Further work on getting everything working again

Mainly working on decompression interface
This commit is contained in:
Caleb Gardner
2026-04-30 07:00:46 -05:00
parent b67d02074d
commit 274d088490
4 changed files with 106 additions and 17 deletions
+55
View File
@@ -0,0 +1,55 @@
const std = @import("std");
const Reader = std.Io.Reader;
const flate = std.compress.flate;
const Decompressor = @import("../util/decompressor.zig");
const Error = Decompressor.Error;
pub fn Zlib(stateless: bool) type {
return if (stateless)
struct {
const Self = @This();
interface: Decompressor = .{ .decomp_fn = decomp },
const init: Self = .{};
fn decomp(_: *?Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8) Error!usize {
const buf = try alloc.alloc(u8, in.len * 2);
defer alloc.free(buf);
return zlibDecomp(buf, in, out);
}
}
else
struct {
const Self = @This();
interface: Decompressor = .{ .decomp_fn = decomp },
alloc: std.mem.Allocator,
block_size: u32,
buffers: std.ArrayList([]u8),
buffer_queue: std.SinglyLinkedList,
pub fn init(alloc: std.mem.Allocator, block_size: u32) !Self {
return .{
.alloc = alloc,
.block_size = block_size,
.buffers = try .initCapacity(alloc, 20),
};
}
pub fn deinit(self: Self) void {
for (self.buffers) |buf|
self.alloc.free(buf);
}
};
}
inline fn zlibDecomp(buffer: []u8, in: []u8, out: []u8) !usize {
var rdr: Reader = .fixed(in);
var decomp = flate.Decompress.init(&rdr, .zlib, buffer);
return decomp.reader.readSliceShort(out);
}
+28 -13
View File
@@ -1,9 +1,29 @@
const std = @import("std"); const std = @import("std");
const Io = std.Io; const Io = std.Io;
const Decompressor = @import("util/decompressor.zig");
const OffsetFile = @import("util/offset_file.zig"); const OffsetFile = @import("util/offset_file.zig");
const MetadataReader = @import("util/metadata.zig"); const MetadataReader = @import("util/metadata.zig");
pub fn lookupValue(comptime T: anytype, alloc: std.mem.Allocator, io: Io, decomp: *Decompressor, file: OffsetFile, table_start: u64, idx: u16) !T {
const T_PER_BLOCK: u16 = 8192 / @sizeOf(T);
const block = idx / T_PER_BLOCK;
const block_offset = idx % T_PER_BLOCK;
var rdr = try file.readerAt(io, table_start + (8 * block), &[0]u8{});
var offset: u64 = undefined;
try rdr.interface.readSliceEndian(u64, @ptrCast(&offset), .little);
rdr = try file.readerAt(io, offset, &[0]u8{});
var meta: MetadataReader = .init(alloc, &rdr, decomp);
try meta.interface.discardAll(@sizeOf(T) * block_offset);
var out: T = undefined;
try meta.interface.readSliceEndian(T, @ptrCast(&out), .little);
return out;
}
pub fn CachedTable(comptime T: anytype) type { pub fn CachedTable(comptime T: anytype) type {
return struct { return struct {
const T_PER_BLOCK: u16 = 8192 / @sizeOf(T); const T_PER_BLOCK: u16 = 8192 / @sizeOf(T);
@@ -11,7 +31,6 @@ pub fn CachedTable(comptime T: anytype) type {
const Table = @This(); const Table = @This();
alloc: std.mem.Allocator, alloc: std.mem.Allocator,
io: Io,
fil: OffsetFile, fil: OffsetFile,
table_start: u64, table_start: u64,
total_num: u32, total_num: u32,
@@ -20,10 +39,9 @@ pub fn CachedTable(comptime T: anytype) type {
mut: Io.Mutex = .init, mut: Io.Mutex = .init,
pub fn init(alloc: std.mem.Allocator, io: Io, fil: OffsetFile, offset: u64, total_num: u32) Table { pub fn init(alloc: std.mem.Allocator, fil: OffsetFile, offset: u64, total_num: u32) Table {
return .{ return .{
.alloc = alloc, .alloc = alloc,
.io = io,
.fil = fil, .fil = fil,
.table_start = offset, .table_start = offset,
.total_num = total_num, .total_num = total_num,
@@ -32,34 +50,31 @@ pub fn CachedTable(comptime T: anytype) type {
}; };
} }
pub fn get(self: *Table, idx: u32) !T { pub fn get(self: *Table, io: Io, idx: u32) !T {
const block = idx / T_PER_BLOCK; const block = idx / T_PER_BLOCK;
const block_offset = idx % T_PER_BLOCK; const block_offset = idx % T_PER_BLOCK;
if (self.table.contains(block)) if (self.table.contains(block))
return self.table.get(block).?[block_offset]; return self.table.get(block).?[block_offset];
try self.mut.lock(self.io); try self.mut.lock(io);
defer self.mut.unlock(self.io); defer self.mut.unlock(io);
if (self.table.contains(block)) if (self.table.contains(block))
return self.table.get(block).?[block_offset]; return self.table.get(block).?[block_offset];
var rdr = try self.fil.readerAt(self.io, self.table_start + (8 * block), &[0]u8{}); var rdr = try self.fil.readerAt(io, self.table_start + (8 * block), &[0]u8{});
var offset: u64 = undefined; var offset: u64 = undefined;
try rdr.interface.readSliceEndian(u64, @ptrCast(&offset), .little); try rdr.interface.readSliceEndian(u64, @ptrCast(&offset), .little);
const arr_num: u16 = if (self.total_num % T_PER_BLOCK != 0 and block == (self.total_num - 1) / T_PER_BLOCK) const len: u16 = if (self.total_num % T_PER_BLOCK != 0 and block == (self.total_num - 1) / T_PER_BLOCK)
self.total_num % T_PER_BLOCK self.total_num % T_PER_BLOCK
else else
T_PER_BLOCK; T_PER_BLOCK;
rdr = try self.fil.readerAt(self.io, offset, &[0]u8{}); rdr = try self.fil.readerAt(io, offset, &[0]u8{});
var meta: MetadataReader = .init(self.alloc, &rdr, self.decomp); var meta: MetadataReader = .init(self.alloc, &rdr, self.decomp);
try self.table.put( try self.table.put(block, try meta.interface.readSliceEndianAlloc(self.alloc, T, len, .little));
block,
try meta.interface.readSliceEndianAlloc(self.alloc, T, arr_num, .little),
);
} }
}; };
} }
+19
View File
@@ -0,0 +1,19 @@
//! A decompression interface
const std = @import("std");
const Decompressor = @This();
pub const Error = std.Io.Reader.StreamError || std.mem.Allocator.Error;
/// The actual decompression function.
/// If the given decompressor is null, then the decompression should be done "stateless" without lasting allocations.
decomp_fn: *fn (?*Decompressor, std.mem.Allocator, in: []u8, out: []u8) Error!usize,
pub fn Decompress(self: *Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8) Error!usize {
return self.decomp_fn(self, alloc, in, out);
}
pub fn StatelessDecompression(self: Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8) Error!usize {
return self.decomp_fn(null, alloc, in, out);
}
+4 -4
View File
@@ -4,7 +4,7 @@ const Writer = std.Io.Writer;
const Limit = std.Io.Limit; const Limit = std.Io.Limit;
const StreamError = std.Io.Reader.StreamError; const StreamError = std.Io.Reader.StreamError;
const DecompFn = @import("../decomp.zig").DecompFn; const Decompressor = @import("decompressor.zig");
const BlockHeader = packed struct { const BlockHeader = packed struct {
size: u15, size: u15,
@@ -15,14 +15,14 @@ const This = @This();
alloc: std.mem.Allocator, alloc: std.mem.Allocator,
rdr: *Reader, rdr: *Reader,
decomp: DecompFn, decomp: *Decompressor,
buf: [8192]u8 = undefined, buf: [8192]u8 = undefined,
interface: Reader, interface: Reader,
err: ?anyerror = null, err: ?anyerror = null,
pub fn init(alloc: std.mem.Allocator, rdr: *Reader, decomp: DecompFn) This { pub fn init(alloc: std.mem.Allocator, rdr: *Reader, decomp: *Decompressor) This {
return .{ return .{
.alloc = alloc, .alloc = alloc,
.rdr = rdr, .rdr = rdr,
@@ -52,7 +52,7 @@ fn advance(self: *This) !void {
} }
var tmp_buf: [8192]u8 = undefined; var tmp_buf: [8192]u8 = undefined;
try self.rdr.readSliceAll(tmp_buf[0..hdr.size]); try self.rdr.readSliceAll(tmp_buf[0..hdr.size]);
self.interface.end = try self.decomp(self.alloc, tmp_buf[0..hdr.size], &self.buf); self.interface.end = try self.decomp.Decompress(self.alloc, tmp_buf[0..hdr.size], &self.buf);
self.interface.buffer = self.buf[0..self.interface.end]; self.interface.buffer = self.buf[0..self.interface.end];
} }