DescriptorIndexAllocator.cpp

Go to the documentation of this file.

#include <Engine/Mesh/DescriptorIndexAllocator.h>
#include <algorithm>
#include <stdexcept>
#include <Engine/Core/Exceptions.h>
 
namespace EngineCore
{
 
DescriptorIndexAllocator::DescriptorIndexAllocator(int descriptorCount): capacity(descriptorCount) {
    for (int i = descriptorCount - 1; i >= 0; i--) {
        freeIndices.push(i);
    }
}
 
uint32_t DescriptorIndexAllocator::allocate() 
{
    if (freeIndices.empty())
    {
        throw AllocationException("Failed to acquire a free index");
    }
 
    const uint32_t index = freeIndices.top();
    freeIndices.pop();
    usedIndices.push_back(index);
    return index;
}
 
void DescriptorIndexAllocator::free(uint32_t index)
{
    usedIndices.erase(std::remove(usedIndices.begin(), usedIndices.end(), index), usedIndices.end());
    freeIndices.push(index);
}
 
const std::vector<uint32_t> &DescriptorIndexAllocator::getUsedIndices() const
{
    return usedIndices;
}
 
void DescriptorIndexAllocator::reset(const int newSize) {
    usedIndices.clear();
    for (int i = 0; i < freeIndices.size(); i++) {
        freeIndices.pop();
    }
    for(int i = newSize-1; i >= 0; i--)
    {
        freeIndices.push(i);
    }
 
}
 
size_t DescriptorIndexAllocator::getCapacity() const {
    return capacity;
}
 
} // namespace EngineCore