| /* |
| * Copyright 2013 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #if SK_SUPPORT_GPU |
| |
| #include "GrContext.h" |
| #include "GrContextFactory.h" |
| #include "GrGpu.h" |
| #include "GrResourceCache2.h" |
| #include "SkCanvas.h" |
| #include "SkGr.h" |
| #include "SkMessageBus.h" |
| #include "SkSurface.h" |
| #include "Test.h" |
| |
| static const int gWidth = 640; |
| static const int gHeight = 480; |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| static void test_cache(skiatest::Reporter* reporter, GrContext* context, SkCanvas* canvas) { |
| const SkIRect size = SkIRect::MakeWH(gWidth, gHeight); |
| |
| SkBitmap src; |
| src.allocN32Pixels(size.width(), size.height()); |
| src.eraseColor(SK_ColorBLACK); |
| size_t srcSize = src.getSize(); |
| |
| size_t initialCacheSize; |
| context->getResourceCacheUsage(NULL, &initialCacheSize); |
| |
| int oldMaxNum; |
| size_t oldMaxBytes; |
| context->getResourceCacheLimits(&oldMaxNum, &oldMaxBytes); |
| |
| // Set the cache limits so we can fit 10 "src" images and the |
| // max number of textures doesn't matter |
| size_t maxCacheSize = initialCacheSize + 10*srcSize; |
| context->setResourceCacheLimits(1000, maxCacheSize); |
| |
| SkBitmap readback; |
| readback.allocN32Pixels(size.width(), size.height()); |
| |
| for (int i = 0; i < 100; ++i) { |
| canvas->drawBitmap(src, 0, 0); |
| canvas->readPixels(size, &readback); |
| |
| // "modify" the src texture |
| src.notifyPixelsChanged(); |
| |
| size_t curCacheSize; |
| context->getResourceCacheUsage(NULL, &curCacheSize); |
| |
| // we should never go over the size limit |
| REPORTER_ASSERT(reporter, curCacheSize <= maxCacheSize); |
| } |
| |
| context->setResourceCacheLimits(oldMaxNum, oldMaxBytes); |
| } |
| |
| class TestResource : public GrGpuResource { |
| static const size_t kDefaultSize = 100; |
| |
| public: |
| SK_DECLARE_INST_COUNT(TestResource); |
| TestResource(GrGpu* gpu, size_t size, GrGpuResource::LifeCycle lifeCycle) |
| : INHERITED(gpu, lifeCycle) |
| , fToDelete(NULL) |
| , fSize(size) { |
| ++fNumAlive; |
| this->registerWithCache(); |
| } |
| |
| TestResource(GrGpu* gpu, GrGpuResource::LifeCycle lifeCycle) |
| : INHERITED(gpu, lifeCycle) |
| , fToDelete(NULL) |
| , fSize(kDefaultSize) { |
| ++fNumAlive; |
| this->registerWithCache(); |
| } |
| |
| TestResource(GrGpu* gpu) |
| : INHERITED(gpu, kCached_LifeCycle) |
| , fToDelete(NULL) |
| , fSize(kDefaultSize) { |
| ++fNumAlive; |
| this->registerWithCache(); |
| } |
| |
| TestResource(GrGpu* gpu, const GrScratchKey& scratchKey) |
| : INHERITED(gpu, kCached_LifeCycle) |
| , fToDelete(NULL) |
| , fSize(kDefaultSize) { |
| this->setScratchKey(scratchKey); |
| ++fNumAlive; |
| this->registerWithCache(); |
| } |
| |
| ~TestResource() { |
| --fNumAlive; |
| SkSafeUnref(fToDelete); |
| } |
| |
| void setSize(size_t size) { |
| fSize = size; |
| this->didChangeGpuMemorySize(); |
| } |
| |
| static int NumAlive() { return fNumAlive; } |
| |
| void setUnrefWhenDestroyed(TestResource* resource) { |
| SkRefCnt_SafeAssign(fToDelete, resource); |
| } |
| |
| private: |
| size_t onGpuMemorySize() const SK_OVERRIDE { return fSize; } |
| |
| TestResource* fToDelete; |
| size_t fSize; |
| static int fNumAlive; |
| |
| typedef GrGpuResource INHERITED; |
| }; |
| int TestResource::fNumAlive = 0; |
| |
| static void test_no_key(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| context->setResourceCacheLimits(10, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| // Create a bunch of resources with no keys |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* c = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* d = SkNEW_ARGS(TestResource, (context->getGpu())); |
| a->setSize(11); |
| b->setSize(12); |
| c->setSize(13); |
| d->setSize(14); |
| |
| REPORTER_ASSERT(reporter, 4 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 4 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, a->gpuMemorySize() + b->gpuMemorySize() + c->gpuMemorySize() + |
| d->gpuMemorySize() == cache2->getResourceBytes()); |
| |
| // Should be safe to purge without deleting the resources since we still have refs. |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 4 == TestResource::NumAlive()); |
| |
| // Since the resources have neither content nor scratch keys, delete immediately upon unref. |
| |
| a->unref(); |
| REPORTER_ASSERT(reporter, 3 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 3 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, b->gpuMemorySize() + c->gpuMemorySize() + d->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| |
| c->unref(); |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, b->gpuMemorySize() + d->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| |
| d->unref(); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, b->gpuMemorySize() == cache2->getResourceBytes()); |
| |
| b->unref(); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceBytes()); |
| } |
| |
| static void make_scratch_key(GrScratchKey* key) { |
| static GrScratchKey::ResourceType t = GrScratchKey::GenerateResourceType(); |
| GrScratchKey::Builder builder(key, t, 0); |
| } |
| |
| static void test_budgeting(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| context->setResourceCacheLimits(10, 300); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| SkASSERT(0 == cache2->getBudgetedResourceCount() && 0 == cache2->getBudgetedResourceBytes()); |
| |
| GrScratchKey scratchKey; |
| make_scratch_key(&scratchKey); |
| |
| GrCacheID::Key keyData; |
| memset(&keyData, 0, sizeof(keyData)); |
| GrResourceKey contentKey(GrCacheID(GrCacheID::GenerateDomain(), keyData), 0); |
| |
| // Create a scratch, a content, and a wrapped resource |
| TestResource* scratch = SkNEW_ARGS(TestResource, (context->getGpu(), scratchKey)); |
| scratch->setSize(10); |
| TestResource* content = SkNEW_ARGS(TestResource, (context->getGpu())); |
| content->setSize(11); |
| REPORTER_ASSERT(reporter, content->cacheAccess().setContentKey(contentKey)); |
| TestResource* wrapped = SkNEW_ARGS(TestResource, |
| (context->getGpu(), GrGpuResource::kWrapped_LifeCycle)); |
| wrapped->setSize(12); |
| TestResource* unbudgeted = SkNEW_ARGS(TestResource, |
| (context->getGpu(), GrGpuResource::kUncached_LifeCycle)); |
| unbudgeted->setSize(13); |
| |
| // Make sure we can't add a content key to the wrapped resource |
| keyData.fData8[0] = 1; |
| GrResourceKey contentKey2(GrCacheID(GrCacheID::GenerateDomain(), keyData), 0); |
| REPORTER_ASSERT(reporter, !wrapped->cacheAccess().setContentKey(contentKey2)); |
| REPORTER_ASSERT(reporter, NULL == cache2->findAndRefContentResource(contentKey2)); |
| |
| // Make sure sizes are as we expect |
| REPORTER_ASSERT(reporter, 4 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() + content->gpuMemorySize() + |
| wrapped->gpuMemorySize() + unbudgeted->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() + content->gpuMemorySize() == |
| cache2->getBudgetedResourceBytes()); |
| |
| // Our refs mean that the resources are non purgable. |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 4 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() + content->gpuMemorySize() + |
| wrapped->gpuMemorySize() + unbudgeted->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() + content->gpuMemorySize() == |
| cache2->getBudgetedResourceBytes()); |
| |
| // Unreffing the wrapped resource should free it right away. |
| wrapped->unref(); |
| REPORTER_ASSERT(reporter, 3 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() + content->gpuMemorySize() + |
| unbudgeted->gpuMemorySize() == cache2->getResourceBytes()); |
| |
| // Now try freeing the budgeted resources first |
| wrapped = SkNEW_ARGS(TestResource, (context->getGpu(), GrGpuResource::kWrapped_LifeCycle)); |
| scratch->setSize(12); |
| content->unref(); |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 3 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() + wrapped->gpuMemorySize() + |
| unbudgeted->gpuMemorySize() == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 1 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, scratch->gpuMemorySize() == cache2->getBudgetedResourceBytes()); |
| |
| scratch->unref(); |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, unbudgeted->gpuMemorySize() + wrapped->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceBytes()); |
| |
| wrapped->unref(); |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, unbudgeted->gpuMemorySize() == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceBytes()); |
| |
| unbudgeted->unref(); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceBytes()); |
| } |
| |
| static void test_unbudgeted(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| context->setResourceCacheLimits(10, 300); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| SkASSERT(0 == cache2->getBudgetedResourceCount() && 0 == cache2->getBudgetedResourceBytes()); |
| |
| GrScratchKey scratchKey; |
| make_scratch_key(&scratchKey); |
| |
| GrCacheID::Key keyData; |
| memset(&keyData, 0, sizeof(keyData)); |
| GrResourceKey contentKey(GrCacheID(GrCacheID::GenerateDomain(), keyData), 0); |
| |
| TestResource* scratch; |
| TestResource* content; |
| TestResource* wrapped; |
| TestResource* unbudgeted; |
| |
| // A large uncached or wrapped resource shouldn't evict anything. |
| scratch = SkNEW_ARGS(TestResource, (context->getGpu(), scratchKey)); |
| scratch->setSize(10); |
| scratch->unref(); |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 10 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 1 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 10 == cache2->getBudgetedResourceBytes()); |
| |
| content = SkNEW_ARGS(TestResource, (context->getGpu())); |
| content->setSize(11); |
| REPORTER_ASSERT(reporter, content->cacheAccess().setContentKey(contentKey)); |
| content->unref(); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getBudgetedResourceBytes()); |
| |
| size_t large = 2 * cache2->getResourceBytes(); |
| unbudgeted = SkNEW_ARGS(TestResource, |
| (context->getGpu(), large, GrGpuResource::kUncached_LifeCycle)); |
| REPORTER_ASSERT(reporter, 3 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 21 + large == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getBudgetedResourceBytes()); |
| |
| unbudgeted->unref(); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getBudgetedResourceBytes()); |
| |
| wrapped = SkNEW_ARGS(TestResource, |
| (context->getGpu(), large, GrGpuResource::kWrapped_LifeCycle)); |
| REPORTER_ASSERT(reporter, 3 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 21 + large == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getBudgetedResourceBytes()); |
| |
| wrapped->unref(); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 21 == cache2->getBudgetedResourceBytes()); |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getBudgetedResourceBytes()); |
| } |
| |
| static void test_duplicate_scratch_key(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| context->setResourceCacheLimits(5, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| GrScratchKey scratchKey; |
| make_scratch_key(&scratchKey); |
| |
| // Create two resources that have the same scratch key. |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu(), scratchKey)); |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu(), scratchKey)); |
| a->setSize(11); |
| b->setSize(12); |
| // Scratch resources are registered with GrResourceCache2 just by existing. There are 2. |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 2 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, a->gpuMemorySize() + b->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| |
| // Our refs mean that the resources are non purgable. |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| |
| // Unref but don't purge |
| a->unref(); |
| b->unref(); |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 2 == cache2->countScratchEntriesForKey(scratchKey));) |
| |
| // Purge again. This time resources should be purgable. |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 0 == cache2->countScratchEntriesForKey(scratchKey));) |
| } |
| |
| static void test_remove_scratch_key(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| context->setResourceCacheLimits(5, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| GrScratchKey scratchKey; |
| make_scratch_key(&scratchKey); |
| |
| // Create two resources that have the same scratch key. |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu(), scratchKey)); |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu(), scratchKey)); |
| a->unref(); |
| b->unref(); |
| |
| // Scratch resources are registered with GrResourceCache2 just by existing. There are 2. |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 2 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| |
| // Find the first resource and remove its scratch key |
| GrGpuResource* find; |
| find = cache2->findAndRefScratchResource(scratchKey); |
| find->cacheAccess().removeScratchKey(); |
| // It's still alive, but not cached by scratch key anymore |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 1 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| |
| // The cache should immediately delete it when it's unrefed since it isn't accessible. |
| find->unref(); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 1 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| |
| // Repeat for the second resource. |
| find = cache2->findAndRefScratchResource(scratchKey); |
| find->cacheAccess().removeScratchKey(); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 0 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| |
| // Should be able to call this multiple times with no problem. |
| find->cacheAccess().removeScratchKey(); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 0 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| |
| find->unref(); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| SkDEBUGCODE(REPORTER_ASSERT(reporter, 0 == cache2->countScratchEntriesForKey(scratchKey));) |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| } |
| |
| static void test_duplicate_content_key(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| context->setResourceCacheLimits(5, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| GrCacheID::Domain domain = GrCacheID::GenerateDomain(); |
| GrCacheID::Key keyData; |
| memset(&keyData, 0, sizeof(keyData)); |
| GrResourceKey key(GrCacheID(domain, keyData), 0); |
| |
| // Create two resources that we will attempt to register with the same content key. |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu())); |
| a->setSize(11); |
| b->setSize(12); |
| |
| // Can't set the same content key on two resources. |
| REPORTER_ASSERT(reporter, a->cacheAccess().setContentKey(key)); |
| REPORTER_ASSERT(reporter, !b->cacheAccess().setContentKey(key)); |
| |
| // Still have two resources because b is still reffed. |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, a->gpuMemorySize() + b->gpuMemorySize() == |
| cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| |
| b->unref(); |
| // Now b should be gone. |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, a->gpuMemorySize() == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, a->gpuMemorySize() == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| |
| // Drop the ref on a but it isn't immediately purged as it still has a valid scratch key. |
| a->unref(); |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, a->gpuMemorySize() == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| } |
| |
| static void test_purge_invalidated(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| |
| GrCacheID::Domain domain = GrCacheID::GenerateDomain(); |
| GrCacheID::Key keyData; |
| memset(&keyData, 0, sizeof(keyData)); |
| |
| keyData.fData64[0] = 1; |
| GrResourceKey key1(GrCacheID(domain, keyData), 0); |
| keyData.fData64[0] = 2; |
| GrResourceKey key2(GrCacheID(domain, keyData), 0); |
| keyData.fData64[0] = 3; |
| GrResourceKey key3(GrCacheID(domain, keyData), 0); |
| |
| context->setResourceCacheLimits(5, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| // Add three resources to the cache. |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* c = SkNEW_ARGS(TestResource, (context->getGpu())); |
| a->cacheAccess().setContentKey(key1); |
| b->cacheAccess().setContentKey(key2); |
| c->cacheAccess().setContentKey(key3); |
| a->unref(); |
| b->unref(); |
| c->unref(); |
| |
| REPORTER_ASSERT(reporter, cache2->hasContentKey(key1)); |
| REPORTER_ASSERT(reporter, cache2->hasContentKey(key2)); |
| REPORTER_ASSERT(reporter, cache2->hasContentKey(key3)); |
| |
| // Invalidate two of the three, they should be purged and destroyed. |
| REPORTER_ASSERT(reporter, 3 == TestResource::NumAlive()); |
| const GrResourceInvalidatedMessage msg1 = { key1 }; |
| SkMessageBus<GrResourceInvalidatedMessage>::Post(msg1); |
| const GrResourceInvalidatedMessage msg2 = { key2 }; |
| SkMessageBus<GrResourceInvalidatedMessage>::Post(msg2); |
| #if 0 // Disabled until reimplemented in GrResourceCache2. |
| cache2->purgeAsNeeded(); |
| REPORTER_ASSERT(reporter, 1 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, !cache2->hasContentKey(key1)); |
| REPORTER_ASSERT(reporter, !cache2->hasContentKey(key2)); |
| REPORTER_ASSERT(reporter, cache2->hasContentKey(key3)); |
| #endif |
| |
| // Invalidate the third. |
| const GrResourceInvalidatedMessage msg3 = { key3 }; |
| SkMessageBus<GrResourceInvalidatedMessage>::Post(msg3); |
| #if 0 // Disabled until reimplemented in GrResourceCache2. |
| cache2->purgeAsNeeded(); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, !cache2->hasContentKey(key3)); |
| #endif |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceCount()); |
| REPORTER_ASSERT(reporter, 0 == cache2->getResourceBytes()); |
| } |
| |
| static void test_cache_chained_purge(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| |
| GrCacheID::Domain domain = GrCacheID::GenerateDomain(); |
| GrCacheID::Key keyData; |
| memset(&keyData, 0, sizeof(keyData)); |
| |
| keyData.fData64[0] = 1; |
| GrResourceKey key1(GrCacheID(domain, keyData), 0); |
| |
| keyData.fData64[0] = 2; |
| GrResourceKey key2(GrCacheID(domain, keyData), 0); |
| |
| { |
| context->setResourceCacheLimits(3, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu())); |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu())); |
| a->cacheAccess().setContentKey(key1); |
| b->cacheAccess().setContentKey(key2); |
| |
| // Make a cycle |
| a->setUnrefWhenDestroyed(b); |
| b->setUnrefWhenDestroyed(a); |
| |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| |
| a->unref(); |
| b->unref(); |
| |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| |
| // Break the cycle |
| a->setUnrefWhenDestroyed(NULL); |
| REPORTER_ASSERT(reporter, 2 == TestResource::NumAlive()); |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, 0 == TestResource::NumAlive()); |
| } |
| } |
| |
| static void test_resource_size_changed(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| |
| GrCacheID::Domain domain = GrCacheID::GenerateDomain(); |
| |
| GrCacheID::Key key1Data; |
| key1Data.fData64[0] = 0; |
| key1Data.fData64[1] = 0; |
| GrResourceKey key1(GrCacheID(domain, key1Data), 0); |
| |
| GrCacheID::Key key2Data; |
| key2Data.fData64[0] = 1; |
| key2Data.fData64[1] = 0; |
| GrResourceKey key2(GrCacheID(domain, key2Data), 0); |
| |
| // Test changing resources sizes (both increase & decrease). |
| { |
| context->setResourceCacheLimits(3, 30000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu())); |
| a->cacheAccess().setContentKey(key1); |
| a->unref(); |
| |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu())); |
| b->cacheAccess().setContentKey(key2); |
| b->unref(); |
| |
| REPORTER_ASSERT(reporter, 200 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| { |
| SkAutoTUnref<TestResource> find2(static_cast<TestResource*>(cache2->findAndRefContentResource(key2))); |
| find2->setSize(200); |
| SkAutoTUnref<TestResource> find1(static_cast<TestResource*>(cache2->findAndRefContentResource(key1))); |
| find1->setSize(50); |
| } |
| |
| REPORTER_ASSERT(reporter, 250 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| } |
| |
| // Test increasing a resources size beyond the cache budget. |
| { |
| context->setResourceCacheLimits(2, 300); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| TestResource* a = SkNEW_ARGS(TestResource, (context->getGpu())); |
| a->setSize(100); |
| a->cacheAccess().setContentKey(key1); |
| a->unref(); |
| |
| TestResource* b = SkNEW_ARGS(TestResource, (context->getGpu())); |
| b->setSize(100); |
| b->cacheAccess().setContentKey(key2); |
| b->unref(); |
| |
| REPORTER_ASSERT(reporter, 200 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 2 == cache2->getResourceCount()); |
| |
| { |
| SkAutoTUnref<TestResource> find2(static_cast<TestResource*>(cache2->findAndRefContentResource(key2))); |
| find2->setSize(201); |
| } |
| REPORTER_ASSERT(reporter, !cache2->hasContentKey(key1)); |
| |
| REPORTER_ASSERT(reporter, 201 == cache2->getResourceBytes()); |
| REPORTER_ASSERT(reporter, 1 == cache2->getResourceCount()); |
| } |
| } |
| |
| static void test_large_resource_count(skiatest::Reporter* reporter) { |
| SkAutoTUnref<GrContext> context(GrContext::CreateMockContext()); |
| REPORTER_ASSERT(reporter, SkToBool(context)); |
| if (NULL == context) { |
| return; |
| } |
| |
| static const int kResourceCnt = 2000; |
| // Set the cache size to double the resource count because we're going to create 2x that number |
| // resources, using two different key domains. Add a little slop to the bytes because we resize |
| // down to 1 byte after creating the resource. |
| context->setResourceCacheLimits(2 * kResourceCnt, 2 * kResourceCnt + 1000); |
| GrResourceCache2* cache2 = context->getResourceCache2(); |
| cache2->purgeAllUnlocked(); |
| SkASSERT(0 == cache2->getResourceCount() && 0 == cache2->getResourceBytes()); |
| |
| GrCacheID::Domain domain0 = GrCacheID::GenerateDomain(); |
| GrCacheID::Domain domain1 = GrCacheID::GenerateDomain(); |
| |
| GrCacheID::Key keyData; |
| memset(&keyData, 0, sizeof(keyData)); |
| |
| for (int i = 0; i < kResourceCnt; ++i) { |
| TestResource* resource; |
| keyData.fData32[0] = i; |
| |
| GrResourceKey key0(GrCacheID(domain0, keyData), 0); |
| resource = SkNEW_ARGS(TestResource, (context->getGpu())); |
| resource->cacheAccess().setContentKey(key0); |
| resource->setSize(1); |
| resource->unref(); |
| |
| GrResourceKey key1(GrCacheID(domain1, keyData), 0); |
| resource = SkNEW_ARGS(TestResource, (context->getGpu())); |
| resource->cacheAccess().setContentKey(key1); |
| resource->setSize(1); |
| resource->unref(); |
| } |
| |
| REPORTER_ASSERT(reporter, TestResource::NumAlive() == 2 * kResourceCnt); |
| REPORTER_ASSERT(reporter, cache2->getBudgetedResourceBytes() == 2 * kResourceCnt); |
| REPORTER_ASSERT(reporter, cache2->getBudgetedResourceCount() == 2 * kResourceCnt); |
| REPORTER_ASSERT(reporter, cache2->getResourceBytes() == 2 * kResourceCnt); |
| REPORTER_ASSERT(reporter, cache2->getResourceCount() == 2 * kResourceCnt); |
| for (int i = 0; i < kResourceCnt; ++i) { |
| keyData.fData32[0] = i; |
| GrResourceKey key0(GrCacheID(domain0, keyData), 0); |
| REPORTER_ASSERT(reporter, cache2->hasContentKey(key0)); |
| GrResourceKey key1(GrCacheID(domain0, keyData), 0); |
| REPORTER_ASSERT(reporter, cache2->hasContentKey(key1)); |
| } |
| |
| cache2->purgeAllUnlocked(); |
| REPORTER_ASSERT(reporter, TestResource::NumAlive() == 0); |
| REPORTER_ASSERT(reporter, cache2->getBudgetedResourceBytes() == 0); |
| REPORTER_ASSERT(reporter, cache2->getBudgetedResourceCount() == 0); |
| REPORTER_ASSERT(reporter, cache2->getResourceBytes() == 0); |
| REPORTER_ASSERT(reporter, cache2->getResourceCount() == 0); |
| |
| for (int i = 0; i < kResourceCnt; ++i) { |
| keyData.fData32[0] = i; |
| GrResourceKey key0(GrCacheID(domain0, keyData), 0); |
| REPORTER_ASSERT(reporter, !cache2->hasContentKey(key0)); |
| GrResourceKey key1(GrCacheID(domain0, keyData), 0); |
| REPORTER_ASSERT(reporter, !cache2->hasContentKey(key1)); |
| } |
| } |
| |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| DEF_GPUTEST(ResourceCache, reporter, factory) { |
| for (int type = 0; type < GrContextFactory::kLastGLContextType; ++type) { |
| GrContextFactory::GLContextType glType = static_cast<GrContextFactory::GLContextType>(type); |
| if (!GrContextFactory::IsRenderingGLContext(glType)) { |
| continue; |
| } |
| GrContext* context = factory->get(glType); |
| if (NULL == context) { |
| continue; |
| } |
| GrSurfaceDesc desc; |
| desc.fConfig = kSkia8888_GrPixelConfig; |
| desc.fFlags = kRenderTarget_GrSurfaceFlag; |
| desc.fWidth = gWidth; |
| desc.fHeight = gHeight; |
| SkImageInfo info = SkImageInfo::MakeN32Premul(gWidth, gHeight); |
| SkAutoTUnref<SkSurface> surface(SkSurface::NewRenderTarget(context, |
| SkSurface::kNo_Budgeted, info)); |
| test_cache(reporter, context, surface->getCanvas()); |
| } |
| |
| // The below tests create their own mock contexts. |
| test_no_key(reporter); |
| test_budgeting(reporter); |
| test_unbudgeted(reporter); |
| test_duplicate_content_key(reporter); |
| test_duplicate_scratch_key(reporter); |
| test_remove_scratch_key(reporter); |
| test_purge_invalidated(reporter); |
| test_cache_chained_purge(reporter); |
| test_resource_size_changed(reporter); |
| test_large_resource_count(reporter); |
| } |
| |
| #endif |
