CommandEncoderImplVulkan.h

 
#pragma once
 
#include <RendererVulkan/RendererVulkanDLL.h>
 
#include <Foundation/Types/Bitflags.h>
#include <RendererFoundation/CommandEncoder/CommandEncoderPlatformInterface.h>
#include <RendererFoundation/Resources/RenderTargetSetup.h>
#include <RendererVulkan/Cache/ResourceCacheVulkan.h>
 
#include <vulkan/vulkan.hpp>
 
class plGALBlendStateVulkan;
class plGALBufferVulkan;
class plGALDepthStencilStateVulkan;
class plGALRasterizerStateVulkan;
class plGALTextureResourceViewVulkan;
class plGALBufferResourceViewVulkan;
class plGALSamplerStateVulkan;
class plGALShaderVulkan;
class plGALTextureUnorderedAccessViewVulkan;
class plGALBufferUnorderedAccessViewVulkan;
class plGALDeviceVulkan;
class plFenceQueueVulkan;
 
 
class PL_RENDERERVULKAN_DLL plGALCommandEncoderImplVulkan : public plGALCommandEncoderCommonPlatformInterface
{
public:
  plGALCommandEncoderImplVulkan(plGALDeviceVulkan& device);
  ~plGALCommandEncoderImplVulkan();
 
  void Reset();
 
  void SetCurrentCommandBuffer(vk::CommandBuffer* commandBuffer, plPipelineBarrierVulkan* pipelineBarrier);
  void CommandBufferSubmitted(vk::Fence submitFence);
 
  // plGALCommandEncoderCommonPlatformInterface
  // State setting functions
 
  virtual void SetShaderPlatform(const plGALShader* pShader) override;
 
  virtual void SetConstantBufferPlatform(const plShaderResourceBinding& binding, const plGALBuffer* pBuffer) override;
  virtual void SetSamplerStatePlatform(const plShaderResourceBinding& binding, const plGALSamplerState* pSamplerState) override;
  virtual void SetResourceViewPlatform(const plShaderResourceBinding& binding, const plGALTextureResourceView* pResourceView) override;
  virtual void SetResourceViewPlatform(const plShaderResourceBinding& binding, const plGALBufferResourceView* pResourceView) override;
  virtual void SetUnorderedAccessViewPlatform(const plShaderResourceBinding& binding, const plGALTextureUnorderedAccessView* pUnorderedAccessView) override;
  virtual void SetUnorderedAccessViewPlatform(const plShaderResourceBinding& binding, const plGALBufferUnorderedAccessView* pUnorderedAccessView) override;
  virtual void SetPushConstantsPlatform(plArrayPtr<const plUInt8> data) override;
 
  // GPU -> CPU query functions
 
  virtual plGALTimestampHandle InsertTimestampPlatform() override;
  virtual plGALOcclusionHandle BeginOcclusionQueryPlatform(plEnum<plGALQueryType> type) override;
  virtual void EndOcclusionQueryPlatform(plGALOcclusionHandle hOcclusion) override;
  virtual plGALFenceHandle InsertFencePlatform() override;
 
 
  // Resource update functions
 
  virtual void ClearUnorderedAccessViewPlatform(const plGALTextureUnorderedAccessView* pUnorderedAccessView, plVec4 clearValues) override;
  virtual void ClearUnorderedAccessViewPlatform(const plGALBufferUnorderedAccessView* pUnorderedAccessView, plVec4 clearValues) override;
 
  virtual void ClearUnorderedAccessViewPlatform(const plGALTextureUnorderedAccessView* pUnorderedAccessView, plVec4U32 clearValues) override;
  virtual void ClearUnorderedAccessViewPlatform(const plGALBufferUnorderedAccessView* pUnorderedAccessView, plVec4U32 clearValues) override;
 
  virtual void CopyBufferPlatform(const plGALBuffer* pDestination, const plGALBuffer* pSource) override;
  virtual void CopyBufferRegionPlatform(const plGALBuffer* pDestination, plUInt32 uiDestOffset, const plGALBuffer* pSource, plUInt32 uiSourceOffset, plUInt32 uiByteCount) override;
 
  virtual void UpdateBufferPlatform(const plGALBuffer* pDestination, plUInt32 uiDestOffset, plArrayPtr<const plUInt8> pSourceData, plGALUpdateMode::Enum updateMode) override;
 
