| // Copyright (c) 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "gpu/vulkan/vulkan_swap_chain.h" |
| |
| #include "gpu/vulkan/vulkan_command_buffer.h" |
| #include "gpu/vulkan/vulkan_command_pool.h" |
| #include "gpu/vulkan/vulkan_device_queue.h" |
| #include "gpu/vulkan/vulkan_function_pointers.h" |
| |
| namespace gpu { |
| |
| namespace { |
| |
| VkPipelineStageFlags GetPipelineStageFlags(const VkImageLayout layout) { |
| switch (layout) { |
| case VK_IMAGE_LAYOUT_UNDEFINED: |
| return VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT; |
| case VK_IMAGE_LAYOUT_GENERAL: |
| return VK_PIPELINE_STAGE_ALL_COMMANDS_BIT; |
| case VK_IMAGE_LAYOUT_PREINITIALIZED: |
| return VK_PIPELINE_STAGE_HOST_BIT; |
| case VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL: |
| case VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL: |
| return VK_PIPELINE_STAGE_TRANSFER_BIT; |
| case VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL: |
| return VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT; |
| case VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL: |
| return VK_PIPELINE_STAGE_TESSELLATION_CONTROL_SHADER_BIT | |
| VK_PIPELINE_STAGE_TESSELLATION_EVALUATION_SHADER_BIT | |
| VK_PIPELINE_STAGE_GEOMETRY_SHADER_BIT | |
| VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT; |
| case VK_IMAGE_LAYOUT_PRESENT_SRC_KHR: |
| return VK_PIPELINE_STAGE_BOTTOM_OF_PIPE_BIT; |
| default: |
| NOTREACHED() << "layout=" << layout; |
| } |
| return 0; |
| } |
| |
| VkAccessFlags GetAccessMask(const VkImageLayout layout) { |
| switch (layout) { |
| case VK_IMAGE_LAYOUT_UNDEFINED: |
| return 0; |
| case VK_IMAGE_LAYOUT_GENERAL: |
| DLOG(WARNING) << "VK_IMAGE_LAYOUT_GENERAL is used."; |
| return VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT | |
| VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT | |
| VK_ACCESS_TRANSFER_WRITE_BIT | VK_ACCESS_TRANSFER_READ_BIT | |
| VK_ACCESS_SHADER_READ_BIT | VK_ACCESS_HOST_WRITE_BIT | |
| VK_ACCESS_HOST_READ_BIT; |
| case VK_IMAGE_LAYOUT_PREINITIALIZED: |
| return VK_ACCESS_HOST_WRITE_BIT; |
| case VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL: |
| return VK_ACCESS_COLOR_ATTACHMENT_READ_BIT | |
| VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT; |
| case VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL: |
| return VK_ACCESS_SHADER_READ_BIT | VK_ACCESS_INPUT_ATTACHMENT_READ_BIT; |
| case VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL: |
| return VK_ACCESS_TRANSFER_READ_BIT; |
| case VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL: |
| return VK_ACCESS_TRANSFER_WRITE_BIT; |
| case VK_IMAGE_LAYOUT_PRESENT_SRC_KHR: |
| return 0; |
| default: |
| NOTREACHED() << "layout=" << layout; |
| } |
| return 0; |
| } |
| |
| void CmdSetImageLayout(VulkanCommandBuffer* command_buffer, |
| VkImage image, |
| VkImageLayout layout, |
| VkImageLayout old_layout) { |
| DCHECK_NE(layout, old_layout); |
| VkImageMemoryBarrier image_memory_barrier = { |
| .sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER, |
| .pNext = nullptr, |
| .srcAccessMask = GetAccessMask(old_layout), |
| .dstAccessMask = GetAccessMask(layout), |
| .oldLayout = old_layout, |
| .newLayout = layout, |
| .srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED, |
| .dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED, |
| .image = image, |
| .subresourceRange = |
| { |
| .aspectMask = VK_IMAGE_ASPECT_COLOR_BIT, |
| .baseMipLevel = 0, |
| .levelCount = 1, |
| .baseArrayLayer = 0, |
| .layerCount = 1, |
| }, |
| }; |
| |
| ScopedSingleUseCommandBufferRecorder recorder(*command_buffer); |
| vkCmdPipelineBarrier(recorder.handle(), GetPipelineStageFlags(old_layout), |
| GetPipelineStageFlags(layout), 0, 0, nullptr, 0, nullptr, |
| 1, &image_memory_barrier); |
| } |
| |
| } // namespace |
| |
| VulkanSwapChain::VulkanSwapChain() {} |
| |
| VulkanSwapChain::~VulkanSwapChain() { |
| DCHECK(images_.empty()); |
| DCHECK_EQ(static_cast<VkSwapchainKHR>(VK_NULL_HANDLE), swap_chain_); |
| DCHECK_EQ(static_cast<VkSemaphore>(VK_NULL_HANDLE), next_present_semaphore_); |
| } |
| |
| bool VulkanSwapChain::Initialize( |
| VulkanDeviceQueue* device_queue, |
| VkSurfaceKHR surface, |
| const VkSurfaceCapabilitiesKHR& surface_caps, |
| const VkSurfaceFormatKHR& surface_format, |
| std::unique_ptr<VulkanSwapChain> old_swap_chain) { |
| DCHECK(device_queue); |
| device_queue_ = device_queue; |
| return InitializeSwapChain(surface, surface_caps, surface_format, |
| std::move(old_swap_chain)) && |
| InitializeSwapImages(surface_caps, surface_format); |
| } |
| |
| void VulkanSwapChain::Destroy() { |
| DestroySwapImages(); |
| DestroySwapChain(); |
| } |
| |
| gfx::SwapResult VulkanSwapChain::SwapBuffers() { |
| VkResult result = VK_SUCCESS; |
| |
| VkDevice device = device_queue_->GetVulkanDevice(); |
| VkQueue queue = device_queue_->GetVulkanQueue(); |
| |
| auto& current_image_data = images_[current_image_]; |
| |
| current_image_data->post_raster_command_buffer->Clear(); |
| CmdSetImageLayout(current_image_data->post_raster_command_buffer.get(), |
| current_image_data->image, |
| VK_IMAGE_LAYOUT_PRESENT_SRC_KHR /* layout */, |
| current_image_data->layout /* old_layout */); |
| current_image_data->layout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR; |
| |
| // Submit our post_raster_command_buffer for the current buffer. It sets the |
| // image layout for presenting. |
| if (!current_image_data->post_raster_command_buffer->Submit( |
| 0, nullptr, 1, ¤t_image_data->render_semaphore)) { |
| return gfx::SwapResult::SWAP_FAILED; |
| } |
| |
| // Queue the present. |
| VkPresentInfoKHR present_info = {}; |
| present_info.sType = VK_STRUCTURE_TYPE_PRESENT_INFO_KHR; |
| present_info.waitSemaphoreCount = 1; |
| present_info.pWaitSemaphores = ¤t_image_data->render_semaphore; |
| present_info.swapchainCount = 1; |
| present_info.pSwapchains = &swap_chain_; |
| present_info.pImageIndices = ¤t_image_; |
| |
| result = vkQueuePresentKHR(queue, &present_info); |
| if (VK_SUCCESS != result) { |
| return gfx::SwapResult::SWAP_FAILED; |
| } |
| |
| uint32_t next_image = 0; |
| // Acquire then next image. |
| result = vkAcquireNextImageKHR(device, swap_chain_, UINT64_MAX, |
| next_present_semaphore_, VK_NULL_HANDLE, |
| &next_image); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkAcquireNextImageKHR() failed: " << result; |
| return gfx::SwapResult::SWAP_FAILED; |
| } |
| |
| auto& next_image_data = images_[next_image]; |
| // Swap in the "next_present_semaphore" into the newly acquired image. The |
| // old "present_semaphore" for the image becomes the place holder for the next |
| // present semaphore for the next image. |
| std::swap(next_image_data->present_semaphore, next_present_semaphore_); |
| |
| // Submit our pre_raster_command_buffer for the next buffer. It sets the image |
| // layout for rastering. |
| next_image_data->pre_raster_command_buffer->Clear(); |
| CmdSetImageLayout(next_image_data->pre_raster_command_buffer.get(), |
| next_image_data->image, |
| VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL /* layout */, |
| next_image_data->layout /* old_layout */); |
| next_image_data->layout = VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL; |
| |
| if (!next_image_data->pre_raster_command_buffer->Submit( |
| 1, &next_image_data->present_semaphore, 0, nullptr)) { |
| return gfx::SwapResult::SWAP_FAILED; |
| } |
| |
| current_image_ = next_image; |
| return gfx::SwapResult::SWAP_ACK; |
| } |
| |
| bool VulkanSwapChain::InitializeSwapChain( |
| VkSurfaceKHR surface, |
| const VkSurfaceCapabilitiesKHR& surface_caps, |
| const VkSurfaceFormatKHR& surface_format, |
| std::unique_ptr<VulkanSwapChain> old_swap_chain) { |
| VkDevice device = device_queue_->GetVulkanDevice(); |
| VkResult result = VK_SUCCESS; |
| |
| VkSwapchainCreateInfoKHR swap_chain_create_info = {}; |
| swap_chain_create_info.sType = VK_STRUCTURE_TYPE_SWAPCHAIN_CREATE_INFO_KHR; |
| swap_chain_create_info.surface = surface; |
| swap_chain_create_info.minImageCount = |
| std::max(3u, surface_caps.minImageCount); |
| swap_chain_create_info.imageFormat = surface_format.format; |
| swap_chain_create_info.imageColorSpace = surface_format.colorSpace; |
| swap_chain_create_info.imageExtent = surface_caps.currentExtent; |
| swap_chain_create_info.imageArrayLayers = 1; |
| swap_chain_create_info.imageUsage = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT; |
| swap_chain_create_info.imageSharingMode = VK_SHARING_MODE_EXCLUSIVE; |
| swap_chain_create_info.preTransform = surface_caps.currentTransform; |
| swap_chain_create_info.compositeAlpha = VK_COMPOSITE_ALPHA_OPAQUE_BIT_KHR; |
| swap_chain_create_info.presentMode = VK_PRESENT_MODE_FIFO_KHR; |
| swap_chain_create_info.clipped = true; |
| swap_chain_create_info.oldSwapchain = |
| old_swap_chain ? old_swap_chain->swap_chain_ : VK_NULL_HANDLE; |
| |
| VkSwapchainKHR new_swap_chain = VK_NULL_HANDLE; |
| result = vkCreateSwapchainKHR(device, &swap_chain_create_info, nullptr, |
| &new_swap_chain); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkCreateSwapchainKHR() failed: " << result; |
| return false; |
| } |
| |
| if (old_swap_chain) { |
| result = vkQueueWaitIdle(device_queue_->GetVulkanQueue()); |
| DLOG_IF(ERROR, VK_SUCCESS != result) |
| << "vkQueueWaitIdle failed: " << result; |
| old_swap_chain->Destroy(); |
| old_swap_chain = nullptr; |
| } |
| |
| swap_chain_ = new_swap_chain; |
| size_ = gfx::Size(swap_chain_create_info.imageExtent.width, |
| swap_chain_create_info.imageExtent.height); |
| |
| return true; |
| } |
| |
| void VulkanSwapChain::DestroySwapChain() { |
| VkDevice device = device_queue_->GetVulkanDevice(); |
| |
| if (swap_chain_ != VK_NULL_HANDLE) { |
| vkDestroySwapchainKHR(device, swap_chain_, nullptr); |
| swap_chain_ = VK_NULL_HANDLE; |
| } |
| } |
| |
| bool VulkanSwapChain::InitializeSwapImages( |
| const VkSurfaceCapabilitiesKHR& surface_caps, |
| const VkSurfaceFormatKHR& surface_format) { |
| VkDevice device = device_queue_->GetVulkanDevice(); |
| VkResult result = VK_SUCCESS; |
| |
| uint32_t image_count = 0; |
| result = vkGetSwapchainImagesKHR(device, swap_chain_, &image_count, nullptr); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkGetSwapchainImagesKHR(NULL) failed: " << result; |
| return false; |
| } |
| |
| std::vector<VkImage> images(image_count); |
| result = |
| vkGetSwapchainImagesKHR(device, swap_chain_, &image_count, images.data()); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkGetSwapchainImagesKHR(images) failed: " << result; |
| return false; |
| } |
| |
| // Generic semaphore creation structure. |
| VkSemaphoreCreateInfo semaphore_create_info = {}; |
| semaphore_create_info.sType = VK_STRUCTURE_TYPE_SEMAPHORE_CREATE_INFO; |
| |
| command_pool_ = device_queue_->CreateCommandPool(); |
| if (!command_pool_) |
| return false; |
| |
| images_.resize(image_count); |
| for (uint32_t i = 0; i < image_count; ++i) { |
| images_[i].reset(new ImageData); |
| std::unique_ptr<ImageData>& image_data = images_[i]; |
| image_data->image = images[i]; |
| |
| // Setup semaphores. |
| result = vkCreateSemaphore(device, &semaphore_create_info, nullptr, |
| &image_data->render_semaphore); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkCreateSemaphore(render) failed: " << result; |
| return false; |
| } |
| |
| result = vkCreateSemaphore(device, &semaphore_create_info, nullptr, |
| &image_data->present_semaphore); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkCreateSemaphore(present) failed: " << result; |
| return false; |
| } |
| |
| // Initialize the command buffer for this buffer data. |
| image_data->pre_raster_command_buffer = |
| command_pool_->CreatePrimaryCommandBuffer(); |
| image_data->post_raster_command_buffer = |
| command_pool_->CreatePrimaryCommandBuffer(); |
| } |
| |
| result = vkCreateSemaphore(device, &semaphore_create_info, nullptr, |
| &next_present_semaphore_); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkCreateSemaphore(next_present) failed: " << result; |
| return false; |
| } |
| |
| // Acquire the initial buffer. |
| result = vkAcquireNextImageKHR(device, swap_chain_, UINT64_MAX, |
| next_present_semaphore_, VK_NULL_HANDLE, |
| ¤t_image_); |
| if (VK_SUCCESS != result) { |
| DLOG(ERROR) << "vkAcquireNextImageKHR() failed: " << result; |
| return false; |
| } |
| |
| std::swap(images_[current_image_]->present_semaphore, |
| next_present_semaphore_); |
| |
| return true; |
| } |
| |
| void VulkanSwapChain::DestroySwapImages() { |
| VkDevice device = device_queue_->GetVulkanDevice(); |
| |
| if (VK_NULL_HANDLE != next_present_semaphore_) { |
| vkDestroySemaphore(device, next_present_semaphore_, nullptr); |
| next_present_semaphore_ = VK_NULL_HANDLE; |
| } |
| |
| for (const std::unique_ptr<ImageData>& image_data : images_) { |
| if (image_data->post_raster_command_buffer) { |
| // Make sure command buffer is done processing. |
| image_data->pre_raster_command_buffer->Wait(UINT64_MAX); |
| image_data->pre_raster_command_buffer->Destroy(); |
| image_data->pre_raster_command_buffer.reset(); |
| |
| // Make sure command buffer is done processing. |
| image_data->post_raster_command_buffer->Wait(UINT64_MAX); |
| image_data->post_raster_command_buffer->Destroy(); |
| image_data->post_raster_command_buffer.reset(); |
| } |
| |
| // Destroy Semaphores. |
| if (VK_NULL_HANDLE != image_data->present_semaphore) { |
| vkDestroySemaphore(device, image_data->present_semaphore, nullptr); |
| image_data->present_semaphore = VK_NULL_HANDLE; |
| } |
| |
| if (VK_NULL_HANDLE != image_data->render_semaphore) { |
| vkDestroySemaphore(device, image_data->render_semaphore, nullptr); |
| image_data->render_semaphore = VK_NULL_HANDLE; |
| } |
| |
| image_data->image = VK_NULL_HANDLE; |
| } |
| images_.clear(); |
| |
| if (command_pool_) { |
| command_pool_->Destroy(); |
| command_pool_.reset(); |
| } |
| } |
| |
| VulkanSwapChain::ImageData::ImageData() {} |
| |
| VulkanSwapChain::ImageData::~ImageData() {} |
| |
| } // namespace gpu |