2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
89 lines
3.3 KiB
C++
89 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include <mln/gfx/uniform_buffer.hpp>
|
|
#include <mln/vulkan/buffer_resource.hpp>
|
|
#include <mln/vulkan/descriptor_set.hpp>
|
|
|
|
namespace mln {
|
|
namespace vulkan {
|
|
|
|
class UniformBuffer final : public gfx::UniformBuffer {
|
|
public:
|
|
UniformBuffer(BufferResource&&);
|
|
UniformBuffer(const UniformBuffer&) = delete;
|
|
UniformBuffer(UniformBuffer&&);
|
|
~UniformBuffer() override;
|
|
|
|
const BufferResource& getBufferResource() const { return buffer; }
|
|
BufferResource& mutableBufferResource() { return buffer; }
|
|
|
|
UniformBuffer clone() const { return {buffer.clone()}; }
|
|
|
|
void update(const void* data, std::size_t dataSize) override;
|
|
|
|
protected:
|
|
BufferResource buffer;
|
|
};
|
|
|
|
/// Stores a collection of uniform buffers by name
|
|
class UniformBufferArray final : public gfx::UniformBufferArray {
|
|
public:
|
|
UniformBufferArray() = delete;
|
|
UniformBufferArray(DescriptorSetType descriptorSetType_,
|
|
uint32_t descriptorStartIndex_,
|
|
uint32_t descriptorStorageCount_,
|
|
uint32_t descriptorUniformCount_)
|
|
: descriptorSetType(descriptorSetType_),
|
|
descriptorStartIndex(descriptorStartIndex_),
|
|
descriptorStorageCount(descriptorStorageCount_),
|
|
descriptorUniformCount(descriptorUniformCount_) {}
|
|
|
|
UniformBufferArray(UniformBufferArray&& other) noexcept
|
|
: gfx::UniformBufferArray(std::move(other)),
|
|
descriptorSetType(other.descriptorSetType), // NOLINT(bugprone-use-after-move)
|
|
descriptorStartIndex(other.descriptorStartIndex), // NOLINT(bugprone-use-after-move)
|
|
descriptorStorageCount(other.descriptorStorageCount), // NOLINT(bugprone-use-after-move)
|
|
descriptorUniformCount(other.descriptorUniformCount), // NOLINT(bugprone-use-after-move)
|
|
descriptorSet(std::move(other.descriptorSet)) {} // NOLINT(bugprone-use-after-move)
|
|
|
|
UniformBufferArray(const UniformBufferArray&) = delete;
|
|
|
|
UniformBufferArray& operator=(UniformBufferArray&& other) noexcept {
|
|
gfx::UniformBufferArray::operator=(std::move(other));
|
|
return *this;
|
|
}
|
|
|
|
UniformBufferArray& operator=(const UniformBufferArray& other) {
|
|
gfx::UniformBufferArray::operator=(other);
|
|
return *this;
|
|
}
|
|
|
|
~UniformBufferArray() override = default;
|
|
|
|
const std::shared_ptr<gfx::UniformBuffer>& set(const size_t id,
|
|
std::shared_ptr<gfx::UniformBuffer> uniformBuffer) override;
|
|
|
|
void createOrUpdate(
|
|
const size_t id, const void* data, std::size_t size, gfx::Context& context, bool persistent = false) override;
|
|
|
|
void bind(gfx::RenderPass& renderPass) override;
|
|
|
|
void bindDescriptorSets(CommandEncoder& encoder);
|
|
void freeDescriptorSets() { descriptorSet.reset(); }
|
|
|
|
private:
|
|
gfx::UniqueUniformBuffer copy(const gfx::UniformBuffer& buffer) override {
|
|
return std::make_unique<UniformBuffer>(static_cast<const UniformBuffer&>(buffer).clone());
|
|
}
|
|
|
|
const DescriptorSetType descriptorSetType{DescriptorSetType::DrawableUniform};
|
|
const uint32_t descriptorStartIndex{0};
|
|
const uint32_t descriptorStorageCount{0};
|
|
const uint32_t descriptorUniformCount{0};
|
|
|
|
std::unique_ptr<UniformDescriptorSet> descriptorSet;
|
|
};
|
|
|
|
} // namespace vulkan
|
|
} // namespace mln
|