Thraix
52 lines
2.2 KiB
C++
52 lines
2.2 KiB
C++
#include "copium/pipeline/DescriptorPool.h"
|
|
|
|
#include "copium/core/Device.h"
|
|
#include "copium/core/SwapChain.h"
|
|
|
|
namespace Copium
|
|
{
|
|
DescriptorPool::DescriptorPool(Vulkan& vulkan)
|
|
: vulkan{vulkan}
|
|
{
|
|
std::vector<VkDescriptorPoolSize> poolSizes{2};
|
|
poolSizes[0].type = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
|
|
poolSizes[0].descriptorCount = DESCRIPTOR_SET_COUNT * SwapChain::MAX_FRAMES_IN_FLIGHT; // TODO: how should this actually be determined?
|
|
|
|
poolSizes[1].type = VK_DESCRIPTOR_TYPE_SAMPLER;
|
|
poolSizes[1].descriptorCount = DESCRIPTOR_SET_COUNT * SwapChain::MAX_FRAMES_IN_FLIGHT; // TODO: how should this actually be determined?
|
|
|
|
VkDescriptorPoolCreateInfo createInfo{};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO;
|
|
createInfo.poolSizeCount = poolSizes.size();
|
|
createInfo.pPoolSizes = poolSizes.data();
|
|
createInfo.maxSets = DESCRIPTOR_SET_COUNT * SwapChain::MAX_FRAMES_IN_FLIGHT;
|
|
createInfo.flags = VK_DESCRIPTOR_POOL_CREATE_FREE_DESCRIPTOR_SET_BIT;
|
|
|
|
CP_VK_ASSERT(vkCreateDescriptorPool(vulkan.GetDevice(), &createInfo, nullptr, &descriptorPool), "DescriptorPool : Failed to initialize descriptor pool");
|
|
}
|
|
|
|
DescriptorPool::~DescriptorPool()
|
|
{
|
|
vkDestroyDescriptorPool(vulkan.GetDevice(), descriptorPool, nullptr);
|
|
}
|
|
|
|
std::vector<VkDescriptorSet> DescriptorPool::AllocateDescriptorSets(VkDescriptorSetLayout descriptorSetLayout)
|
|
{
|
|
std::vector<VkDescriptorSet> descriptorSets{SwapChain::MAX_FRAMES_IN_FLIGHT};
|
|
std::vector<VkDescriptorSetLayout> layouts{SwapChain::MAX_FRAMES_IN_FLIGHT, descriptorSetLayout};
|
|
VkDescriptorSetAllocateInfo allocateInfo{};
|
|
allocateInfo.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO;
|
|
allocateInfo.descriptorPool = descriptorPool;
|
|
allocateInfo.descriptorSetCount = descriptorSets.size();
|
|
allocateInfo.pSetLayouts = layouts.data();
|
|
|
|
CP_VK_ASSERT(vkAllocateDescriptorSets(vulkan.GetDevice(), &allocateInfo, descriptorSets.data()), "AllocateDescriptorSets : Failed to allocate descriptor sets");
|
|
|
|
return descriptorSets;
|
|
}
|
|
|
|
void DescriptorPool::FreeDescriptorSets(const std::vector<VkDescriptorSet>& descriptorSets)
|
|
{
|
|
vkFreeDescriptorSets(vulkan.GetDevice(), descriptorPool, descriptorSets.size(), descriptorSets.data());
|
|
}
|
|
} |