mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2025-12-16 11:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extract setBits

Browse Source
This commit is contained in:
Benjamin Feng 2019-12-03 17:41:05 -06:00
parent b33211ed51
commit baffaf7986
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lib/std/heap.zig
View File

@ -272,6 +272,13 @@ const WasmPageAllocator = struct {
return self.packed_data.int_count; return self.packed_data.int_count;
} }
fn setBits(self: *FreeBlock, start_idx: usize, len: usize, val: u1) void {
var i: usize = 0;
while (i < len) : (i += 1) {
self.packed_data.set(i + start_idx + self.offset, val);
}
}
fn useRecycled(self: *FreeBlock, num_pages: usize) ?[*]u8 { fn useRecycled(self: *FreeBlock, num_pages: usize) ?[*]u8 {
var found_idx: usize = 0; var found_idx: usize = 0;
var found_size: usize = 0; var found_size: usize = 0;
@ -296,10 +303,7 @@ const WasmPageAllocator = struct {
found_size += 1; found_size += 1;
if (found_size >= num_pages) { if (found_size >= num_pages) {
while (found_size > 0) { self.setBits(found_idx, found_size, 0);
found_size -= 1;
self.packed_data.set(found_idx + found_size, 0);
}
return @intToPtr([*]u8, (found_idx + self.offset) * std.mem.page_size); return @intToPtr([*]u8, (found_idx + self.offset) * std.mem.page_size);
} }
} }
@ -308,12 +312,8 @@ const WasmPageAllocator = struct {
return null; return null;
} }
fn recycle(self: *FreeBlock, start_index: usize, end_index: usize) void { fn recycle(self: *FreeBlock, start_idx: usize, len: usize) void {
var i = start_index - self.offset; self.setBits(start_idx, len, 1);
while (i < end_index - self.offset) : (i += 1) {
std.debug.assert(self.packed_data.get(i) == 0);
self.packed_data.set(i, 1);
}
} }
}; };
@ -369,7 +369,7 @@ const WasmPageAllocator = struct {
} }
if (free_start < conventional.totalPages()) { if (free_start < conventional.totalPages()) {
conventional.recycle(free_start, free_end); conventional.recycle(free_start, free_end - free_start);
} else { } else {
if (extended.totalPages() == 0) { if (extended.totalPages() == 0) {
extended.offset = conventional.offset + conventional.totalPages(); extended.offset = conventional.offset + conventional.totalPages();
@ -378,7 +378,7 @@ const WasmPageAllocator = struct {
free_end -= 1; free_end -= 1;
extended.initData(@intToPtr([*]align(16) u8, free_end * std.mem.page_size)[0..std.mem.page_size]); extended.initData(@intToPtr([*]align(16) u8, free_end * std.mem.page_size)[0..std.mem.page_size]);
} }
extended.recycle(free_start, free_end); extended.recycle(free_start, free_end - free_start);
} }
} }