OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2015 Google Inc. | 2 * Copyright 2015 Google Inc. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
6 */ | 6 */ |
7 | 7 |
8 #ifndef GrVkResource_DEFINED | 8 #ifndef GrVkResource_DEFINED |
9 #define GrVkResource_DEFINED | 9 #define GrVkResource_DEFINED |
10 | 10 |
11 #include "SkAtomics.h" | 11 #include "SkAtomics.h" |
12 #include "SkTDynamicHash.h" | |
13 #include "SkRandom.h" | 12 #include "SkRandom.h" |
| 13 #include "SkTHash.h" |
14 | 14 |
15 class GrVkGpu; | 15 class GrVkGpu; |
16 | 16 |
17 // uncomment to enable tracing of resource refs | 17 // uncomment to enable tracing of resource refs |
18 #ifdef SK_DEBUG | 18 #ifdef SK_DEBUG |
19 #define SK_TRACE_VK_RESOURCES | 19 #define SK_TRACE_VK_RESOURCES |
20 #endif | 20 #endif |
21 | 21 |
22 /** \class GrVkResource | 22 /** \class GrVkResource |
23 | 23 |
24 GrVkResource is the base class for Vulkan resources that may be shared by mult
iple | 24 GrVkResource is the base class for Vulkan resources that may be shared by mult
iple |
25 objects. When an existing owner wants to share a reference, it calls ref(). | 25 objects. When an existing owner wants to share a reference, it calls ref(). |
26 When an owner wants to release its reference, it calls unref(). When the | 26 When an owner wants to release its reference, it calls unref(). When the |
27 shared object's reference count goes to zero as the result of an unref() | 27 shared object's reference count goes to zero as the result of an unref() |
28 call, its (virtual) destructor is called. It is an error for the | 28 call, its (virtual) destructor is called. It is an error for the |
29 destructor to be called explicitly (or via the object going out of scope on | 29 destructor to be called explicitly (or via the object going out of scope on |
30 the stack or calling delete) if getRefCnt() > 1. | 30 the stack or calling delete) if getRefCnt() > 1. |
31 | 31 |
32 This is nearly identical to SkRefCntBase. The exceptions are that unref() | 32 This is nearly identical to SkRefCntBase. The exceptions are that unref() |
33 takes a GrVkGpu, and any derived classes must implement freeGPUData() and | 33 takes a GrVkGpu, and any derived classes must implement freeGPUData() and |
34 possibly abandonSubResources(). | 34 possibly abandonSubResources(). |
35 */ | 35 */ |
36 | 36 |
37 class GrVkResource : SkNoncopyable { | 37 class GrVkResource : SkNoncopyable { |
38 public: | 38 public: |
39 // Simple refCount tracing, to ensure that everything ref'ed is unref'ed. | 39 // Simple refCount tracing, to ensure that everything ref'ed is unref'ed. |
40 #ifdef SK_TRACE_VK_RESOURCES | 40 #ifdef SK_TRACE_VK_RESOURCES |
41 static const uint32_t& GetKey(const GrVkResource& r) { return r.fKey; } | 41 struct Hash { |
42 static uint32_t Hash(const uint32_t& k) { return k; } | 42 uint32_t operator()(const GrVkResource* const& r) const { |
| 43 SkASSERT(r); |
| 44 return r->fKey; |
| 45 } |
| 46 }; |
43 | 47 |
44 class Trace { | 48 class Trace { |
45 public: | 49 public: |
46 ~Trace() { | 50 ~Trace() { |
47 if (fHash.count()) { | 51 fHashSet.foreach([](const GrVkResource* r) { |
48 SkTDynamicHash<GrVkResource, uint32_t>::Iter iter(&fHash); | 52 r->dumpInfo(); |
49 for (; !iter.done(); ++iter) { | 53 }); |
50 (*iter).dumpInfo(); | 54 SkASSERT(0 == fHashSet.count()); |
51 } | |
52 } | |
53 SkASSERT(0 == fHash.count()); | |
54 } | 55 } |
55 void add(GrVkResource* r) { fHash.add(r); } | 56 void add(const GrVkResource* r) { fHashSet.add(r); } |
56 void remove(const GrVkResource* r) { fHash.remove(GetKey(*r)); } | 57 void remove(const GrVkResource* r) { fHashSet.remove(r); } |
57 | 58 |
58 private: | 59 private: |
59 SkTDynamicHash<GrVkResource, uint32_t> fHash; | 60 SkTHashSet<const GrVkResource*, GrVkResource::Hash> fHashSet; |
60 }; | 61 }; |
61 static Trace fTrace; | 62 static Trace fTrace; |
62 | 63 |
63 static uint32_t fKeyCounter; | 64 static uint32_t fKeyCounter; |
64 #endif | 65 #endif |
65 | 66 |
66 /** Default construct, initializing the reference count to 1. | 67 /** Default construct, initializing the reference count to 1. |
67 */ | 68 */ |
68 GrVkResource() : fRefCnt(1) { | 69 GrVkResource() : fRefCnt(1) { |
69 #ifdef SK_TRACE_VK_RESOURCES | 70 #ifdef SK_TRACE_VK_RESOURCES |
(...skipping 133 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
203 } else { | 204 } else { |
204 this->unref(gpu); | 205 this->unref(gpu); |
205 } | 206 } |
206 } | 207 } |
207 | 208 |
208 private: | 209 private: |
209 virtual void onRecycle(GrVkGpu* gpu) const = 0; | 210 virtual void onRecycle(GrVkGpu* gpu) const = 0; |
210 }; | 211 }; |
211 | 212 |
212 #endif | 213 #endif |
OLD | NEW |