Reduce memory usage of StructureMapFragments by constructing all the lists on the stackAllocator before transfering them tightly packed into the arena.

progress towards #813
This commit is contained in:
IntegratedQuantum 2024-12-07 14:34:43 +01:00
parent 3d67c9023e
commit 8a097804ab
2 changed files with 51 additions and 15 deletions

View File

@ -13,14 +13,18 @@ const Vec3i = vec.Vec3i;
const terrain = @import("terrain.zig");
const TerrainGenerationProfile = terrain.TerrainGenerationProfile;
pub const Structure = struct {
const StructureInternal = struct {
generateFn: *const fn(self: *const anyopaque, chunk: *ServerChunk, caveMap: terrain.CaveMap.CaveMapView) void,
data: *const anyopaque,
priority: f32,
pub fn generate(self: Structure, chunk: *ServerChunk, caveMap: terrain.CaveMap.CaveMapView) void {
pub fn generate(self: StructureInternal, chunk: *ServerChunk, caveMap: terrain.CaveMap.CaveMapView) void {
self.generateFn(self.data, chunk, caveMap);
}
};
pub const Structure = struct {
internal: StructureInternal,
priority: f32,
fn lessThan(_: void, lhs: Structure, rhs: Structure) bool {
return lhs.priority < rhs.priority;
@ -32,7 +36,7 @@ pub const StructureMapFragment = struct {
pub const sizeMask = size - 1;
pub const chunkedSize = size >> main.chunk.chunkShift;
data: [chunkedSize*chunkedSize*chunkedSize]main.ListUnmanaged(Structure) = undefined,
data: [chunkedSize*chunkedSize*chunkedSize][]StructureInternal = undefined,
pos: ChunkPosition,
voxelShift: u5,
@ -40,8 +44,13 @@ pub const StructureMapFragment = struct {
arena: main.utils.NeverFailingArenaAllocator,
allocator: main.utils.NeverFailingAllocator,
tempData: struct {
lists: *[chunkedSize*chunkedSize*chunkedSize]main.ListUnmanaged(Structure),
allocator: NeverFailingAllocator,
},
pub fn init(self: *StructureMapFragment, wx: i32, wy: i32, wz: i32, voxelSize: u31) void {
pub fn init(self: *StructureMapFragment, tempAllocator: NeverFailingAllocator, wx: i32, wy: i32, wz: i32, voxelSize: u31) void {
self.* = .{
.pos = .{
.wx = wx, .wy = wy, .wz = wz,
@ -50,19 +59,42 @@ pub const StructureMapFragment = struct {
.voxelShift = @ctz(voxelSize),
.arena = .init(main.globalAllocator),
.allocator = self.arena.allocator(),
.tempData = .{
.lists = tempAllocator.create([chunkedSize*chunkedSize*chunkedSize]main.ListUnmanaged(Structure)),
.allocator = tempAllocator,
},
};
@memset(&self.data, .{});
@memset(self.tempData.lists, .{});
}
pub fn deinit(self: *StructureMapFragment) void {
var actual: usize = 0;
for(self.data) |i| {
actual += i.len*@sizeOf(StructureInternal);
}
var expected: usize = 0;
var it = self.arena.arena.state.buffer_list.first;
while(it) |node| {
expected += node.data;
it = node.next;
}
std.log.err("{} {}", .{expected, actual});
self.arena.deinit();
main.globalAllocator.destroy(self);
}
fn finishGeneration(self: *StructureMapFragment) void {
for(&self.data) |list| {
std.sort.insertion(Structure, list.items, {}, Structure.lessThan);
for(0..self.data.len) |i| {
std.sort.insertion(Structure, self.tempData.lists[i].items, {}, Structure.lessThan);
self.data[i] = self.allocator.alloc(StructureInternal, self.tempData.lists[i].items.len);
for(0..self.tempData.lists[i].items.len) |j| {
self.data[i][j] = self.tempData.lists[i].items[j].internal;
}
self.tempData.lists[i].deinit(self.tempData.allocator);
self.tempData.lists[i] = undefined;
}
self.tempData.allocator.destroy(self.tempData.lists);
self.tempData = undefined;
self.arena.shrinkAndFree();
}
@ -86,7 +118,7 @@ pub const StructureMapFragment = struct {
pub fn generateStructuresInChunk(self: *const StructureMapFragment, chunk: *ServerChunk, caveMap: terrain.CaveMap.CaveMapView) void {
const index = self.getIndex(chunk.super.pos.wx - self.pos.wx, chunk.super.pos.wy - self.pos.wy, chunk.super.pos.wz - self.pos.wz);
for(self.data[index].items) |structure| {
for(self.data[index]) |structure| {
structure.generate(chunk, caveMap);
}
}
@ -101,7 +133,7 @@ pub const StructureMapFragment = struct {
var z = min[2] & ~@as(i32, main.chunk.chunkMask << self.voxelShift | self.pos.voxelSize-1);
while(z < max[2]) : (z += main.chunk.chunkSize << self.voxelShift) {
if(z < 0 or z >= size*self.pos.voxelSize) continue;
self.data[self.getIndex(x, y, z)].append(self.allocator, structure);
self.tempData.lists[self.getIndex(x, y, z)].append(self.tempData.allocator, structure);
}
}
}
@ -159,7 +191,7 @@ var profile: TerrainGenerationProfile = undefined;
fn cacheInit(pos: ChunkPosition) *StructureMapFragment {
const mapFragment = main.globalAllocator.create(StructureMapFragment);
mapFragment.init(pos.wx, pos.wy, pos.wz, pos.voxelSize);
mapFragment.init(main.stackAllocator, pos.wx, pos.wy, pos.wz, pos.voxelSize);
for(profile.structureMapGenerators) |generator| {
generator.generate(mapFragment, profile.seed ^ generator.generatorSeed);
}

View File

@ -62,8 +62,10 @@ pub fn generate(map: *StructureMapFragment, worldSeed: u64) void {
.model = model,
};
map.addStructure(.{
.data = @ptrCast(data),
.generateFn = &SimpleStructure.generate,
.internal = .{
.data = @ptrCast(data),
.generateFn = &SimpleStructure.generate,
},
.priority = model.priority,
}, .{px -% margin, py -% margin, relZ -% margin -% 15}, .{px +% margin, py +% margin, relZ +% margin +% 15});
break;
@ -101,8 +103,10 @@ pub fn generate(map: *StructureMapFragment, worldSeed: u64) void {
.model = model,
};
map.addStructure(.{
.data = @ptrCast(data),
.generateFn = &SimpleStructure.generate,
.internal = .{
.data = @ptrCast(data),
.generateFn = &SimpleStructure.generate,
},
.priority = model.priority,
}, .{px -% margin, py -% margin, relZ -% margin -% 15}, .{px +% margin, py +% margin, relZ +% margin +% 15});
break;