voxelcoreprojects/voxelcore
1
0
Fork 0
mirror of https://github.com/MihailRis/voxelcore.git synced 2026-10-13 15:01:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

BufferPool docs + fix

Browse source
This commit is contained in:
MihailRis 2024-04-15 00:55:35 +03:00
parent 4e5f6a5a7d
commit 0cbaa1ccc6
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/util/BufferPool.h
View file

@ -9,6 +9,8 @@
#include <memory> #include <memory>
namespace util { namespace util {
/// @brief Thread-safe pool of same-sized buffers
/// @tparam T array type
template<class T> template<class T>
class BufferPool { class BufferPool {
std::vector<std::unique_ptr<T[]>> buffers; std::vector<std::unique_ptr<T[]>> buffers;
@ -19,6 +21,8 @@ namespace util {
BufferPool(size_t bufferSize) : bufferSize(bufferSize) { BufferPool(size_t bufferSize) : bufferSize(bufferSize) {
} }
/// @brief Retrieve a buffer for use
/// @return pointer that brings buffer back to the pool when destroyed
std::shared_ptr<T[]> get() { std::shared_ptr<T[]> get() {
std::lock_guard lock(mutex); std::lock_guard lock(mutex);
if (freeBuffers.empty()) { if (freeBuffers.empty()) {
@ -28,6 +32,7 @@ namespace util {
auto* buffer = freeBuffers.front(); auto* buffer = freeBuffers.front();
freeBuffers.pop(); freeBuffers.pop();
return std::shared_ptr<T[]>(buffer, [this](T* ptr) { return std::shared_ptr<T[]>(buffer, [this](T* ptr) {
std::lock_guard lock(mutex);
freeBuffers.push(ptr); freeBuffers.push(ptr);
}); });
} }