OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "gpu/vulkan/vulkan_command_buffer.h" |
| 6 |
| 7 #include "gpu/vulkan/vulkan_command_pool.h" |
| 8 #include "gpu/vulkan/vulkan_implementation.h" |
| 9 |
| 10 namespace gpu { |
| 11 |
| 12 VulkanCommandBuffer::VulkanCommandBuffer(VulkanCommandPool* command_pool, |
| 13 bool primary) |
| 14 : primary_(primary), command_pool_(command_pool) { |
| 15 command_pool_->IncrementCommandBufferCount(); |
| 16 } |
| 17 |
| 18 VulkanCommandBuffer::~VulkanCommandBuffer() { |
| 19 DCHECK_EQ(static_cast<VkCommandBuffer>(VK_NULL_HANDLE), command_buffer_); |
| 20 DCHECK(!recording_); |
| 21 command_pool_->DecrementCommandBufferCount(); |
| 22 } |
| 23 |
| 24 bool VulkanCommandBuffer::Initialize() { |
| 25 VkCommandBufferAllocateInfo command_buffer_info = {}; |
| 26 command_buffer_info.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO; |
| 27 command_buffer_info.commandPool = command_pool_->handle(); |
| 28 command_buffer_info.level = primary_ ? VK_COMMAND_BUFFER_LEVEL_PRIMARY |
| 29 : VK_COMMAND_BUFFER_LEVEL_SECONDARY; |
| 30 command_buffer_info.commandBufferCount = 1; |
| 31 |
| 32 VkResult result = vkAllocateCommandBuffers( |
| 33 GetVulkanDevice(), &command_buffer_info, &command_buffer_); |
| 34 if (VK_SUCCESS != result) { |
| 35 DLOG(ERROR) << "vkAllocateCommandBuffers() failed: " << result; |
| 36 return false; |
| 37 } |
| 38 return true; |
| 39 } |
| 40 |
| 41 void VulkanCommandBuffer::Destroy() { |
| 42 if (VK_NULL_HANDLE != command_buffer_) { |
| 43 vkFreeCommandBuffers(GetVulkanDevice(), command_pool_->handle(), 1, |
| 44 &command_buffer_); |
| 45 command_buffer_ = VK_NULL_HANDLE; |
| 46 } |
| 47 } |
| 48 |
| 49 bool VulkanCommandBuffer::Submit(VkQueue queue, |
| 50 uint32_t num_wait_semaphores, |
| 51 VkSemaphore* wait_semaphores, |
| 52 uint32_t num_signal_semaphores, |
| 53 VkSemaphore* signal_semaphores) { |
| 54 DCHECK(primary_); |
| 55 VkSubmitInfo submit_info = {}; |
| 56 submit_info.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO; |
| 57 submit_info.commandBufferCount = 1; |
| 58 submit_info.pCommandBuffers = &command_buffer_; |
| 59 |
| 60 if (num_wait_semaphores) { |
| 61 submit_info.waitSemaphoreCount = num_wait_semaphores; |
| 62 submit_info.pWaitSemaphores = wait_semaphores; |
| 63 } |
| 64 |
| 65 if (num_signal_semaphores) { |
| 66 submit_info.signalSemaphoreCount = num_signal_semaphores; |
| 67 submit_info.pSignalSemaphores = signal_semaphores; |
| 68 } |
| 69 |
| 70 VkResult result = vkQueueSubmit(queue, 1, &submit_info, VK_NULL_HANDLE); |
| 71 PostExecution(); |
| 72 if (VK_SUCCESS != result) { |
| 73 DLOG(ERROR) << "vkQueueSubmit() failed: " << result; |
| 74 return false; |
| 75 } |
| 76 |
| 77 return true; |
| 78 } |
| 79 |
| 80 void VulkanCommandBuffer::Enqueue(VkCommandBuffer primary_command_buffer) { |
| 81 DCHECK(!primary_); |
| 82 vkCmdExecuteCommands(primary_command_buffer, 1, &command_buffer_); |
| 83 PostExecution(); |
| 84 } |
| 85 |
| 86 void VulkanCommandBuffer::PostExecution() { |
| 87 if (record_type_ == RECORD_TYPE_SINGLE_USE) { |
| 88 // Clear upon next use. |
| 89 record_type_ = RECORD_TYPE_DIRTY; |
| 90 } else if (record_type_ == RECORD_TYPE_MULTI_USE) { |
| 91 // Can no longer record new items unless marked as clear. |
| 92 record_type_ = RECORD_TYPE_RECORDED; |
| 93 } |
| 94 } |
| 95 |
| 96 void VulkanCommandBuffer::ResetIfDirty() { |
| 97 DCHECK(!recording_); |
| 98 if (record_type_ == RECORD_TYPE_DIRTY) { |
| 99 vkResetCommandPool(GetVulkanDevice(), command_pool_->handle(), |
| 100 VK_COMMAND_POOL_RESET_RELEASE_RESOURCES_BIT); |
| 101 record_type_ = RECORD_TYPE_EMPTY; |
| 102 } |
| 103 } |
| 104 |
| 105 CommandBufferRecorderBase::~CommandBufferRecorderBase() { |
| 106 vkEndCommandBuffer(handle_); |
| 107 }; |
| 108 |
| 109 } // namespace gpu |
OLD | NEW |