Chromium Code Reviews| 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_descriptor_set.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 #include "gpu/vulkan/vulkan_descriptor_pool.h" | |
| 9 #include "gpu/vulkan/vulkan_device_queue.h" | |
| 10 | |
| 11 namespace gpu { | |
| 12 | |
| 13 VulkanDescriptorSet::~VulkanDescriptorSet() { | |
| 14 DCHECK_EQ(static_cast<VkDescriptorSet>(VK_NULL_HANDLE), handle_); | |
| 15 DCHECK_EQ(static_cast<VkDescriptorSetLayout>(VK_NULL_HANDLE), layout_); | |
| 16 descriptor_pool_->DecrementDescriptorSetCount(); | |
| 17 } | |
| 18 | |
| 19 bool VulkanDescriptorSet::Initialize( | |
| 20 const std::vector<VkDescriptorSetLayoutBinding>& layout) { | |
| 21 VkResult result = VK_SUCCESS; | |
| 22 VkDevice device = device_queue_->GetVulkanDevice(); | |
| 23 | |
| 24 VkDescriptorSetLayoutCreateInfo layout_create_info = {}; | |
| 25 layout_create_info.sType = | |
| 26 VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO; | |
| 27 layout_create_info.bindingCount = static_cast<uint32_t>(layout.size()); | |
| 28 layout_create_info.pBindings = layout.data(); | |
| 29 | |
| 30 result = vkCreateDescriptorSetLayout(device, &layout_create_info, nullptr, | |
| 31 &layout_); | |
|
piman
2016/05/18 19:02:18
I think we'll want to separate out the DescriptorS
David Yen
2016/05/18 21:08:35
Done.
| |
| 32 if (VK_SUCCESS != result) { | |
| 33 DLOG(ERROR) << "vkCreateDescriptorSetLayout() failed: " << result; | |
| 34 return false; | |
| 35 } | |
| 36 | |
| 37 VkDescriptorSetAllocateInfo set_allocate_info = {}; | |
| 38 set_allocate_info.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO; | |
| 39 set_allocate_info.descriptorPool = descriptor_pool_->handle(); | |
| 40 set_allocate_info.descriptorSetCount = 1; | |
| 41 set_allocate_info.pSetLayouts = &layout_; | |
| 42 | |
| 43 result = vkAllocateDescriptorSets(device, &set_allocate_info, &handle_); | |
| 44 if (VK_SUCCESS != result) { | |
| 45 DLOG(ERROR) << "vkAllocateDescriptorSets() failed: " << result; | |
| 46 return false; | |
| 47 } | |
| 48 | |
| 49 return true; | |
| 50 } | |
| 51 | |
| 52 void VulkanDescriptorSet::Destroy() { | |
| 53 VkDevice device = device_queue_->GetVulkanDevice(); | |
| 54 if (VK_NULL_HANDLE != handle_) { | |
| 55 vkFreeDescriptorSets(device, descriptor_pool_->handle(), 1, &handle_); | |
| 56 handle_ = VK_NULL_HANDLE; | |
| 57 } | |
| 58 | |
| 59 if (VK_NULL_HANDLE != layout_) { | |
| 60 vkDestroyDescriptorSetLayout(device, layout_, nullptr); | |
| 61 layout_ = VK_NULL_HANDLE; | |
| 62 } | |
| 63 } | |
| 64 | |
| 65 void VulkanDescriptorSet::WriteToDescriptorSet( | |
| 66 uint32_t dst_binding, | |
| 67 uint32_t dst_array_element, | |
| 68 uint32_t descriptor_count, | |
| 69 VkDescriptorType descriptor_type, | |
| 70 const VkDescriptorImageInfo* image_info, | |
| 71 const VkDescriptorBufferInfo* buffer_info, | |
| 72 const VkBufferView* texel_buffer_view) { | |
| 73 VkWriteDescriptorSet write_descriptor_set = {}; | |
| 74 write_descriptor_set.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET; | |
| 75 write_descriptor_set.dstSet = handle_; | |
| 76 write_descriptor_set.dstBinding = dst_binding; | |
| 77 write_descriptor_set.dstArrayElement = dst_array_element; | |
| 78 write_descriptor_set.descriptorCount = descriptor_count; | |
| 79 write_descriptor_set.descriptorType = descriptor_type; | |
| 80 write_descriptor_set.pImageInfo = image_info; | |
| 81 write_descriptor_set.pBufferInfo = buffer_info; | |
| 82 write_descriptor_set.pTexelBufferView = texel_buffer_view; | |
| 83 | |
| 84 vkUpdateDescriptorSets(device_queue_->GetVulkanDevice(), 1, | |
| 85 &write_descriptor_set, 0, nullptr); | |
| 86 } | |
| 87 | |
| 88 void VulkanDescriptorSet::CopyFromDescriptorSet( | |
| 89 const VulkanDescriptorSet* source_set, | |
| 90 uint32_t src_binding, | |
| 91 uint32_t src_array_element, | |
| 92 uint32_t dst_binding, | |
| 93 uint32_t dst_array_element, | |
| 94 uint32_t descriptor_count) { | |
| 95 VkCopyDescriptorSet copy_descriptor_set = {}; | |
| 96 copy_descriptor_set.sType = VK_STRUCTURE_TYPE_COPY_DESCRIPTOR_SET; | |
| 97 copy_descriptor_set.srcSet = source_set->handle(); | |
| 98 copy_descriptor_set.srcBinding = src_binding; | |
| 99 copy_descriptor_set.srcArrayElement = src_array_element; | |
| 100 copy_descriptor_set.dstSet = handle_; | |
| 101 copy_descriptor_set.dstBinding = dst_binding; | |
| 102 copy_descriptor_set.dstArrayElement = dst_array_element; | |
| 103 copy_descriptor_set.descriptorCount = descriptor_count; | |
| 104 | |
| 105 vkUpdateDescriptorSets(device_queue_->GetVulkanDevice(), 0, nullptr, 1, | |
| 106 ©_descriptor_set); | |
| 107 } | |
| 108 | |
| 109 VulkanDescriptorSet::VulkanDescriptorSet(VulkanDeviceQueue* device_queue, | |
| 110 VulkanDescriptorPool* descriptor_pool) | |
| 111 : device_queue_(device_queue), descriptor_pool_(descriptor_pool) { | |
| 112 descriptor_pool_->IncrementDescriptorSetCount(); | |
| 113 } | |
| 114 | |
| 115 } // namespace gpu | |
| OLD | NEW |