gl_staging_buffer_pool: Refactor allocation variables into a struct

This commit is contained in:
Ameer J 2023-08-13 21:38:23 -04:00
parent 43be2bfe33
commit 48b87d64de
2 changed files with 28 additions and 30 deletions

View File

@ -32,12 +32,12 @@ StagingBufferMap StagingBuffers::RequestMap(size_t requested_size, bool insert_f
MICROPROFILE_SCOPE(OpenGL_BufferRequest); MICROPROFILE_SCOPE(OpenGL_BufferRequest);
const size_t index = RequestBuffer(requested_size); const size_t index = RequestBuffer(requested_size);
OGLSync* const sync = insert_fence ? &syncs[index] : nullptr; OGLSync* const sync = insert_fence ? &allocs[index].sync : nullptr;
sync_indices[index] = insert_fence ? ++current_sync_index : 0; allocs[index].sync_index = insert_fence ? ++current_sync_index : 0;
return StagingBufferMap{ return StagingBufferMap{
.mapped_span = std::span(maps[index], requested_size), .mapped_span = std::span(allocs[index].map, requested_size),
.sync = sync, .sync = sync,
.buffer = buffers[index].handle, .buffer = allocs[index].buffer.handle,
}; };
} }
@ -45,46 +45,41 @@ size_t StagingBuffers::RequestBuffer(size_t requested_size) {
if (const std::optional<size_t> index = FindBuffer(requested_size); index) { if (const std::optional<size_t> index = FindBuffer(requested_size); index) {
return *index; return *index;
} }
StagingBufferAlloc alloc;
OGLBuffer& buffer = buffers.emplace_back(); alloc.buffer.Create();
buffer.Create();
const auto next_pow2_size = Common::NextPow2(requested_size); const auto next_pow2_size = Common::NextPow2(requested_size);
glNamedBufferStorage(buffer.handle, next_pow2_size, nullptr, glNamedBufferStorage(alloc.buffer.handle, next_pow2_size, nullptr,
storage_flags | GL_MAP_PERSISTENT_BIT); storage_flags | GL_MAP_PERSISTENT_BIT);
maps.push_back(static_cast<u8*>(glMapNamedBufferRange(buffer.handle, 0, next_pow2_size, alloc.map = static_cast<u8*>(glMapNamedBufferRange(alloc.buffer.handle, 0, next_pow2_size,
map_flags | GL_MAP_PERSISTENT_BIT))); map_flags | GL_MAP_PERSISTENT_BIT));
syncs.emplace_back(); alloc.size = next_pow2_size;
sync_indices.emplace_back(); allocs.emplace_back(std::move(alloc));
sizes.push_back(next_pow2_size); return allocs.size() - 1;
ASSERT(syncs.size() == buffers.size() && buffers.size() == maps.size() &&
maps.size() == sizes.size());
return buffers.size() - 1;
} }
std::optional<size_t> StagingBuffers::FindBuffer(size_t requested_size) { std::optional<size_t> StagingBuffers::FindBuffer(size_t requested_size) {
size_t known_unsignaled_index = current_sync_index + 1; size_t known_unsignaled_index = current_sync_index + 1;
size_t smallest_buffer = std::numeric_limits<size_t>::max(); size_t smallest_buffer = std::numeric_limits<size_t>::max();
std::optional<size_t> found; std::optional<size_t> found;
const size_t num_buffers = sizes.size(); const size_t num_buffers = allocs.size();
for (size_t index = 0; index < num_buffers; ++index) { for (size_t index = 0; index < num_buffers; ++index) {
const size_t buffer_size = sizes[index]; StagingBufferAlloc& alloc = allocs[index];
const size_t buffer_size = alloc.size;
if (buffer_size < requested_size || buffer_size >= smallest_buffer) { if (buffer_size < requested_size || buffer_size >= smallest_buffer) {
continue; continue;
} }
if (syncs[index].handle != 0) { if (alloc.sync.handle != 0) {
if (sync_indices[index] >= known_unsignaled_index) { if (alloc.sync_index >= known_unsignaled_index) {
// This fence is later than a fence that is known to not be signaled // This fence is later than a fence that is known to not be signaled
continue; continue;
} }
if (!syncs[index].IsSignaled()) { if (!alloc.sync.IsSignaled()) {
// Since this fence hasn't been signaled, it's safe to assume all later // Since this fence hasn't been signaled, it's safe to assume all later
// fences haven't been signaled either // fences haven't been signaled either
known_unsignaled_index = std::min(known_unsignaled_index, sync_indices[index]); known_unsignaled_index = std::min(known_unsignaled_index, alloc.sync_index);
continue; continue;
} }
syncs[index].Release(); alloc.sync.Release();
} }
smallest_buffer = buffer_size; smallest_buffer = buffer_size;
found = index; found = index;

View File

@ -38,11 +38,14 @@ struct StagingBuffers {
std::optional<size_t> FindBuffer(size_t requested_size); std::optional<size_t> FindBuffer(size_t requested_size);
std::vector<OGLSync> syncs; struct StagingBufferAlloc {
std::vector<OGLBuffer> buffers; OGLSync sync;
std::vector<u8*> maps; OGLBuffer buffer;
std::vector<size_t> sizes; u8* map;
std::vector<size_t> sync_indices; size_t size;
size_t sync_index;
};
std::vector<StagingBufferAlloc> allocs;
GLenum storage_flags; GLenum storage_flags;
GLenum map_flags; GLenum map_flags;
size_t current_sync_index = 0; size_t current_sync_index = 0;