Added Vulkan Descriptor Sets and Samplers.

gpu/vulkan/vulkan_descriptor_set.cc

+// 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_descriptor_set.h"
+
+#include "base/logging.h"
+#include "gpu/vulkan/vulkan_descriptor_layout.h"
+#include "gpu/vulkan/vulkan_descriptor_pool.h"
+#include "gpu/vulkan/vulkan_device_queue.h"
+
+namespace gpu {
+
+VulkanDescriptorSet::~VulkanDescriptorSet() {
+  DCHECK_EQ(static_cast<VkDescriptorSet>(VK_NULL_HANDLE), handle_);
+  descriptor_pool_->DecrementDescriptorSetCount();
+}
+
+bool VulkanDescriptorSet::Initialize(const VulkanDescriptorLayout* layout) {
+  VkResult result = VK_SUCCESS;
+  VkDevice device = device_queue_->GetVulkanDevice();
+
+  VkDescriptorSetLayout layout_handle = layout->handle();
+
+  VkDescriptorSetAllocateInfo set_allocate_info = {};
+  set_allocate_info.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO;
+  set_allocate_info.descriptorPool = descriptor_pool_->handle();
+  set_allocate_info.descriptorSetCount = 1;
+  set_allocate_info.pSetLayouts = &layout_handle;
+
+  result = vkAllocateDescriptorSets(device, &set_allocate_info, &handle_);
+  if (VK_SUCCESS != result) {
+    DLOG(ERROR) << "vkAllocateDescriptorSets() failed: " << result;
+    return false;
+  }
+
+  return true;
+}
+
+void VulkanDescriptorSet::Destroy() {
+  VkDevice device = device_queue_->GetVulkanDevice();
+  if (VK_NULL_HANDLE != handle_) {
+    vkFreeDescriptorSets(device, descriptor_pool_->handle(), 1, &handle_);
+    handle_ = VK_NULL_HANDLE;
+  }
+}
+
+void VulkanDescriptorSet::WriteToDescriptorSet(
+    uint32_t dst_binding,
+    uint32_t dst_array_element,
+    uint32_t descriptor_count,
+    VkDescriptorType descriptor_type,
+    const VkDescriptorImageInfo* image_info,
+    const VkDescriptorBufferInfo* buffer_info,
+    const VkBufferView* texel_buffer_view) {
+  VkWriteDescriptorSet write_descriptor_set = {};
+  write_descriptor_set.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
+  write_descriptor_set.dstSet = handle_;
+  write_descriptor_set.dstBinding = dst_binding;
+  write_descriptor_set.dstArrayElement = dst_array_element;
+  write_descriptor_set.descriptorCount = descriptor_count;
+  write_descriptor_set.descriptorType = descriptor_type;
+  write_descriptor_set.pImageInfo = image_info;
+  write_descriptor_set.pBufferInfo = buffer_info;
+  write_descriptor_set.pTexelBufferView = texel_buffer_view;
+
+  vkUpdateDescriptorSets(device_queue_->GetVulkanDevice(), 1,
+                         &write_descriptor_set, 0, nullptr);
+}
+
+void VulkanDescriptorSet::CopyFromDescriptorSet(
+    const VulkanDescriptorSet* source_set,
+    uint32_t src_binding,
+    uint32_t src_array_element,
+    uint32_t dst_binding,
+    uint32_t dst_array_element,
+    uint32_t descriptor_count) {
+  VkCopyDescriptorSet copy_descriptor_set = {};
+  copy_descriptor_set.sType = VK_STRUCTURE_TYPE_COPY_DESCRIPTOR_SET;
+  copy_descriptor_set.srcSet = source_set->handle();
+  copy_descriptor_set.srcBinding = src_binding;
+  copy_descriptor_set.srcArrayElement = src_array_element;
+  copy_descriptor_set.dstSet = handle_;
+  copy_descriptor_set.dstBinding = dst_binding;
+  copy_descriptor_set.dstArrayElement = dst_array_element;
+  copy_descriptor_set.descriptorCount = descriptor_count;
+
+  vkUpdateDescriptorSets(device_queue_->GetVulkanDevice(), 0, nullptr, 1,
+                         &copy_descriptor_set);
+}
+
+VulkanDescriptorSet::VulkanDescriptorSet(VulkanDeviceQueue* device_queue,
+                                         VulkanDescriptorPool* descriptor_pool)
+    : device_queue_(device_queue), descriptor_pool_(descriptor_pool) {
+  descriptor_pool_->IncrementDescriptorSetCount();
+}
+
+}  // namespace gpu