descriptorsetallocator.h

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
 
#pragma once
 
// External Includes
#include <unordered_map>
#include <utility/dllexport.h>
#include <vulkan/vulkan_core.h>
#include <vector>
 
namespace nap
{
    class NAPAPI DescriptorSetAllocator final
    {
    public:
        DescriptorSetAllocator(VkDevice device);
 
        // Destructor
        ~DescriptorSetAllocator();
 
        VkDescriptorSet allocate(VkDescriptorSetLayout layout, int numUBODescriptors, int numSSBODescriptors, int numSamplerDescriptors);
 
    private:
        struct DescriptorPool
        {
            using DescriptorSetList = std::vector<VkDescriptorSet>;
            VkDescriptorPool    mPool = VK_NULL_HANDLE;                 
            DescriptorSetList   mAllocatedDescriptorSets;               
            int                 mMaxNumSets = 0;                        
        };
 
        using DescriptorPoolMap = std::unordered_map<uint64_t, std::vector<DescriptorPool>>;
        VkDevice                mDevice;
        DescriptorPoolMap       mDescriptorPools;
    };
} // nap