TextureVulkan.h

#pragma once
 
#include <RendererFoundation/Resources/Texture.h>
 
#include <vulkan/vulkan.hpp>
 
class plGALBufferVulkan;
class plGALDeviceVulkan;
 
class plGALTextureVulkan : public plGALTexture
{
public:
  enum class StagingMode : plUInt8
  {
    None,
    Buffer,          
    Texture,         
    TextureAndBuffer 
  };
  struct SubResourceOffset
  {
    PL_DECLARE_POD_TYPE();
    plUInt32 m_uiOffset;
    plUInt32 m_uiSize;
    plUInt32 m_uiRowLength;
    plUInt32 m_uiImageHeight;
  };
 
  PL_ALWAYS_INLINE vk::Image GetImage() const;
  PL_ALWAYS_INLINE vk::Format GetImageFormat() const { return m_imageFormat; }
  PL_ALWAYS_INLINE vk::ImageLayout GetPreferredLayout() const;
  PL_ALWAYS_INLINE vk::ImageLayout GetPreferredLayout(vk::ImageLayout targetLayout) const;
  PL_ALWAYS_INLINE vk::PipelineStageFlags GetUsedByPipelineStage() const;
  PL_ALWAYS_INLINE vk::AccessFlags GetAccessMask() const;
 
  PL_ALWAYS_INLINE plVulkanAllocation GetAllocation() const;
  PL_ALWAYS_INLINE const plVulkanAllocationInfo& GetAllocationInfo() const;
 
  PL_ALWAYS_INLINE bool IsLinearLayout() const;
 
  vk::Extent3D GetMipLevelSize(plUInt32 uiMipLevel) const;
  vk::ImageSubresourceRange GetFullRange() const;
  vk::ImageAspectFlags GetAspectMask() const;
 
  // Read-back staging resources
  PL_ALWAYS_INLINE StagingMode GetStagingMode() const;
  PL_ALWAYS_INLINE plGALTextureHandle GetStagingTexture() const;
  PL_ALWAYS_INLINE plGALBufferHandle GetStagingBuffer() const;
  plUInt32 ComputeSubResourceOffsets(plDynamicArray<SubResourceOffset>& out_subResourceOffsets) const;
 
protected:
  friend class plGALDeviceVulkan;
  friend class plMemoryUtils;
 
  plGALTextureVulkan(const plGALTextureCreationDescription& Description, bool bLinearCPU, bool bStaging);
 
  ~plGALTextureVulkan();
 
  virtual plResult InitPlatform(plGALDevice* pDevice, plArrayPtr<plGALSystemMemoryDescription> pInitialData) override;
  virtual plResult DeInitPlatform(plGALDevice* pDevice) override;
  virtual void SetDebugNamePlatform(const char* szName) const override;
 
  static vk::Format ComputeImageFormat(plGALDeviceVulkan* pDevice, plEnum<plGALResourceFormat> galFormat, vk::ImageCreateInfo& ref_createInfo, vk::ImageFormatListCreateInfo& ref_imageFormats, bool bStaging);
  static void ComputeCreateInfo(plGALDeviceVulkan* pDevice, const plGALTextureCreationDescription& description, vk::ImageCreateInfo& ref_createInfo, vk::PipelineStageFlags& ref_stages, vk::AccessFlags& ref_access, vk::ImageLayout& ref_preferredLayout);
  static void ComputeCreateInfoLinear(vk::ImageCreateInfo& ref_createInfo, vk::PipelineStageFlags& ref_stages, vk::AccessFlags& ref_access);
  static void ComputeAllocInfo(bool bLinearCPU, plVulkanAllocationCreateInfo& ref_allocInfo);
  static StagingMode ComputeStagingMode(plGALDeviceVulkan* pDevice, const plGALTextureCreationDescription& description, const vk::ImageCreateInfo& createInfo);
 
  plResult CreateStagingBuffer(const vk::ImageCreateInfo& createInfo);
 
  vk::Image m_image;
  vk::Format m_imageFormat = vk::Format::eUndefined;
  vk::ImageLayout m_preferredLayout = vk::ImageLayout::eUndefined;
  vk::PipelineStageFlags m_stages = {};
  vk::AccessFlags m_access = {};
 
  plVulkanAllocation m_alloc = nullptr;
  plVulkanAllocationInfo m_allocInfo;
 
  plGALDeviceVulkan* m_pDevice = nullptr;
 
  bool m_bLinearCPU = false;
  bool m_bStaging = false;
 
  StagingMode m_stagingMode = StagingMode::None;
  plGALTextureHandle m_hStagingTexture;
  plGALBufferHandle m_hStagingBuffer;
};
 
 
 
#include <RendererVulkan/Resources/Implementation/TextureVulkan_inl.h>