PipelineBarrierVulkan.h

#pragma once
 
#include <RendererVulkan/RendererVulkanDLL.h>
 
#include <Foundation/Containers/Bitfield.h>
 
#include <vulkan/vulkan.hpp>
 
class plGALBufferVulkan;
class plGALTextureVulkan;
class plGALRenderTargetViewVulkan;
class plGALTextureResourceViewVulkan;
class plGALBufferResourceViewVulkan;
class plGALTextureUnorderedAccessViewVulkan;
class plGALBufferUnorderedAccessViewVulkan;
 
class PL_RENDERERVULKAN_DLL plPipelineBarrierVulkan
{
public:
  plPipelineBarrierVulkan(plAllocator* pAllocator);
 
 
  void SetCommandBuffer(vk::CommandBuffer* pCommandBuffer);
 
  void Flush();
 
  void Submit();
 
  void FullBarrier();
 
  bool IsDirty() const;
 
 
  void BufferDestroyed(const plGALBufferVulkan* pBuffer);
 
  void AccessBuffer(const plGALBufferVulkan* pBuffer, vk::DeviceSize offset, vk::DeviceSize length, vk::PipelineStageFlags srcStages, vk::AccessFlags srcAccess, vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess);
  bool IsDirty(vk::Buffer buffer, vk::DeviceSize offset, vk::DeviceSize length, vk::AccessFlags dstAccess);
 
  void SetInitialImageState(const plGALTextureVulkan* pTexture, vk::ImageLayout dstLayout, vk::PipelineStageFlags dstStages = vk::PipelineStageFlagBits::eTopOfPipe, vk::AccessFlags dstAccess = {});
 
  void TextureDestroyed(const plGALTextureVulkan* pTexture);
 
  void EnsureImageLayout(const plGALTextureVulkan* pTexture, vk::ImageLayout dstLayout, vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess, bool bDiscardSource = false);
  void EnsureImageLayout(const plGALRenderTargetViewVulkan* pTextureView, vk::ImageLayout dstLayout, vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess, bool bDiscardSource = false);
  void EnsureImageLayout(const plGALTextureResourceViewVulkan* pTextureView, vk::ImageLayout dstLayout, vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess, bool bDiscardSource = false);
  void EnsureImageLayout(const plGALTextureUnorderedAccessViewVulkan* pTextureView, vk::ImageLayout dstLayout, vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess, bool bDiscardSource = false);
  void EnsureImageLayout(const plGALTextureVulkan* pTexture, vk::ImageSubresourceRange subResources, vk::ImageLayout dstLayout, vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess, bool bDiscardSource = false);
 
  bool IsDirty(vk::Image image, const vk::ImageSubresourceRange& subResources) const;
 
private:
  struct SubElementState
  {
    vk::PipelineStageFlags m_stages;
    vk::AccessFlags m_accessMask;
    vk::ImageLayout m_layout;
 
    bool operator==(const SubElementState& rhs) const
    {
      return m_stages == rhs.m_stages && m_accessMask == rhs.m_accessMask && m_layout == rhs.m_layout;
    }
  };
 
  struct ImageState
  {
    const plGALTextureVulkan* m_pTexture = nullptr;
    plBitfield<plHybridArray<plUInt32, 1>> m_dirty;
    plHybridArray<SubElementState, 1> m_subElementLayout;
  };
 
  struct SubBufferState
  {
    vk::DeviceSize m_offset;
    vk::DeviceSize m_length;
    vk::PipelineStageFlags m_stages;
    vk::AccessFlags m_accessMask;
  };
 
  struct BufferState
  {
    const plGALBufferVulkan* m_pBuffer = nullptr;
    plBitfield<plHybridArray<plUInt32, 1>> m_dirty;
    plHybridArray<SubBufferState, 1> m_subBufferState;
  };
 
  bool AddBufferBarrierInternal(vk::Buffer buffer, vk::DeviceSize offset, vk::DeviceSize length,
    vk::PipelineStageFlags srcStages, vk::AccessFlags srcAccess,
    vk::PipelineStageFlags dstStages, vk::AccessFlags dstAccess);
  bool IsDirtyInternal(const BufferState& state, const SubBufferState& subState) const;
 
  bool AddImageBarrierInternal(vk::Image image, const vk::ImageSubresourceRange& subResources,
    vk::ImageLayout srcLayout, vk::AccessFlags srcAccess,
    vk::ImageLayout dstLayout, vk::AccessFlags dstAccess, bool bDiscardSource);
  bool IsDirtyInternal(const ImageState& state, const vk::ImageSubresourceRange& subResources) const;
 
private:
  static constexpr vk::AccessFlags s_readAccess = vk::AccessFlagBits::eIndirectCommandRead | vk::AccessFlagBits::eIndexRead |
                                                  vk::AccessFlagBits::eVertexAttributeRead | vk::AccessFlagBits::eUniformRead |
                                                  vk::AccessFlagBits::eInputAttachmentRead | vk::AccessFlagBits::eShaderRead |
                                                  vk::AccessFlagBits::eColorAttachmentRead | vk::AccessFlagBits::eDepthStencilAttachmentRead |
                                                  vk::AccessFlagBits::eTransferRead | vk::AccessFlagBits::eHostRead |
                                                  vk::AccessFlagBits::eMemoryRead // | vk::AccessFlagBits::eTransformFeedbackCounterReadEXT |
                                                                                  // vk::AccessFlagBits::eConditionalRenderingReadEXT | vk::AccessFlagBits::eColorAttachmentReadNoncoherentEXT |
                                                                                  // vk::AccessFlagBits::eAccelerationStructureReadKHR | vk::AccessFlagBits::eFragmentDensityMapReadEXT |
                                                                                  // vk::AccessFlagBits::eFragmentShadingRateAttachmentReadKHR | vk::AccessFlagBits::eCommandPreprocessReadNV |
                                                                                  // vk::AccessFlagBits::eAccelerationStructureReadKHR | vk::AccessFlagBits::eFragmentDensityMapReadEXT |
    /*vk::AccessFlagBits::eFragmentShadingRateAttachmentReadKHR | vk::AccessFlagBits::eCommandPreprocessReadNV*/;
  static constexpr vk::AccessFlags s_writeAccess = vk::AccessFlagBits::eShaderWrite | vk::AccessFlagBits::eColorAttachmentWrite |
                                                   vk::AccessFlagBits::eDepthStencilAttachmentWrite | vk::AccessFlagBits::eTransferWrite |
                                                   vk::AccessFlagBits::eHostWrite | vk::AccessFlagBits::eMemoryWrite
    // vk::AccessFlagBits::eTransformFeedbackWriteEXT | vk::AccessFlagBits::eTransformFeedbackCounterWriteEXT |
    /*vk::AccessFlagBits::eAccelerationStructureWriteKHR | vk::AccessFlagBits::eCommandPreprocessWriteNV*/;
 
  vk::CommandBuffer* m_pCommandBuffer = nullptr;
 
  vk::PipelineStageFlags m_srcStageMask;
  vk::PipelineStageFlags m_dstStageMask;
 
  vk::AccessFlags m_srcAccess;
  vk::AccessFlags m_dstAccess;
 
  plHybridArray<vk::BufferMemoryBarrier, 8> m_bufferBarriers;
  plHybridArray<vk::ImageMemoryBarrier, 8> m_imageBarriers;
 
  plMap<vk::Image, ImageState> m_imageState;
  plMap<vk::Buffer, BufferState> m_bufferState;
};