  virtual void CopyTexturePlatform(const plGALTexture* pDestination, const plGALTexture* pSource) override;
  virtual void CopyTextureRegionPlatform(const plGALTexture* pDestination, const plGALTextureSubresource& DestinationSubResource, const plVec3U32& DestinationPoint, const plGALTexture* pSource, const plGALTextureSubresource& SourceSubResource, const plBoundingBoxu32& Box) override;
 
  virtual void UpdateTexturePlatform(const plGALTexture* pDestination, const plGALTextureSubresource& DestinationSubResource, const plBoundingBoxu32& DestinationBox, const plGALSystemMemoryDescription& pSourceData) override;
 
  virtual void ResolveTexturePlatform(const plGALTexture* pDestination, const plGALTextureSubresource& DestinationSubResource, const plGALTexture* pSource, const plGALTextureSubresource& SourceSubResource) override;
 
  virtual void ReadbackTexturePlatform(const plGALTexture* pTexture) override;
 
  virtual void CopyTextureReadbackResultPlatform(const plGALTexture* pTexture, plArrayPtr<plGALTextureSubresource> SourceSubResource, plArrayPtr<plGALSystemMemoryDescription> TargetData) override;
 
  virtual void GenerateMipMapsPlatform(const plGALTextureResourceView* pResourceView) override;
 
  void CopyImageToBuffer(const plGALTextureVulkan* pSource, const plGALBufferVulkan* pDestination);
 
  // Misc
 
  virtual void FlushPlatform() override;
 
  // Debug helper functions
 
  virtual void PushMarkerPlatform(const char* szMarker) override;
  virtual void PopMarkerPlatform() override;
  virtual void InsertEventMarkerPlatform(const char* szMarker) override;
 
 
  // plGALCommandEncoderComputePlatformInterface
  // Dispatch
  virtual void BeginComputePlatform() override;
  virtual void EndComputePlatform() override;
 
  virtual plResult DispatchPlatform(plUInt32 uiThreadGroupCountX, plUInt32 uiThreadGroupCountY, plUInt32 uiThreadGroupCountZ) override;
  virtual plResult DispatchIndirectPlatform(const plGALBuffer* pIndirectArgumentBuffer, plUInt32 uiArgumentOffsetInBytes) override;
 
 
  // plGALCommandEncoderRenderPlatformInterface
  virtual void BeginRenderingPlatform(const plGALRenderingSetup& renderingSetup) override;
  virtual void EndRenderingPlatform() override;
 
  // Draw functions
 
  virtual void ClearPlatform(const plColor& ClearColor, plUInt32 uiRenderTargetClearMask, bool bClearDepth, bool bClearStencil, float fDepthClear, plUInt8 uiStencilClear) override;
 
  virtual plResult DrawPlatform(plUInt32 uiVertexCount, plUInt32 uiStartVertex) override;
  virtual plResult DrawIndexedPlatform(plUInt32 uiIndexCount, plUInt32 uiStartIndex) override;
  virtual plResult DrawIndexedInstancedPlatform(plUInt32 uiIndexCountPerInstance, plUInt32 uiInstanceCount, plUInt32 uiStartIndex) override;
  virtual plResult DrawIndexedInstancedIndirectPlatform(const plGALBuffer* pIndirectArgumentBuffer, plUInt32 uiArgumentOffsetInBytes) override;
  virtual plResult DrawInstancedPlatform(plUInt32 uiVertexCountPerInstance, plUInt32 uiInstanceCount, plUInt32 uiStartVertex) override;
  virtual plResult DrawInstancedIndirectPlatform(const plGALBuffer* pIndirectArgumentBuffer, plUInt32 uiArgumentOffsetInBytes) override;
 
  // State functions
 
  virtual void SetIndexBufferPlatform(const plGALBuffer* pIndexBuffer) override;
  virtual void SetVertexBufferPlatform(plUInt32 uiSlot, const plGALBuffer* pVertexBuffer) override;
  virtual void SetVertexDeclarationPlatform(const plGALVertexDeclaration* pVertexDeclaration) override;
  virtual void SetPrimitiveTopologyPlatform(plGALPrimitiveTopology::Enum Topology) override;
 
