|
|
@@ -6,11 +6,13 @@
|
|
|
#include "common/settings.h"
|
|
|
|
|
|
#include "video_core/fsr.h"
|
|
|
-#include "video_core/host_shaders/vulkan_fidelityfx_fsr_easu_fp16_comp_spv.h"
|
|
|
-#include "video_core/host_shaders/vulkan_fidelityfx_fsr_easu_fp32_comp_spv.h"
|
|
|
-#include "video_core/host_shaders/vulkan_fidelityfx_fsr_rcas_fp16_comp_spv.h"
|
|
|
-#include "video_core/host_shaders/vulkan_fidelityfx_fsr_rcas_fp32_comp_spv.h"
|
|
|
+#include "video_core/host_shaders/vulkan_fidelityfx_fsr_easu_fp16_frag_spv.h"
|
|
|
+#include "video_core/host_shaders/vulkan_fidelityfx_fsr_easu_fp32_frag_spv.h"
|
|
|
+#include "video_core/host_shaders/vulkan_fidelityfx_fsr_rcas_fp16_frag_spv.h"
|
|
|
+#include "video_core/host_shaders/vulkan_fidelityfx_fsr_rcas_fp32_frag_spv.h"
|
|
|
+#include "video_core/host_shaders/vulkan_fidelityfx_fsr_vert_spv.h"
|
|
|
#include "video_core/renderer_vulkan/present/fsr.h"
|
|
|
+#include "video_core/renderer_vulkan/present/util.h"
|
|
|
#include "video_core/renderer_vulkan/vk_scheduler.h"
|
|
|
#include "video_core/renderer_vulkan/vk_shader_util.h"
|
|
|
#include "video_core/vulkan_common/vulkan_device.h"
|
|
|
@@ -18,403 +20,207 @@
|
|
|
namespace Vulkan {
|
|
|
using namespace FSR;
|
|
|
|
|
|
-FSR::FSR(const Device& device_, MemoryAllocator& memory_allocator_, size_t image_count_,
|
|
|
- VkExtent2D output_size_)
|
|
|
- : device{device_}, memory_allocator{memory_allocator_}, image_count{image_count_},
|
|
|
- output_size{output_size_} {
|
|
|
+using PushConstants = std::array<u32, 4 * 4>;
|
|
|
+
|
|
|
+FSR::FSR(const Device& device, MemoryAllocator& memory_allocator, size_t image_count,
|
|
|
+ VkExtent2D extent)
|
|
|
+ : m_device{device}, m_memory_allocator{memory_allocator},
|
|
|
+ m_image_count{image_count}, m_extent{extent} {
|
|
|
|
|
|
CreateImages();
|
|
|
+ CreateRenderPasses();
|
|
|
CreateSampler();
|
|
|
CreateShaders();
|
|
|
CreateDescriptorPool();
|
|
|
CreateDescriptorSetLayout();
|
|
|
CreateDescriptorSets();
|
|
|
- CreatePipelineLayout();
|
|
|
- CreatePipeline();
|
|
|
+ CreatePipelineLayouts();
|
|
|
+ CreatePipelines();
|
|
|
}
|
|
|
|
|
|
-VkImageView FSR::Draw(Scheduler& scheduler, size_t image_index, VkImageView image_view,
|
|
|
- VkExtent2D input_image_extent, const Common::Rectangle<f32>& crop_rect) {
|
|
|
-
|
|
|
- UpdateDescriptorSet(image_index, image_view);
|
|
|
-
|
|
|
- scheduler.Record([this, image_index, input_image_extent, crop_rect](vk::CommandBuffer cmdbuf) {
|
|
|
- const VkImageMemoryBarrier base_barrier{
|
|
|
- .sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER,
|
|
|
- .pNext = nullptr,
|
|
|
- .srcAccessMask = 0,
|
|
|
- .dstAccessMask = 0,
|
|
|
- .oldLayout = VK_IMAGE_LAYOUT_GENERAL,
|
|
|
- .newLayout = VK_IMAGE_LAYOUT_GENERAL,
|
|
|
- .srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
- .dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
- .image = {},
|
|
|
- .subresourceRange =
|
|
|
- {
|
|
|
- .aspectMask = VK_IMAGE_ASPECT_COLOR_BIT,
|
|
|
- .baseMipLevel = 0,
|
|
|
- .levelCount = 1,
|
|
|
- .baseArrayLayer = 0,
|
|
|
- .layerCount = 1,
|
|
|
- },
|
|
|
- };
|
|
|
-
|
|
|
- cmdbuf.BindPipeline(VK_PIPELINE_BIND_POINT_COMPUTE, *easu_pipeline);
|
|
|
-
|
|
|
- const f32 input_image_width = static_cast<f32>(input_image_extent.width);
|
|
|
- const f32 input_image_height = static_cast<f32>(input_image_extent.height);
|
|
|
- const f32 output_image_width = static_cast<f32>(output_size.width);
|
|
|
- const f32 output_image_height = static_cast<f32>(output_size.height);
|
|
|
- const f32 viewport_width = (crop_rect.right - crop_rect.left) * input_image_width;
|
|
|
- const f32 viewport_x = crop_rect.left * input_image_width;
|
|
|
- const f32 viewport_height = (crop_rect.bottom - crop_rect.top) * input_image_height;
|
|
|
- const f32 viewport_y = crop_rect.top * input_image_height;
|
|
|
-
|
|
|
- std::array<u32, 4 * 4> push_constants;
|
|
|
- FsrEasuConOffset(push_constants.data() + 0, push_constants.data() + 4,
|
|
|
- push_constants.data() + 8, push_constants.data() + 12,
|
|
|
-
|
|
|
- viewport_width, viewport_height, input_image_width, input_image_height,
|
|
|
- output_image_width, output_image_height, viewport_x, viewport_y);
|
|
|
- cmdbuf.PushConstants(*pipeline_layout, VK_SHADER_STAGE_COMPUTE_BIT, push_constants);
|
|
|
-
|
|
|
- {
|
|
|
- VkImageMemoryBarrier fsr_write_barrier = base_barrier;
|
|
|
- fsr_write_barrier.image = *images[image_index];
|
|
|
- fsr_write_barrier.oldLayout = VK_IMAGE_LAYOUT_UNDEFINED;
|
|
|
-
|
|
|
- cmdbuf.PipelineBarrier(VK_PIPELINE_STAGE_ALL_COMMANDS_BIT,
|
|
|
- VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT, 0, fsr_write_barrier);
|
|
|
- }
|
|
|
-
|
|
|
- cmdbuf.BindDescriptorSets(VK_PIPELINE_BIND_POINT_COMPUTE, *pipeline_layout, 0,
|
|
|
- descriptor_sets[image_index * 2], {});
|
|
|
- cmdbuf.Dispatch(Common::DivCeil(output_size.width, 16u),
|
|
|
- Common::DivCeil(output_size.height, 16u), 1);
|
|
|
-
|
|
|
- cmdbuf.BindPipeline(VK_PIPELINE_BIND_POINT_COMPUTE, *rcas_pipeline);
|
|
|
-
|
|
|
- const float sharpening =
|
|
|
- static_cast<float>(Settings::values.fsr_sharpening_slider.GetValue()) / 100.0f;
|
|
|
-
|
|
|
- FsrRcasCon(push_constants.data(), sharpening);
|
|
|
- cmdbuf.PushConstants(*pipeline_layout, VK_SHADER_STAGE_COMPUTE_BIT, push_constants);
|
|
|
-
|
|
|
- {
|
|
|
- std::array<VkImageMemoryBarrier, 2> barriers;
|
|
|
- auto& fsr_read_barrier = barriers[0];
|
|
|
- auto& blit_write_barrier = barriers[1];
|
|
|
-
|
|
|
- fsr_read_barrier = base_barrier;
|
|
|
- fsr_read_barrier.image = *images[image_index];
|
|
|
- fsr_read_barrier.srcAccessMask = VK_ACCESS_SHADER_WRITE_BIT;
|
|
|
- fsr_read_barrier.dstAccessMask = VK_ACCESS_SHADER_READ_BIT;
|
|
|
-
|
|
|
- blit_write_barrier = base_barrier;
|
|
|
- blit_write_barrier.image = *images[image_count + image_index];
|
|
|
- blit_write_barrier.oldLayout = VK_IMAGE_LAYOUT_UNDEFINED;
|
|
|
- blit_write_barrier.newLayout = VK_IMAGE_LAYOUT_GENERAL;
|
|
|
-
|
|
|
- cmdbuf.PipelineBarrier(VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
|
|
|
- VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT, 0, {}, {}, barriers);
|
|
|
- }
|
|
|
+void FSR::CreateImages() {
|
|
|
+ m_dynamic_images.resize(m_image_count);
|
|
|
+ for (auto& images : m_dynamic_images) {
|
|
|
+ images.images[Easu] =
|
|
|
+ CreateWrappedImage(m_memory_allocator, m_extent, VK_FORMAT_R16G16B16A16_SFLOAT);
|
|
|
+ images.images[Rcas] =
|
|
|
+ CreateWrappedImage(m_memory_allocator, m_extent, VK_FORMAT_R16G16B16A16_SFLOAT);
|
|
|
+ images.image_views[Easu] =
|
|
|
+ CreateWrappedImageView(m_device, images.images[Easu], VK_FORMAT_R16G16B16A16_SFLOAT);
|
|
|
+ images.image_views[Rcas] =
|
|
|
+ CreateWrappedImageView(m_device, images.images[Rcas], VK_FORMAT_R16G16B16A16_SFLOAT);
|
|
|
+ }
|
|
|
+}
|
|
|
|
|
|
- cmdbuf.BindDescriptorSets(VK_PIPELINE_BIND_POINT_COMPUTE, *pipeline_layout, 0,
|
|
|
- descriptor_sets[image_index * 2 + 1], {});
|
|
|
- cmdbuf.Dispatch(Common::DivCeil(output_size.width, 16u),
|
|
|
- Common::DivCeil(output_size.height, 16u), 1);
|
|
|
+void FSR::CreateRenderPasses() {
|
|
|
+ m_renderpass = CreateWrappedRenderPass(m_device, VK_FORMAT_R16G16B16A16_SFLOAT);
|
|
|
|
|
|
- {
|
|
|
- std::array<VkImageMemoryBarrier, 1> barriers;
|
|
|
- auto& blit_read_barrier = barriers[0];
|
|
|
+ for (auto& images : m_dynamic_images) {
|
|
|
+ images.framebuffers[Easu] =
|
|
|
+ CreateWrappedFramebuffer(m_device, m_renderpass, images.image_views[Easu], m_extent);
|
|
|
+ images.framebuffers[Rcas] =
|
|
|
+ CreateWrappedFramebuffer(m_device, m_renderpass, images.image_views[Rcas], m_extent);
|
|
|
+ }
|
|
|
+}
|
|
|
|
|
|
- blit_read_barrier = base_barrier;
|
|
|
- blit_read_barrier.image = *images[image_count + image_index];
|
|
|
- blit_read_barrier.srcAccessMask = VK_ACCESS_SHADER_WRITE_BIT;
|
|
|
- blit_read_barrier.dstAccessMask = VK_ACCESS_SHADER_READ_BIT;
|
|
|
+void FSR::CreateSampler() {
|
|
|
+ m_sampler = CreateBilinearSampler(m_device);
|
|
|
+}
|
|
|
|
|
|
- cmdbuf.PipelineBarrier(VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
|
|
|
- VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT, 0, {}, {}, barriers);
|
|
|
- }
|
|
|
- });
|
|
|
+void FSR::CreateShaders() {
|
|
|
+ m_vert_shader = BuildShader(m_device, VULKAN_FIDELITYFX_FSR_VERT_SPV);
|
|
|
|
|
|
- return *image_views[image_count + image_index];
|
|
|
+ if (m_device.IsFloat16Supported()) {
|
|
|
+ m_easu_shader = BuildShader(m_device, VULKAN_FIDELITYFX_FSR_EASU_FP16_FRAG_SPV);
|
|
|
+ m_rcas_shader = BuildShader(m_device, VULKAN_FIDELITYFX_FSR_RCAS_FP16_FRAG_SPV);
|
|
|
+ } else {
|
|
|
+ m_easu_shader = BuildShader(m_device, VULKAN_FIDELITYFX_FSR_EASU_FP32_FRAG_SPV);
|
|
|
+ m_rcas_shader = BuildShader(m_device, VULKAN_FIDELITYFX_FSR_RCAS_FP32_FRAG_SPV);
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
void FSR::CreateDescriptorPool() {
|
|
|
- const std::array<VkDescriptorPoolSize, 2> pool_sizes{{
|
|
|
- {
|
|
|
- .type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
|
|
|
- .descriptorCount = static_cast<u32>(image_count * 2),
|
|
|
- },
|
|
|
- {
|
|
|
- .type = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
|
|
|
- .descriptorCount = static_cast<u32>(image_count * 2),
|
|
|
- },
|
|
|
- }};
|
|
|
-
|
|
|
- const VkDescriptorPoolCreateInfo ci{
|
|
|
- .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .maxSets = static_cast<u32>(image_count * 2),
|
|
|
- .poolSizeCount = static_cast<u32>(pool_sizes.size()),
|
|
|
- .pPoolSizes = pool_sizes.data(),
|
|
|
- };
|
|
|
- descriptor_pool = device.GetLogical().CreateDescriptorPool(ci);
|
|
|
+ // EASU: 1 descriptor
|
|
|
+ // RCAS: 1 descriptor
|
|
|
+ // 2 descriptors, 2 descriptor sets per invocation
|
|
|
+ m_descriptor_pool = CreateWrappedDescriptorPool(m_device, 2 * m_image_count, 2 * m_image_count);
|
|
|
}
|
|
|
|
|
|
void FSR::CreateDescriptorSetLayout() {
|
|
|
- const std::array<VkDescriptorSetLayoutBinding, 2> layout_bindings{{
|
|
|
- {
|
|
|
- .binding = 0,
|
|
|
- .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
|
|
|
- .descriptorCount = 1,
|
|
|
- .stageFlags = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
- .pImmutableSamplers = sampler.address(),
|
|
|
- },
|
|
|
- {
|
|
|
- .binding = 1,
|
|
|
- .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
|
|
|
- .descriptorCount = 1,
|
|
|
- .stageFlags = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
- .pImmutableSamplers = sampler.address(),
|
|
|
- },
|
|
|
- }};
|
|
|
-
|
|
|
- const VkDescriptorSetLayoutCreateInfo ci{
|
|
|
- .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .bindingCount = static_cast<u32>(layout_bindings.size()),
|
|
|
- .pBindings = layout_bindings.data(),
|
|
|
- };
|
|
|
-
|
|
|
- descriptor_set_layout = device.GetLogical().CreateDescriptorSetLayout(ci);
|
|
|
+ m_descriptor_set_layout =
|
|
|
+ CreateWrappedDescriptorSetLayout(m_device, {VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER});
|
|
|
}
|
|
|
|
|
|
void FSR::CreateDescriptorSets() {
|
|
|
- const u32 sets = static_cast<u32>(image_count * 2);
|
|
|
- const std::vector layouts(sets, *descriptor_set_layout);
|
|
|
-
|
|
|
- const VkDescriptorSetAllocateInfo ai{
|
|
|
- .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .descriptorPool = *descriptor_pool,
|
|
|
- .descriptorSetCount = sets,
|
|
|
- .pSetLayouts = layouts.data(),
|
|
|
- };
|
|
|
+ std::vector<VkDescriptorSetLayout> layouts(MaxFsrStage, *m_descriptor_set_layout);
|
|
|
|
|
|
- descriptor_sets = descriptor_pool.Allocate(ai);
|
|
|
-}
|
|
|
-
|
|
|
-void FSR::CreateImages() {
|
|
|
- images.resize(image_count * 2);
|
|
|
- image_views.resize(image_count * 2);
|
|
|
-
|
|
|
- for (size_t i = 0; i < image_count * 2; ++i) {
|
|
|
- images[i] = memory_allocator.CreateImage(VkImageCreateInfo{
|
|
|
- .sType = VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .imageType = VK_IMAGE_TYPE_2D,
|
|
|
- .format = VK_FORMAT_R16G16B16A16_SFLOAT,
|
|
|
- .extent =
|
|
|
- {
|
|
|
- .width = output_size.width,
|
|
|
- .height = output_size.height,
|
|
|
- .depth = 1,
|
|
|
- },
|
|
|
- .mipLevels = 1,
|
|
|
- .arrayLayers = 1,
|
|
|
- .samples = VK_SAMPLE_COUNT_1_BIT,
|
|
|
- .tiling = VK_IMAGE_TILING_OPTIMAL,
|
|
|
- .usage = VK_IMAGE_USAGE_TRANSFER_SRC_BIT | VK_IMAGE_USAGE_STORAGE_BIT |
|
|
|
- VK_IMAGE_USAGE_SAMPLED_BIT,
|
|
|
- .sharingMode = VK_SHARING_MODE_EXCLUSIVE,
|
|
|
- .queueFamilyIndexCount = 0,
|
|
|
- .pQueueFamilyIndices = nullptr,
|
|
|
- .initialLayout = VK_IMAGE_LAYOUT_UNDEFINED,
|
|
|
- });
|
|
|
- image_views[i] = device.GetLogical().CreateImageView(VkImageViewCreateInfo{
|
|
|
- .sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .image = *images[i],
|
|
|
- .viewType = VK_IMAGE_VIEW_TYPE_2D,
|
|
|
- .format = VK_FORMAT_R16G16B16A16_SFLOAT,
|
|
|
- .components =
|
|
|
- {
|
|
|
- .r = VK_COMPONENT_SWIZZLE_IDENTITY,
|
|
|
- .g = VK_COMPONENT_SWIZZLE_IDENTITY,
|
|
|
- .b = VK_COMPONENT_SWIZZLE_IDENTITY,
|
|
|
- .a = VK_COMPONENT_SWIZZLE_IDENTITY,
|
|
|
- },
|
|
|
- .subresourceRange =
|
|
|
- {
|
|
|
- .aspectMask = VK_IMAGE_ASPECT_COLOR_BIT,
|
|
|
- .baseMipLevel = 0,
|
|
|
- .levelCount = 1,
|
|
|
- .baseArrayLayer = 0,
|
|
|
- .layerCount = 1,
|
|
|
- },
|
|
|
- });
|
|
|
+ for (auto& images : m_dynamic_images) {
|
|
|
+ images.descriptor_sets = CreateWrappedDescriptorSets(m_descriptor_pool, layouts);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-void FSR::CreatePipelineLayout() {
|
|
|
- VkPushConstantRange push_const{
|
|
|
- .stageFlags = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
+void FSR::CreatePipelineLayouts() {
|
|
|
+ const VkPushConstantRange range{
|
|
|
+ .stageFlags = VK_SHADER_STAGE_FRAGMENT_BIT,
|
|
|
.offset = 0,
|
|
|
- .size = sizeof(std::array<u32, 4 * 4>),
|
|
|
+ .size = sizeof(PushConstants),
|
|
|
};
|
|
|
VkPipelineLayoutCreateInfo ci{
|
|
|
.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO,
|
|
|
.pNext = nullptr,
|
|
|
.flags = 0,
|
|
|
.setLayoutCount = 1,
|
|
|
- .pSetLayouts = descriptor_set_layout.address(),
|
|
|
+ .pSetLayouts = m_descriptor_set_layout.address(),
|
|
|
.pushConstantRangeCount = 1,
|
|
|
- .pPushConstantRanges = &push_const,
|
|
|
+ .pPushConstantRanges = &range,
|
|
|
};
|
|
|
|
|
|
- pipeline_layout = device.GetLogical().CreatePipelineLayout(ci);
|
|
|
+ m_pipeline_layout = m_device.GetLogical().CreatePipelineLayout(ci);
|
|
|
}
|
|
|
|
|
|
-void FSR::UpdateDescriptorSet(std::size_t image_index, VkImageView image_view) const {
|
|
|
- const auto fsr_image_view = *image_views[image_index];
|
|
|
- const auto blit_image_view = *image_views[image_count + image_index];
|
|
|
-
|
|
|
- const VkDescriptorImageInfo image_info{
|
|
|
- .sampler = VK_NULL_HANDLE,
|
|
|
- .imageView = image_view,
|
|
|
- .imageLayout = VK_IMAGE_LAYOUT_GENERAL,
|
|
|
- };
|
|
|
- const VkDescriptorImageInfo fsr_image_info{
|
|
|
- .sampler = VK_NULL_HANDLE,
|
|
|
- .imageView = fsr_image_view,
|
|
|
- .imageLayout = VK_IMAGE_LAYOUT_GENERAL,
|
|
|
- };
|
|
|
- const VkDescriptorImageInfo blit_image_info{
|
|
|
- .sampler = VK_NULL_HANDLE,
|
|
|
- .imageView = blit_image_view,
|
|
|
- .imageLayout = VK_IMAGE_LAYOUT_GENERAL,
|
|
|
- };
|
|
|
-
|
|
|
- VkWriteDescriptorSet sampler_write{
|
|
|
- .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
|
|
|
- .pNext = nullptr,
|
|
|
- .dstSet = descriptor_sets[image_index * 2],
|
|
|
- .dstBinding = 0,
|
|
|
- .dstArrayElement = 0,
|
|
|
- .descriptorCount = 1,
|
|
|
- .descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
|
|
|
- .pImageInfo = &image_info,
|
|
|
- .pBufferInfo = nullptr,
|
|
|
- .pTexelBufferView = nullptr,
|
|
|
- };
|
|
|
-
|
|
|
- VkWriteDescriptorSet output_write{
|
|
|
- .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
|
|
|
- .pNext = nullptr,
|
|
|
- .dstSet = descriptor_sets[image_index * 2],
|
|
|
- .dstBinding = 1,
|
|
|
- .dstArrayElement = 0,
|
|
|
- .descriptorCount = 1,
|
|
|
- .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
|
|
|
- .pImageInfo = &fsr_image_info,
|
|
|
- .pBufferInfo = nullptr,
|
|
|
- .pTexelBufferView = nullptr,
|
|
|
- };
|
|
|
-
|
|
|
- device.GetLogical().UpdateDescriptorSets(std::array{sampler_write, output_write}, {});
|
|
|
-
|
|
|
- sampler_write.dstSet = descriptor_sets[image_index * 2 + 1];
|
|
|
- sampler_write.pImageInfo = &fsr_image_info;
|
|
|
- output_write.dstSet = descriptor_sets[image_index * 2 + 1];
|
|
|
- output_write.pImageInfo = &blit_image_info;
|
|
|
-
|
|
|
- device.GetLogical().UpdateDescriptorSets(std::array{sampler_write, output_write}, {});
|
|
|
+void FSR::CreatePipelines() {
|
|
|
+ m_easu_pipeline = CreateWrappedPipeline(m_device, m_renderpass, m_pipeline_layout,
|
|
|
+ std::tie(m_vert_shader, m_easu_shader));
|
|
|
+ m_rcas_pipeline = CreateWrappedPipeline(m_device, m_renderpass, m_pipeline_layout,
|
|
|
+ std::tie(m_vert_shader, m_rcas_shader));
|
|
|
}
|
|
|
|
|
|
-void FSR::CreateSampler() {
|
|
|
- const VkSamplerCreateInfo ci{
|
|
|
- .sType = VK_STRUCTURE_TYPE_SAMPLER_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .magFilter = VK_FILTER_LINEAR,
|
|
|
- .minFilter = VK_FILTER_LINEAR,
|
|
|
- .mipmapMode = VK_SAMPLER_MIPMAP_MODE_LINEAR,
|
|
|
- .addressModeU = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_EDGE,
|
|
|
- .addressModeV = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_EDGE,
|
|
|
- .addressModeW = VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_EDGE,
|
|
|
- .mipLodBias = 0.0f,
|
|
|
- .anisotropyEnable = VK_FALSE,
|
|
|
- .maxAnisotropy = 0.0f,
|
|
|
- .compareEnable = VK_FALSE,
|
|
|
- .compareOp = VK_COMPARE_OP_NEVER,
|
|
|
- .minLod = 0.0f,
|
|
|
- .maxLod = 0.0f,
|
|
|
- .borderColor = VK_BORDER_COLOR_FLOAT_OPAQUE_BLACK,
|
|
|
- .unnormalizedCoordinates = VK_FALSE,
|
|
|
- };
|
|
|
+void FSR::UpdateDescriptorSets(VkImageView image_view, size_t image_index) {
|
|
|
+ Images& images = m_dynamic_images[image_index];
|
|
|
+ std::vector<VkDescriptorImageInfo> image_infos;
|
|
|
+ std::vector<VkWriteDescriptorSet> updates;
|
|
|
+ image_infos.reserve(2);
|
|
|
|
|
|
- sampler = device.GetLogical().CreateSampler(ci);
|
|
|
-}
|
|
|
+ updates.push_back(CreateWriteDescriptorSet(image_infos, *m_sampler, image_view,
|
|
|
+ images.descriptor_sets[Easu], 0));
|
|
|
+ updates.push_back(CreateWriteDescriptorSet(image_infos, *m_sampler, *images.image_views[Easu],
|
|
|
+ images.descriptor_sets[Rcas], 0));
|
|
|
|
|
|
-void FSR::CreateShaders() {
|
|
|
- if (device.IsFloat16Supported()) {
|
|
|
- easu_shader = BuildShader(device, VULKAN_FIDELITYFX_FSR_EASU_FP16_COMP_SPV);
|
|
|
- rcas_shader = BuildShader(device, VULKAN_FIDELITYFX_FSR_RCAS_FP16_COMP_SPV);
|
|
|
- } else {
|
|
|
- easu_shader = BuildShader(device, VULKAN_FIDELITYFX_FSR_EASU_FP32_COMP_SPV);
|
|
|
- rcas_shader = BuildShader(device, VULKAN_FIDELITYFX_FSR_RCAS_FP32_COMP_SPV);
|
|
|
- }
|
|
|
+ m_device.GetLogical().UpdateDescriptorSets(updates, {});
|
|
|
}
|
|
|
|
|
|
-void FSR::CreatePipeline() {
|
|
|
- VkPipelineShaderStageCreateInfo shader_stage_easu{
|
|
|
- .sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .stage = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
- .module = *easu_shader,
|
|
|
- .pName = "main",
|
|
|
- .pSpecializationInfo = nullptr,
|
|
|
- };
|
|
|
+void FSR::UploadImages(Scheduler& scheduler) {
|
|
|
+ if (m_images_ready) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
|
|
|
- VkPipelineShaderStageCreateInfo shader_stage_rcas{
|
|
|
- .sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .stage = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
- .module = *rcas_shader,
|
|
|
- .pName = "main",
|
|
|
- .pSpecializationInfo = nullptr,
|
|
|
- };
|
|
|
+ scheduler.Record([&](vk::CommandBuffer cmdbuf) {
|
|
|
+ for (auto& image : m_dynamic_images) {
|
|
|
+ ClearColorImage(cmdbuf, *image.images[Easu]);
|
|
|
+ ClearColorImage(cmdbuf, *image.images[Rcas]);
|
|
|
+ }
|
|
|
+ });
|
|
|
+ scheduler.Finish();
|
|
|
|
|
|
- VkComputePipelineCreateInfo pipeline_ci_easu{
|
|
|
- .sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .stage = shader_stage_easu,
|
|
|
- .layout = *pipeline_layout,
|
|
|
- .basePipelineHandle = VK_NULL_HANDLE,
|
|
|
- .basePipelineIndex = 0,
|
|
|
- };
|
|
|
+ m_images_ready = true;
|
|
|
+}
|
|
|
|
|
|
- VkComputePipelineCreateInfo pipeline_ci_rcas{
|
|
|
- .sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO,
|
|
|
- .pNext = nullptr,
|
|
|
- .flags = 0,
|
|
|
- .stage = shader_stage_rcas,
|
|
|
- .layout = *pipeline_layout,
|
|
|
- .basePipelineHandle = VK_NULL_HANDLE,
|
|
|
- .basePipelineIndex = 0,
|
|
|
- };
|
|
|
+VkImageView FSR::Draw(Scheduler& scheduler, size_t image_index, VkImage source_image,
|
|
|
+ VkImageView source_image_view, VkExtent2D input_image_extent,
|
|
|
+ const Common::Rectangle<f32>& crop_rect) {
|
|
|
+ Images& images = m_dynamic_images[image_index];
|
|
|
+
|
|
|
+ VkImage easu_image = *images.images[Easu];
|
|
|
+ VkImage rcas_image = *images.images[Rcas];
|
|
|
+ VkDescriptorSet easu_descriptor_set = images.descriptor_sets[Easu];
|
|
|
+ VkDescriptorSet rcas_descriptor_set = images.descriptor_sets[Rcas];
|
|
|
+ VkFramebuffer easu_framebuffer = *images.framebuffers[Easu];
|
|
|
+ VkFramebuffer rcas_framebuffer = *images.framebuffers[Rcas];
|
|
|
+ VkPipeline easu_pipeline = *m_easu_pipeline;
|
|
|
+ VkPipeline rcas_pipeline = *m_rcas_pipeline;
|
|
|
+ VkPipelineLayout pipeline_layout = *m_pipeline_layout;
|
|
|
+ VkRenderPass renderpass = *m_renderpass;
|
|
|
+ VkExtent2D extent = m_extent;
|
|
|
+
|
|
|
+ const f32 input_image_width = static_cast<f32>(input_image_extent.width);
|
|
|
+ const f32 input_image_height = static_cast<f32>(input_image_extent.height);
|
|
|
+ const f32 output_image_width = static_cast<f32>(extent.width);
|
|
|
+ const f32 output_image_height = static_cast<f32>(extent.height);
|
|
|
+ const f32 viewport_width = (crop_rect.right - crop_rect.left) * input_image_width;
|
|
|
+ const f32 viewport_x = crop_rect.left * input_image_width;
|
|
|
+ const f32 viewport_height = (crop_rect.bottom - crop_rect.top) * input_image_height;
|
|
|
+ const f32 viewport_y = crop_rect.top * input_image_height;
|
|
|
+
|
|
|
+ PushConstants easu_con{};
|
|
|
+ PushConstants rcas_con{};
|
|
|
+ FsrEasuConOffset(easu_con.data() + 0, easu_con.data() + 4, easu_con.data() + 8,
|
|
|
+ easu_con.data() + 12, viewport_width, viewport_height, input_image_width,
|
|
|
+ input_image_height, output_image_width, output_image_height, viewport_x,
|
|
|
+ viewport_y);
|
|
|
+
|
|
|
+ const float sharpening =
|
|
|
+ static_cast<float>(Settings::values.fsr_sharpening_slider.GetValue()) / 100.0f;
|
|
|
+ FsrRcasCon(rcas_con.data(), sharpening);
|
|
|
+
|
|
|
+ UploadImages(scheduler);
|
|
|
+ UpdateDescriptorSets(source_image_view, image_index);
|
|
|
+
|
|
|
+ scheduler.RequestOutsideRenderPassOperationContext();
|
|
|
+ scheduler.Record([=](vk::CommandBuffer cmdbuf) {
|
|
|
+ TransitionImageLayout(cmdbuf, source_image, VK_IMAGE_LAYOUT_GENERAL);
|
|
|
+ TransitionImageLayout(cmdbuf, easu_image, VK_IMAGE_LAYOUT_GENERAL);
|
|
|
+ BeginRenderPass(cmdbuf, renderpass, easu_framebuffer, extent);
|
|
|
+ cmdbuf.BindPipeline(VK_PIPELINE_BIND_POINT_GRAPHICS, easu_pipeline);
|
|
|
+ cmdbuf.BindDescriptorSets(VK_PIPELINE_BIND_POINT_GRAPHICS, pipeline_layout, 0,
|
|
|
+ easu_descriptor_set, {});
|
|
|
+ cmdbuf.PushConstants(pipeline_layout, VK_SHADER_STAGE_FRAGMENT_BIT, easu_con);
|
|
|
+ cmdbuf.Draw(3, 1, 0, 0);
|
|
|
+ cmdbuf.EndRenderPass();
|
|
|
+
|
|
|
+ TransitionImageLayout(cmdbuf, easu_image, VK_IMAGE_LAYOUT_GENERAL);
|
|
|
+ TransitionImageLayout(cmdbuf, rcas_image, VK_IMAGE_LAYOUT_GENERAL);
|
|
|
+ BeginRenderPass(cmdbuf, renderpass, rcas_framebuffer, extent);
|
|
|
+ cmdbuf.BindPipeline(VK_PIPELINE_BIND_POINT_GRAPHICS, rcas_pipeline);
|
|
|
+ cmdbuf.BindDescriptorSets(VK_PIPELINE_BIND_POINT_GRAPHICS, pipeline_layout, 0,
|
|
|
+ rcas_descriptor_set, {});
|
|
|
+ cmdbuf.PushConstants(pipeline_layout, VK_SHADER_STAGE_FRAGMENT_BIT, rcas_con);
|
|
|
+ cmdbuf.Draw(3, 1, 0, 0);
|
|
|
+ cmdbuf.EndRenderPass();
|
|
|
+
|
|
|
+ TransitionImageLayout(cmdbuf, rcas_image, VK_IMAGE_LAYOUT_GENERAL);
|
|
|
+ });
|
|
|
|
|
|
- easu_pipeline = device.GetLogical().CreateComputePipeline(pipeline_ci_easu);
|
|
|
- rcas_pipeline = device.GetLogical().CreateComputePipeline(pipeline_ci_rcas);
|
|
|
+ return *images.image_views[Rcas];
|
|
|
}
|
|
|
|
|
|
} // namespace Vulkan
|