reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
| 8 | #include "SkImage_Base.h" |
| 9 | #include "SkBitmap.h" |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 10 | #include "SkCanvas.h" |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 11 | #include "SkData.h" |
| 12 | #include "SkImageCacherator.h" |
| 13 | #include "SkImagePriv.h" |
| 14 | #include "SkPixelRef.h" |
| 15 | #include "SkSurface.h" |
| 16 | |
| 17 | class SkImage_Generator : public SkImage_Base { |
| 18 | public: |
| 19 | SkImage_Generator(SkImageCacherator* cache) |
reed | af3fbfc | 2015-10-04 11:28:36 -0700 | [diff] [blame] | 20 | : INHERITED(cache->info().width(), cache->info().height(), cache->uniqueID()) |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 21 | , fCache(cache) // take ownership |
| 22 | {} |
| 23 | |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 24 | bool onReadPixels(const SkImageInfo&, void*, size_t, int srcX, int srcY, CachingHint) const override; |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 25 | const void* onPeekPixels(SkImageInfo*, size_t* /*rowBytes*/) const override; |
bsalomon | 1cf6f9b | 2015-12-08 10:53:43 -0800 | [diff] [blame] | 26 | SkImageCacherator* peekCacherator() const override { return fCache; } |
reed | 05dd251 | 2016-01-05 09:16:19 -0800 | [diff] [blame] | 27 | SkData* onRefEncoded(GrContext*) const override; |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 28 | bool isOpaque() const override { return fCache->info().isOpaque(); } |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 29 | SkImage* onNewSubset(const SkIRect&) const override; |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 30 | bool getROPixels(SkBitmap*, CachingHint) const override; |
bsalomon | afa95e2 | 2015-10-12 10:39:46 -0700 | [diff] [blame] | 31 | GrTexture* asTextureRef(GrContext*, const GrTextureParams&) const override; |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 32 | bool onIsLazyGenerated() const override { return true; } |
| 33 | |
| 34 | private: |
| 35 | SkAutoTDelete<SkImageCacherator> fCache; |
| 36 | |
| 37 | typedef SkImage_Base INHERITED; |
| 38 | }; |
| 39 | |
| 40 | /////////////////////////////////////////////////////////////////////////////// |
| 41 | |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 42 | bool SkImage_Generator::onReadPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRB, |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 43 | int srcX, int srcY, CachingHint chint) const { |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 44 | SkBitmap bm; |
reed | 6868c3f | 2015-11-24 11:44:47 -0800 | [diff] [blame] | 45 | if (kDisallow_CachingHint == chint) { |
| 46 | if (fCache->lockAsBitmapOnlyIfAlreadyCached(&bm)) { |
| 47 | return bm.readPixels(dstInfo, dstPixels, dstRB, srcX, srcY); |
| 48 | } else { |
| 49 | // Try passing the caller's buffer directly down to the generator. If this fails we |
| 50 | // may still succeed in the general case, as the generator may prefer some other |
| 51 | // config, which we could then convert via SkBitmap::readPixels. |
| 52 | if (fCache->directGeneratePixels(dstInfo, dstPixels, dstRB, srcX, srcY)) { |
| 53 | return true; |
| 54 | } |
| 55 | // else fall through |
| 56 | } |
| 57 | } |
| 58 | |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 59 | if (this->getROPixels(&bm, chint)) { |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 60 | return bm.readPixels(dstInfo, dstPixels, dstRB, srcX, srcY); |
| 61 | } |
| 62 | return false; |
| 63 | } |
| 64 | |
| 65 | const void* SkImage_Generator::onPeekPixels(SkImageInfo* infoPtr, size_t* rowBytesPtr) const { |
| 66 | return NULL; |
| 67 | } |
| 68 | |
reed | 05dd251 | 2016-01-05 09:16:19 -0800 | [diff] [blame] | 69 | SkData* SkImage_Generator::onRefEncoded(GrContext* ctx) const { |
| 70 | return fCache->refEncoded(ctx); |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 71 | } |
| 72 | |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 73 | bool SkImage_Generator::getROPixels(SkBitmap* bitmap, CachingHint chint) const { |
| 74 | return fCache->lockAsBitmap(bitmap, this, chint); |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 75 | } |
| 76 | |
bsalomon | afa95e2 | 2015-10-12 10:39:46 -0700 | [diff] [blame] | 77 | GrTexture* SkImage_Generator::asTextureRef(GrContext* ctx, const GrTextureParams& params) const { |
| 78 | return fCache->lockAsTexture(ctx, params, this); |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 79 | } |
| 80 | |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 81 | SkImage* SkImage_Generator::onNewSubset(const SkIRect& subset) const { |
| 82 | // TODO: make this lazy, by wrapping the subset inside a new generator or something |
| 83 | // For now, we do effectively what we did before, make it a raster |
| 84 | |
| 85 | const SkImageInfo info = SkImageInfo::MakeN32(subset.width(), subset.height(), |
| 86 | this->isOpaque() ? kOpaque_SkAlphaType : kPremul_SkAlphaType); |
| 87 | SkAutoTUnref<SkSurface> surface(SkSurface::NewRaster(info)); |
| 88 | if (!surface) { |
| 89 | return nullptr; |
| 90 | } |
| 91 | surface->getCanvas()->clear(0); |
| 92 | surface->getCanvas()->drawImage(this, SkIntToScalar(-subset.x()), SkIntToScalar(-subset.y()), |
| 93 | nullptr); |
| 94 | return surface->newImageSnapshot(); |
| 95 | } |
| 96 | |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 97 | SkImage* SkImage::NewFromGenerator(SkImageGenerator* generator, const SkIRect* subset) { |
| 98 | SkImageCacherator* cache = SkImageCacherator::NewFromGenerator(generator, subset); |
| 99 | if (!cache) { |
| 100 | return nullptr; |
| 101 | } |
mdempsky | 106b124 | 2015-09-22 06:10:35 -0700 | [diff] [blame] | 102 | return new SkImage_Generator(cache); |
reed | 85d9178 | 2015-09-10 14:33:38 -0700 | [diff] [blame] | 103 | } |