mirror of
https://github.com/RPCSX/rpcsx.git
synced 2026-03-19 11:45:03 +01:00
303 lines
8.4 KiB
C++
303 lines
8.4 KiB
C++
#pragma once
|
|
#include "stdafx.h"
|
|
#include "VKRenderTargets.h"
|
|
#include "VKGSRender.h"
|
|
#include "../Common/TextureUtils.h"
|
|
|
|
namespace vk
|
|
{
|
|
class cached_texture_section : public rsx::buffered_section
|
|
{
|
|
u16 width;
|
|
u16 height;
|
|
u16 depth;
|
|
u16 mipmaps;
|
|
|
|
std::unique_ptr<vk::image_view> uploaded_image_view;
|
|
std::unique_ptr<vk::image> uploaded_texture;
|
|
|
|
public:
|
|
|
|
cached_texture_section() {}
|
|
|
|
void create(u16 w, u16 h, u16 depth, u16 mipmaps, vk::image_view *view, vk::image *image)
|
|
{
|
|
width = w;
|
|
height = h;
|
|
this->depth = depth;
|
|
this->mipmaps = mipmaps;
|
|
|
|
uploaded_image_view.reset(view);
|
|
uploaded_texture.reset(image);
|
|
}
|
|
|
|
bool matches(u32 rsx_address, u32 rsx_size) const
|
|
{
|
|
return rsx::buffered_section::matches(rsx_address, rsx_size);
|
|
}
|
|
|
|
bool matches(u32 rsx_address, u32 width, u32 height, u32 mipmaps) const
|
|
{
|
|
if (rsx_address == cpu_address_base)
|
|
{
|
|
if (!width && !height && !mipmaps)
|
|
return true;
|
|
|
|
return (width == this->width && height == this->height && mipmaps == this->mipmaps);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool exists() const
|
|
{
|
|
return (uploaded_texture.get() != nullptr);
|
|
}
|
|
|
|
u16 get_width() const
|
|
{
|
|
return width;
|
|
}
|
|
|
|
u16 get_height() const
|
|
{
|
|
return height;
|
|
}
|
|
|
|
std::unique_ptr<vk::image_view>& get_view()
|
|
{
|
|
return uploaded_image_view;
|
|
}
|
|
|
|
std::unique_ptr<vk::image>& get_texture()
|
|
{
|
|
return uploaded_texture;
|
|
}
|
|
};
|
|
|
|
class texture_cache
|
|
{
|
|
private:
|
|
std::vector<cached_texture_section> m_cache;
|
|
std::pair<u64, u64> texture_cache_range = std::make_pair(0xFFFFFFFF, 0);
|
|
std::vector<std::unique_ptr<vk::image_view> > m_temporary_image_view;
|
|
std::vector<std::unique_ptr<vk::image>> m_dirty_textures;
|
|
|
|
cached_texture_section& find_cached_texture(u32 rsx_address, u32 rsx_size, bool confirm_dimensions = false, u16 width = 0, u16 height = 0, u16 mipmaps = 0)
|
|
{
|
|
for (auto &tex : m_cache)
|
|
{
|
|
if (tex.matches(rsx_address, rsx_size) && !tex.is_dirty())
|
|
{
|
|
if (!confirm_dimensions) return tex;
|
|
|
|
if (tex.matches(rsx_address, width, height, mipmaps))
|
|
return tex;
|
|
else
|
|
{
|
|
LOG_ERROR(RSX, "Cached object for address 0x%X was found, but it does not match stored parameters.");
|
|
LOG_ERROR(RSX, "%d x %d vs %d x %d", width, height, tex.get_width(), tex.get_height());
|
|
}
|
|
}
|
|
}
|
|
|
|
for (auto &tex : m_cache)
|
|
{
|
|
if (tex.is_dirty())
|
|
{
|
|
if (tex.exists())
|
|
{
|
|
m_dirty_textures.push_back(std::move(tex.get_texture()));
|
|
m_temporary_image_view.push_back(std::move(tex.get_view()));
|
|
}
|
|
|
|
return tex;
|
|
}
|
|
}
|
|
|
|
m_cache.push_back(cached_texture_section());
|
|
|
|
return m_cache[m_cache.size() - 1];
|
|
}
|
|
|
|
void purge_cache()
|
|
{
|
|
for (auto &tex : m_cache)
|
|
{
|
|
if (tex.exists())
|
|
{
|
|
m_dirty_textures.push_back(std::move(tex.get_texture()));
|
|
m_temporary_image_view.push_back(std::move(tex.get_view()));
|
|
}
|
|
|
|
if (tex.is_locked())
|
|
tex.unprotect();
|
|
}
|
|
|
|
m_temporary_image_view.clear();
|
|
m_dirty_textures.clear();
|
|
|
|
m_cache.resize(0);
|
|
}
|
|
|
|
//Helpers
|
|
VkComponentMapping get_component_map(rsx::fragment_texture &tex, u32 gcm_format)
|
|
{
|
|
return vk::get_component_mapping(gcm_format, tex.remap());
|
|
}
|
|
|
|
VkComponentMapping get_component_map(rsx::vertex_texture &tex, u32 gcm_format)
|
|
{
|
|
return vk::get_component_mapping(gcm_format, (0 | 1 << 2 | 2 << 4 | 3 << 6));
|
|
}
|
|
|
|
public:
|
|
|
|
texture_cache() {}
|
|
~texture_cache() {}
|
|
|
|
void destroy()
|
|
{
|
|
purge_cache();
|
|
}
|
|
|
|
template <typename RsxTextureType>
|
|
vk::image_view* upload_texture(command_buffer cmd, RsxTextureType &tex, rsx::vk_render_targets &m_rtts, const vk::memory_type_mapping &memory_type_mapping, vk_data_heap& upload_heap, vk::buffer* upload_buffer)
|
|
{
|
|
const u32 texaddr = rsx::get_address(tex.offset(), tex.location());
|
|
const u32 range = (u32)get_texture_size(tex);
|
|
|
|
//First check if it exists as an rtt...
|
|
vk::image *rtt_texture = nullptr;
|
|
if (rtt_texture = m_rtts.get_texture_from_render_target_if_applicable(texaddr))
|
|
{
|
|
m_temporary_image_view.push_back(std::make_unique<vk::image_view>(*vk::get_current_renderer(), rtt_texture->value, VK_IMAGE_VIEW_TYPE_2D, rtt_texture->info.format,
|
|
rtt_texture->native_layout,
|
|
vk::get_image_subresource_range(0, 0, 1, 1, VK_IMAGE_ASPECT_COLOR_BIT)));
|
|
return m_temporary_image_view.back().get();
|
|
}
|
|
|
|
if (rtt_texture = m_rtts.get_texture_from_depth_stencil_if_applicable(texaddr))
|
|
{
|
|
m_temporary_image_view.push_back(std::make_unique<vk::image_view>(*vk::get_current_renderer(), rtt_texture->value, VK_IMAGE_VIEW_TYPE_2D, rtt_texture->info.format,
|
|
rtt_texture->native_layout,
|
|
vk::get_image_subresource_range(0, 0, 1, 1, VK_IMAGE_ASPECT_DEPTH_BIT)));
|
|
return m_temporary_image_view.back().get();
|
|
}
|
|
|
|
u32 raw_format = tex.format();
|
|
u32 format = raw_format & ~(CELL_GCM_TEXTURE_LN | CELL_GCM_TEXTURE_UN);
|
|
|
|
VkComponentMapping mapping = get_component_map(tex, format);
|
|
VkFormat vk_format = get_compatible_sampler_format(format);
|
|
|
|
VkImageType image_type;
|
|
VkImageViewType image_view_type;
|
|
u16 height = 0;
|
|
u16 depth = 0;
|
|
u8 layer = 0;
|
|
|
|
switch (tex.get_extended_texture_dimension())
|
|
{
|
|
case rsx::texture_dimension_extended::texture_dimension_1d:
|
|
image_type = VK_IMAGE_TYPE_1D;
|
|
image_view_type = VK_IMAGE_VIEW_TYPE_1D;
|
|
height = 1;
|
|
depth = 1;
|
|
layer = 1;
|
|
break;
|
|
case rsx::texture_dimension_extended::texture_dimension_2d:
|
|
image_type = VK_IMAGE_TYPE_2D;
|
|
image_view_type = VK_IMAGE_VIEW_TYPE_2D;
|
|
height = tex.height();
|
|
depth = 1;
|
|
layer = 1;
|
|
break;
|
|
case rsx::texture_dimension_extended::texture_dimension_cubemap:
|
|
image_type = VK_IMAGE_TYPE_2D;
|
|
image_view_type = VK_IMAGE_VIEW_TYPE_CUBE;
|
|
height = tex.height();
|
|
depth = 1;
|
|
layer = 6;
|
|
break;
|
|
case rsx::texture_dimension_extended::texture_dimension_3d:
|
|
image_type = VK_IMAGE_TYPE_3D;
|
|
image_view_type = VK_IMAGE_VIEW_TYPE_3D;
|
|
height = tex.height();
|
|
depth = tex.depth();
|
|
layer = 1;
|
|
break;
|
|
}
|
|
|
|
cached_texture_section& region = find_cached_texture(texaddr, range, true, tex.width(), height, tex.get_exact_mipmap_count());
|
|
if (region.exists() && !region.is_dirty())
|
|
{
|
|
return region.get_view().get();
|
|
}
|
|
|
|
bool is_cubemap = tex.get_extended_texture_dimension() == rsx::texture_dimension_extended::texture_dimension_cubemap;
|
|
VkImageSubresourceRange subresource_range = vk::get_image_subresource_range(0, 0, is_cubemap ? 6 : 1, tex.get_exact_mipmap_count(), VK_IMAGE_ASPECT_COLOR_BIT);
|
|
|
|
//If for some reason invalid dimensions are requested, fail
|
|
if (!height || !depth || !layer || !tex.width())
|
|
{
|
|
LOG_ERROR(RSX, "Texture upload requested but invalid texture dimensions passed");
|
|
return nullptr;
|
|
}
|
|
|
|
vk::image *image = new vk::image(*vk::get_current_renderer(), memory_type_mapping.device_local, VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT,
|
|
image_type,
|
|
vk_format,
|
|
tex.width(), height, depth, tex.get_exact_mipmap_count(), layer, VK_SAMPLE_COUNT_1_BIT, VK_IMAGE_LAYOUT_UNDEFINED,
|
|
VK_IMAGE_TILING_OPTIMAL, VK_IMAGE_USAGE_TRANSFER_DST_BIT | VK_IMAGE_USAGE_SAMPLED_BIT, is_cubemap ? VK_IMAGE_CREATE_CUBE_COMPATIBLE_BIT : 0);
|
|
change_image_layout(cmd, image->value, VK_IMAGE_LAYOUT_UNDEFINED, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, subresource_range);
|
|
|
|
vk::image_view *view = new vk::image_view(*vk::get_current_renderer(), image->value, image_view_type, vk_format,
|
|
mapping,
|
|
subresource_range);
|
|
|
|
copy_mipmaped_image_using_buffer(cmd, image->value, get_subresources_layout(tex), format, !(tex.format() & CELL_GCM_TEXTURE_LN), tex.get_exact_mipmap_count(),
|
|
upload_heap, upload_buffer);
|
|
|
|
change_image_layout(cmd, image->value, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, subresource_range);
|
|
|
|
region.reset(texaddr, range);
|
|
region.create(tex.width(), height, depth, tex.get_exact_mipmap_count(), view, image);
|
|
region.protect(0, vm::page_writable);
|
|
region.set_dirty(false);
|
|
|
|
return view;
|
|
}
|
|
|
|
bool invalidate_address(u32 rsx_address)
|
|
{
|
|
if (rsx_address < texture_cache_range.first ||
|
|
rsx_address > texture_cache_range.second)
|
|
return false;
|
|
|
|
bool response = false;
|
|
|
|
for (auto &tex : m_cache)
|
|
{
|
|
if (tex.is_dirty()) continue;
|
|
|
|
if (tex.overlaps(rsx_address))
|
|
{
|
|
tex.set_dirty(true);
|
|
tex.unprotect();
|
|
|
|
response = true;
|
|
}
|
|
}
|
|
|
|
return response;
|
|
}
|
|
|
|
void flush()
|
|
{
|
|
m_dirty_textures.clear();
|
|
m_temporary_image_view.clear();
|
|
}
|
|
};
|
|
}
|