137 lines
5.1 KiB
C++
137 lines
5.1 KiB
C++
// Copyright 2017 Dolphin Emulator Project
|
|
// Licensed under GPLv2+
|
|
// Refer to the license.txt file included.
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
|
|
#include "VideoBackends/Vulkan/VulkanLoader.h"
|
|
#include "VideoCommon/AbstractFramebuffer.h"
|
|
#include "VideoCommon/AbstractStagingTexture.h"
|
|
#include "VideoCommon/AbstractTexture.h"
|
|
|
|
namespace Vulkan
|
|
{
|
|
class StagingBuffer;
|
|
class Texture2D;
|
|
|
|
class VKTexture final : public AbstractTexture
|
|
{
|
|
public:
|
|
// Custom image layouts, mainly used for switching to/from compute
|
|
enum class ComputeImageLayout
|
|
{
|
|
Undefined,
|
|
ReadOnly,
|
|
WriteOnly,
|
|
ReadWrite
|
|
};
|
|
|
|
VKTexture() = delete;
|
|
VKTexture(const TextureConfig& tex_config, VkDeviceMemory device_memory, VkImage image,
|
|
VkImageLayout layout = VK_IMAGE_LAYOUT_UNDEFINED,
|
|
ComputeImageLayout compute_layout = ComputeImageLayout::Undefined);
|
|
~VKTexture();
|
|
|
|
static VkFormat GetLinearFormat(VkFormat format);
|
|
static VkFormat GetVkFormatForHostTextureFormat(AbstractTextureFormat format);
|
|
static VkImageAspectFlags GetImageAspectForFormat(AbstractTextureFormat format);
|
|
|
|
void CopyRectangleFromTexture(const AbstractTexture* src,
|
|
const MathUtil::Rectangle<int>& src_rect, u32 src_layer,
|
|
u32 src_level, const MathUtil::Rectangle<int>& dst_rect,
|
|
u32 dst_layer, u32 dst_level) override;
|
|
void ResolveFromTexture(const AbstractTexture* src, const MathUtil::Rectangle<int>& rect,
|
|
u32 layer, u32 level) override;
|
|
void Load(u32 level, u32 width, u32 height, u32 row_length, const u8* buffer,
|
|
size_t buffer_size) override;
|
|
void FinishedRendering() override;
|
|
|
|
VkImage GetImage() const { return m_image; }
|
|
VkDeviceMemory GetDeviceMemory() const { return m_device_memory; }
|
|
VkImageView GetView() const { return m_view; }
|
|
VkImageLayout GetLayout() const { return m_layout; }
|
|
VkFormat GetVkFormat() const { return GetVkFormatForHostTextureFormat(m_config.format); }
|
|
bool IsAdopted() const { return m_device_memory != nullptr; }
|
|
|
|
static std::unique_ptr<VKTexture> Create(const TextureConfig& tex_config);
|
|
static std::unique_ptr<VKTexture>
|
|
CreateAdopted(const TextureConfig& tex_config, VkImage image,
|
|
VkImageViewType view_type = VK_IMAGE_VIEW_TYPE_2D_ARRAY,
|
|
VkImageLayout layout = VK_IMAGE_LAYOUT_UNDEFINED);
|
|
|
|
// Used when the render pass is changing the image layout, or to force it to
|
|
// VK_IMAGE_LAYOUT_UNDEFINED, if the existing contents of the image is
|
|
// irrelevant and will not be loaded.
|
|
void OverrideImageLayout(VkImageLayout new_layout);
|
|
|
|
void TransitionToLayout(VkCommandBuffer command_buffer, VkImageLayout new_layout) const;
|
|
void TransitionToLayout(VkCommandBuffer command_buffer, ComputeImageLayout new_layout) const;
|
|
|
|
private:
|
|
bool CreateView(VkImageViewType type);
|
|
|
|
VkDeviceMemory m_device_memory;
|
|
VkImage m_image;
|
|
VkImageView m_view = VK_NULL_HANDLE;
|
|
mutable VkImageLayout m_layout = VK_IMAGE_LAYOUT_UNDEFINED;
|
|
mutable ComputeImageLayout m_compute_layout = ComputeImageLayout::Undefined;
|
|
};
|
|
|
|
class VKStagingTexture final : public AbstractStagingTexture
|
|
{
|
|
public:
|
|
VKStagingTexture() = delete;
|
|
~VKStagingTexture();
|
|
|
|
void CopyFromTexture(const AbstractTexture* src, const MathUtil::Rectangle<int>& src_rect,
|
|
u32 src_layer, u32 src_level,
|
|
const MathUtil::Rectangle<int>& dst_rect) override;
|
|
void CopyToTexture(const MathUtil::Rectangle<int>& src_rect, AbstractTexture* dst,
|
|
const MathUtil::Rectangle<int>& dst_rect, u32 dst_layer,
|
|
u32 dst_level) override;
|
|
|
|
bool Map() override;
|
|
void Unmap() override;
|
|
void Flush() override;
|
|
|
|
static std::unique_ptr<VKStagingTexture> Create(StagingTextureType type,
|
|
const TextureConfig& config);
|
|
|
|
private:
|
|
VKStagingTexture(StagingTextureType type, const TextureConfig& config,
|
|
std::unique_ptr<StagingBuffer> buffer);
|
|
|
|
std::unique_ptr<StagingBuffer> m_staging_buffer;
|
|
u64 m_flush_fence_counter = 0;
|
|
};
|
|
|
|
class VKFramebuffer final : public AbstractFramebuffer
|
|
{
|
|
public:
|
|
VKFramebuffer(VKTexture* color_attachment, VKTexture* depth_attachment, u32 width, u32 height,
|
|
u32 layers, u32 samples, VkFramebuffer fb, VkRenderPass load_render_pass,
|
|
VkRenderPass discard_render_pass, VkRenderPass clear_render_pass);
|
|
~VKFramebuffer() override;
|
|
|
|
VkFramebuffer GetFB() const { return m_fb; }
|
|
VkRect2D GetRect() const { return VkRect2D{{0, 0}, {m_width, m_height}}; }
|
|
|
|
VkRenderPass GetLoadRenderPass() const { return m_load_render_pass; }
|
|
VkRenderPass GetDiscardRenderPass() const { return m_discard_render_pass; }
|
|
VkRenderPass GetClearRenderPass() const { return m_clear_render_pass; }
|
|
void TransitionForRender();
|
|
|
|
static std::unique_ptr<VKFramebuffer> Create(VKTexture* color_attachments,
|
|
VKTexture* depth_attachment);
|
|
|
|
protected:
|
|
VkFramebuffer m_fb;
|
|
VkRenderPass m_load_render_pass;
|
|
VkRenderPass m_discard_render_pass;
|
|
VkRenderPass m_clear_render_pass;
|
|
};
|
|
|
|
} // namespace Vulkan
|