descriptor.h

#pragma once
 
#include <volk.h>
 
#include <cstdint>
#include <memory>
#include <optional>
 
namespace pixelbullet
{
class OffsetSize
{
public:
    OffsetSize(uint32_t offset, uint32_t size)
        : offset_(offset)
        , size_(size)
    {
    }
 
    uint32_t GetOffset() const
    {
        return offset_;
    }
    uint32_t GetSize() const
    {
        return size_;
    }
 
    bool operator==(const OffsetSize& rhs) const
    {
        return offset_ == rhs.offset_ && size_ == rhs.size_;
    }
 
    bool operator!=(const OffsetSize& rhs) const
    {
        return !operator==(rhs);
    }
 
private:
    uint32_t offset_;
    uint32_t size_;
};
 
class WriteDescriptorSet
{
public:
    WriteDescriptorSet(const VkWriteDescriptorSet& writeDescriptorSet, const VkDescriptorImageInfo& imageInfo)
        : write_descriptor_set_(writeDescriptorSet)
        , image_info_(std::make_unique<VkDescriptorImageInfo>(imageInfo))
    {
        this->write_descriptor_set_.descriptorCount = 1;
        this->write_descriptor_set_.pImageInfo = image_info_ ? image_info_.get() : nullptr;
    }
 
    WriteDescriptorSet(const VkWriteDescriptorSet& writeDescriptorSet, const VkDescriptorBufferInfo& bufferInfo)
        : write_descriptor_set_(writeDescriptorSet)
        , buffer_info_(std::make_unique<VkDescriptorBufferInfo>(bufferInfo))
    {
        this->write_descriptor_set_.descriptorCount = 1;
        this->write_descriptor_set_.pBufferInfo = buffer_info_ ? buffer_info_.get() : nullptr;
    }
 
    const VkWriteDescriptorSet& GetWriteDescriptorSet() const
    {
        return write_descriptor_set_;
    }
 
private:
    VkWriteDescriptorSet write_descriptor_set_;
    std::unique_ptr<VkDescriptorImageInfo> image_info_;
    std::unique_ptr<VkDescriptorBufferInfo> buffer_info_;
};
 
class Descriptor
{
public:
    Descriptor() = default;
    virtual ~Descriptor() = default;
 
    virtual WriteDescriptorSet GetWriteDescriptor(uint32_t binding, VkDescriptorType descriptorType,
                                                  const std::optional<OffsetSize>& offsetSize) const = 0;
 
    virtual uint64_t GetRevision() const
    {
        return 0;
    }
};
} // namespace pixelbullet