|  | /* | 
|  | * Copyright 2014 Google Inc. | 
|  | * | 
|  | * Use of this source code is governed by a BSD-style license that can be | 
|  | * found in the LICENSE file. | 
|  | */ | 
|  |  | 
|  | #include "GrAtlas.h" | 
|  | #include "GrGpu.h" | 
|  | #include "GrLayerCache.h" | 
|  | #include "GrSurfacePriv.h" | 
|  |  | 
|  | #ifdef SK_DEBUG | 
|  | void GrCachedLayer::validate(const GrTexture* backingTexture) const { | 
|  | SkASSERT(SK_InvalidGenID != fKey.pictureID()); | 
|  |  | 
|  | if (fTexture) { | 
|  | // If the layer is in some texture then it must occupy some rectangle | 
|  | SkASSERT(!fRect.isEmpty()); | 
|  | if (!this->isAtlased()) { | 
|  | // If it isn't atlased then the rectangle should start at the origin | 
|  | SkASSERT(0.0f == fRect.fLeft && 0.0f == fRect.fTop); | 
|  | } | 
|  | } else { | 
|  | SkASSERT(fRect.isEmpty()); | 
|  | SkASSERT(NULL == fPlot); | 
|  | SkASSERT(!fLocked);     // layers without a texture cannot be locked | 
|  | } | 
|  |  | 
|  | if (fPlot) { | 
|  | // If a layer has a plot (i.e., is atlased) then it must point to | 
|  | // the backing texture. Additionally, its rect should be non-empty. | 
|  | SkASSERT(fTexture && backingTexture == fTexture); | 
|  | SkASSERT(!fRect.isEmpty()); | 
|  | } | 
|  |  | 
|  | if (fLocked) { | 
|  | // If a layer is locked it must have a texture (though it need not be | 
|  | // the atlas-backing texture) and occupy some space. | 
|  | SkASSERT(fTexture); | 
|  | SkASSERT(!fRect.isEmpty()); | 
|  | } | 
|  |  | 
|  | // Unfortunately there is a brief time where a layer can be locked | 
|  | // but not used, so we can only check the "used implies locked" | 
|  | // invariant. | 
|  | if (fUses > 0) { | 
|  | SkASSERT(fLocked); | 
|  | } else { | 
|  | SkASSERT(0 == fUses); | 
|  | } | 
|  | } | 
|  |  | 
|  | class GrAutoValidateLayer : ::SkNoncopyable { | 
|  | public: | 
|  | GrAutoValidateLayer(GrTexture* backingTexture, const GrCachedLayer* layer) | 
|  | : fBackingTexture(backingTexture) | 
|  | , fLayer(layer) { | 
|  | if (fLayer) { | 
|  | fLayer->validate(backingTexture); | 
|  | } | 
|  | } | 
|  | ~GrAutoValidateLayer() { | 
|  | if (fLayer) { | 
|  | fLayer->validate(fBackingTexture); | 
|  | } | 
|  | } | 
|  | void setBackingTexture(GrTexture* backingTexture) { | 
|  | SkASSERT(NULL == fBackingTexture || fBackingTexture == backingTexture); | 
|  | fBackingTexture = backingTexture; | 
|  | } | 
|  |  | 
|  | private: | 
|  | const GrTexture* fBackingTexture; | 
|  | const GrCachedLayer* fLayer; | 
|  | }; | 
|  | #endif | 
|  |  | 
|  | GrLayerCache::GrLayerCache(GrContext* context) | 
|  | : fContext(context) { | 
|  | memset(fPlotLocks, 0, sizeof(fPlotLocks)); | 
|  | } | 
|  |  | 
|  | GrLayerCache::~GrLayerCache() { | 
|  |  | 
|  | SkTDynamicHash<GrCachedLayer, GrCachedLayer::Key>::Iter iter(&fLayerHash); | 
|  | for (; !iter.done(); ++iter) { | 
|  | GrCachedLayer* layer = &(*iter); | 
|  | SkASSERT(0 == layer->uses()); | 
|  | this->unlock(layer); | 
|  | SkDELETE(layer); | 
|  | } | 
|  |  | 
|  | SkASSERT(0 == fPictureHash.count()); | 
|  |  | 
|  | // The atlas only lets go of its texture when the atlas is deleted. | 
|  | fAtlas.free(); | 
|  | } | 
|  |  | 
|  | void GrLayerCache::initAtlas() { | 
|  | SkASSERT(NULL == fAtlas.get()); | 
|  | GR_STATIC_ASSERT(kNumPlotsX*kNumPlotsX == GrPictureInfo::kNumPlots); | 
|  |  | 
|  | SkISize textureSize = SkISize::Make(kAtlasTextureWidth, kAtlasTextureHeight); | 
|  | fAtlas.reset(SkNEW_ARGS(GrAtlas, (fContext->getGpu(), kSkia8888_GrPixelConfig, | 
|  | kRenderTarget_GrSurfaceFlag, | 
|  | textureSize, kNumPlotsX, kNumPlotsY, false))); | 
|  | } | 
|  |  | 
|  | void GrLayerCache::freeAll() { | 
|  |  | 
|  | SkTDynamicHash<GrCachedLayer, GrCachedLayer::Key>::Iter iter(&fLayerHash); | 
|  | for (; !iter.done(); ++iter) { | 
|  | GrCachedLayer* layer = &(*iter); | 
|  | this->unlock(layer); | 
|  | SkDELETE(layer); | 
|  | } | 
|  | fLayerHash.rewind(); | 
|  |  | 
|  | // The atlas only lets go of its texture when the atlas is deleted. | 
|  | fAtlas.free(); | 
|  | } | 
|  |  | 
|  | GrCachedLayer* GrLayerCache::createLayer(uint32_t pictureID, | 
|  | int start, int stop, | 
|  | const SkIRect& srcIR, | 
|  | const SkIRect& dstIR, | 
|  | const SkMatrix& initialMat, | 
|  | const unsigned* key, | 
|  | int keySize, | 
|  | const SkPaint* paint) { | 
|  | SkASSERT(pictureID != SK_InvalidGenID && start >= 0 && stop > 0); | 
|  |  | 
|  | GrCachedLayer* layer = SkNEW_ARGS(GrCachedLayer, (pictureID, start, stop, | 
|  | srcIR, dstIR, initialMat, | 
|  | key, keySize, paint)); | 
|  | fLayerHash.add(layer); | 
|  | return layer; | 
|  | } | 
|  |  | 
|  | GrCachedLayer* GrLayerCache::findLayer(uint32_t pictureID, const SkMatrix& initialMat, | 
|  | const unsigned* key, int keySize) { | 
|  | SkASSERT(pictureID != SK_InvalidGenID); | 
|  | return fLayerHash.find(GrCachedLayer::Key(pictureID, initialMat, key, keySize)); | 
|  | } | 
|  |  | 
|  | GrCachedLayer* GrLayerCache::findLayerOrCreate(uint32_t pictureID, | 
|  | int start, int stop, | 
|  | const SkIRect& srcIR, | 
|  | const SkIRect& dstIR, | 
|  | const SkMatrix& initialMat, | 
|  | const unsigned* key, | 
|  | int keySize, | 
|  | const SkPaint* paint) { | 
|  | SkASSERT(pictureID != SK_InvalidGenID && start >= 0 && stop > 0); | 
|  | GrCachedLayer* layer = fLayerHash.find(GrCachedLayer::Key(pictureID, initialMat, key, keySize)); | 
|  | if (NULL == layer) { | 
|  | layer = this->createLayer(pictureID, start, stop, | 
|  | srcIR, dstIR, initialMat, | 
|  | key, keySize, paint); | 
|  | } | 
|  |  | 
|  | return layer; | 
|  | } | 
|  |  | 
|  | bool GrLayerCache::tryToAtlas(GrCachedLayer* layer, | 
|  | const GrSurfaceDesc& desc, | 
|  | bool* needsRendering) { | 
|  | SkDEBUGCODE(GrAutoValidateLayer avl(fAtlas ? fAtlas->getTexture() : NULL, layer);) | 
|  |  | 
|  | SkASSERT(PlausiblyAtlasable(desc.fWidth, desc.fHeight)); | 
|  | SkASSERT(0 == desc.fSampleCnt); | 
|  |  | 
|  | if (layer->locked()) { | 
|  | // This layer is already locked | 
|  | SkASSERT(fAtlas); | 
|  | SkASSERT(layer->isAtlased()); | 
|  | SkASSERT(layer->rect().width() == desc.fWidth); | 
|  | SkASSERT(layer->rect().height() == desc.fHeight); | 
|  | *needsRendering = false; | 
|  | return true; | 
|  | } | 
|  |  | 
|  | if (layer->isAtlased()) { | 
|  | SkASSERT(fAtlas); | 
|  | // Hooray it is still in the atlas - make sure it stays there | 
|  | layer->setLocked(true); | 
|  | this->incPlotLock(layer->plot()->id()); | 
|  | *needsRendering = false; | 
|  | return true; | 
|  | } else { | 
|  | if (!fAtlas) { | 
|  | this->initAtlas(); | 
|  | if (!fAtlas) { | 
|  | return false; | 
|  | } | 
|  | } | 
|  | // Not in the atlas - will it fit? | 
|  | GrPictureInfo* pictInfo = fPictureHash.find(layer->pictureID()); | 
|  | if (NULL == pictInfo) { | 
|  | pictInfo = SkNEW_ARGS(GrPictureInfo, (layer->pictureID())); | 
|  | fPictureHash.add(pictInfo); | 
|  | } | 
|  |  | 
|  | SkIPoint16 loc; | 
|  | for (int i = 0; i < 2; ++i) { // extra pass in case we fail to add but are able to purge | 
|  | GrPlot* plot = fAtlas->addToAtlas(&pictInfo->fPlotUsage, | 
|  | desc.fWidth, desc.fHeight, | 
|  | NULL, &loc); | 
|  | // addToAtlas can allocate the backing texture | 
|  | SkDEBUGCODE(avl.setBackingTexture(fAtlas->getTexture())); | 
|  | if (plot) { | 
|  | #if !GR_CACHE_HOISTED_LAYERS | 
|  | pictInfo->incPlotUsage(plot->id()); | 
|  | #endif | 
|  | // The layer was successfully added to the atlas | 
|  | const SkIRect bounds = SkIRect::MakeXYWH(loc.fX, loc.fY, | 
|  | desc.fWidth, desc.fHeight); | 
|  | layer->setTexture(fAtlas->getTexture(), bounds); | 
|  | layer->setPlot(plot); | 
|  | layer->setLocked(true); | 
|  | this->incPlotLock(layer->plot()->id()); | 
|  | *needsRendering = true; | 
|  | return true; | 
|  | } | 
|  |  | 
|  | // The layer was rejected by the atlas (even though we know it is | 
|  | // plausibly atlas-able). See if a plot can be purged and try again. | 
|  | if (!this->purgePlot()) { | 
|  | break;  // We weren't able to purge any plots | 
|  | } | 
|  | } | 
|  |  | 
|  | if (pictInfo->fPlotUsage.isEmpty()) { | 
|  | fPictureHash.remove(pictInfo->fPictureID); | 
|  | SkDELETE(pictInfo); | 
|  | } | 
|  | } | 
|  |  | 
|  | return false; | 
|  | } | 
|  |  | 
|  | bool GrLayerCache::lock(GrCachedLayer* layer, const GrSurfaceDesc& desc, bool* needsRendering) { | 
|  | if (layer->locked()) { | 
|  | // This layer is already locked | 
|  | *needsRendering = false; | 
|  | return true; | 
|  | } | 
|  |  | 
|  | // TODO: make the test for exact match depend on the image filters themselves | 
|  | GrContext::ScratchTexMatch usage = GrContext::kApprox_ScratchTexMatch; | 
|  | if (layer->fFilter) { | 
|  | usage = GrContext::kExact_ScratchTexMatch; | 
|  | } | 
|  |  | 
|  | SkAutoTUnref<GrTexture> tex(fContext->refScratchTexture(desc, usage)); | 
|  | if (!tex) { | 
|  | return false; | 
|  | } | 
|  |  | 
|  | layer->setTexture(tex, SkIRect::MakeWH(desc.fWidth, desc.fHeight)); | 
|  | layer->setLocked(true); | 
|  | *needsRendering = true; | 
|  | return true; | 
|  | } | 
|  |  | 
|  | void GrLayerCache::unlock(GrCachedLayer* layer) { | 
|  | SkDEBUGCODE(GrAutoValidateLayer avl(fAtlas ? fAtlas->getTexture() : NULL, layer);) | 
|  |  | 
|  | if (NULL == layer || !layer->locked()) { | 
|  | // invalid or not locked | 
|  | return; | 
|  | } | 
|  |  | 
|  | if (layer->isAtlased()) { | 
|  | const int plotID = layer->plot()->id(); | 
|  |  | 
|  | this->decPlotLock(plotID); | 
|  | // At this point we could aggressively clear out un-locked plots but | 
|  | // by delaying we may be able to reuse some of the atlased layers later. | 
|  | #if !GR_CACHE_HOISTED_LAYERS | 
|  | // This testing code aggressively removes the atlased layers. This | 
|  | // can be used to separate the performance contribution of less | 
|  | // render target pingponging from that due to the re-use of cached layers | 
|  | GrPictureInfo* pictInfo = fPictureHash.find(layer->pictureID()); | 
|  | SkASSERT(pictInfo); | 
|  |  | 
|  | pictInfo->decPlotUsage(plotID); | 
|  |  | 
|  | if (0 == pictInfo->plotUsage(plotID)) { | 
|  | GrAtlas::RemovePlot(&pictInfo->fPlotUsage, layer->plot()); | 
|  |  | 
|  | if (pictInfo->fPlotUsage.isEmpty()) { | 
|  | fPictureHash.remove(pictInfo->fPictureID); | 
|  | SkDELETE(pictInfo); | 
|  | } | 
|  | } | 
|  |  | 
|  | layer->setPlot(NULL); | 
|  | layer->setTexture(NULL, SkIRect::MakeEmpty()); | 
|  | #endif | 
|  |  | 
|  | } else { | 
|  | layer->setTexture(NULL, SkIRect::MakeEmpty()); | 
|  | } | 
|  |  | 
|  | layer->setLocked(false); | 
|  | } | 
|  |  | 
|  | #ifdef SK_DEBUG | 
|  | void GrLayerCache::validate() const { | 
|  | int plotLocks[kNumPlotsX * kNumPlotsY]; | 
|  | memset(plotLocks, 0, sizeof(plotLocks)); | 
|  |  | 
|  | SkTDynamicHash<GrCachedLayer, GrCachedLayer::Key>::ConstIter iter(&fLayerHash); | 
|  | for (; !iter.done(); ++iter) { | 
|  | const GrCachedLayer* layer = &(*iter); | 
|  |  | 
|  | layer->validate(fAtlas.get() ? fAtlas->getTexture() : NULL); | 
|  |  | 
|  | const GrPictureInfo* pictInfo = fPictureHash.find(layer->pictureID()); | 
|  | if (!pictInfo) { | 
|  | // If there is no picture info for this picture then all of its | 
|  | // layers should be non-atlased. | 
|  | SkASSERT(!layer->isAtlased()); | 
|  | } | 
|  |  | 
|  | if (layer->plot()) { | 
|  | SkASSERT(pictInfo); | 
|  | SkASSERT(pictInfo->fPictureID == layer->pictureID()); | 
|  |  | 
|  | SkASSERT(pictInfo->fPlotUsage.contains(layer->plot())); | 
|  | #if !GR_CACHE_HOISTED_LAYERS | 
|  | SkASSERT(pictInfo->plotUsage(layer->plot()->id()) > 0); | 
|  | #endif | 
|  |  | 
|  | if (layer->locked()) { | 
|  | plotLocks[layer->plot()->id()]++; | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | for (int i = 0; i < kNumPlotsX*kNumPlotsY; ++i) { | 
|  | SkASSERT(plotLocks[i] == fPlotLocks[i]); | 
|  | } | 
|  | } | 
|  |  | 
|  | class GrAutoValidateCache : ::SkNoncopyable { | 
|  | public: | 
|  | explicit GrAutoValidateCache(GrLayerCache* cache) | 
|  | : fCache(cache) { | 
|  | fCache->validate(); | 
|  | } | 
|  | ~GrAutoValidateCache() { | 
|  | fCache->validate(); | 
|  | } | 
|  | private: | 
|  | GrLayerCache* fCache; | 
|  | }; | 
|  | #endif | 
|  |  | 
|  | void GrLayerCache::purge(uint32_t pictureID) { | 
|  |  | 
|  | SkDEBUGCODE(GrAutoValidateCache avc(this);) | 
|  |  | 
|  | // We need to find all the layers associated with 'picture' and remove them. | 
|  | SkTDArray<GrCachedLayer*> toBeRemoved; | 
|  |  | 
|  | SkTDynamicHash<GrCachedLayer, GrCachedLayer::Key>::Iter iter(&fLayerHash); | 
|  | for (; !iter.done(); ++iter) { | 
|  | if (pictureID == (*iter).pictureID()) { | 
|  | *toBeRemoved.append() = &(*iter); | 
|  | } | 
|  | } | 
|  |  | 
|  | for (int i = 0; i < toBeRemoved.count(); ++i) { | 
|  | SkASSERT(0 == toBeRemoved[i]->uses()); | 
|  | this->unlock(toBeRemoved[i]); | 
|  | fLayerHash.remove(GrCachedLayer::GetKey(*toBeRemoved[i])); | 
|  | SkDELETE(toBeRemoved[i]); | 
|  | } | 
|  |  | 
|  | GrPictureInfo* pictInfo = fPictureHash.find(pictureID); | 
|  | if (pictInfo) { | 
|  | fPictureHash.remove(pictureID); | 
|  | SkDELETE(pictInfo); | 
|  | } | 
|  | } | 
|  |  | 
|  | bool GrLayerCache::purgePlot() { | 
|  | SkDEBUGCODE(GrAutoValidateCache avc(this);) | 
|  | SkASSERT(fAtlas); | 
|  |  | 
|  | GrAtlas::PlotIter iter; | 
|  | GrPlot* plot; | 
|  | for (plot = fAtlas->iterInit(&iter, GrAtlas::kLRUFirst_IterOrder); | 
|  | plot; | 
|  | plot = iter.prev()) { | 
|  | if (fPlotLocks[plot->id()] > 0) { | 
|  | continue; | 
|  | } | 
|  |  | 
|  | this->purgePlot(plot); | 
|  | return true; | 
|  | } | 
|  |  | 
|  | return false; | 
|  | } | 
|  |  | 
|  | void GrLayerCache::purgePlot(GrPlot* plot) { | 
|  | SkASSERT(0 == fPlotLocks[plot->id()]); | 
|  |  | 
|  | // We need to find all the layers in 'plot' and remove them. | 
|  | SkTDArray<GrCachedLayer*> toBeRemoved; | 
|  |  | 
|  | SkTDynamicHash<GrCachedLayer, GrCachedLayer::Key>::Iter iter(&fLayerHash); | 
|  | for (; !iter.done(); ++iter) { | 
|  | if (plot == (*iter).plot()) { | 
|  | *toBeRemoved.append() = &(*iter); | 
|  | } | 
|  | } | 
|  |  | 
|  | for (int i = 0; i < toBeRemoved.count(); ++i) { | 
|  | SkASSERT(0 == toBeRemoved[i]->uses()); | 
|  | SkASSERT(!toBeRemoved[i]->locked()); | 
|  |  | 
|  | uint32_t pictureIDToRemove = toBeRemoved[i]->pictureID(); | 
|  |  | 
|  | // Aggressively remove layers and, if it becomes totally uncached, delete the picture info | 
|  | fLayerHash.remove(GrCachedLayer::GetKey(*toBeRemoved[i])); | 
|  | SkDELETE(toBeRemoved[i]); | 
|  |  | 
|  | GrPictureInfo* pictInfo = fPictureHash.find(pictureIDToRemove); | 
|  | if (pictInfo) { | 
|  | #if !GR_CACHE_HOISTED_LAYERS | 
|  | SkASSERT(0 == pictInfo->plotUsage(plot->id())); | 
|  | #endif | 
|  | GrAtlas::RemovePlot(&pictInfo->fPlotUsage, plot); | 
|  |  | 
|  | if (pictInfo->fPlotUsage.isEmpty()) { | 
|  | fPictureHash.remove(pictInfo->fPictureID); | 
|  | SkDELETE(pictInfo); | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | plot->resetRects(); | 
|  | } | 
|  |  | 
|  | #if !GR_CACHE_HOISTED_LAYERS | 
|  | void GrLayerCache::purgeAll() { | 
|  | if (!fAtlas) { | 
|  | return; | 
|  | } | 
|  |  | 
|  | GrAtlas::PlotIter iter; | 
|  | GrPlot* plot; | 
|  | for (plot = fAtlas->iterInit(&iter, GrAtlas::kLRUFirst_IterOrder); | 
|  | plot; | 
|  | plot = iter.prev()) { | 
|  | SkASSERT(0 == fPlotLocks[plot->id()]); | 
|  |  | 
|  | this->purgePlot(plot); | 
|  | } | 
|  |  | 
|  | SkASSERT(0 == fPictureHash.count()); | 
|  |  | 
|  | fContext->discardRenderTarget(fAtlas->getTexture()->asRenderTarget()); | 
|  | } | 
|  | #endif | 
|  |  | 
|  | void GrLayerCache::processDeletedPictures() { | 
|  | SkTArray<SkPicture::DeletionMessage> deletedPictures; | 
|  | fPictDeletionInbox.poll(&deletedPictures); | 
|  |  | 
|  | for (int i = 0; i < deletedPictures.count(); i++) { | 
|  | this->purge(deletedPictures[i].fUniqueID); | 
|  | } | 
|  | } | 
|  |  | 
|  | #ifdef SK_DEVELOPER | 
|  | void GrLayerCache::writeLayersToDisk(const SkString& dirName) { | 
|  |  | 
|  | if (fAtlas) { | 
|  | GrTexture* atlasTexture = fAtlas->getTexture(); | 
|  | if (NULL != atlasTexture) { | 
|  | SkString fileName(dirName); | 
|  | fileName.append("\\atlas.png"); | 
|  |  | 
|  | atlasTexture->surfacePriv().savePixels(fileName.c_str()); | 
|  | } | 
|  | } | 
|  |  | 
|  | SkTDynamicHash<GrCachedLayer, GrCachedLayer::Key>::Iter iter(&fLayerHash); | 
|  | for (; !iter.done(); ++iter) { | 
|  | GrCachedLayer* layer = &(*iter); | 
|  |  | 
|  | if (layer->isAtlased() || !layer->texture()) { | 
|  | continue; | 
|  | } | 
|  |  | 
|  | SkString fileName(dirName); | 
|  | fileName.appendf("\\%d", layer->fKey.pictureID()); | 
|  | for (int i = 0; i < layer->fKey.keySize(); ++i) { | 
|  | fileName.appendf("-%d", layer->fKey.key()[i]); | 
|  | } | 
|  | fileName.appendf(".png"); | 
|  |  | 
|  | layer->texture()->surfacePriv().savePixels(fileName.c_str()); | 
|  | } | 
|  | } | 
|  | #endif |