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