124 lines
3.1 KiB
C++
124 lines
3.1 KiB
C++
#pragma once
|
|
#include "VulkanGraphicsResources.h"
|
|
|
|
namespace Seele
|
|
{
|
|
namespace Vulkan
|
|
{
|
|
DECLARE_REF(Graphics);
|
|
class DescriptorLayout : public Gfx::DescriptorLayout
|
|
{
|
|
public:
|
|
DescriptorLayout(PGraphics graphics);
|
|
virtual ~DescriptorLayout();
|
|
virtual void create();
|
|
inline VkDescriptorSetLayout getHandle() const
|
|
{
|
|
return layoutHandle;
|
|
}
|
|
|
|
private:
|
|
uint32 hash;
|
|
PGraphics graphics;
|
|
Array<VkDescriptorSetLayoutBinding> bindings;
|
|
VkDescriptorSetLayout layoutHandle;
|
|
friend class DescriptorAllocator;
|
|
};
|
|
DEFINE_REF(DescriptorLayout);
|
|
class PipelineLayout : public Gfx::PipelineLayout
|
|
{
|
|
public:
|
|
PipelineLayout(PGraphics graphics)
|
|
: graphics(graphics), layoutHash(0), layoutHandle(VK_NULL_HANDLE)
|
|
{
|
|
}
|
|
virtual ~PipelineLayout();
|
|
virtual void create();
|
|
virtual void reset();
|
|
inline VkPipelineLayout getHandle() const
|
|
{
|
|
return layoutHandle;
|
|
}
|
|
virtual uint32 getHash() const
|
|
{
|
|
return layoutHash;
|
|
}
|
|
|
|
private:
|
|
Array<VkDescriptorSetLayout> vulkanDescriptorLayouts;
|
|
uint32 layoutHash;
|
|
VkPipelineLayout layoutHandle;
|
|
PGraphics graphics;
|
|
};
|
|
DEFINE_REF(PipelineLayout);
|
|
|
|
class DescriptorAllocator : public Gfx::DescriptorAllocator
|
|
{
|
|
public:
|
|
DescriptorAllocator(PGraphics graphics, DescriptorLayout &layout);
|
|
virtual ~DescriptorAllocator();
|
|
virtual void allocateDescriptorSet(Gfx::PDescriptorSet &descriptorSet);
|
|
virtual void reset();
|
|
|
|
inline VkDescriptorPool getHandle() const
|
|
{
|
|
return poolHandle;
|
|
}
|
|
inline DescriptorLayout getLayout() const
|
|
{
|
|
return layout;
|
|
}
|
|
|
|
private:
|
|
PGraphics graphics;
|
|
const static int maxSets = 512;
|
|
VkDescriptorSet cachedHandles[maxSets];
|
|
uint32 currentCachedIndex;
|
|
VkDescriptorPool poolHandle;
|
|
DescriptorLayout &layout;
|
|
};
|
|
DEFINE_REF(DescriptorAllocator);
|
|
|
|
class DescriptorSet : public Gfx::DescriptorSet
|
|
{
|
|
public:
|
|
DescriptorSet(PGraphics graphics, PDescriptorAllocator owner)
|
|
: graphics(graphics), owner(owner), currentFrameSet(0)
|
|
{
|
|
}
|
|
virtual ~DescriptorSet();
|
|
virtual void beginFrame();
|
|
virtual void endFrame();
|
|
virtual void writeChanges();
|
|
virtual void updateBuffer(uint32_t binding, Gfx::PUniformBuffer uniformBuffer);
|
|
virtual void updateBuffer(uint32_t binding, Gfx::PStructuredBuffer uniformBuffer);
|
|
virtual void updateSampler(uint32_t binding, Gfx::PSamplerState samplerState);
|
|
virtual void updateTexture(uint32_t binding, Gfx::PTexture texture, Gfx::PSamplerState sampler = nullptr);
|
|
virtual bool operator<(Gfx::PDescriptorSet other);
|
|
inline VkDescriptorSet getHandle() const
|
|
{
|
|
return setHandle[currentFrameSet];
|
|
}
|
|
virtual uint32 getSetIndex() const
|
|
{
|
|
return owner->getLayout().getSetIndex();
|
|
}
|
|
|
|
private:
|
|
Array<VkDescriptorImageInfo> imageInfos;
|
|
Array<VkDescriptorBufferInfo> bufferInfos;
|
|
Array<VkWriteDescriptorSet> writeDescriptors;
|
|
// contains the previously bound resources at every binding
|
|
// since the layout is fixed, trying to bind a texture to a buffer
|
|
// would not work anyways, so casts should be safe
|
|
Array<void*> cachedData[Gfx::numFramesBuffered];
|
|
VkDescriptorSet setHandle[Gfx::numFramesBuffered];
|
|
uint32 currentFrameSet;
|
|
PDescriptorAllocator owner;
|
|
PGraphics graphics;
|
|
friend class DescriptorAllocator;
|
|
};
|
|
DEFINE_REF(DescriptorSet);
|
|
|
|
} // namespace Vulkan
|
|
} // namespace Seele
|