reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2012 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" |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 9 | #include "SkBitmap.h" |
reed | 4c1abdc | 2016-08-05 14:50:19 -0700 | [diff] [blame] | 10 | #include "SkBitmapProcShader.h" |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 11 | #include "SkCanvas.h" |
Matt Sarett | 5b1dba8 | 2017-04-03 11:42:52 -0400 | [diff] [blame] | 12 | #include "SkColorSpaceXform_Base.h" |
Matt Sarett | dc79270 | 2017-06-08 09:37:01 -0400 | [diff] [blame] | 13 | #include "SkColorSpaceXformImageGenerator.h" |
Matt Sarett | 5b1dba8 | 2017-04-03 11:42:52 -0400 | [diff] [blame] | 14 | #include "SkColorSpaceXformPriv.h" |
reed | 759373a | 2015-07-03 21:01:10 -0700 | [diff] [blame] | 15 | #include "SkColorTable.h" |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 16 | #include "SkConvertPixels.h" |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 17 | #include "SkData.h" |
Matt Sarett | d21fd4b | 2017-06-08 09:08:34 -0400 | [diff] [blame] | 18 | #include "SkImageInfoPriv.h" |
kkinnunen | 73953e7 | 2015-02-23 22:12:12 -0800 | [diff] [blame] | 19 | #include "SkImagePriv.h" |
| 20 | #include "SkPixelRef.h" |
reed | 4af267b | 2014-11-21 08:46:37 -0800 | [diff] [blame] | 21 | #include "SkSurface.h" |
Matt Sarett | 0887c70 | 2017-03-21 13:05:41 -0400 | [diff] [blame] | 22 | #include "SkTLazy.h" |
Matt Sarett | 5b1dba8 | 2017-04-03 11:42:52 -0400 | [diff] [blame] | 23 | #include "SkUnPreMultiplyPriv.h" |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 24 | |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 25 | #if SK_SUPPORT_GPU |
| 26 | #include "GrContext.h" |
Brian Osman | e8e5458 | 2016-11-28 10:06:27 -0500 | [diff] [blame] | 27 | #include "GrTextureAdjuster.h" |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 28 | #include "SkGr.h" |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 29 | #endif |
| 30 | |
halcanary | 8d3ad5a | 2016-03-24 09:18:14 -0700 | [diff] [blame] | 31 | // fixes https://bug.skia.org/5096 |
| 32 | static bool is_not_subset(const SkBitmap& bm) { |
| 33 | SkASSERT(bm.pixelRef()); |
Matt Sarett | a6e976a | 2017-04-28 15:43:35 -0400 | [diff] [blame] | 34 | SkISize dim = SkISize::Make(bm.pixelRef()->width(), bm.pixelRef()->height()); |
halcanary | 8d3ad5a | 2016-03-24 09:18:14 -0700 | [diff] [blame] | 35 | SkASSERT(dim != bm.dimensions() || bm.pixelRefOrigin().isZero()); |
| 36 | return dim == bm.dimensions(); |
| 37 | } |
| 38 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 39 | class SkImage_Raster : public SkImage_Base { |
| 40 | public: |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 41 | static bool ValidArgs(const Info& info, size_t rowBytes, size_t* minSize) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 42 | const int maxDimension = SK_MaxS32 >> 2; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 43 | |
reed | b2497c2 | 2014-12-31 12:31:43 -0800 | [diff] [blame] | 44 | if (info.width() <= 0 || info.height() <= 0) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 45 | return false; |
| 46 | } |
reed | e5ea500 | 2014-09-03 11:54:58 -0700 | [diff] [blame] | 47 | if (info.width() > maxDimension || info.height() > maxDimension) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 48 | return false; |
| 49 | } |
reed | e5ea500 | 2014-09-03 11:54:58 -0700 | [diff] [blame] | 50 | if ((unsigned)info.colorType() > (unsigned)kLastEnum_SkColorType) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 51 | return false; |
| 52 | } |
reed | e5ea500 | 2014-09-03 11:54:58 -0700 | [diff] [blame] | 53 | if ((unsigned)info.alphaType() > (unsigned)kLastEnum_SkAlphaType) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 54 | return false; |
| 55 | } |
| 56 | |
commit-bot@chromium.org | a3264e5 | 2014-05-30 13:26:10 +0000 | [diff] [blame] | 57 | if (kUnknown_SkColorType == info.colorType()) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 58 | return false; |
| 59 | } |
Mike Reed | cd284c5 | 2017-09-29 15:22:56 -0400 | [diff] [blame^] | 60 | if (!info.validRowBytes(rowBytes)) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 61 | return false; |
| 62 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 63 | |
Mike Reed | cd284c5 | 2017-09-29 15:22:56 -0400 | [diff] [blame^] | 64 | size_t size = info.computeByteSize(rowBytes); |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 65 | if (0 == size) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 66 | return false; |
| 67 | } |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 68 | |
| 69 | if (minSize) { |
| 70 | *minSize = size; |
| 71 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 72 | return true; |
| 73 | } |
| 74 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 75 | SkImage_Raster(const SkImageInfo&, sk_sp<SkData>, size_t rb, |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 76 | uint32_t id = kNeedNewImageUniqueID); |
Brian Salomon | d3b6597 | 2017-03-22 12:05:03 -0400 | [diff] [blame] | 77 | ~SkImage_Raster() override; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 78 | |
herb | a7c9d63 | 2016-04-19 12:30:22 -0700 | [diff] [blame] | 79 | SkImageInfo onImageInfo() const override { |
| 80 | return fBitmap.info(); |
| 81 | } |
brianosman | 69c166d | 2016-08-17 14:01:05 -0700 | [diff] [blame] | 82 | SkAlphaType onAlphaType() const override { |
| 83 | return fBitmap.alphaType(); |
| 84 | } |
herb | a7c9d63 | 2016-04-19 12:30:22 -0700 | [diff] [blame] | 85 | |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 86 | bool onReadPixels(const SkImageInfo&, void*, size_t, int srcX, int srcY, CachingHint) const override; |
reed | 6ceeebd | 2016-03-09 14:26:26 -0800 | [diff] [blame] | 87 | bool onPeekPixels(SkPixmap*) const override; |
reed | f1ac182 | 2016-08-01 11:24:14 -0700 | [diff] [blame] | 88 | const SkBitmap* onPeekBitmap() const override { return &fBitmap; } |
| 89 | |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 90 | #if SK_SUPPORT_GPU |
Brian Salomon | 2bbdcc4 | 2017-09-07 12:36:34 -0400 | [diff] [blame] | 91 | sk_sp<GrTextureProxy> asTextureProxyRef(GrContext*, const GrSamplerState&, SkColorSpace*, |
| 92 | sk_sp<SkColorSpace>*, |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 93 | SkScalar scaleAdjust[2]) const override; |
| 94 | #endif |
| 95 | |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 96 | bool getROPixels(SkBitmap*, SkColorSpace* dstColorSpace, CachingHint) const override; |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 97 | sk_sp<SkImage> onMakeSubset(const SkIRect&) const override; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 98 | |
reed@google.com | 97af1a6 | 2012-08-28 12:19:02 +0000 | [diff] [blame] | 99 | SkPixelRef* getPixelRef() const { return fBitmap.pixelRef(); } |
| 100 | |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 101 | bool onAsLegacyBitmap(SkBitmap*, LegacyBitmapMode) const override; |
piotaixr | d2a3522 | 2014-08-19 14:29:02 -0700 | [diff] [blame] | 102 | |
reed | 2dad769 | 2016-08-01 11:12:58 -0700 | [diff] [blame] | 103 | SkImage_Raster(const SkBitmap& bm, bool bitmapMayBeMutable = false) |
halcanary | 8d3ad5a | 2016-03-24 09:18:14 -0700 | [diff] [blame] | 104 | : INHERITED(bm.width(), bm.height(), |
| 105 | is_not_subset(bm) ? bm.getGenerationID() |
| 106 | : (uint32_t)kNeedNewImageUniqueID) |
reed | 02d91d1 | 2015-07-30 20:13:43 -0700 | [diff] [blame] | 107 | , fBitmap(bm) |
| 108 | { |
reed | 2dad769 | 2016-08-01 11:12:58 -0700 | [diff] [blame] | 109 | SkASSERT(bitmapMayBeMutable || fBitmap.isImmutable()); |
fmalita | 9a5d1ab | 2015-07-27 10:27:28 -0700 | [diff] [blame] | 110 | } |
halcanary | ea4673f | 2014-08-18 08:27:09 -0700 | [diff] [blame] | 111 | |
Matt Sarett | 9f3dcb3 | 2017-05-04 08:53:32 -0400 | [diff] [blame] | 112 | sk_sp<SkImage> onMakeColorSpace(sk_sp<SkColorSpace>, SkColorType, |
| 113 | SkTransferFunctionBehavior) const override; |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 114 | |
Brian Osman | 5bbd076 | 2017-05-08 11:07:42 -0400 | [diff] [blame] | 115 | bool onIsValid(GrContext* context) const override { return true; } |
| 116 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 117 | #if SK_SUPPORT_GPU |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 118 | sk_sp<GrTextureProxy> refPinnedTextureProxy(uint32_t* uniqueID) const override; |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 119 | bool onPinAsTexture(GrContext*) const override; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 120 | void onUnpinAsTexture(GrContext*) const override; |
| 121 | #endif |
| 122 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 123 | private: |
reed | af3fbfc | 2015-10-04 11:28:36 -0700 | [diff] [blame] | 124 | SkBitmap fBitmap; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 125 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 126 | #if SK_SUPPORT_GPU |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 127 | mutable sk_sp<GrTextureProxy> fPinnedProxy; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 128 | mutable int32_t fPinnedCount = 0; |
| 129 | mutable uint32_t fPinnedUniqueID = 0; |
| 130 | #endif |
| 131 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 132 | typedef SkImage_Base INHERITED; |
| 133 | }; |
| 134 | |
| 135 | /////////////////////////////////////////////////////////////////////////////// |
| 136 | |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 137 | static void release_data(void* addr, void* context) { |
| 138 | SkData* data = static_cast<SkData*>(context); |
| 139 | data->unref(); |
| 140 | } |
| 141 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 142 | SkImage_Raster::SkImage_Raster(const Info& info, sk_sp<SkData> data, size_t rowBytes, uint32_t id) |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 143 | : INHERITED(info.width(), info.height(), id) |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 144 | { |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 145 | void* addr = const_cast<void*>(data->data()); |
| 146 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 147 | fBitmap.installPixels(info, addr, rowBytes, release_data, data.release()); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 148 | fBitmap.setImmutable(); |
| 149 | } |
| 150 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 151 | SkImage_Raster::~SkImage_Raster() { |
| 152 | #if SK_SUPPORT_GPU |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 153 | SkASSERT(nullptr == fPinnedProxy.get()); // want the caller to have manually unpinned |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 154 | #endif |
| 155 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 156 | |
reed | 96472de | 2014-12-10 09:53:42 -0800 | [diff] [blame] | 157 | bool SkImage_Raster::onReadPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRowBytes, |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 158 | int srcX, int srcY, CachingHint) const { |
reed | b4f629c | 2014-12-25 13:55:08 -0800 | [diff] [blame] | 159 | SkBitmap shallowCopy(fBitmap); |
| 160 | return shallowCopy.readPixels(dstInfo, dstPixels, dstRowBytes, srcX, srcY); |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 161 | } |
| 162 | |
reed | 6ceeebd | 2016-03-09 14:26:26 -0800 | [diff] [blame] | 163 | bool SkImage_Raster::onPeekPixels(SkPixmap* pm) const { |
| 164 | return fBitmap.peekPixels(pm); |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 165 | } |
| 166 | |
Brian Osman | 61624f0 | 2016-12-09 14:51:59 -0500 | [diff] [blame] | 167 | bool SkImage_Raster::getROPixels(SkBitmap* dst, SkColorSpace* dstColorSpace, CachingHint) const { |
reed@google.com | 4b0757b | 2013-05-20 16:33:41 +0000 | [diff] [blame] | 168 | *dst = fBitmap; |
| 169 | return true; |
| 170 | } |
| 171 | |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 172 | #if SK_SUPPORT_GPU |
| 173 | sk_sp<GrTextureProxy> SkImage_Raster::asTextureProxyRef(GrContext* context, |
Brian Salomon | 2bbdcc4 | 2017-09-07 12:36:34 -0400 | [diff] [blame] | 174 | const GrSamplerState& params, |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 175 | SkColorSpace* dstColorSpace, |
| 176 | sk_sp<SkColorSpace>* texColorSpace, |
| 177 | SkScalar scaleAdjust[2]) const { |
Robert Phillips | 30a38ff | 2017-03-22 10:31:11 -0400 | [diff] [blame] | 178 | if (!context) { |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 179 | return nullptr; |
| 180 | } |
| 181 | |
Brian Osman | 7992da3 | 2016-11-18 11:28:24 -0500 | [diff] [blame] | 182 | if (texColorSpace) { |
| 183 | *texColorSpace = sk_ref_sp(fBitmap.colorSpace()); |
| 184 | } |
| 185 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 186 | uint32_t uniqueID; |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 187 | sk_sp<GrTextureProxy> tex = this->refPinnedTextureProxy(&uniqueID); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 188 | if (tex) { |
Brian Salomon | 4df0092 | 2017-09-07 16:34:11 +0000 | [diff] [blame] | 189 | GrTextureAdjuster adjuster(context, fPinnedProxy, |
| 190 | fBitmap.alphaType(), fBitmap.bounds(), |
| 191 | fPinnedUniqueID, fBitmap.colorSpace()); |
Greg Daniel | 4faa040 | 2017-09-29 15:21:28 -0400 | [diff] [blame] | 192 | return adjuster.refTextureProxySafeForParams(params, scaleAdjust); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 193 | } |
| 194 | |
Robert Phillips | 7807580 | 2017-03-23 11:11:59 -0400 | [diff] [blame] | 195 | return GrRefCachedBitmapTextureProxy(context, fBitmap, params, scaleAdjust); |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 196 | } |
Robert Phillips | 30a38ff | 2017-03-22 10:31:11 -0400 | [diff] [blame] | 197 | #endif |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 198 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 199 | #if SK_SUPPORT_GPU |
| 200 | |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 201 | sk_sp<GrTextureProxy> SkImage_Raster::refPinnedTextureProxy(uint32_t* uniqueID) const { |
| 202 | if (fPinnedProxy) { |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 203 | SkASSERT(fPinnedCount > 0); |
| 204 | SkASSERT(fPinnedUniqueID != 0); |
| 205 | *uniqueID = fPinnedUniqueID; |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 206 | return fPinnedProxy; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 207 | } |
| 208 | return nullptr; |
| 209 | } |
| 210 | |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 211 | bool SkImage_Raster::onPinAsTexture(GrContext* ctx) const { |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 212 | if (fPinnedProxy) { |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 213 | SkASSERT(fPinnedCount > 0); |
| 214 | SkASSERT(fPinnedUniqueID != 0); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 215 | } else { |
| 216 | SkASSERT(fPinnedCount == 0); |
| 217 | SkASSERT(fPinnedUniqueID == 0); |
Brian Salomon | 2bbdcc4 | 2017-09-07 12:36:34 -0400 | [diff] [blame] | 218 | fPinnedProxy = GrRefCachedBitmapTextureProxy(ctx, fBitmap, GrSamplerState::ClampNearest(), |
| 219 | nullptr); |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 220 | if (!fPinnedProxy) { |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 221 | return false; |
| 222 | } |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 223 | fPinnedUniqueID = fBitmap.getGenerationID(); |
| 224 | } |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 225 | // Note: we only increment if the texture was successfully pinned |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 226 | ++fPinnedCount; |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 227 | return true; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 228 | } |
| 229 | |
| 230 | void SkImage_Raster::onUnpinAsTexture(GrContext* ctx) const { |
| 231 | // Note: we always decrement, even if fPinnedTexture is null |
| 232 | SkASSERT(fPinnedCount > 0); |
| 233 | SkASSERT(fPinnedUniqueID != 0); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 234 | |
| 235 | if (0 == --fPinnedCount) { |
Robert Phillips | 3798c86 | 2017-03-27 11:08:16 -0400 | [diff] [blame] | 236 | fPinnedProxy.reset(nullptr); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 237 | fPinnedUniqueID = 0; |
| 238 | } |
| 239 | } |
| 240 | #endif |
| 241 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 242 | sk_sp<SkImage> SkImage_Raster::onMakeSubset(const SkIRect& subset) const { |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 243 | SkImageInfo info = fBitmap.info().makeWH(subset.width(), subset.height()); |
| 244 | SkBitmap bitmap; |
| 245 | if (!bitmap.tryAllocPixels(info)) { |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 246 | return nullptr; |
| 247 | } |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 248 | |
| 249 | void* dst = bitmap.getPixels(); |
| 250 | void* src = fBitmap.getAddr(subset.x(), subset.y()); |
| 251 | if (!dst || !src) { |
| 252 | SkDEBUGFAIL("SkImage_Raster::onMakeSubset with nullptr src or dst"); |
| 253 | return nullptr; |
| 254 | } |
| 255 | |
| 256 | SkRectMemcpy(dst, bitmap.rowBytes(), src, fBitmap.rowBytes(), bitmap.rowBytes(), |
| 257 | subset.height()); |
| 258 | |
| 259 | bitmap.setImmutable(); |
| 260 | return MakeFromBitmap(bitmap); |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 261 | } |
| 262 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 263 | /////////////////////////////////////////////////////////////////////////////// |
| 264 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 265 | sk_sp<SkImage> MakeRasterCopyPriv(const SkPixmap& pmap, uint32_t id) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 266 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 267 | if (!SkImage_Raster::ValidArgs(pmap.info(), pmap.rowBytes(), &size) || !pmap.addr()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 268 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 269 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 270 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 271 | // Here we actually make a copy of the caller's pixel data |
bungeman | 38d909e | 2016-08-02 14:40:46 -0700 | [diff] [blame] | 272 | sk_sp<SkData> data(SkData::MakeWithCopy(pmap.addr(), size)); |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 273 | return sk_make_sp<SkImage_Raster>(pmap.info(), std::move(data), pmap.rowBytes(), id); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 274 | } |
| 275 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 276 | sk_sp<SkImage> SkImage::MakeRasterCopy(const SkPixmap& pmap) { |
| 277 | return MakeRasterCopyPriv(pmap, kNeedNewImageUniqueID); |
| 278 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 279 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 280 | sk_sp<SkImage> SkImage::MakeRasterData(const SkImageInfo& info, sk_sp<SkData> data, |
| 281 | size_t rowBytes) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 282 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 283 | if (!SkImage_Raster::ValidArgs(info, rowBytes, &size) || !data) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 284 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 285 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 286 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 287 | // did they give us enough data? |
reed@google.com | 999da9c | 2014-02-06 13:43:07 +0000 | [diff] [blame] | 288 | if (data->size() < size) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 289 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 290 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 291 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 292 | return sk_make_sp<SkImage_Raster>(info, std::move(data), rowBytes); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 293 | } |
| 294 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 295 | sk_sp<SkImage> SkImage::MakeFromRaster(const SkPixmap& pmap, RasterReleaseProc proc, |
| 296 | ReleaseContext ctx) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 297 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 298 | if (!SkImage_Raster::ValidArgs(pmap.info(), pmap.rowBytes(), &size) || !pmap.addr()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 299 | return nullptr; |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 300 | } |
| 301 | |
bungeman | 38d909e | 2016-08-02 14:40:46 -0700 | [diff] [blame] | 302 | sk_sp<SkData> data(SkData::MakeWithProc(pmap.addr(), size, proc, ctx)); |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 303 | return sk_make_sp<SkImage_Raster>(pmap.info(), std::move(data), pmap.rowBytes()); |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 304 | } |
| 305 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 306 | sk_sp<SkImage> SkMakeImageFromRasterBitmapPriv(const SkBitmap& bm, SkCopyPixelsMode cpm, |
| 307 | uint32_t idForCopy) { |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 308 | if (kAlways_SkCopyPixelsMode == cpm || (!bm.isImmutable() && kNever_SkCopyPixelsMode != cpm)) { |
| 309 | SkPixmap pmap; |
| 310 | if (bm.peekPixels(&pmap)) { |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 311 | return MakeRasterCopyPriv(pmap, idForCopy); |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 312 | } else { |
| 313 | return sk_sp<SkImage>(); |
| 314 | } |
halcanary | c56c6ef | 2015-09-28 11:55:28 -0700 | [diff] [blame] | 315 | } |
| 316 | |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 317 | return sk_make_sp<SkImage_Raster>(bm, kNever_SkCopyPixelsMode == cpm); |
| 318 | } |
| 319 | |
| 320 | sk_sp<SkImage> SkMakeImageFromRasterBitmap(const SkBitmap& bm, SkCopyPixelsMode cpm) { |
| 321 | if (!SkImageInfoIsValidAllowNumericalCS(bm.info()) || bm.rowBytes() < bm.info().minRowBytes()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 322 | return nullptr; |
kkinnunen | 73953e7 | 2015-02-23 22:12:12 -0800 | [diff] [blame] | 323 | } |
| 324 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 325 | return SkMakeImageFromRasterBitmapPriv(bm, cpm, kNeedNewImageUniqueID); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 326 | } |
| 327 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 328 | sk_sp<SkImage> SkMakeImageInColorSpace(const SkBitmap& bm, sk_sp<SkColorSpace> dstCS, uint32_t id, |
| 329 | SkCopyPixelsMode cpm) { |
Matt Sarett | dc79270 | 2017-06-08 09:37:01 -0400 | [diff] [blame] | 330 | if (!SkImageInfoIsValidAllowNumericalCS(bm.info()) || !bm.getPixels() || |
| 331 | bm.rowBytes() < bm.info().minRowBytes() || !dstCS) { |
| 332 | return nullptr; |
| 333 | } |
| 334 | |
| 335 | sk_sp<SkColorSpace> srcCS = bm.info().refColorSpace(); |
| 336 | if (!srcCS) { |
| 337 | // Treat nullptr as sRGB. |
| 338 | srcCS = SkColorSpace::MakeSRGB(); |
| 339 | } |
| 340 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 341 | sk_sp<SkImage> image = nullptr; |
| 342 | |
Matt Sarett | dc79270 | 2017-06-08 09:37:01 -0400 | [diff] [blame] | 343 | // For the Android use case, this is very likely to be true. |
| 344 | if (SkColorSpace::Equals(srcCS.get(), dstCS.get())) { |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 345 | SkASSERT(kNeedNewImageUniqueID == id || bm.getGenerationID() == id); |
| 346 | image = SkMakeImageFromRasterBitmapPriv(bm, cpm, id); |
| 347 | } else { |
| 348 | image = SkImage::MakeFromGenerator(SkColorSpaceXformImageGenerator::Make(bm, dstCS, cpm, |
| 349 | id)); |
Matt Sarett | dc79270 | 2017-06-08 09:37:01 -0400 | [diff] [blame] | 350 | } |
| 351 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 352 | // If the caller suplied an id, we must propagate that to the image we return |
| 353 | SkASSERT(kNeedNewImageUniqueID == id || image->uniqueID() == id); |
| 354 | return image; |
Matt Sarett | dc79270 | 2017-06-08 09:37:01 -0400 | [diff] [blame] | 355 | } |
| 356 | |
piotaixr | 6515175 | 2014-10-16 11:58:39 -0700 | [diff] [blame] | 357 | const SkPixelRef* SkBitmapImageGetPixelRef(const SkImage* image) { |
| 358 | return ((const SkImage_Raster*)image)->getPixelRef(); |
reed@google.com | 97af1a6 | 2012-08-28 12:19:02 +0000 | [diff] [blame] | 359 | } |
piotaixr | d2a3522 | 2014-08-19 14:29:02 -0700 | [diff] [blame] | 360 | |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 361 | bool SkImage_Raster::onAsLegacyBitmap(SkBitmap* bitmap, LegacyBitmapMode mode) const { |
| 362 | if (kRO_LegacyBitmapMode == mode) { |
| 363 | // When we're a snapshot from a surface, our bitmap may not be marked immutable |
| 364 | // even though logically always we are, but in that case we can't physically share our |
| 365 | // pixelref since the caller might call setImmutable() themselves |
| 366 | // (thus changing our state). |
| 367 | if (fBitmap.isImmutable()) { |
fmalita | ea561bf | 2015-07-09 17:10:31 -0700 | [diff] [blame] | 368 | bitmap->setInfo(fBitmap.info(), fBitmap.rowBytes()); |
Hal Canary | 1b3387b | 2016-12-12 13:48:12 -0500 | [diff] [blame] | 369 | bitmap->setPixelRef(sk_ref_sp(fBitmap.pixelRef()), |
| 370 | fBitmap.pixelRefOrigin().x(), |
| 371 | fBitmap.pixelRefOrigin().y()); |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 372 | return true; |
| 373 | } |
| 374 | } |
| 375 | return this->INHERITED::onAsLegacyBitmap(bitmap, mode); |
| 376 | } |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 377 | |
Matt Sarett | 5b1dba8 | 2017-04-03 11:42:52 -0400 | [diff] [blame] | 378 | /////////////////////////////////////////////////////////////////////////////// |
| 379 | |
Matt Sarett | 9f3dcb3 | 2017-05-04 08:53:32 -0400 | [diff] [blame] | 380 | sk_sp<SkImage> SkImage_Raster::onMakeColorSpace(sk_sp<SkColorSpace> target, |
| 381 | SkColorType targetColorType, |
| 382 | SkTransferFunctionBehavior premulBehavior) const { |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 383 | SkPixmap src; |
Matt Sarett | 9f3dcb3 | 2017-05-04 08:53:32 -0400 | [diff] [blame] | 384 | SkAssertResult(fBitmap.peekPixels(&src)); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 385 | |
Mike Klein | 919cc45 | 2017-03-18 15:36:52 +0000 | [diff] [blame] | 386 | // Treat nullptr srcs as sRGB. |
| 387 | if (!src.colorSpace()) { |
Matt Sarett | d3df9ec | 2017-06-05 10:45:30 -0400 | [diff] [blame] | 388 | if (target->isSRGB()) { |
| 389 | return sk_ref_sp(const_cast<SkImage*>((SkImage*)this)); |
| 390 | } |
| 391 | |
Mike Klein | 919cc45 | 2017-03-18 15:36:52 +0000 | [diff] [blame] | 392 | src.setColorSpace(SkColorSpace::MakeSRGB()); |
| 393 | } |
| 394 | |
Matt Sarett | d3df9ec | 2017-06-05 10:45:30 -0400 | [diff] [blame] | 395 | SkImageInfo dstInfo = fBitmap.info().makeColorType(targetColorType).makeColorSpace(target); |
| 396 | SkBitmap dst; |
| 397 | dst.allocPixels(dstInfo); |
| 398 | |
Matt Sarett | 9f3dcb3 | 2017-05-04 08:53:32 -0400 | [diff] [blame] | 399 | SkAssertResult(dst.writePixels(src, 0, 0, premulBehavior)); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 400 | dst.setImmutable(); |
| 401 | return SkImage::MakeFromBitmap(dst); |
| 402 | } |