  virtual void SetBlendStatePlatform(const plGALBlendState* pBlendState, const plColor& BlendFactor, plUInt32 uiSampleMask) override;
  virtual void SetDepthStencilStatePlatform(const plGALDepthStencilState* pDepthStencilState, plUInt8 uiStencilRefValue) override;
  virtual void SetRasterizerStatePlatform(const plGALRasterizerState* pRasterizerState) override;
 
  virtual void SetViewportPlatform(const plRectFloat& rect, float fMinDepth, float fMaxDepth) override;
  virtual void SetScissorRectPlatform(const plRectU32& rect) override;
 
 
 
private:
  // Map resources from sets then slots to pointer.
  struct SetResources
  {
    plDynamicArray<const plGALBufferVulkan*> m_pBoundConstantBuffers;
    plDynamicArray<const plGALTextureResourceViewVulkan*> m_pBoundTextureResourceViews;
    plDynamicArray<const plGALBufferResourceViewVulkan*> m_pBoundBufferResourceViews;
    plDynamicArray<const plGALTextureUnorderedAccessViewVulkan*> m_pBoundTextureUnorderedAccessViews;
    plDynamicArray<const plGALBufferUnorderedAccessViewVulkan*> m_pBoundBufferUnorderedAccessViews;
    plDynamicArray<const plGALSamplerStateVulkan*> m_pBoundSamplerStates;
  };
 
private:
  plResult FlushDeferredStateChanges();
  const plGALTextureResourceViewVulkan* GetTextureResourceView(const SetResources& resources, const plShaderResourceBinding& mapping);
  const plGALBufferResourceViewVulkan* GetBufferResourceView(const SetResources& resources, const plShaderResourceBinding& mapping);
  const plGALTextureUnorderedAccessViewVulkan* GetTextureUAV(const SetResources& resources, const plShaderResourceBinding& mapping);
  const plGALBufferUnorderedAccessViewVulkan* GetBufferUAV(const SetResources& resources, const plShaderResourceBinding& mapping);
 
private:
  plGALDeviceVulkan& m_GALDeviceVulkan;
  vk::Device m_vkDevice;
 
  vk::CommandBuffer* m_pCommandBuffer = nullptr;
  plPipelineBarrierVulkan* m_pPipelineBarrier = nullptr;
 
  // Cache flags.
  bool m_bPipelineStateDirty = true;
  bool m_bViewportDirty = true;
  bool m_bIndexBufferDirty = false;
  bool m_bDescriptorsDirty = false;
  plGAL::ModifiedRange m_BoundVertexBuffersRange;
  bool m_bRenderPassActive = false; 
  bool m_bClearSubmitted = false;   
  bool m_bInsideCompute = false;    
  bool m_bPushConstantsDirty = false;
 
  // Bound objects for deferred state flushes
  plResourceCacheVulkan::PipelineLayoutDesc m_LayoutDesc;
  plResourceCacheVulkan::GraphicsPipelineDesc m_PipelineDesc;
  plResourceCacheVulkan::ComputePipelineDesc m_ComputeDesc;
  vk::Framebuffer m_frameBuffer;
  vk::RenderPassBeginInfo m_renderPass;
  plHybridArray<vk::ClearValue, PL_GAL_MAX_RENDERTARGET_COUNT + 1> m_clearValues;
  vk::ImageAspectFlags m_depthMask = {};
  plUInt32 m_uiLayers = 0;
 
  vk::Viewport m_viewport;
  vk::Rect2D m_scissor;
  bool m_bScissorEnabled = false;
 
  const plGALRenderTargetView* m_pBoundRenderTargets[PL_GAL_MAX_RENDERTARGET_COUNT] = {};
  const plGALRenderTargetView* m_pBoundDepthStencilTarget = nullptr;
  plUInt32 m_uiBoundRenderTargetCount;
 
  const plGALBufferVulkan* m_pIndexBuffer = nullptr;
  vk::Buffer m_pBoundVertexBuffers[PL_GAL_MAX_VERTEX_BUFFER_COUNT];
  vk::DeviceSize m_VertexBufferOffsets[PL_GAL_MAX_VERTEX_BUFFER_COUNT] = {};
 
  plHybridArray<SetResources, 4> m_Resources;
 
  plDeque<vk::DescriptorImageInfo> m_TextureAndSampler;
  plHybridArray<vk::WriteDescriptorSet, 16> m_DescriptorWrites;
  plHybridArray<vk::DescriptorSet, 4> m_DescriptorSets;
 
  plDynamicArray<plUInt8> m_PushConstants;
};