OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright 2016 Google Inc. |
| 3 * |
| 4 * Use of this source code is governed by a BSD-style license that can be |
| 5 * found in the LICENSE file. |
| 6 */ |
| 7 |
| 8 #include "GrVkDescriptorPool.h" |
| 9 |
| 10 #include "GrVkGpu.h" |
| 11 #include "SkTemplates.h" |
| 12 |
| 13 |
| 14 GrVkDescriptorPool::GrVkDescriptorPool(const GrVkGpu* gpu, const DescriptorTypeC
ounts& typeCounts) |
| 15 : INHERITED() |
| 16 , fTypeCounts(typeCounts) { |
| 17 uint32_t numPools = fTypeCounts.numPoolSizes(); |
| 18 SkAutoTDeleteArray<VkDescriptorPoolSize> poolSizes(new VkDescriptorPoolSize[
numPools]); |
| 19 int currentPool = 0; |
| 20 for (int i = VK_DESCRIPTOR_TYPE_BEGIN_RANGE; i < VK_DESCRIPTOR_TYPE_END_RANG
E; ++i) { |
| 21 if (fTypeCounts.fDescriptorTypeCount[i]) { |
| 22 VkDescriptorPoolSize& poolSize = poolSizes.get()[currentPool++]; |
| 23 poolSize.type = (VkDescriptorType)i; |
| 24 poolSize.descriptorCount = fTypeCounts.fDescriptorTypeCount[i]; |
| 25 } |
| 26 } |
| 27 SkASSERT(currentPool == numPools); |
| 28 |
| 29 VkDescriptorPoolCreateInfo createInfo; |
| 30 memset(&createInfo, 0, sizeof(VkDescriptorPoolCreateInfo)); |
| 31 createInfo.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO; |
| 32 createInfo.pNext = nullptr; |
| 33 createInfo.flags = 0; |
| 34 createInfo.maxSets = 2; // Currently we allow one set for samplers and one
set for uniforms |
| 35 createInfo.poolSizeCount = numPools; |
| 36 createInfo.pPoolSizes = poolSizes.get(); |
| 37 |
| 38 GR_VK_CALL_ERRCHECK(gpu->vkInterface(), CreateDescriptorPool(gpu->device(), |
| 39 &createInfo, |
| 40 nullptr, |
| 41 &fDescPool)); |
| 42 } |
| 43 |
| 44 bool GrVkDescriptorPool::isCompatible(const DescriptorTypeCounts& typeCounts) co
nst { |
| 45 return fTypeCounts.isSuperSet(typeCounts); |
| 46 } |
| 47 |
| 48 void GrVkDescriptorPool::reset(const GrVkGpu* gpu) { |
| 49 GR_VK_CALL_ERRCHECK(gpu->vkInterface(), ResetDescriptorPool(gpu->device(), f
DescPool, 0)); |
| 50 } |
| 51 |
| 52 void GrVkDescriptorPool::freeGPUData(const GrVkGpu* gpu) const { |
| 53 // Destroying the VkDescriptorPool will automatically free and delete any Vk
DescriptorSets |
| 54 // allocated from the pool. |
| 55 GR_VK_CALL(gpu->vkInterface(), DestroyDescriptorPool(gpu->device(), fDescPoo
l, nullptr)); |
| 56 } |
| 57 |
| 58 /////////////////////////////////////////////////////////////////////////////// |
| 59 |
| 60 uint32_t GrVkDescriptorPool::DescriptorTypeCounts::numPoolSizes() const { |
| 61 uint32_t count = 0; |
| 62 for (int i = VK_DESCRIPTOR_TYPE_BEGIN_RANGE; i < VK_DESCRIPTOR_TYPE_END_RANG
E; ++i) { |
| 63 count += fDescriptorTypeCount[i] ? 1 : 0; |
| 64 } |
| 65 return count; |
| 66 } |
| 67 |
| 68 bool GrVkDescriptorPool::DescriptorTypeCounts::isSuperSet(const DescriptorTypeCo
unts& that) const { |
| 69 for (int i = VK_DESCRIPTOR_TYPE_BEGIN_RANGE; i < VK_DESCRIPTOR_TYPE_END_RANG
E; ++i) { |
| 70 if (that.fDescriptorTypeCount[i] > fDescriptorTypeCount[i]) { |
| 71 return false; |
| 72 } |
| 73 } |
| 74 return true; |
| 75 } |
| 76 |
| 77 void GrVkDescriptorPool::DescriptorTypeCounts::setTypeCount(VkDescriptorType typ
e, uint8_t count) { |
| 78 fDescriptorTypeCount[type] = count; |
| 79 } |
OLD | NEW |