UntitledGame/Descriptors.cpp at main · minco3/UntitledGame · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#include "Descriptors.hpp"

Descriptors::Descriptors(
    Device& device, std::vector<Buffer<UniformBufferObject>>& uniformBuffers,
    size_t imageCount)
    : m_DescriptorPool(CreateDescriptorPool(device, imageCount)),
      m_DescriptorSetLayouts(CreateDescriptorSetLayouts(device, imageCount)),
      m_DescriptorSets(CreateDescriptorSets(device))
{
    std::vector<vk::WriteDescriptorSet> descriptorWriteSets;
    for (size_t i = 0; i < imageCount; i++)
    {
        // need a more dynamic solution
        vk::DescriptorBufferInfo bufferInfo(
            *uniformBuffers.at(i).Get(), 0, sizeof(UniformBufferObject));
        descriptorWriteSets.emplace_back(
            *m_DescriptorSets.at(i), 0, 0, vk::DescriptorType::eUniformBuffer,
            nullptr, bufferInfo, nullptr);
    }
    device.Get().updateDescriptorSets(descriptorWriteSets, nullptr);
}

vk::raii::DescriptorPool
Descriptors::CreateDescriptorPool(Device& device, size_t imageCount)
{
    vk::DescriptorPoolSize discriptorPoolSize(
        vk::DescriptorType::eUniformBuffer, imageCount);
    vk::DescriptorPoolCreateInfo createInfo(
        vk::DescriptorPoolCreateFlagBits::eFreeDescriptorSet, imageCount,
        discriptorPoolSize);
    return device.Get().createDescriptorPool(createInfo);
}

std::vector<vk::raii::DescriptorSetLayout>
Descriptors::CreateDescriptorSetLayouts(Device& device, size_t imageCount)
{
    vk::DescriptorSetLayoutBinding descriptorSetLayoutBinding(
        0, vk::DescriptorType::eUniformBuffer, 1,
        vk::ShaderStageFlagBits::eVertex);

    m_DescriptorSetLayoutBindings.push_back(descriptorSetLayoutBinding);

    vk::DescriptorSetLayoutCreateInfo createInfo(
        {}, m_DescriptorSetLayoutBindings);

    std::vector<vk::raii::DescriptorSetLayout> descriptorSetLayouts;

    for (size_t i = 0; i < imageCount; i++)
    {
        descriptorSetLayouts.emplace_back(
            device.Get().createDescriptorSetLayout(createInfo));
    }
    return descriptorSetLayouts;
}

vk::raii::DescriptorSets Descriptors::CreateDescriptorSets(Device& device)
{
    std::vector<vk::DescriptorSetLayout> setLayouts;
    for (auto& layout : m_DescriptorSetLayouts)
    {
        setLayouts.push_back(*layout);
    }

    vk::DescriptorSetAllocateInfo allocInfo(*m_DescriptorPool, setLayouts);
    return vk::raii::DescriptorSets(device.Get(), allocInfo);
}