epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | |
reed@google.com | ac10a2d | 2010-12-22 21:39:39 +0000 | [diff] [blame] | 2 | /* |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 3 | * Copyright 2010 Google Inc. |
| 4 | * |
| 5 | * Use of this source code is governed by a BSD-style license that can be |
| 6 | * found in the LICENSE file. |
reed@google.com | ac10a2d | 2010-12-22 21:39:39 +0000 | [diff] [blame] | 7 | */ |
| 8 | |
| 9 | |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 10 | |
robertphillips@google.com | 5088eb4 | 2012-06-28 20:59:13 +0000 | [diff] [blame] | 11 | #include "SkGrPixelRef.h" |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 12 | |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 13 | #include "GrContext.h" |
reed@google.com | ac10a2d | 2010-12-22 21:39:39 +0000 | [diff] [blame] | 14 | #include "GrTexture.h" |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 15 | #include "SkBitmapCache.h" |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 16 | #include "SkGr.h" |
bsalomon@google.com | 669fdc4 | 2011-04-05 17:08:27 +0000 | [diff] [blame] | 17 | #include "SkRect.h" |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 18 | |
| 19 | // since we call lockPixels recursively on fBitmap, we need a distinct mutex, |
| 20 | // to avoid deadlock with the default one provided by SkPixelRef. |
digit@google.com | 1771cbf | 2012-01-26 21:26:40 +0000 | [diff] [blame] | 21 | SK_DECLARE_STATIC_MUTEX(gROLockPixelsPixelRefMutex); |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 22 | |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 23 | SkROLockPixelsPixelRef::SkROLockPixelsPixelRef(const SkImageInfo& info) |
| 24 | : INHERITED(info, &gROLockPixelsPixelRefMutex) {} |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 25 | |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 26 | SkROLockPixelsPixelRef::~SkROLockPixelsPixelRef() {} |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 27 | |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame] | 28 | bool SkROLockPixelsPixelRef::onNewLockPixels(LockRec* rec) { |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 29 | fBitmap.reset(); |
| 30 | // SkDebugf("---------- calling readpixels in support of lockpixels\n"); |
| 31 | if (!this->onReadPixels(&fBitmap, NULL)) { |
| 32 | SkDebugf("SkROLockPixelsPixelRef::onLockPixels failed!\n"); |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame] | 33 | return false; |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 34 | } |
| 35 | fBitmap.lockPixels(); |
reed@google.com | d0419b1 | 2014-01-06 17:08:27 +0000 | [diff] [blame] | 36 | if (NULL == fBitmap.getPixels()) { |
| 37 | return false; |
| 38 | } |
| 39 | |
| 40 | rec->fPixels = fBitmap.getPixels(); |
| 41 | rec->fColorTable = NULL; |
| 42 | rec->fRowBytes = fBitmap.rowBytes(); |
| 43 | return true; |
reed@google.com | 9c49bc3 | 2011-07-07 13:42:37 +0000 | [diff] [blame] | 44 | } |
| 45 | |
| 46 | void SkROLockPixelsPixelRef::onUnlockPixels() { |
| 47 | fBitmap.unlockPixels(); |
| 48 | } |
| 49 | |
| 50 | bool SkROLockPixelsPixelRef::onLockPixelsAreWritable() const { |
| 51 | return false; |
| 52 | } |
| 53 | |
| 54 | /////////////////////////////////////////////////////////////////////////////// |
bsalomon@google.com | 669fdc4 | 2011-04-05 17:08:27 +0000 | [diff] [blame] | 55 | |
junov | 96c118e | 2014-09-26 13:09:47 -0700 | [diff] [blame] | 56 | static SkGrPixelRef* copy_to_new_texture_pixelref(GrTexture* texture, SkColorType dstCT, |
junov | 2bb5210 | 2014-09-29 10:18:59 -0700 | [diff] [blame] | 57 | const SkIRect* subset) { |
reed | e4538f5 | 2014-06-11 06:09:50 -0700 | [diff] [blame] | 58 | if (NULL == texture || kUnknown_SkColorType == dstCT) { |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 59 | return NULL; |
| 60 | } |
| 61 | GrContext* context = texture->getContext(); |
| 62 | if (NULL == context) { |
| 63 | return NULL; |
| 64 | } |
bsalomon | f2703d8 | 2014-10-28 14:33:06 -0700 | [diff] [blame] | 65 | GrSurfaceDesc desc; |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 66 | |
bsalomon | f80bfed | 2014-10-07 05:56:02 -0700 | [diff] [blame] | 67 | SkIRect srcRect; |
| 68 | |
| 69 | if (!subset) { |
| 70 | desc.fWidth = texture->width(); |
| 71 | desc.fHeight = texture->height(); |
| 72 | srcRect = SkIRect::MakeWH(texture->width(), texture->height()); |
| 73 | } else { |
scroggo@google.com | a2a3192 | 2012-12-07 19:14:45 +0000 | [diff] [blame] | 74 | SkASSERT(SkIRect::MakeWH(texture->width(), texture->height()).contains(*subset)); |
| 75 | // Create a new texture that is the size of subset. |
| 76 | desc.fWidth = subset->width(); |
| 77 | desc.fHeight = subset->height(); |
bsalomon | f80bfed | 2014-10-07 05:56:02 -0700 | [diff] [blame] | 78 | srcRect = *subset; |
scroggo@google.com | a2a3192 | 2012-12-07 19:14:45 +0000 | [diff] [blame] | 79 | } |
bsalomon | f2703d8 | 2014-10-28 14:33:06 -0700 | [diff] [blame] | 80 | desc.fFlags = kRenderTarget_GrSurfaceFlag | kNoStencil_GrSurfaceFlag; |
reed | e4538f5 | 2014-06-11 06:09:50 -0700 | [diff] [blame] | 81 | desc.fConfig = SkImageInfo2GrPixelConfig(dstCT, kPremul_SkAlphaType); |
skia.committer@gmail.com | 96f5fa0 | 2013-12-16 07:01:40 +0000 | [diff] [blame] | 82 | |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 83 | GrTexture* dst = context->createUncachedTexture(desc, NULL, 0); |
| 84 | if (NULL == dst) { |
| 85 | return NULL; |
| 86 | } |
| 87 | |
junov | 96c118e | 2014-09-26 13:09:47 -0700 | [diff] [blame] | 88 | // Blink is relying on the above copy being sent to GL immediately in the case when the source |
bsalomon | f80bfed | 2014-10-07 05:56:02 -0700 | [diff] [blame] | 89 | // is a WebGL canvas backing store. We could have a TODO to remove this flush flag, but we have |
| 90 | // a larger TODO to remove SkGrPixelRef entirely. |
| 91 | context->copySurface(dst->asRenderTarget(), texture, srcRect, SkIPoint::Make(0,0), |
| 92 | GrContext::kFlushWrites_PixelOp); |
| 93 | |
reed | e4538f5 | 2014-06-11 06:09:50 -0700 | [diff] [blame] | 94 | SkImageInfo info = SkImageInfo::Make(desc.fWidth, desc.fHeight, dstCT, kPremul_SkAlphaType); |
reed@google.com | bf79023 | 2013-12-13 19:45:58 +0000 | [diff] [blame] | 95 | SkGrPixelRef* pixelRef = SkNEW_ARGS(SkGrPixelRef, (info, dst)); |
commit-bot@chromium.org | a4de8c2 | 2013-09-09 13:38:37 +0000 | [diff] [blame] | 96 | SkSafeUnref(dst); |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 97 | return pixelRef; |
| 98 | } |
| 99 | |
| 100 | /////////////////////////////////////////////////////////////////////////////// |
| 101 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 102 | SkGrPixelRef::SkGrPixelRef(const SkImageInfo& info, GrSurface* surface) : INHERITED(info) { |
bsalomon | 7187593 | 2014-08-29 08:08:35 -0700 | [diff] [blame] | 103 | // For surfaces that are both textures and render targets, the texture owns the |
| 104 | // render target but not vice versa. So we ref the texture to keep both alive for |
| 105 | // the lifetime of this pixel ref. |
| 106 | fSurface = SkSafeRef(surface->asTexture()); |
reed@google.com | 6186787 | 2013-12-09 13:41:06 +0000 | [diff] [blame] | 107 | if (NULL == fSurface) { |
bsalomon | 7187593 | 2014-08-29 08:08:35 -0700 | [diff] [blame] | 108 | fSurface = SkSafeRef(surface); |
reed@google.com | 6186787 | 2013-12-09 13:41:06 +0000 | [diff] [blame] | 109 | } |
commit-bot@chromium.org | 8e0993d | 2014-01-28 15:16:45 +0000 | [diff] [blame] | 110 | |
| 111 | if (fSurface) { |
reed | e5ea500 | 2014-09-03 11:54:58 -0700 | [diff] [blame] | 112 | SkASSERT(info.width() <= fSurface->width()); |
| 113 | SkASSERT(info.height() <= fSurface->height()); |
commit-bot@chromium.org | 8e0993d | 2014-01-28 15:16:45 +0000 | [diff] [blame] | 114 | } |
reed@google.com | 6186787 | 2013-12-09 13:41:06 +0000 | [diff] [blame] | 115 | } |
| 116 | |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 117 | SkGrPixelRef::~SkGrPixelRef() { |
commit-bot@chromium.org | a4de8c2 | 2013-09-09 13:38:37 +0000 | [diff] [blame] | 118 | SkSafeUnref(fSurface); |
bsalomon@google.com | 669fdc4 | 2011-04-05 17:08:27 +0000 | [diff] [blame] | 119 | } |
| 120 | |
commit-bot@chromium.org | b8d00db | 2013-06-26 19:18:23 +0000 | [diff] [blame] | 121 | GrTexture* SkGrPixelRef::getTexture() { |
bsalomon | 49f085d | 2014-09-05 13:34:00 -0700 | [diff] [blame] | 122 | if (fSurface) { |
commit-bot@chromium.org | b8d00db | 2013-06-26 19:18:23 +0000 | [diff] [blame] | 123 | return fSurface->asTexture(); |
bsalomon@google.com | 669fdc4 | 2011-04-05 17:08:27 +0000 | [diff] [blame] | 124 | } |
| 125 | return NULL; |
| 126 | } |
| 127 | |
reed | e4538f5 | 2014-06-11 06:09:50 -0700 | [diff] [blame] | 128 | SkPixelRef* SkGrPixelRef::deepCopy(SkColorType dstCT, const SkIRect* subset) { |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 129 | if (NULL == fSurface) { |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 130 | return NULL; |
| 131 | } |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 132 | |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 133 | // Note that when copying a render-target-backed pixel ref, we |
| 134 | // return a texture-backed pixel ref instead. This is because |
| 135 | // render-target pixel refs are usually created in conjunction with |
| 136 | // a GrTexture owned elsewhere (e.g., SkGpuDevice), and cannot live |
| 137 | // independently of that texture. Texture-backed pixel refs, on the other |
| 138 | // hand, own their GrTextures, and are thus self-contained. |
junov | 96c118e | 2014-09-26 13:09:47 -0700 | [diff] [blame] | 139 | return copy_to_new_texture_pixelref(fSurface->asTexture(), dstCT, subset); |
senorblanco@chromium.org | ef843cd | 2011-12-02 19:11:17 +0000 | [diff] [blame] | 140 | } |
| 141 | |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 142 | static bool tryAllocBitmapPixels(SkBitmap* bitmap) { |
| 143 | SkBitmap::Allocator* allocator = SkBitmapCache::GetAllocator(); |
| 144 | if (NULL != allocator) { |
| 145 | return allocator->allocPixelRef(bitmap, 0); |
| 146 | } else { |
| 147 | // DiscardableMemory is not available, fallback to default allocator |
| 148 | return bitmap->tryAllocPixels(); |
| 149 | } |
| 150 | } |
| 151 | |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 152 | bool SkGrPixelRef::onReadPixels(SkBitmap* dst, const SkIRect* subset) { |
commit-bot@chromium.org | 089a780 | 2014-05-02 21:38:22 +0000 | [diff] [blame] | 153 | if (NULL == fSurface || fSurface->wasDestroyed()) { |
bsalomon@google.com | 669fdc4 | 2011-04-05 17:08:27 +0000 | [diff] [blame] | 154 | return false; |
| 155 | } |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 156 | |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 157 | SkIRect bounds; |
bsalomon | 49f085d | 2014-09-05 13:34:00 -0700 | [diff] [blame] | 158 | if (subset) { |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 159 | bounds = *subset; |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 160 | } else { |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 161 | bounds = SkIRect::MakeWH(this->info().width(), this->info().height()); |
robertphillips@google.com | d881bc1 | 2012-06-28 20:02:39 +0000 | [diff] [blame] | 162 | } |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 163 | |
| 164 | //Check the cache |
| 165 | if(!SkBitmapCache::Find(this->getGenerationID(), bounds, dst)) { |
| 166 | //Cache miss |
| 167 | |
| 168 | SkBitmap cachedBitmap; |
| 169 | cachedBitmap.setInfo(this->info().makeWH(bounds.width(), bounds.height())); |
| 170 | |
| 171 | // If we can't alloc the pixels, then fail |
| 172 | if (!tryAllocBitmapPixels(&cachedBitmap)) { |
| 173 | return false; |
| 174 | } |
| 175 | |
| 176 | // Try to read the pixels from the surface |
| 177 | void* buffer = cachedBitmap.getPixels(); |
| 178 | bool readPixelsOk = fSurface->readPixels(bounds.fLeft, bounds.fTop, |
| 179 | bounds.width(), bounds.height(), |
bsalomon@google.com | fec0bc3 | 2013-02-07 14:43:04 +0000 | [diff] [blame] | 180 | kSkia8888_GrPixelConfig, |
piotaixr | 0e97705 | 2014-09-17 16:24:04 -0700 | [diff] [blame] | 181 | buffer, cachedBitmap.rowBytes()); |
| 182 | |
| 183 | if (!readPixelsOk) { |
| 184 | return false; |
| 185 | } |
| 186 | |
| 187 | // If we are here, pixels were read correctly from the surface. |
| 188 | cachedBitmap.setImmutable(); |
| 189 | //Add to the cache |
| 190 | SkBitmapCache::Add(this->getGenerationID(), bounds, cachedBitmap); |
| 191 | |
| 192 | dst->swap(cachedBitmap); |
| 193 | } |
| 194 | |
| 195 | return true; |
| 196 | |
reed@google.com | 4281d65 | 2011-04-08 18:54:20 +0000 | [diff] [blame] | 197 | } |