commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014 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 "src/shaders/SkPictureShader.h" |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 9 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 10 | #include "include/core/SkBitmap.h" |
| 11 | #include "include/core/SkCanvas.h" |
| 12 | #include "include/core/SkImage.h" |
Ben Wagner | 729a23f | 2019-05-17 16:29:34 -0400 | [diff] [blame] | 13 | #include "src/core/SkArenaAlloc.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 14 | #include "src/core/SkMatrixUtils.h" |
| 15 | #include "src/core/SkPicturePriv.h" |
| 16 | #include "src/core/SkReadBuffer.h" |
| 17 | #include "src/core/SkResourceCache.h" |
| 18 | #include "src/shaders/SkBitmapProcShader.h" |
| 19 | #include "src/shaders/SkImageShader.h" |
Mike Klein | 0ec1c57 | 2018-12-04 11:52:51 -0500 | [diff] [blame] | 20 | #include <atomic> |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 21 | |
| 22 | #if SK_SUPPORT_GPU |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 23 | #include "include/private/GrRecordingContext.h" |
| 24 | #include "src/gpu/GrCaps.h" |
Brian Salomon | 4bc0c1f | 2019-09-30 15:12:27 -0400 | [diff] [blame] | 25 | #include "src/gpu/GrColorInfo.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 26 | #include "src/gpu/GrFragmentProcessor.h" |
| 27 | #include "src/gpu/GrRecordingContextPriv.h" |
| 28 | #include "src/gpu/SkGr.h" |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 29 | #endif |
| 30 | |
Mike Reed | e25b447 | 2019-04-02 17:49:12 -0400 | [diff] [blame] | 31 | sk_sp<SkShader> SkPicture::makeShader(SkTileMode tmx, SkTileMode tmy, const SkMatrix* localMatrix, |
| 32 | const SkRect* tile) const { |
| 33 | if (localMatrix && !localMatrix->invert(nullptr)) { |
| 34 | return nullptr; |
| 35 | } |
| 36 | return SkPictureShader::Make(sk_ref_sp(this), tmx, tmy, localMatrix, tile); |
| 37 | } |
| 38 | |
| 39 | sk_sp<SkShader> SkPicture::makeShader(SkTileMode tmx, SkTileMode tmy, |
| 40 | const SkMatrix* localMatrix) const { |
| 41 | return this->makeShader(tmx, tmy, localMatrix, nullptr); |
| 42 | } |
| 43 | |
fmalita | 171e5b7 | 2014-10-22 11:20:40 -0700 | [diff] [blame] | 44 | namespace { |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 45 | static unsigned gBitmapShaderKeyNamespaceLabel; |
fmalita | 171e5b7 | 2014-10-22 11:20:40 -0700 | [diff] [blame] | 46 | |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 47 | struct BitmapShaderKey : public SkResourceCache::Key { |
| 48 | public: |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 49 | BitmapShaderKey(SkColorSpace* colorSpace, |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 50 | SkImage::BitDepth bitDepth, |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 51 | uint32_t shaderID, |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 52 | const SkSize& scale) |
| 53 | : fColorSpaceXYZHash(colorSpace->toXYZD50Hash()) |
| 54 | , fColorSpaceTransferFnHash(colorSpace->transferFnHash()) |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 55 | , fBitDepth(bitDepth) |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 56 | , fScale(scale) { |
fmalita | 04b49c3 | 2014-12-10 13:01:43 -0800 | [diff] [blame] | 57 | |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 58 | static const size_t keySize = sizeof(fColorSpaceXYZHash) + |
| 59 | sizeof(fColorSpaceTransferFnHash) + |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 60 | sizeof(fBitDepth) + |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 61 | sizeof(fScale); |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 62 | // This better be packed. |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 63 | SkASSERT(sizeof(uint32_t) * (&fEndOfStruct - &fColorSpaceXYZHash) == keySize); |
| 64 | this->init(&gBitmapShaderKeyNamespaceLabel, MakeSharedID(shaderID), keySize); |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 65 | } |
| 66 | |
| 67 | static uint64_t MakeSharedID(uint32_t shaderID) { |
| 68 | uint64_t sharedID = SkSetFourByteTag('p', 's', 'd', 'r'); |
| 69 | return (sharedID << 32) | shaderID; |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 70 | } |
| 71 | |
| 72 | private: |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 73 | uint32_t fColorSpaceXYZHash; |
| 74 | uint32_t fColorSpaceTransferFnHash; |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 75 | SkImage::BitDepth fBitDepth; |
Matt Sarett | 28a7ad2 | 2017-04-21 15:12:34 -0400 | [diff] [blame] | 76 | SkSize fScale; |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 77 | |
| 78 | SkDEBUGCODE(uint32_t fEndOfStruct;) |
| 79 | }; |
| 80 | |
| 81 | struct BitmapShaderRec : public SkResourceCache::Rec { |
fmalita | c9a9ca9 | 2016-10-12 13:43:43 -0700 | [diff] [blame] | 82 | BitmapShaderRec(const BitmapShaderKey& key, SkShader* tileShader) |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 83 | : fKey(key) |
fmalita | c9a9ca9 | 2016-10-12 13:43:43 -0700 | [diff] [blame] | 84 | , fShader(SkRef(tileShader)) {} |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 85 | |
Hal Canary | 704cd32 | 2016-11-07 14:13:52 -0500 | [diff] [blame] | 86 | BitmapShaderKey fKey; |
| 87 | sk_sp<SkShader> fShader; |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 88 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 89 | const Key& getKey() const override { return fKey; } |
| 90 | size_t bytesUsed() const override { |
Brian Osman | bd65955 | 2018-09-11 10:03:19 -0400 | [diff] [blame] | 91 | // Just the record overhead -- the actual pixels are accounted by SkImage_Lazy. |
fmalita | c9a9ca9 | 2016-10-12 13:43:43 -0700 | [diff] [blame] | 92 | return sizeof(fKey) + sizeof(SkImageShader); |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 93 | } |
reed | 216b643 | 2015-08-19 12:25:40 -0700 | [diff] [blame] | 94 | const char* getCategory() const override { return "bitmap-shader"; } |
| 95 | SkDiscardableMemory* diagnostic_only_getDiscardable() const override { return nullptr; } |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 96 | |
| 97 | static bool Visitor(const SkResourceCache::Rec& baseRec, void* contextShader) { |
| 98 | const BitmapShaderRec& rec = static_cast<const BitmapShaderRec&>(baseRec); |
Hal Canary | 704cd32 | 2016-11-07 14:13:52 -0500 | [diff] [blame] | 99 | sk_sp<SkShader>* result = reinterpret_cast<sk_sp<SkShader>*>(contextShader); |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 100 | |
Hal Canary | 704cd32 | 2016-11-07 14:13:52 -0500 | [diff] [blame] | 101 | *result = rec.fShader; |
fmalita | 387a01a | 2014-12-10 12:17:58 -0800 | [diff] [blame] | 102 | |
fmalita | ddc4b46 | 2015-08-31 19:54:03 -0700 | [diff] [blame] | 103 | // The bitmap shader is backed by an image generator, thus it can always re-generate its |
| 104 | // pixels if discarded. |
| 105 | return true; |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 106 | } |
| 107 | }; |
| 108 | |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 109 | uint32_t next_id() { |
Mike Klein | 0ec1c57 | 2018-12-04 11:52:51 -0500 | [diff] [blame] | 110 | static std::atomic<uint32_t> nextID{1}; |
| 111 | |
| 112 | uint32_t id; |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 113 | do { |
Mike Klein | 0ec1c57 | 2018-12-04 11:52:51 -0500 | [diff] [blame] | 114 | id = nextID++; |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 115 | } while (id == SK_InvalidGenID); |
Mike Klein | 0ec1c57 | 2018-12-04 11:52:51 -0500 | [diff] [blame] | 116 | return id; |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 117 | } |
| 118 | |
fmalita | 171e5b7 | 2014-10-22 11:20:40 -0700 | [diff] [blame] | 119 | } // namespace |
| 120 | |
Mike Reed | fae8fce | 2019-04-03 10:27:45 -0400 | [diff] [blame] | 121 | SkPictureShader::SkPictureShader(sk_sp<SkPicture> picture, SkTileMode tmx, SkTileMode tmy, |
Brian Osman | f9666f5 | 2019-03-21 12:35:02 -0400 | [diff] [blame] | 122 | const SkMatrix* localMatrix, const SkRect* tile) |
commit-bot@chromium.org | 5aacfe9 | 2014-05-02 21:23:52 +0000 | [diff] [blame] | 123 | : INHERITED(localMatrix) |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 124 | , fPicture(std::move(picture)) |
| 125 | , fTile(tile ? *tile : fPicture->cullRect()) |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 126 | , fTmx(tmx) |
Matt Sarett | 28a7ad2 | 2017-04-21 15:12:34 -0400 | [diff] [blame] | 127 | , fTmy(tmy) |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 128 | , fUniqueID(next_id()) |
| 129 | , fAddedToCache(false) {} |
| 130 | |
| 131 | SkPictureShader::~SkPictureShader() { |
| 132 | if (fAddedToCache.load()) { |
| 133 | SkResourceCache::PostPurgeSharedID(BitmapShaderKey::MakeSharedID(fUniqueID)); |
| 134 | } |
| 135 | } |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 136 | |
Mike Reed | e25b447 | 2019-04-02 17:49:12 -0400 | [diff] [blame] | 137 | sk_sp<SkShader> SkPictureShader::Make(sk_sp<SkPicture> picture, SkTileMode tmx, SkTileMode tmy, |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 138 | const SkMatrix* localMatrix, const SkRect* tile) { |
Robert Phillips | 54be5c9 | 2017-02-11 01:19:30 +0000 | [diff] [blame] | 139 | if (!picture || picture->cullRect().isEmpty() || (tile && tile->isEmpty())) { |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 140 | return SkShaders::Empty(); |
commit-bot@chromium.org | 855e88e | 2014-04-21 19:33:12 +0000 | [diff] [blame] | 141 | } |
Mike Reed | fae8fce | 2019-04-03 10:27:45 -0400 | [diff] [blame] | 142 | return sk_sp<SkShader>(new SkPictureShader(std::move(picture), tmx, tmy, localMatrix, tile)); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 143 | } |
| 144 | |
Hal Canary | 8a64fa6 | 2019-05-30 10:29:40 -0400 | [diff] [blame] | 145 | SkPicture* SkPictureShader::isAPicture(SkMatrix* matrix, |
| 146 | SkTileMode tileModes[2], |
| 147 | SkRect* tile) const { |
| 148 | if (matrix) { |
| 149 | *matrix = this->getLocalMatrix(); |
| 150 | } |
| 151 | if (tileModes) { |
| 152 | tileModes[0] = fTmx; |
| 153 | tileModes[1] = fTmy; |
| 154 | } |
| 155 | if (tile) { |
| 156 | *tile = fTile; |
| 157 | } |
| 158 | return fPicture.get(); |
| 159 | } |
| 160 | |
reed | 60c9b58 | 2016-04-03 09:11:13 -0700 | [diff] [blame] | 161 | sk_sp<SkFlattenable> SkPictureShader::CreateProc(SkReadBuffer& buffer) { |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 162 | SkMatrix lm; |
| 163 | buffer.readMatrix(&lm); |
Mike Reed | e25b447 | 2019-04-02 17:49:12 -0400 | [diff] [blame] | 164 | auto tmx = buffer.read32LE(SkTileMode::kLastTileMode); |
| 165 | auto tmy = buffer.read32LE(SkTileMode::kLastTileMode); |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 166 | SkRect tile; |
| 167 | buffer.readRect(&tile); |
mtklein | 76be9c8 | 2015-05-20 12:05:15 -0700 | [diff] [blame] | 168 | |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 169 | sk_sp<SkPicture> picture; |
hendrikw | 446ee67 | 2015-06-16 09:28:37 -0700 | [diff] [blame] | 170 | |
Mike Reed | 5a0f345 | 2018-01-19 08:25:11 -0500 | [diff] [blame] | 171 | bool didSerialize = buffer.readBool(); |
| 172 | if (didSerialize) { |
Cary Clark | efd99cc | 2018-06-11 16:25:43 -0400 | [diff] [blame] | 173 | picture = SkPicturePriv::MakeFromBuffer(buffer); |
mtklein | 76be9c8 | 2015-05-20 12:05:15 -0700 | [diff] [blame] | 174 | } |
Mike Reed | e25b447 | 2019-04-02 17:49:12 -0400 | [diff] [blame] | 175 | return SkPictureShader::Make(picture, tmx, tmy, &lm, &tile); |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 176 | } |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 177 | |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 178 | void SkPictureShader::flatten(SkWriteBuffer& buffer) const { |
| 179 | buffer.writeMatrix(this->getLocalMatrix()); |
Mike Reed | fae8fce | 2019-04-03 10:27:45 -0400 | [diff] [blame] | 180 | buffer.write32((unsigned)fTmx); |
| 181 | buffer.write32((unsigned)fTmy); |
fmalita | b5f7826 | 2014-08-06 13:07:15 -0700 | [diff] [blame] | 182 | buffer.writeRect(fTile); |
mtklein | 76be9c8 | 2015-05-20 12:05:15 -0700 | [diff] [blame] | 183 | |
Mike Reed | 5a0f345 | 2018-01-19 08:25:11 -0500 | [diff] [blame] | 184 | buffer.writeBool(true); |
Cary Clark | efd99cc | 2018-06-11 16:25:43 -0400 | [diff] [blame] | 185 | SkPicturePriv::Flatten(fPicture, buffer); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 186 | } |
| 187 | |
Florin Malita | 8f88d89 | 2018-04-19 21:07:19 -0400 | [diff] [blame] | 188 | // Returns a cached image shader, which wraps a single picture tile at the given |
| 189 | // CTM/local matrix. Also adjusts the local matrix for tile scaling. |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 190 | sk_sp<SkShader> SkPictureShader::refBitmapShader(const SkMatrix& viewMatrix, |
Florin Malita | 8f88d89 | 2018-04-19 21:07:19 -0400 | [diff] [blame] | 191 | SkTCopyOnFirstWrite<SkMatrix>* localMatrix, |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 192 | SkColorType dstColorType, |
Brian Osman | 138ea97 | 2016-12-16 11:55:18 -0500 | [diff] [blame] | 193 | SkColorSpace* dstColorSpace, |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 194 | const int maxTextureSize) const { |
robertphillips | a8d7f0b | 2014-08-29 08:03:56 -0700 | [diff] [blame] | 195 | SkASSERT(fPicture && !fPicture->cullRect().isEmpty()); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 196 | |
Florin Malita | 8f88d89 | 2018-04-19 21:07:19 -0400 | [diff] [blame] | 197 | const SkMatrix m = SkMatrix::Concat(viewMatrix, **localMatrix); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 198 | |
| 199 | // Use a rotation-invariant scale |
| 200 | SkPoint scale; |
reed | adf9990 | 2015-03-19 16:10:54 -0700 | [diff] [blame] | 201 | // |
| 202 | // TODO: replace this with decomposeScale() -- but beware LayoutTest rebaselines! |
| 203 | // |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 204 | if (!SkDecomposeUpper2x2(m, nullptr, &scale, nullptr)) { |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 205 | // Decomposition failed, use an approximation. |
| 206 | scale.set(SkScalarSqrt(m.getScaleX() * m.getScaleX() + m.getSkewX() * m.getSkewX()), |
| 207 | SkScalarSqrt(m.getScaleY() * m.getScaleY() + m.getSkewY() * m.getSkewY())); |
| 208 | } |
fmalita | b087879 | 2015-01-15 10:45:56 -0800 | [diff] [blame] | 209 | SkSize scaledSize = SkSize::Make(SkScalarAbs(scale.x() * fTile.width()), |
| 210 | SkScalarAbs(scale.y() * fTile.height())); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 211 | |
fmalita | 511005b | 2015-03-13 06:50:44 -0700 | [diff] [blame] | 212 | // Clamp the tile size to about 4M pixels |
| 213 | static const SkScalar kMaxTileArea = 2048 * 2048; |
Mike Reed | a99b6ce | 2017-02-04 11:04:26 -0500 | [diff] [blame] | 214 | SkScalar tileArea = scaledSize.width() * scaledSize.height(); |
fmalita | bb204f4 | 2014-08-07 08:39:24 -0700 | [diff] [blame] | 215 | if (tileArea > kMaxTileArea) { |
reed | 80ea19c | 2015-05-12 10:37:34 -0700 | [diff] [blame] | 216 | SkScalar clampScale = SkScalarSqrt(kMaxTileArea / tileArea); |
Mike Reed | a99b6ce | 2017-02-04 11:04:26 -0500 | [diff] [blame] | 217 | scaledSize.set(scaledSize.width() * clampScale, |
| 218 | scaledSize.height() * clampScale); |
fmalita | bb204f4 | 2014-08-07 08:39:24 -0700 | [diff] [blame] | 219 | } |
gen.kim | b9ed884 | 2015-05-03 22:36:30 -0700 | [diff] [blame] | 220 | #if SK_SUPPORT_GPU |
| 221 | // Scale down the tile size if larger than maxTextureSize for GPU Path or it should fail on create texture |
| 222 | if (maxTextureSize) { |
| 223 | if (scaledSize.width() > maxTextureSize || scaledSize.height() > maxTextureSize) { |
ericrk | f469fc0 | 2015-10-14 17:33:29 -0700 | [diff] [blame] | 224 | SkScalar downScale = maxTextureSize / SkMaxScalar(scaledSize.width(), scaledSize.height()); |
Mike Reed | a99b6ce | 2017-02-04 11:04:26 -0500 | [diff] [blame] | 225 | scaledSize.set(SkScalarFloorToScalar(scaledSize.width() * downScale), |
| 226 | SkScalarFloorToScalar(scaledSize.height() * downScale)); |
gen.kim | b9ed884 | 2015-05-03 22:36:30 -0700 | [diff] [blame] | 227 | } |
| 228 | } |
| 229 | #endif |
fmalita | bb204f4 | 2014-08-07 08:39:24 -0700 | [diff] [blame] | 230 | |
fmalita | 85c6f98 | 2016-02-29 09:18:31 -0800 | [diff] [blame] | 231 | const SkISize tileSize = scaledSize.toCeil(); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 232 | if (tileSize.isEmpty()) { |
Mike Reed | c8bea7d | 2019-04-09 13:55:36 -0400 | [diff] [blame] | 233 | return SkShaders::Empty(); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 234 | } |
| 235 | |
fmalita | bb204f4 | 2014-08-07 08:39:24 -0700 | [diff] [blame] | 236 | // The actual scale, compensating for rounding & clamping. |
fmalita | 85c6f98 | 2016-02-29 09:18:31 -0800 | [diff] [blame] | 237 | const SkSize tileScale = SkSize::Make(SkIntToScalar(tileSize.width()) / fTile.width(), |
| 238 | SkIntToScalar(tileSize.height()) / fTile.height()); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 239 | |
Matt Sarett | 28a7ad2 | 2017-04-21 15:12:34 -0400 | [diff] [blame] | 240 | |
Brian Osman | f9666f5 | 2019-03-21 12:35:02 -0400 | [diff] [blame] | 241 | sk_sp<SkColorSpace> imgCS = dstColorSpace ? sk_ref_sp(dstColorSpace): SkColorSpace::MakeSRGB(); |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 242 | SkImage::BitDepth bitDepth = |
Brian Osman | dfddf34 | 2019-03-07 13:33:45 -0500 | [diff] [blame] | 243 | dstColorType >= kRGBA_F16Norm_SkColorType |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 244 | ? SkImage::BitDepth::kF16 : SkImage::BitDepth::kU8; |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 245 | |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 246 | BitmapShaderKey key(imgCS.get(), bitDepth, fUniqueID, tileScale); |
| 247 | |
| 248 | sk_sp<SkShader> tileShader; |
fmalita | 171e5b7 | 2014-10-22 11:20:40 -0700 | [diff] [blame] | 249 | if (!SkResourceCache::Find(key, BitmapShaderRec::Visitor, &tileShader)) { |
robertphillips | 25a5b0d | 2015-09-28 09:32:50 -0700 | [diff] [blame] | 250 | SkMatrix tileMatrix; |
| 251 | tileMatrix.setRectToRect(fTile, SkRect::MakeIWH(tileSize.width(), tileSize.height()), |
fmalita | 1b46a57 | 2016-02-01 02:34:03 -0800 | [diff] [blame] | 252 | SkMatrix::kFill_ScaleToFit); |
| 253 | |
Brian Osman | e1552a5 | 2018-10-18 10:40:37 -0400 | [diff] [blame] | 254 | sk_sp<SkImage> tileImage = SkImage::MakeFromPicture(fPicture, tileSize, &tileMatrix, |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 255 | nullptr, bitDepth, std::move(imgCS)); |
fmalita | 1b46a57 | 2016-02-01 02:34:03 -0800 | [diff] [blame] | 256 | if (!tileImage) { |
fmalita | ddc4b46 | 2015-08-31 19:54:03 -0700 | [diff] [blame] | 257 | return nullptr; |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 258 | } |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 259 | |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 260 | tileShader = tileImage->makeShader(fTmx, fTmy); |
fmalita | 23df2d6 | 2014-10-22 07:39:08 -0700 | [diff] [blame] | 261 | |
fmalita | c9a9ca9 | 2016-10-12 13:43:43 -0700 | [diff] [blame] | 262 | SkResourceCache::Add(new BitmapShaderRec(key, tileShader.get())); |
Florin Malita | b00a360 | 2017-07-13 22:34:04 -0400 | [diff] [blame] | 263 | fAddedToCache.store(true); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 264 | } |
| 265 | |
Florin Malita | 8f88d89 | 2018-04-19 21:07:19 -0400 | [diff] [blame] | 266 | if (tileScale.width() != 1 || tileScale.height() != 1) { |
| 267 | localMatrix->writable()->preScale(1 / tileScale.width(), 1 / tileScale.height()); |
| 268 | } |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 269 | |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 270 | return tileShader; |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 271 | } |
| 272 | |
Mike Reed | 1386b2d | 2019-03-13 21:15:05 -0400 | [diff] [blame] | 273 | bool SkPictureShader::onAppendStages(const SkStageRec& rec) const { |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 274 | auto lm = this->totalLocalMatrix(rec.fLocalM); |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 275 | |
Yuqian Li | 8d2fb47 | 2017-01-30 11:33:46 -0500 | [diff] [blame] | 276 | // Keep bitmapShader alive by using alloc instead of stack memory |
Mike Reed | 1d8c42e | 2017-08-29 14:58:19 -0400 | [diff] [blame] | 277 | auto& bitmapShader = *rec.fAlloc->make<sk_sp<SkShader>>(); |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 278 | bitmapShader = this->refBitmapShader(rec.fCTM, &lm, rec.fDstColorType, rec.fDstCS); |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 279 | |
| 280 | if (!bitmapShader) { |
| 281 | return false; |
| 282 | } |
| 283 | |
Mike Reed | 1386b2d | 2019-03-13 21:15:05 -0400 | [diff] [blame] | 284 | SkStageRec localRec = rec; |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 285 | localRec.fLocalM = lm->isIdentity() ? nullptr : lm.get(); |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 286 | |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 287 | return as_SB(bitmapShader)->appendStages(localRec); |
Yuqian Li | 8d2fb47 | 2017-01-30 11:33:46 -0500 | [diff] [blame] | 288 | } |
| 289 | |
commit-bot@chromium.org | ce56d96 | 2014-05-05 18:39:18 +0000 | [diff] [blame] | 290 | ///////////////////////////////////////////////////////////////////////////////////////// |
Mike Reed | e92aae6 | 2018-10-17 10:21:51 -0400 | [diff] [blame] | 291 | |
| 292 | #ifdef SK_ENABLE_LEGACY_SHADERCONTEXT |
Florin Malita | 4aed138 | 2017-05-25 10:38:07 -0400 | [diff] [blame] | 293 | SkShaderBase::Context* SkPictureShader::onMakeContext(const ContextRec& rec, SkArenaAlloc* alloc) |
Herb Derby | 83e939b | 2017-02-07 14:25:11 -0500 | [diff] [blame] | 294 | const { |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 295 | auto lm = this->totalLocalMatrix(rec.fLocalMatrix); |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 296 | sk_sp<SkShader> bitmapShader = this->refBitmapShader(*rec.fMatrix, &lm, rec.fDstColorType, |
| 297 | rec.fDstColorSpace); |
Herb Derby | 83e939b | 2017-02-07 14:25:11 -0500 | [diff] [blame] | 298 | if (!bitmapShader) { |
| 299 | return nullptr; |
| 300 | } |
commit-bot@chromium.org | ce56d96 | 2014-05-05 18:39:18 +0000 | [diff] [blame] | 301 | |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 302 | ContextRec localRec = rec; |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 303 | localRec.fLocalMatrix = lm->isIdentity() ? nullptr : lm.get(); |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 304 | |
Herb Derby | 83e939b | 2017-02-07 14:25:11 -0500 | [diff] [blame] | 305 | PictureShaderContext* ctx = |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 306 | alloc->make<PictureShaderContext>(*this, localRec, std::move(bitmapShader), alloc); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 307 | if (nullptr == ctx->fBitmapShaderContext) { |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 308 | ctx = nullptr; |
commit-bot@chromium.org | ce56d96 | 2014-05-05 18:39:18 +0000 | [diff] [blame] | 309 | } |
| 310 | return ctx; |
| 311 | } |
Mike Reed | e92aae6 | 2018-10-17 10:21:51 -0400 | [diff] [blame] | 312 | #endif |
commit-bot@chromium.org | ce56d96 | 2014-05-05 18:39:18 +0000 | [diff] [blame] | 313 | |
Herb Derby | 83e939b | 2017-02-07 14:25:11 -0500 | [diff] [blame] | 314 | ///////////////////////////////////////////////////////////////////////////////////////// |
| 315 | |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 316 | SkPictureShader::PictureShaderContext::PictureShaderContext( |
Herb Derby | 83e939b | 2017-02-07 14:25:11 -0500 | [diff] [blame] | 317 | const SkPictureShader& shader, const ContextRec& rec, sk_sp<SkShader> bitmapShader, |
| 318 | SkArenaAlloc* alloc) |
commit-bot@chromium.org | e901b6d | 2014-05-01 19:31:31 +0000 | [diff] [blame] | 319 | : INHERITED(shader, rec) |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 320 | , fBitmapShader(std::move(bitmapShader)) |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 321 | { |
Florin Malita | 4aed138 | 2017-05-25 10:38:07 -0400 | [diff] [blame] | 322 | fBitmapShaderContext = as_SB(fBitmapShader)->makeContext(rec, alloc); |
commit-bot@chromium.org | ce56d96 | 2014-05-05 18:39:18 +0000 | [diff] [blame] | 323 | //if fBitmapShaderContext is null, we are invalid |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 324 | } |
| 325 | |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 326 | uint32_t SkPictureShader::PictureShaderContext::getFlags() const { |
commit-bot@chromium.org | ce56d96 | 2014-05-05 18:39:18 +0000 | [diff] [blame] | 327 | SkASSERT(fBitmapShaderContext); |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 328 | return fBitmapShaderContext->getFlags(); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 329 | } |
| 330 | |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 331 | void SkPictureShader::PictureShaderContext::shadeSpan(int x, int y, SkPMColor dstC[], int count) { |
| 332 | SkASSERT(fBitmapShaderContext); |
| 333 | fBitmapShaderContext->shadeSpan(x, y, dstC, count); |
| 334 | } |
| 335 | |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 336 | #if SK_SUPPORT_GPU |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 337 | #include "include/gpu/GrContext.h" |
| 338 | #include "src/gpu/GrContextPriv.h" |
Robert Phillips | 1efecea | 2019-02-15 16:58:40 -0500 | [diff] [blame] | 339 | |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 340 | std::unique_ptr<GrFragmentProcessor> SkPictureShader::asFragmentProcessor( |
Mike Reed | e3429e6 | 2018-01-19 11:43:34 -0500 | [diff] [blame] | 341 | const GrFPArgs& args) const { |
gen.kim | b9ed884 | 2015-05-03 22:36:30 -0700 | [diff] [blame] | 342 | int maxTextureSize = 0; |
brianosman | 839345d | 2016-07-22 11:04:53 -0700 | [diff] [blame] | 343 | if (args.fContext) { |
Robert Phillips | 9da87e0 | 2019-02-04 13:26:26 -0500 | [diff] [blame] | 344 | maxTextureSize = args.fContext->priv().caps()->maxTextureSize(); |
gen.kim | b9ed884 | 2015-05-03 22:36:30 -0700 | [diff] [blame] | 345 | } |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 346 | |
| 347 | auto lm = this->totalLocalMatrix(args.fPreLocalMatrix, args.fPostLocalMatrix); |
Brian Salomon | 4bc0c1f | 2019-09-30 15:12:27 -0400 | [diff] [blame] | 348 | SkColorType dstColorType = GrColorTypeToSkColorType(args.fDstColorInfo->colorType()); |
Brian Salomon | bd3d8d3 | 2019-07-02 09:16:28 -0400 | [diff] [blame] | 349 | if (dstColorType == kUnknown_SkColorType) { |
| 350 | dstColorType = kRGBA_8888_SkColorType; |
| 351 | } |
Brian Osman | 2e8f48e | 2018-10-19 13:50:48 -0400 | [diff] [blame] | 352 | sk_sp<SkShader> bitmapShader(this->refBitmapShader(*args.fViewMatrix, &lm, dstColorType, |
Brian Salomon | 4bc0c1f | 2019-09-30 15:12:27 -0400 | [diff] [blame] | 353 | args.fDstColorInfo->colorSpace(), |
Florin Malita | 8f88d89 | 2018-04-19 21:07:19 -0400 | [diff] [blame] | 354 | maxTextureSize)); |
commit-bot@chromium.org | 87fcd95 | 2014-04-23 19:10:51 +0000 | [diff] [blame] | 355 | if (!bitmapShader) { |
bsalomon | c21b09e | 2015-08-28 18:46:56 -0700 | [diff] [blame] | 356 | return nullptr; |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 357 | } |
Florin Malita | 18f7e0a | 2018-02-06 16:49:40 -0500 | [diff] [blame] | 358 | |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 359 | // We want to *reset* args.fPreLocalMatrix, not compose it. |
Brian Salomon | 4bc0c1f | 2019-09-30 15:12:27 -0400 | [diff] [blame] | 360 | GrFPArgs newArgs(args.fContext, args.fViewMatrix, args.fFilterQuality, args.fDstColorInfo); |
Florin Malita | c6c5ead | 2018-04-11 15:33:40 -0400 | [diff] [blame] | 361 | newArgs.fPreLocalMatrix = lm.get(); |
| 362 | |
| 363 | return as_SB(bitmapShader)->asFragmentProcessor(newArgs); |
commit-bot@chromium.org | c5d9bb0 | 2014-04-08 15:19:34 +0000 | [diff] [blame] | 364 | } |
| 365 | #endif |