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 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 8 | #include "include/core/SkBitmap.h" |
| 9 | #include "include/core/SkCanvas.h" |
| 10 | #include "include/core/SkData.h" |
| 11 | #include "include/core/SkPixelRef.h" |
| 12 | #include "include/core/SkSurface.h" |
| 13 | #include "include/private/SkImageInfoPriv.h" |
| 14 | #include "src/codec/SkColorTable.h" |
Robert Phillips | 99dead9 | 2020-01-27 16:11:57 -0500 | [diff] [blame] | 15 | #include "src/core/SkCompressedDataUtils.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 16 | #include "src/core/SkConvertPixels.h" |
| 17 | #include "src/core/SkImagePriv.h" |
| 18 | #include "src/core/SkTLazy.h" |
| 19 | #include "src/image/SkImage_Base.h" |
| 20 | #include "src/shaders/SkBitmapProcShader.h" |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 21 | |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 22 | #if SK_SUPPORT_GPU |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 23 | #include "include/gpu/GrContext.h" |
| 24 | #include "src/gpu/GrTextureAdjuster.h" |
| 25 | #include "src/gpu/SkGr.h" |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 26 | #endif |
| 27 | |
halcanary | 8d3ad5a | 2016-03-24 09:18:14 -0700 | [diff] [blame] | 28 | // fixes https://bug.skia.org/5096 |
| 29 | static bool is_not_subset(const SkBitmap& bm) { |
| 30 | SkASSERT(bm.pixelRef()); |
Matt Sarett | a6e976a | 2017-04-28 15:43:35 -0400 | [diff] [blame] | 31 | SkISize dim = SkISize::Make(bm.pixelRef()->width(), bm.pixelRef()->height()); |
halcanary | 8d3ad5a | 2016-03-24 09:18:14 -0700 | [diff] [blame] | 32 | SkASSERT(dim != bm.dimensions() || bm.pixelRefOrigin().isZero()); |
| 33 | return dim == bm.dimensions(); |
| 34 | } |
| 35 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 36 | class SkImage_Raster : public SkImage_Base { |
| 37 | public: |
Cary Clark | cc45cc7 | 2018-05-15 16:06:12 -0400 | [diff] [blame] | 38 | static bool ValidArgs(const SkImageInfo& info, size_t rowBytes, size_t* minSize) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 39 | const int maxDimension = SK_MaxS32 >> 2; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 40 | |
Khushal | b11ce19 | 2020-02-14 14:07:16 -0800 | [diff] [blame^] | 41 | // TODO(mtklein): eliminate anything here that setInfo() has already checked. |
| 42 | SkBitmap dummy; |
| 43 | if (!dummy.setInfo(info, rowBytes)) { |
| 44 | return false; |
| 45 | } |
| 46 | |
reed | b2497c2 | 2014-12-31 12:31:43 -0800 | [diff] [blame] | 47 | if (info.width() <= 0 || info.height() <= 0) { |
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 (info.width() > maxDimension || info.height() > maxDimension) { |
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.colorType() > (unsigned)kLastEnum_SkColorType) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 54 | return false; |
| 55 | } |
reed | e5ea500 | 2014-09-03 11:54:58 -0700 | [diff] [blame] | 56 | if ((unsigned)info.alphaType() > (unsigned)kLastEnum_SkAlphaType) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 57 | return false; |
| 58 | } |
| 59 | |
commit-bot@chromium.org | a3264e5 | 2014-05-30 13:26:10 +0000 | [diff] [blame] | 60 | if (kUnknown_SkColorType == info.colorType()) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 61 | return false; |
| 62 | } |
Mike Reed | f0ffb89 | 2017-10-03 14:47:21 -0400 | [diff] [blame] | 63 | if (!info.validRowBytes(rowBytes)) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 64 | return false; |
| 65 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 66 | |
Mike Reed | f0ffb89 | 2017-10-03 14:47:21 -0400 | [diff] [blame] | 67 | size_t size = info.computeByteSize(rowBytes); |
Mike Reed | c5eb97d | 2017-10-09 10:42:51 -0400 | [diff] [blame] | 68 | if (SkImageInfo::ByteSizeOverflowed(size)) { |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 69 | return false; |
| 70 | } |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 71 | |
| 72 | if (minSize) { |
| 73 | *minSize = size; |
| 74 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 75 | return true; |
| 76 | } |
| 77 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 78 | SkImage_Raster(const SkImageInfo&, sk_sp<SkData>, size_t rb, |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 79 | uint32_t id = kNeedNewImageUniqueID); |
Brian Salomon | d3b6597 | 2017-03-22 12:05:03 -0400 | [diff] [blame] | 80 | ~SkImage_Raster() override; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 81 | |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 82 | bool onReadPixels(const SkImageInfo&, void*, size_t, int srcX, int srcY, |
| 83 | CachingHint) const override; |
reed | 6ceeebd | 2016-03-09 14:26:26 -0800 | [diff] [blame] | 84 | bool onPeekPixels(SkPixmap*) const override; |
reed | f1ac182 | 2016-08-01 11:24:14 -0700 | [diff] [blame] | 85 | const SkBitmap* onPeekBitmap() const override { return &fBitmap; } |
| 86 | |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 87 | #if SK_SUPPORT_GPU |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 88 | GrSurfaceProxyView refView(GrRecordingContext*, GrSamplerState, |
| 89 | SkScalar scaleAdjust[2]) const override; |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 90 | #endif |
| 91 | |
Brian Osman | e50cdf0 | 2018-10-19 13:02:14 -0400 | [diff] [blame] | 92 | bool getROPixels(SkBitmap*, CachingHint) const override; |
Robert Phillips | 6603a17 | 2019-03-05 12:35:44 -0500 | [diff] [blame] | 93 | sk_sp<SkImage> onMakeSubset(GrRecordingContext*, const SkIRect&) const override; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 94 | |
reed@google.com | 97af1a6 | 2012-08-28 12:19:02 +0000 | [diff] [blame] | 95 | SkPixelRef* getPixelRef() const { return fBitmap.pixelRef(); } |
| 96 | |
Cary Clark | 4f5a79c | 2018-02-07 15:51:00 -0500 | [diff] [blame] | 97 | bool onAsLegacyBitmap(SkBitmap*) const override; |
piotaixr | d2a3522 | 2014-08-19 14:29:02 -0700 | [diff] [blame] | 98 | |
reed | 2dad769 | 2016-08-01 11:12:58 -0700 | [diff] [blame] | 99 | SkImage_Raster(const SkBitmap& bm, bool bitmapMayBeMutable = false) |
Brian Salomon | 5ad6fd3 | 2019-03-21 15:30:08 -0400 | [diff] [blame] | 100 | : INHERITED(bm.info(), |
| 101 | is_not_subset(bm) ? bm.getGenerationID() : (uint32_t)kNeedNewImageUniqueID) |
| 102 | , fBitmap(bm) { |
reed | 2dad769 | 2016-08-01 11:12:58 -0700 | [diff] [blame] | 103 | SkASSERT(bitmapMayBeMutable || fBitmap.isImmutable()); |
fmalita | 9a5d1ab | 2015-07-27 10:27:28 -0700 | [diff] [blame] | 104 | } |
halcanary | ea4673f | 2014-08-18 08:27:09 -0700 | [diff] [blame] | 105 | |
Robert Phillips | 6603a17 | 2019-03-05 12:35:44 -0500 | [diff] [blame] | 106 | sk_sp<SkImage> onMakeColorTypeAndColorSpace(GrRecordingContext*, |
| 107 | SkColorType, sk_sp<SkColorSpace>) const override; |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 108 | |
Brian Osman | d514837 | 2019-08-14 16:14:51 -0400 | [diff] [blame] | 109 | sk_sp<SkImage> onReinterpretColorSpace(sk_sp<SkColorSpace>) const override; |
| 110 | |
Brian Osman | 5bbd076 | 2017-05-08 11:07:42 -0400 | [diff] [blame] | 111 | bool onIsValid(GrContext* context) const override { return true; } |
Brian Osman | 3a16073 | 2018-10-15 15:32:06 -0400 | [diff] [blame] | 112 | void notifyAddedToRasterCache() const override { |
| 113 | // We explicitly DON'T want to call INHERITED::notifyAddedToRasterCache. That ties the |
| 114 | // lifetime of derived/cached resources to the image. In this case, we only want cached |
| 115 | // data (eg mips) tied to the lifetime of the underlying pixelRef. |
| 116 | SkASSERT(fBitmap.pixelRef()); |
| 117 | fBitmap.pixelRef()->notifyAddedToCache(); |
| 118 | } |
Brian Osman | 5bbd076 | 2017-05-08 11:07:42 -0400 | [diff] [blame] | 119 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 120 | #if SK_SUPPORT_GPU |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 121 | GrSurfaceProxyView refPinnedView(GrRecordingContext* context, |
| 122 | uint32_t* uniqueID) const override; |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 123 | bool onPinAsTexture(GrContext*) const override; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 124 | void onUnpinAsTexture(GrContext*) const override; |
| 125 | #endif |
| 126 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 127 | private: |
reed | af3fbfc | 2015-10-04 11:28:36 -0700 | [diff] [blame] | 128 | SkBitmap fBitmap; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 129 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 130 | #if SK_SUPPORT_GPU |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 131 | mutable GrSurfaceProxyView fPinnedView; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 132 | mutable int32_t fPinnedCount = 0; |
| 133 | mutable uint32_t fPinnedUniqueID = 0; |
| 134 | #endif |
| 135 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 136 | typedef SkImage_Base INHERITED; |
| 137 | }; |
| 138 | |
| 139 | /////////////////////////////////////////////////////////////////////////////// |
| 140 | |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 141 | static void release_data(void* addr, void* context) { |
| 142 | SkData* data = static_cast<SkData*>(context); |
| 143 | data->unref(); |
| 144 | } |
| 145 | |
Cary Clark | cc45cc7 | 2018-05-15 16:06:12 -0400 | [diff] [blame] | 146 | SkImage_Raster::SkImage_Raster(const SkImageInfo& info, sk_sp<SkData> data, size_t rowBytes, |
| 147 | uint32_t id) |
Brian Salomon | 5ad6fd3 | 2019-03-21 15:30:08 -0400 | [diff] [blame] | 148 | : INHERITED(info, id) { |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 149 | void* addr = const_cast<void*>(data->data()); |
| 150 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 151 | fBitmap.installPixels(info, addr, rowBytes, release_data, data.release()); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 152 | fBitmap.setImmutable(); |
| 153 | } |
| 154 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 155 | SkImage_Raster::~SkImage_Raster() { |
| 156 | #if SK_SUPPORT_GPU |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 157 | SkASSERT(!fPinnedView); // want the caller to have manually unpinned |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 158 | #endif |
| 159 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 160 | |
reed | 96472de | 2014-12-10 09:53:42 -0800 | [diff] [blame] | 161 | bool SkImage_Raster::onReadPixels(const SkImageInfo& dstInfo, void* dstPixels, size_t dstRowBytes, |
reed | 0955303 | 2015-11-23 12:32:16 -0800 | [diff] [blame] | 162 | int srcX, int srcY, CachingHint) const { |
reed | b4f629c | 2014-12-25 13:55:08 -0800 | [diff] [blame] | 163 | SkBitmap shallowCopy(fBitmap); |
| 164 | return shallowCopy.readPixels(dstInfo, dstPixels, dstRowBytes, srcX, srcY); |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 165 | } |
| 166 | |
reed | 6ceeebd | 2016-03-09 14:26:26 -0800 | [diff] [blame] | 167 | bool SkImage_Raster::onPeekPixels(SkPixmap* pm) const { |
| 168 | return fBitmap.peekPixels(pm); |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 169 | } |
| 170 | |
Brian Osman | e50cdf0 | 2018-10-19 13:02:14 -0400 | [diff] [blame] | 171 | bool SkImage_Raster::getROPixels(SkBitmap* dst, CachingHint) const { |
reed@google.com | 4b0757b | 2013-05-20 16:33:41 +0000 | [diff] [blame] | 172 | *dst = fBitmap; |
| 173 | return true; |
| 174 | } |
| 175 | |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 176 | #if SK_SUPPORT_GPU |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 177 | GrSurfaceProxyView SkImage_Raster::refView(GrRecordingContext* context, GrSamplerState params, |
| 178 | SkScalar scaleAdjust[2]) const { |
Robert Phillips | 30a38ff | 2017-03-22 10:31:11 -0400 | [diff] [blame] | 179 | if (!context) { |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 180 | return {}; |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 181 | } |
| 182 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 183 | uint32_t uniqueID; |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 184 | if (GrSurfaceProxyView view = this->refPinnedView(context, &uniqueID)) { |
Greg Daniel | cc104db | 2020-02-03 14:17:08 -0500 | [diff] [blame] | 185 | GrTextureAdjuster adjuster(context, std::move(view), fBitmap.info().colorInfo(), |
Brian Salomon | 1a217eb | 2019-10-24 10:50:36 -0400 | [diff] [blame] | 186 | fPinnedUniqueID); |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 187 | return adjuster.viewForParams(params, scaleAdjust); |
Brian Osman | a507392 | 2018-10-18 15:55:15 -0400 | [diff] [blame] | 188 | } |
| 189 | |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 190 | return GrRefCachedBitmapView(context, fBitmap, params, scaleAdjust); |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 191 | } |
Robert Phillips | 30a38ff | 2017-03-22 10:31:11 -0400 | [diff] [blame] | 192 | #endif |
reed | 746f31f | 2015-09-16 12:53:29 -0700 | [diff] [blame] | 193 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 194 | #if SK_SUPPORT_GPU |
| 195 | |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 196 | GrSurfaceProxyView SkImage_Raster::refPinnedView(GrRecordingContext*, uint32_t* uniqueID) const { |
| 197 | if (fPinnedView) { |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 198 | SkASSERT(fPinnedCount > 0); |
| 199 | SkASSERT(fPinnedUniqueID != 0); |
| 200 | *uniqueID = fPinnedUniqueID; |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 201 | return fPinnedView; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 202 | } |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 203 | return {}; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 204 | } |
| 205 | |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 206 | bool SkImage_Raster::onPinAsTexture(GrContext* ctx) const { |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 207 | if (fPinnedView) { |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 208 | SkASSERT(fPinnedCount > 0); |
| 209 | SkASSERT(fPinnedUniqueID != 0); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 210 | } else { |
| 211 | SkASSERT(fPinnedCount == 0); |
| 212 | SkASSERT(fPinnedUniqueID == 0); |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 213 | fPinnedView = |
| 214 | GrRefCachedBitmapView(ctx, fBitmap, GrSamplerState::Filter::kNearest, nullptr); |
| 215 | if (!fPinnedView) { |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 216 | return false; |
| 217 | } |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 218 | SkASSERT(fPinnedView.asTextureProxy()); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 219 | fPinnedUniqueID = fBitmap.getGenerationID(); |
| 220 | } |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 221 | // Note: we only increment if the texture was successfully pinned |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 222 | ++fPinnedCount; |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 223 | return true; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 224 | } |
| 225 | |
| 226 | void SkImage_Raster::onUnpinAsTexture(GrContext* ctx) const { |
| 227 | // Note: we always decrement, even if fPinnedTexture is null |
| 228 | SkASSERT(fPinnedCount > 0); |
| 229 | SkASSERT(fPinnedUniqueID != 0); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 230 | |
| 231 | if (0 == --fPinnedCount) { |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 232 | fPinnedView = GrSurfaceProxyView(); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 233 | fPinnedUniqueID = 0; |
| 234 | } |
| 235 | } |
| 236 | #endif |
| 237 | |
Robert Phillips | 6603a17 | 2019-03-05 12:35:44 -0500 | [diff] [blame] | 238 | sk_sp<SkImage> SkImage_Raster::onMakeSubset(GrRecordingContext*, const SkIRect& subset) const { |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 239 | SkImageInfo info = fBitmap.info().makeDimensions(subset.size()); |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 240 | SkBitmap bitmap; |
| 241 | if (!bitmap.tryAllocPixels(info)) { |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 242 | return nullptr; |
| 243 | } |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 244 | |
| 245 | void* dst = bitmap.getPixels(); |
| 246 | void* src = fBitmap.getAddr(subset.x(), subset.y()); |
| 247 | if (!dst || !src) { |
| 248 | SkDEBUGFAIL("SkImage_Raster::onMakeSubset with nullptr src or dst"); |
| 249 | return nullptr; |
| 250 | } |
| 251 | |
| 252 | SkRectMemcpy(dst, bitmap.rowBytes(), src, fBitmap.rowBytes(), bitmap.rowBytes(), |
| 253 | subset.height()); |
| 254 | |
| 255 | bitmap.setImmutable(); |
| 256 | return MakeFromBitmap(bitmap); |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 257 | } |
| 258 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 259 | /////////////////////////////////////////////////////////////////////////////// |
| 260 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 261 | sk_sp<SkImage> MakeRasterCopyPriv(const SkPixmap& pmap, uint32_t id) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 262 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 263 | if (!SkImage_Raster::ValidArgs(pmap.info(), pmap.rowBytes(), &size) || !pmap.addr()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 264 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 265 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 266 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 267 | // Here we actually make a copy of the caller's pixel data |
bungeman | 38d909e | 2016-08-02 14:40:46 -0700 | [diff] [blame] | 268 | sk_sp<SkData> data(SkData::MakeWithCopy(pmap.addr(), size)); |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 269 | 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] | 270 | } |
| 271 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 272 | sk_sp<SkImage> SkImage::MakeRasterCopy(const SkPixmap& pmap) { |
| 273 | return MakeRasterCopyPriv(pmap, kNeedNewImageUniqueID); |
| 274 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 275 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 276 | sk_sp<SkImage> SkImage::MakeRasterData(const SkImageInfo& info, sk_sp<SkData> data, |
| 277 | size_t rowBytes) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 278 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 279 | if (!SkImage_Raster::ValidArgs(info, rowBytes, &size) || !data) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 280 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 281 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 282 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 283 | // did they give us enough data? |
reed@google.com | 999da9c | 2014-02-06 13:43:07 +0000 | [diff] [blame] | 284 | if (data->size() < size) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 285 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 286 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 287 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 288 | return sk_make_sp<SkImage_Raster>(info, std::move(data), rowBytes); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 289 | } |
| 290 | |
Robert Phillips | 99dead9 | 2020-01-27 16:11:57 -0500 | [diff] [blame] | 291 | // TODO: this could be improved to decode and make use of the mipmap |
| 292 | // levels potentially present in the compressed data. For now, any |
| 293 | // mipmap levels are discarded. |
| 294 | sk_sp<SkImage> SkImage::MakeRasterFromCompressed(sk_sp<SkData> data, |
| 295 | int width, int height, |
| 296 | CompressionType type) { |
| 297 | size_t expectedSize = SkCompressedFormatDataSize(type, { width, height }, false); |
| 298 | if (!data || data->size() < expectedSize) { |
| 299 | return nullptr; |
| 300 | } |
| 301 | |
| 302 | SkAlphaType at = SkCompressionTypeIsOpaque(type) ? kOpaque_SkAlphaType |
| 303 | : kPremul_SkAlphaType; |
| 304 | |
Robert Phillips | 162e04b | 2020-01-28 14:22:43 -0500 | [diff] [blame] | 305 | SkImageInfo ii = SkImageInfo::MakeN32(width, height, at); |
Robert Phillips | 99dead9 | 2020-01-27 16:11:57 -0500 | [diff] [blame] | 306 | |
| 307 | if (!SkImage_Raster::ValidArgs(ii, ii.minRowBytes(), nullptr)) { |
| 308 | return nullptr; |
| 309 | } |
| 310 | |
| 311 | SkBitmap bitmap; |
| 312 | if (!bitmap.tryAllocPixels(ii)) { |
| 313 | return nullptr; |
| 314 | } |
| 315 | |
| 316 | if (!SkDecompress(std::move(data), { width, height }, type, &bitmap)) { |
| 317 | return nullptr; |
| 318 | } |
| 319 | |
| 320 | bitmap.setImmutable(); |
| 321 | return MakeFromBitmap(bitmap); |
| 322 | } |
| 323 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 324 | sk_sp<SkImage> SkImage::MakeFromRaster(const SkPixmap& pmap, RasterReleaseProc proc, |
| 325 | ReleaseContext ctx) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 326 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 327 | if (!SkImage_Raster::ValidArgs(pmap.info(), pmap.rowBytes(), &size) || !pmap.addr()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 328 | return nullptr; |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 329 | } |
| 330 | |
bungeman | 38d909e | 2016-08-02 14:40:46 -0700 | [diff] [blame] | 331 | sk_sp<SkData> data(SkData::MakeWithProc(pmap.addr(), size, proc, ctx)); |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 332 | return sk_make_sp<SkImage_Raster>(pmap.info(), std::move(data), pmap.rowBytes()); |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 333 | } |
| 334 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 335 | sk_sp<SkImage> SkMakeImageFromRasterBitmapPriv(const SkBitmap& bm, SkCopyPixelsMode cpm, |
| 336 | uint32_t idForCopy) { |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 337 | if (kAlways_SkCopyPixelsMode == cpm || (!bm.isImmutable() && kNever_SkCopyPixelsMode != cpm)) { |
| 338 | SkPixmap pmap; |
| 339 | if (bm.peekPixels(&pmap)) { |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 340 | return MakeRasterCopyPriv(pmap, idForCopy); |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 341 | } else { |
| 342 | return sk_sp<SkImage>(); |
| 343 | } |
halcanary | c56c6ef | 2015-09-28 11:55:28 -0700 | [diff] [blame] | 344 | } |
| 345 | |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 346 | return sk_make_sp<SkImage_Raster>(bm, kNever_SkCopyPixelsMode == cpm); |
| 347 | } |
| 348 | |
| 349 | sk_sp<SkImage> SkMakeImageFromRasterBitmap(const SkBitmap& bm, SkCopyPixelsMode cpm) { |
Brian Osman | e1adc3a | 2018-06-04 09:21:17 -0400 | [diff] [blame] | 350 | if (!SkImageInfoIsValid(bm.info()) || bm.rowBytes() < bm.info().minRowBytes()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 351 | return nullptr; |
kkinnunen | 73953e7 | 2015-02-23 22:12:12 -0800 | [diff] [blame] | 352 | } |
| 353 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 354 | return SkMakeImageFromRasterBitmapPriv(bm, cpm, kNeedNewImageUniqueID); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [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 | |
Cary Clark | 4f5a79c | 2018-02-07 15:51:00 -0500 | [diff] [blame] | 361 | bool SkImage_Raster::onAsLegacyBitmap(SkBitmap* bitmap) const { |
| 362 | // When we're a snapshot from a surface, our bitmap may not be marked immutable |
| 363 | // even though logically always we are, but in that case we can't physically share our |
| 364 | // pixelref since the caller might call setImmutable() themselves |
| 365 | // (thus changing our state). |
| 366 | if (fBitmap.isImmutable()) { |
| 367 | SkIPoint origin = fBitmap.pixelRefOrigin(); |
| 368 | bitmap->setInfo(fBitmap.info(), fBitmap.rowBytes()); |
| 369 | bitmap->setPixelRef(sk_ref_sp(fBitmap.pixelRef()), origin.x(), origin.y()); |
| 370 | return true; |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 371 | } |
Cary Clark | 4f5a79c | 2018-02-07 15:51:00 -0500 | [diff] [blame] | 372 | return this->INHERITED::onAsLegacyBitmap(bitmap); |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 373 | } |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 374 | |
Matt Sarett | 5b1dba8 | 2017-04-03 11:42:52 -0400 | [diff] [blame] | 375 | /////////////////////////////////////////////////////////////////////////////// |
| 376 | |
Robert Phillips | 6603a17 | 2019-03-05 12:35:44 -0500 | [diff] [blame] | 377 | sk_sp<SkImage> SkImage_Raster::onMakeColorTypeAndColorSpace(GrRecordingContext*, |
| 378 | SkColorType targetCT, |
Brian Osman | f48c996 | 2019-01-14 11:15:50 -0500 | [diff] [blame] | 379 | sk_sp<SkColorSpace> targetCS) const { |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 380 | SkPixmap src; |
Matt Sarett | 9f3dcb3 | 2017-05-04 08:53:32 -0400 | [diff] [blame] | 381 | SkAssertResult(fBitmap.peekPixels(&src)); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 382 | |
Matt Sarett | d3df9ec | 2017-06-05 10:45:30 -0400 | [diff] [blame] | 383 | SkBitmap dst; |
Brian Osman | f48c996 | 2019-01-14 11:15:50 -0500 | [diff] [blame] | 384 | dst.allocPixels(fBitmap.info().makeColorType(targetCT).makeColorSpace(targetCS)); |
Matt Sarett | d3df9ec | 2017-06-05 10:45:30 -0400 | [diff] [blame] | 385 | |
Brian Osman | be686f0 | 2018-10-12 11:18:02 -0400 | [diff] [blame] | 386 | SkAssertResult(dst.writePixels(src)); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 387 | dst.setImmutable(); |
| 388 | return SkImage::MakeFromBitmap(dst); |
| 389 | } |
Brian Osman | d514837 | 2019-08-14 16:14:51 -0400 | [diff] [blame] | 390 | |
| 391 | sk_sp<SkImage> SkImage_Raster::onReinterpretColorSpace(sk_sp<SkColorSpace> newCS) const { |
| 392 | // TODO: If our bitmap is immutable, then we could theoretically create another image sharing |
| 393 | // our pixelRef. That doesn't work (without more invasive logic), because the image gets its |
| 394 | // gen ID from the bitmap, which gets it from the pixelRef. |
| 395 | SkPixmap pixmap = fBitmap.pixmap(); |
| 396 | pixmap.setColorSpace(std::move(newCS)); |
| 397 | return SkImage::MakeRasterCopy(pixmap); |
| 398 | } |