Allow GPU resources to not be counted against the cache budget.

src/gpu/GrGpuResource.cpp

 
 /*
  * Copyright 2011 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 
 #include "GrGpuResource.h"
 #include "GrResourceCache2.h"
 #include "GrGpu.h"
 
 static inline GrResourceCache2* get_resource_cache2(GrGpu* gpu) {
     SkASSERT(gpu);
     SkASSERT(gpu->getContext());
     SkASSERT(gpu->getContext()->getResourceCache2());
     return gpu->getContext()->getResourceCache2();
 }
 
 GrGpuResource::GrGpuResource(GrGpu* gpu, bool isWrapped)
-    : fGpu(gpu)
+    : fScratchKey(GrResourceKey::NullScratchKey())
+    , fGpu(gpu)
     , fGpuMemorySize(kInvalidGpuMemorySize)
-    , fUniqueID(CreateUniqueID())
-    , fScratchKey(GrResourceKey::NullScratchKey())
-    , fContentKeySet(false) {
+    , fUniqueID(CreateUniqueID()) {
     if (isWrapped) {
-        fFlags = kWrapped_FlagBit;
+        fFlags = kWrapped_Flag;
     } else {
-        fFlags = 0;
+        // By default all non-wrapped resources are budgeted.
+        fFlags = kBudgeted_Flag;
     }
 }
 
 void GrGpuResource::registerWithCache() {
     get_resource_cache2(fGpu)->resourceAccess().insertResource(this);
 }
 
 GrGpuResource::~GrGpuResource() {
     // The cache should have released or destroyed this resource.
     SkASSERT(this->wasDestroyed());
@@ skipping 44 matching lines @@
 
 bool GrGpuResource::setContentKey(const GrResourceKey& contentKey) {
     // Currently this can only be called once and can't be called when the resource is scratch.
     SkASSERT(!contentKey.isScratch());
     SkASSERT(this->internalHasRef());
 
     // Wrapped resources can never have a key.
     if (this->isWrapped()) {
         return false;
     }
-    
-    if (fContentKeySet || this->wasDestroyed()) {
+
+    if ((fFlags & kContentKeySet_Flag) || this->wasDestroyed()) {
         return false;
     }
 
     fContentKey = contentKey;
-    fContentKeySet = true;
+    fFlags |= kContentKeySet_Flag;
 
     if (!get_resource_cache2(fGpu)->resourceAccess().didSetContentKey(this)) {
-        fContentKeySet = false;
+         fFlags &= ~kContentKeySet_Flag;
         return false;
     }
     return true;
 }
 
 void GrGpuResource::notifyIsPurgable() const {
     if (this->wasDestroyed()) {
         // We've already been removed from the cache. Goodbye cruel world!
         SkDELETE(this);
     } else {
@@ skipping 14 matching lines @@
 }
 
 uint32_t GrGpuResource::CreateUniqueID() {
     static int32_t gUniqueID = SK_InvalidUniqueID;
     uint32_t id;
     do {
         id = static_cast<uint32_t>(sk_atomic_inc(&gUniqueID) + 1);
     } while (id == SK_InvalidUniqueID);
     return id;
 }
+
+void GrGpuResource::setBudgeted(bool countsAgainstBudget) {
+    // Wrapped resources never count against the budget, nothing to do. No point in changing the
+    // budgeting of destroyed resources.
+    if (this->isWrapped() || this->wasDestroyed()) {
+        return;
+    }
+
+    uint32_t oldFlags = fFlags;
+    if (countsAgainstBudget) {
+        fFlags |= kBudgeted_Flag;
+    } else {
+        fFlags &= ~kBudgeted_Flag;
+    }
+    if (fFlags != oldFlags) {
+        get_resource_cache2(fGpu)->resourceAccess().didChangeBudgetStatus(this);
+    }
+}