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 |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 23 | #include "src/gpu/GrBitmapTextureMaker.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 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 | |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 82 | bool onReadPixels(GrDirectContext*, const SkImageInfo&, void*, size_t, int srcX, int srcY, |
Greg Daniel | febdedf | 2020-02-05 17:06:27 -0500 | [diff] [blame] | 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 | |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 87 | bool getROPixels(GrDirectContext*, SkBitmap*, CachingHint) const override; |
Adlai Holler | 872a32c | 2020-07-10 14:33:22 -0400 | [diff] [blame] | 88 | sk_sp<SkImage> onMakeSubset(const SkIRect&, GrDirectContext*) const override; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 89 | |
reed@google.com | 97af1a6 | 2012-08-28 12:19:02 +0000 | [diff] [blame] | 90 | SkPixelRef* getPixelRef() const { return fBitmap.pixelRef(); } |
| 91 | |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 92 | bool onAsLegacyBitmap(GrDirectContext*, SkBitmap*) const override; |
piotaixr | d2a3522 | 2014-08-19 14:29:02 -0700 | [diff] [blame] | 93 | |
reed | 2dad769 | 2016-08-01 11:12:58 -0700 | [diff] [blame] | 94 | SkImage_Raster(const SkBitmap& bm, bool bitmapMayBeMutable = false) |
Brian Salomon | 5ad6fd3 | 2019-03-21 15:30:08 -0400 | [diff] [blame] | 95 | : INHERITED(bm.info(), |
| 96 | is_not_subset(bm) ? bm.getGenerationID() : (uint32_t)kNeedNewImageUniqueID) |
| 97 | , fBitmap(bm) { |
reed | 2dad769 | 2016-08-01 11:12:58 -0700 | [diff] [blame] | 98 | SkASSERT(bitmapMayBeMutable || fBitmap.isImmutable()); |
fmalita | 9a5d1ab | 2015-07-27 10:27:28 -0700 | [diff] [blame] | 99 | } |
halcanary | ea4673f | 2014-08-18 08:27:09 -0700 | [diff] [blame] | 100 | |
Adlai Holler | 3a22017 | 2020-07-15 10:37:50 -0400 | [diff] [blame] | 101 | sk_sp<SkImage> onMakeColorTypeAndColorSpace(SkColorType, sk_sp<SkColorSpace>, |
| 102 | GrDirectContext*) const override; |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 103 | |
Brian Osman | d514837 | 2019-08-14 16:14:51 -0400 | [diff] [blame] | 104 | sk_sp<SkImage> onReinterpretColorSpace(sk_sp<SkColorSpace>) const override; |
| 105 | |
Robert Phillips | 4a3ebc2 | 2020-07-10 11:27:43 -0400 | [diff] [blame] | 106 | bool onIsValid(GrRecordingContext* context) const override { return true; } |
Brian Osman | 3a16073 | 2018-10-15 15:32:06 -0400 | [diff] [blame] | 107 | void notifyAddedToRasterCache() const override { |
| 108 | // We explicitly DON'T want to call INHERITED::notifyAddedToRasterCache. That ties the |
| 109 | // lifetime of derived/cached resources to the image. In this case, we only want cached |
| 110 | // data (eg mips) tied to the lifetime of the underlying pixelRef. |
| 111 | SkASSERT(fBitmap.pixelRef()); |
| 112 | fBitmap.pixelRef()->notifyAddedToCache(); |
| 113 | } |
Brian Osman | 5bbd076 | 2017-05-08 11:07:42 -0400 | [diff] [blame] | 114 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 115 | #if SK_SUPPORT_GPU |
Adlai Holler | 2940538 | 2020-07-20 16:02:05 -0400 | [diff] [blame] | 116 | bool onPinAsTexture(GrRecordingContext*) const override; |
| 117 | void onUnpinAsTexture(GrRecordingContext*) const override; |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 118 | bool isPinnedOnContext(GrRecordingContext*) const override; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 119 | #endif |
| 120 | |
Brian Salomon | 7d2757f | 2021-01-26 17:46:50 -0500 | [diff] [blame] | 121 | bool onHasMipmaps() const override { return SkToBool(fBitmap.fMips); } |
| 122 | |
Mike Reed | 13711eb | 2020-07-14 17:16:32 -0400 | [diff] [blame] | 123 | SkMipmap* onPeekMips() const override { return fBitmap.fMips.get(); } |
Mike Reed | 2fe1569 | 2020-07-14 11:29:38 -0400 | [diff] [blame] | 124 | |
Mike Reed | 13711eb | 2020-07-14 17:16:32 -0400 | [diff] [blame] | 125 | sk_sp<SkImage> onMakeWithMipmaps(sk_sp<SkMipmap> mips) const override { |
Mike Reed | 2fe1569 | 2020-07-14 11:29:38 -0400 | [diff] [blame] | 126 | auto img = new SkImage_Raster(fBitmap); |
| 127 | if (mips) { |
| 128 | img->fBitmap.fMips = std::move(mips); |
| 129 | } else { |
Mike Reed | 13711eb | 2020-07-14 17:16:32 -0400 | [diff] [blame] | 130 | img->fBitmap.fMips.reset(SkMipmap::Build(fBitmap.pixmap(), nullptr)); |
Mike Reed | 2fe1569 | 2020-07-14 11:29:38 -0400 | [diff] [blame] | 131 | } |
| 132 | return sk_sp<SkImage>(img); |
| 133 | } |
| 134 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 135 | private: |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 136 | #if SK_SUPPORT_GPU |
| 137 | std::tuple<GrSurfaceProxyView, GrColorType> onAsView(GrRecordingContext*, |
| 138 | GrMipmapped, |
| 139 | GrImageTexGenPolicy) const override; |
| 140 | #endif |
| 141 | |
reed | af3fbfc | 2015-10-04 11:28:36 -0700 | [diff] [blame] | 142 | SkBitmap fBitmap; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 143 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 144 | #if SK_SUPPORT_GPU |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 145 | mutable GrSurfaceProxyView fPinnedView; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 146 | mutable int32_t fPinnedCount = 0; |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 147 | mutable uint32_t fPinnedUniqueID = SK_InvalidUniqueID; |
| 148 | mutable uint32_t fPinnedContextID = SK_InvalidUniqueID; |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 149 | mutable GrColorType fPinnedColorType = GrColorType::kUnknown; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 150 | #endif |
| 151 | |
John Stiles | 7571f9e | 2020-09-02 22:42:33 -0400 | [diff] [blame] | 152 | using INHERITED = SkImage_Base; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 153 | }; |
| 154 | |
| 155 | /////////////////////////////////////////////////////////////////////////////// |
| 156 | |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 157 | static void release_data(void* addr, void* context) { |
| 158 | SkData* data = static_cast<SkData*>(context); |
| 159 | data->unref(); |
| 160 | } |
| 161 | |
Cary Clark | cc45cc7 | 2018-05-15 16:06:12 -0400 | [diff] [blame] | 162 | SkImage_Raster::SkImage_Raster(const SkImageInfo& info, sk_sp<SkData> data, size_t rowBytes, |
| 163 | uint32_t id) |
Brian Salomon | 5ad6fd3 | 2019-03-21 15:30:08 -0400 | [diff] [blame] | 164 | : INHERITED(info, id) { |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 165 | void* addr = const_cast<void*>(data->data()); |
| 166 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 167 | fBitmap.installPixels(info, addr, rowBytes, release_data, data.release()); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 168 | fBitmap.setImmutable(); |
| 169 | } |
| 170 | |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 171 | SkImage_Raster::~SkImage_Raster() { |
| 172 | #if SK_SUPPORT_GPU |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 173 | SkASSERT(!fPinnedView); // want the caller to have manually unpinned |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 174 | #endif |
| 175 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 176 | |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 177 | bool SkImage_Raster::onReadPixels(GrDirectContext*, |
| 178 | const SkImageInfo& dstInfo, |
| 179 | void* dstPixels, |
| 180 | size_t dstRowBytes, |
| 181 | int srcX, |
| 182 | int srcY, |
| 183 | CachingHint) const { |
reed | b4f629c | 2014-12-25 13:55:08 -0800 | [diff] [blame] | 184 | SkBitmap shallowCopy(fBitmap); |
| 185 | return shallowCopy.readPixels(dstInfo, dstPixels, dstRowBytes, srcX, srcY); |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 186 | } |
| 187 | |
reed | 6ceeebd | 2016-03-09 14:26:26 -0800 | [diff] [blame] | 188 | bool SkImage_Raster::onPeekPixels(SkPixmap* pm) const { |
| 189 | return fBitmap.peekPixels(pm); |
reed@google.com | 4f7c615 | 2014-02-06 14:11:56 +0000 | [diff] [blame] | 190 | } |
| 191 | |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 192 | bool SkImage_Raster::getROPixels(GrDirectContext*, SkBitmap* dst, CachingHint) const { |
reed@google.com | 4b0757b | 2013-05-20 16:33:41 +0000 | [diff] [blame] | 193 | *dst = fBitmap; |
| 194 | return true; |
| 195 | } |
| 196 | |
Robert Phillips | b726d58 | 2017-03-09 16:36:32 -0500 | [diff] [blame] | 197 | #if SK_SUPPORT_GPU |
Adlai Holler | 2940538 | 2020-07-20 16:02:05 -0400 | [diff] [blame] | 198 | bool SkImage_Raster::onPinAsTexture(GrRecordingContext* rContext) const { |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 199 | if (fPinnedView) { |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 200 | SkASSERT(fPinnedCount > 0); |
| 201 | SkASSERT(fPinnedUniqueID != 0); |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 202 | if (rContext->priv().contextID() != fPinnedContextID) { |
| 203 | return false; |
| 204 | } |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 205 | } else { |
| 206 | SkASSERT(fPinnedCount == 0); |
| 207 | SkASSERT(fPinnedUniqueID == 0); |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 208 | GrBitmapTextureMaker maker(rContext, fBitmap, GrImageTexGenPolicy::kDraw); |
| 209 | fPinnedView = maker.view(GrMipmapped::kNo); |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 210 | if (!fPinnedView) { |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 211 | return false; |
| 212 | } |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 213 | SkASSERT(fPinnedView.asTextureProxy()); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 214 | fPinnedUniqueID = fBitmap.getGenerationID(); |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 215 | fPinnedContextID = rContext->priv().contextID(); |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 216 | fPinnedColorType = maker.colorType(); |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 217 | } |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 218 | // Note: we only increment if the texture was successfully pinned |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 219 | ++fPinnedCount; |
Derek Sollenberger | d3ea9b7 | 2016-11-09 11:25:13 -0500 | [diff] [blame] | 220 | return true; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 221 | } |
| 222 | |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 223 | void SkImage_Raster::onUnpinAsTexture(GrRecordingContext* rContext) const { |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 224 | // Note: we always decrement, even if fPinnedTexture is null |
| 225 | SkASSERT(fPinnedCount > 0); |
| 226 | SkASSERT(fPinnedUniqueID != 0); |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 227 | #if 0 // This would be better but Android currently calls with an already freed context ptr. |
| 228 | if (rContext->priv().contextID() != fPinnedContextID) { |
| 229 | return; |
| 230 | } |
| 231 | #endif |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 232 | |
| 233 | if (0 == --fPinnedCount) { |
Greg Daniel | cc21d0c | 2020-02-05 16:58:40 -0500 | [diff] [blame] | 234 | fPinnedView = GrSurfaceProxyView(); |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 235 | fPinnedUniqueID = SK_InvalidUniqueID; |
| 236 | fPinnedContextID = SK_InvalidUniqueID; |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 237 | fPinnedColorType = GrColorType::kUnknown; |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 238 | } |
| 239 | } |
Brian Salomon | 490f192 | 2021-02-01 13:14:37 -0500 | [diff] [blame] | 240 | |
| 241 | bool SkImage_Raster::isPinnedOnContext(GrRecordingContext* rContext) const { |
| 242 | return fPinnedContextID == rContext->priv().contextID(); |
| 243 | } |
reed | 2d5b714 | 2016-08-17 11:12:33 -0700 | [diff] [blame] | 244 | #endif |
| 245 | |
Adlai Holler | 872a32c | 2020-07-10 14:33:22 -0400 | [diff] [blame] | 246 | sk_sp<SkImage> SkImage_Raster::onMakeSubset(const SkIRect& subset, GrDirectContext*) const { |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 247 | SkImageInfo info = fBitmap.info().makeDimensions(subset.size()); |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 248 | SkBitmap bitmap; |
| 249 | if (!bitmap.tryAllocPixels(info)) { |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 250 | return nullptr; |
| 251 | } |
Brian Osman | 7314093 | 2017-08-16 16:35:54 -0400 | [diff] [blame] | 252 | |
| 253 | void* dst = bitmap.getPixels(); |
| 254 | void* src = fBitmap.getAddr(subset.x(), subset.y()); |
| 255 | if (!dst || !src) { |
| 256 | SkDEBUGFAIL("SkImage_Raster::onMakeSubset with nullptr src or dst"); |
| 257 | return nullptr; |
| 258 | } |
| 259 | |
| 260 | SkRectMemcpy(dst, bitmap.rowBytes(), src, fBitmap.rowBytes(), bitmap.rowBytes(), |
| 261 | subset.height()); |
| 262 | |
| 263 | bitmap.setImmutable(); |
Mike Reed | dc607e3 | 2020-12-23 11:50:36 -0500 | [diff] [blame] | 264 | return bitmap.asImage(); |
reed | 7b6945b | 2015-09-24 00:50:58 -0700 | [diff] [blame] | 265 | } |
| 266 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 267 | /////////////////////////////////////////////////////////////////////////////// |
| 268 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 269 | sk_sp<SkImage> MakeRasterCopyPriv(const SkPixmap& pmap, uint32_t id) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 270 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 271 | if (!SkImage_Raster::ValidArgs(pmap.info(), pmap.rowBytes(), &size) || !pmap.addr()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 272 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 273 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 274 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 275 | // Here we actually make a copy of the caller's pixel data |
bungeman | 38d909e | 2016-08-02 14:40:46 -0700 | [diff] [blame] | 276 | sk_sp<SkData> data(SkData::MakeWithCopy(pmap.addr(), size)); |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 277 | 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] | 278 | } |
| 279 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 280 | sk_sp<SkImage> SkImage::MakeRasterCopy(const SkPixmap& pmap) { |
| 281 | return MakeRasterCopyPriv(pmap, kNeedNewImageUniqueID); |
| 282 | } |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 283 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 284 | sk_sp<SkImage> SkImage::MakeRasterData(const SkImageInfo& info, sk_sp<SkData> data, |
| 285 | size_t rowBytes) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 286 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 287 | if (!SkImage_Raster::ValidArgs(info, rowBytes, &size) || !data) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 288 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 289 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 290 | |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 291 | // did they give us enough data? |
reed@google.com | 999da9c | 2014-02-06 13:43:07 +0000 | [diff] [blame] | 292 | if (data->size() < size) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 293 | return nullptr; |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 294 | } |
rmistry@google.com | fbfcd56 | 2012-08-23 18:09:54 +0000 | [diff] [blame] | 295 | |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 296 | return sk_make_sp<SkImage_Raster>(info, std::move(data), rowBytes); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 297 | } |
| 298 | |
Robert Phillips | 99dead9 | 2020-01-27 16:11:57 -0500 | [diff] [blame] | 299 | // TODO: this could be improved to decode and make use of the mipmap |
| 300 | // levels potentially present in the compressed data. For now, any |
| 301 | // mipmap levels are discarded. |
| 302 | sk_sp<SkImage> SkImage::MakeRasterFromCompressed(sk_sp<SkData> data, |
| 303 | int width, int height, |
| 304 | CompressionType type) { |
| 305 | size_t expectedSize = SkCompressedFormatDataSize(type, { width, height }, false); |
| 306 | if (!data || data->size() < expectedSize) { |
| 307 | return nullptr; |
| 308 | } |
| 309 | |
| 310 | SkAlphaType at = SkCompressionTypeIsOpaque(type) ? kOpaque_SkAlphaType |
| 311 | : kPremul_SkAlphaType; |
| 312 | |
Robert Phillips | 162e04b | 2020-01-28 14:22:43 -0500 | [diff] [blame] | 313 | SkImageInfo ii = SkImageInfo::MakeN32(width, height, at); |
Robert Phillips | 99dead9 | 2020-01-27 16:11:57 -0500 | [diff] [blame] | 314 | |
| 315 | if (!SkImage_Raster::ValidArgs(ii, ii.minRowBytes(), nullptr)) { |
| 316 | return nullptr; |
| 317 | } |
| 318 | |
| 319 | SkBitmap bitmap; |
| 320 | if (!bitmap.tryAllocPixels(ii)) { |
| 321 | return nullptr; |
| 322 | } |
| 323 | |
| 324 | if (!SkDecompress(std::move(data), { width, height }, type, &bitmap)) { |
| 325 | return nullptr; |
| 326 | } |
| 327 | |
| 328 | bitmap.setImmutable(); |
| 329 | return MakeFromBitmap(bitmap); |
| 330 | } |
| 331 | |
reed | 7fb4f8b | 2016-03-11 04:33:52 -0800 | [diff] [blame] | 332 | sk_sp<SkImage> SkImage::MakeFromRaster(const SkPixmap& pmap, RasterReleaseProc proc, |
| 333 | ReleaseContext ctx) { |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 334 | size_t size; |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 335 | if (!SkImage_Raster::ValidArgs(pmap.info(), pmap.rowBytes(), &size) || !pmap.addr()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 336 | return nullptr; |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 337 | } |
| 338 | |
bungeman | 38d909e | 2016-08-02 14:40:46 -0700 | [diff] [blame] | 339 | sk_sp<SkData> data(SkData::MakeWithProc(pmap.addr(), size, proc, ctx)); |
Mike Reed | 086a427 | 2017-07-18 10:53:11 -0400 | [diff] [blame] | 340 | return sk_make_sp<SkImage_Raster>(pmap.info(), std::move(data), pmap.rowBytes()); |
reed | de49988 | 2015-06-18 13:41:40 -0700 | [diff] [blame] | 341 | } |
| 342 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 343 | sk_sp<SkImage> SkMakeImageFromRasterBitmapPriv(const SkBitmap& bm, SkCopyPixelsMode cpm, |
| 344 | uint32_t idForCopy) { |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 345 | if (kAlways_SkCopyPixelsMode == cpm || (!bm.isImmutable() && kNever_SkCopyPixelsMode != cpm)) { |
| 346 | SkPixmap pmap; |
| 347 | if (bm.peekPixels(&pmap)) { |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 348 | return MakeRasterCopyPriv(pmap, idForCopy); |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 349 | } else { |
| 350 | return sk_sp<SkImage>(); |
| 351 | } |
halcanary | c56c6ef | 2015-09-28 11:55:28 -0700 | [diff] [blame] | 352 | } |
| 353 | |
Matt Sarett | b6759dd | 2017-06-12 13:46:11 -0400 | [diff] [blame] | 354 | return sk_make_sp<SkImage_Raster>(bm, kNever_SkCopyPixelsMode == cpm); |
| 355 | } |
| 356 | |
| 357 | sk_sp<SkImage> SkMakeImageFromRasterBitmap(const SkBitmap& bm, SkCopyPixelsMode cpm) { |
Brian Osman | e1adc3a | 2018-06-04 09:21:17 -0400 | [diff] [blame] | 358 | if (!SkImageInfoIsValid(bm.info()) || bm.rowBytes() < bm.info().minRowBytes()) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 359 | return nullptr; |
kkinnunen | 73953e7 | 2015-02-23 22:12:12 -0800 | [diff] [blame] | 360 | } |
| 361 | |
Brian Osman | e531207 | 2017-06-20 09:35:51 -0400 | [diff] [blame] | 362 | return SkMakeImageFromRasterBitmapPriv(bm, cpm, kNeedNewImageUniqueID); |
reed@google.com | 58b21ec | 2012-07-30 18:20:12 +0000 | [diff] [blame] | 363 | } |
| 364 | |
piotaixr | 6515175 | 2014-10-16 11:58:39 -0700 | [diff] [blame] | 365 | const SkPixelRef* SkBitmapImageGetPixelRef(const SkImage* image) { |
| 366 | return ((const SkImage_Raster*)image)->getPixelRef(); |
reed@google.com | 97af1a6 | 2012-08-28 12:19:02 +0000 | [diff] [blame] | 367 | } |
piotaixr | d2a3522 | 2014-08-19 14:29:02 -0700 | [diff] [blame] | 368 | |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 369 | bool SkImage_Raster::onAsLegacyBitmap(GrDirectContext*, SkBitmap* bitmap) const { |
Cary Clark | 4f5a79c | 2018-02-07 15:51:00 -0500 | [diff] [blame] | 370 | // When we're a snapshot from a surface, our bitmap may not be marked immutable |
| 371 | // even though logically always we are, but in that case we can't physically share our |
| 372 | // pixelref since the caller might call setImmutable() themselves |
| 373 | // (thus changing our state). |
| 374 | if (fBitmap.isImmutable()) { |
| 375 | SkIPoint origin = fBitmap.pixelRefOrigin(); |
| 376 | bitmap->setInfo(fBitmap.info(), fBitmap.rowBytes()); |
| 377 | bitmap->setPixelRef(sk_ref_sp(fBitmap.pixelRef()), origin.x(), origin.y()); |
| 378 | return true; |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 379 | } |
Adlai Holler | bcfc554 | 2020-08-27 12:44:07 -0400 | [diff] [blame] | 380 | return this->INHERITED::onAsLegacyBitmap(nullptr, bitmap); |
reed | 3c06511 | 2015-07-08 12:46:22 -0700 | [diff] [blame] | 381 | } |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 382 | |
Matt Sarett | 5b1dba8 | 2017-04-03 11:42:52 -0400 | [diff] [blame] | 383 | /////////////////////////////////////////////////////////////////////////////// |
| 384 | |
Adlai Holler | 3a22017 | 2020-07-15 10:37:50 -0400 | [diff] [blame] | 385 | sk_sp<SkImage> SkImage_Raster::onMakeColorTypeAndColorSpace(SkColorType targetCT, |
| 386 | sk_sp<SkColorSpace> targetCS, |
| 387 | GrDirectContext*) const { |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 388 | SkPixmap src; |
Matt Sarett | 9f3dcb3 | 2017-05-04 08:53:32 -0400 | [diff] [blame] | 389 | SkAssertResult(fBitmap.peekPixels(&src)); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 390 | |
Matt Sarett | d3df9ec | 2017-06-05 10:45:30 -0400 | [diff] [blame] | 391 | SkBitmap dst; |
Brian Osman | f48c996 | 2019-01-14 11:15:50 -0500 | [diff] [blame] | 392 | dst.allocPixels(fBitmap.info().makeColorType(targetCT).makeColorSpace(targetCS)); |
Matt Sarett | d3df9ec | 2017-06-05 10:45:30 -0400 | [diff] [blame] | 393 | |
Brian Osman | be686f0 | 2018-10-12 11:18:02 -0400 | [diff] [blame] | 394 | SkAssertResult(dst.writePixels(src)); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 395 | dst.setImmutable(); |
Mike Reed | dc607e3 | 2020-12-23 11:50:36 -0500 | [diff] [blame] | 396 | return dst.asImage(); |
Matt Sarett | 6de1310 | 2017-03-14 14:10:48 -0400 | [diff] [blame] | 397 | } |
Brian Osman | d514837 | 2019-08-14 16:14:51 -0400 | [diff] [blame] | 398 | |
| 399 | sk_sp<SkImage> SkImage_Raster::onReinterpretColorSpace(sk_sp<SkColorSpace> newCS) const { |
| 400 | // TODO: If our bitmap is immutable, then we could theoretically create another image sharing |
| 401 | // our pixelRef. That doesn't work (without more invasive logic), because the image gets its |
| 402 | // gen ID from the bitmap, which gets it from the pixelRef. |
| 403 | SkPixmap pixmap = fBitmap.pixmap(); |
| 404 | pixmap.setColorSpace(std::move(newCS)); |
| 405 | return SkImage::MakeRasterCopy(pixmap); |
| 406 | } |
Brian Salomon | d0924f3 | 2021-02-03 10:15:31 -0500 | [diff] [blame] | 407 | |
| 408 | #if SK_SUPPORT_GPU |
| 409 | std::tuple<GrSurfaceProxyView, GrColorType> SkImage_Raster::onAsView( |
| 410 | GrRecordingContext* context, |
| 411 | GrMipmapped mipmapped, |
| 412 | GrImageTexGenPolicy policy) const { |
| 413 | if (fPinnedView) { |
| 414 | if (policy != GrImageTexGenPolicy::kDraw) { |
| 415 | return {CopyView(context, fPinnedView, mipmapped, policy), fPinnedColorType}; |
| 416 | } |
| 417 | GrColorInfo colorInfo(fPinnedColorType, this->alphaType(), this->refColorSpace()); |
| 418 | GrTextureAdjuster adjuster(context, fPinnedView, colorInfo, fPinnedUniqueID); |
| 419 | return {adjuster.view(mipmapped), adjuster.colorType()}; |
| 420 | } |
| 421 | |
| 422 | GrBitmapTextureMaker maker(context, fBitmap, policy); |
| 423 | return {maker.view(mipmapped), maker.colorType()}; |
| 424 | } |
| 425 | #endif |