blob: 495a6aa4de4ebf49516755175201423798d5a090 [file] [log] [blame]
scroggo478652e2015-03-25 07:11:02 -07001/*
2 * Copyright 2015 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "dm/DMSrcSink.h"
Tyler Denniston45f94f82020-02-04 16:09:08 -05009#include "gm/verifiers/gmverifier.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050010#include "include/codec/SkAndroidCodec.h"
11#include "include/codec/SkCodec.h"
12#include "include/core/SkColorSpace.h"
13#include "include/core/SkData.h"
14#include "include/core/SkDeferredDisplayListRecorder.h"
15#include "include/core/SkDocument.h"
16#include "include/core/SkExecutor.h"
17#include "include/core/SkImageGenerator.h"
18#include "include/core/SkMallocPixelRef.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050019#include "include/core/SkPictureRecorder.h"
20#include "include/core/SkStream.h"
21#include "include/core/SkSurface.h"
22#include "include/core/SkSurfaceCharacterization.h"
23#include "include/docs/SkPDFDocument.h"
24#include "include/gpu/GrBackendSurface.h"
Robert Phillipsb87b39b2020-07-01 14:45:24 -040025#include "include/gpu/GrDirectContext.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050026#include "include/ports/SkImageGeneratorCG.h"
Leon Scroggins IIIf21d6b92020-08-05 10:44:17 -040027#include "include/ports/SkImageGeneratorNDK.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050028#include "include/ports/SkImageGeneratorWIC.h"
29#include "include/private/SkImageInfoPriv.h"
30#include "include/private/SkTLogic.h"
Brian Osmanea236bf2019-04-29 10:28:22 -040031#include "include/third_party/skcms/skcms.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050032#include "include/utils/SkNullCanvas.h"
John Stiles65ec1982021-07-20 12:53:54 -040033#include "include/utils/SkPaintFilterCanvas.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050034#include "include/utils/SkRandom.h"
Florin Malitafbddfbb2020-05-06 15:55:18 -040035#include "modules/skottie/utils/SkottieUtils.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050036#include "src/codec/SkCodecImageGenerator.h"
37#include "src/codec/SkSwizzler.h"
38#include "src/core/SkAutoMalloc.h"
39#include "src/core/SkAutoPixmapStorage.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050040#include "src/core/SkOSFile.h"
41#include "src/core/SkOpts.h"
42#include "src/core/SkPictureCommon.h"
43#include "src/core/SkPictureData.h"
44#include "src/core/SkRecordDraw.h"
45#include "src/core/SkRecorder.h"
46#include "src/core/SkTaskGroup.h"
Adlai Hollera0693042020-10-14 11:23:11 -040047#include "src/gpu/GrDirectContextPriv.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050048#include "src/gpu/GrGpu.h"
49#include "src/utils/SkMultiPictureDocumentPriv.h"
50#include "src/utils/SkOSPath.h"
51#include "tools/DDLPromiseImageHelper.h"
52#include "tools/DDLTileHelper.h"
53#include "tools/Resources.h"
John Stiles65ec1982021-07-20 12:53:54 -040054#include "tools/RuntimeBlendUtils.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050055#include "tools/debugger/DebugCanvas.h"
Brian Salomon72c7b982020-10-06 10:07:38 -040056#include "tools/gpu/BackendSurfaceFactory.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050057#include "tools/gpu/MemoryCache.h"
msarettfc0b6d12016-03-17 13:50:17 -070058#if defined(SK_BUILD_FOR_WIN)
Mike Kleinc0bd9f92019-04-23 12:05:21 -050059 #include "include/docs/SkXPSDocument.h"
60 #include "src/utils/win/SkAutoCoInitialize.h"
61 #include "src/utils/win/SkHRESULT.h"
62 #include "src/utils/win/SkTScopedComPtr.h"
Hal Canary5e221e72017-02-06 09:51:42 -050063 #include <XpsObjectModel.h>
msarettfc0b6d12016-03-17 13:50:17 -070064#endif
65
Florin Malita87ccf332018-05-04 12:23:24 -040066#if defined(SK_ENABLE_SKOTTIE)
Mike Kleinc0bd9f92019-04-23 12:05:21 -050067 #include "modules/skottie/include/Skottie.h"
Brian Osman849f4d62019-11-26 08:58:26 -050068 #include "modules/skresources/include/SkResources.h"
Florin Malita124d5af2017-12-31 17:02:26 -050069#endif
70
Florin Malita02616ea2020-06-25 13:33:17 -040071#if defined(SK_ENABLE_SKRIVE)
72 #include "experimental/skrive/include/SkRive.h"
73#endif
74
Robert Phillips2af13c12021-09-01 16:47:01 +000075#if defined(SK_ENABLE_SVG)
Mike Kleinc0bd9f92019-04-23 12:05:21 -050076 #include "include/svg/SkSVGCanvas.h"
Florin Malitab3418102020-10-15 18:10:29 -040077 #include "modules/svg/include/SkSVGDOM.h"
Robert Phillips2af13c12021-09-01 16:47:01 +000078 #include "modules/svg/include/SkSVGNode.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050079 #include "src/xml/SkXMLWriter.h"
fmalita718df0a2016-07-15 10:33:29 -070080#endif
Leon Scroggins III87caae62020-05-04 10:02:45 -040081
Robert Phillips755e00e2021-09-21 11:59:57 -040082#ifdef SK_GRAPHITE_ENABLED
Robert Phillips091694f2021-09-30 11:41:16 -040083#include "experimental/graphite/include/Context.h"
Robert Phillips755e00e2021-09-21 11:59:57 -040084#include "experimental/graphite/include/SkStuff.h"
Robert Phillips091694f2021-09-30 11:41:16 -040085#include "tools/graphite/ContextFactory.h"
86#include "tools/graphite/GraphiteTestContext.h"
Robert Phillips1a4080a2021-09-21 09:57:43 -040087#endif
88
Leon Scroggins III87caae62020-05-04 10:02:45 -040089#if defined(SK_ENABLE_ANDROID_UTILS)
90 #include "client_utils/android/BitmapRegionDecoder.h"
91#endif
Mike Kleinc0bd9f92019-04-23 12:05:21 -050092#include "tests/TestUtils.h"
fmalita718df0a2016-07-15 10:33:29 -070093
Hal Canary8a001442018-09-19 11:31:27 -040094#include <cmath>
95#include <functional>
96
Mike Klein84836b72019-03-21 11:31:36 -050097static DEFINE_bool(multiPage, false,
98 "For document-type backends, render the source into multiple pages");
99static DEFINE_bool(RAW_threading, true, "Allow RAW decodes to run on multiple threads?");
halcanary7e798182015-04-14 14:06:18 -0700100
Mike Klein4932a5e2019-03-22 11:08:02 -0500101DECLARE_int(gpuThreads);
102
bsalomon3724e572016-03-30 18:56:19 -0700103using sk_gpu_test::GrContextFactory;
Robert Phillips291f3402020-02-19 14:14:47 -0500104using sk_gpu_test::ContextInfo;
bsalomon3724e572016-03-30 18:56:19 -0700105
mtklein748ca3b2015-01-15 10:56:12 -0800106namespace DM {
107
Hal Canary972eba32018-07-30 17:07:07 -0400108GMSrc::GMSrc(skiagm::GMFactory factory) : fFactory(factory) {}
mtklein748ca3b2015-01-15 10:56:12 -0800109
Robert Phillipsb87b39b2020-07-01 14:45:24 -0400110Result GMSrc::draw(GrDirectContext* context, SkCanvas* canvas) const {
Hal Canaryedda5652019-08-05 10:28:09 -0400111 std::unique_ptr<skiagm::GM> gm(fFactory());
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500112 SkString msg;
Robert Phillips762cb4e2020-06-15 13:12:32 -0400113
Robert Phillipsd26d25e2020-06-25 13:26:22 -0400114 skiagm::DrawResult gpuSetupResult = gm->gpuSetup(context, canvas, &msg);
Robert Phillips889d6132020-06-16 11:11:33 -0400115 switch (gpuSetupResult) {
116 case skiagm::DrawResult::kOk : break;
117 case skiagm::DrawResult::kFail: return Result(Result::Status::Fatal, msg);
118 case skiagm::DrawResult::kSkip: return Result(Result::Status::Skip, msg);
119 default: SK_ABORT("");
120 }
121
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500122 skiagm::DrawResult drawResult = gm->draw(canvas, &msg);
123 switch (drawResult) {
124 case skiagm::DrawResult::kOk : return Result(Result::Status::Ok, msg);
125 case skiagm::DrawResult::kFail: return Result(Result::Status::Fatal, msg);
126 case skiagm::DrawResult::kSkip: return Result(Result::Status::Skip, msg);
127 default: SK_ABORT("");
Chris Dalton50e24d72019-02-07 16:20:09 -0700128 }
Robert Phillipse9229532020-06-26 10:10:49 -0400129
130 // Note: we don't call "gpuTeardown" here because, when testing DDL recording, we want
131 // the gpu-backed images to live past the lifetime of the GM.
mtklein748ca3b2015-01-15 10:56:12 -0800132}
133
134SkISize GMSrc::size() const {
Hal Canaryedda5652019-08-05 10:28:09 -0400135 std::unique_ptr<skiagm::GM> gm(fFactory());
mtklein748ca3b2015-01-15 10:56:12 -0800136 return gm->getISize();
137}
138
139Name GMSrc::name() const {
Hal Canaryedda5652019-08-05 10:28:09 -0400140 std::unique_ptr<skiagm::GM> gm(fFactory());
mtklein748ca3b2015-01-15 10:56:12 -0800141 return gm->getName();
142}
143
bsalomon4ee6bd82015-05-27 13:23:23 -0700144void GMSrc::modifyGrContextOptions(GrContextOptions* options) const {
Hal Canaryedda5652019-08-05 10:28:09 -0400145 std::unique_ptr<skiagm::GM> gm(fFactory());
bsalomon4ee6bd82015-05-27 13:23:23 -0700146 gm->modifyGrContextOptions(options);
147}
148
Tyler Denniston45f94f82020-02-04 16:09:08 -0500149std::unique_ptr<skiagm::verifiers::VerifierList> GMSrc::getVerifiers() const {
150 std::unique_ptr<skiagm::GM> gm(fFactory());
151 return gm->getVerifiers();
152}
153
mtklein748ca3b2015-01-15 10:56:12 -0800154/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
155
Leon Scroggins III87caae62020-05-04 10:02:45 -0400156static SkString get_scaled_name(const Path& path, float scale) {
157 return SkStringPrintf("%s_%.3f", SkOSPath::Basename(path.c_str()).c_str(), scale);
158}
159
160#ifdef SK_ENABLE_ANDROID_UTILS
msarettd1227a72016-05-18 06:23:57 -0700161BRDSrc::BRDSrc(Path path, Mode mode, CodecSrc::DstColorType dstColorType, uint32_t sampleSize)
msaretta5783ae2015-09-08 15:35:32 -0700162 : fPath(path)
msaretta5783ae2015-09-08 15:35:32 -0700163 , fMode(mode)
164 , fDstColorType(dstColorType)
165 , fSampleSize(sampleSize)
166{}
167
168bool BRDSrc::veto(SinkFlags flags) const {
169 // No need to test to non-raster or indirect backends.
170 return flags.type != SinkFlags::kRaster
171 || flags.approach != SinkFlags::kDirect;
172}
173
Leon Scroggins III87caae62020-05-04 10:02:45 -0400174static std::unique_ptr<android::skia::BitmapRegionDecoder> create_brd(Path path) {
bungeman38d909e2016-08-02 14:40:46 -0700175 sk_sp<SkData> encoded(SkData::MakeFromFileName(path.c_str()));
Leon Scroggins III87caae62020-05-04 10:02:45 -0400176 return android::skia::BitmapRegionDecoder::Make(encoded);
msaretta5783ae2015-09-08 15:35:32 -0700177}
178
Matt Sarett334df3a2016-12-15 18:17:33 -0500179static inline void alpha8_to_gray8(SkBitmap* bitmap) {
180 // Android requires kGray8 bitmaps to be tagged as kAlpha8. Here we convert
181 // them back to kGray8 so our test framework can draw them correctly.
182 if (kAlpha_8_SkColorType == bitmap->info().colorType()) {
183 SkImageInfo newInfo = bitmap->info().makeColorType(kGray_8_SkColorType)
184 .makeAlphaType(kOpaque_SkAlphaType);
185 *const_cast<SkImageInfo*>(&bitmap->info()) = newInfo;
186 }
187}
188
Robert Phillipsb87b39b2020-07-01 14:45:24 -0400189Result BRDSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
msaretta5783ae2015-09-08 15:35:32 -0700190 SkColorType colorType = canvas->imageInfo().colorType();
191 if (kRGB_565_SkColorType == colorType &&
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500192 CodecSrc::kGetFromCanvas_DstColorType != fDstColorType)
193 {
194 return Result::Skip("Testing non-565 to 565 is uninteresting.");
msaretta5783ae2015-09-08 15:35:32 -0700195 }
196 switch (fDstColorType) {
197 case CodecSrc::kGetFromCanvas_DstColorType:
198 break;
msaretta5783ae2015-09-08 15:35:32 -0700199 case CodecSrc::kGrayscale_Always_DstColorType:
200 colorType = kGray_8_SkColorType;
201 break;
msarett34e0ec42016-04-22 16:27:24 -0700202 default:
203 SkASSERT(false);
204 break;
msaretta5783ae2015-09-08 15:35:32 -0700205 }
206
Leon Scroggins III87caae62020-05-04 10:02:45 -0400207 auto brd = create_brd(fPath);
John Stilesa008b0f2020-08-16 08:48:02 -0400208 if (nullptr == brd) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500209 return Result::Skip("Could not create brd for %s.", fPath.c_str());
msaretta5783ae2015-09-08 15:35:32 -0700210 }
211
Leon Scroggins III0118e972018-03-13 11:14:33 -0400212 auto recommendedCT = brd->computeOutputColorType(colorType);
213 if (kRGB_565_SkColorType == colorType && recommendedCT != colorType) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500214 return Result::Skip("Skip decoding non-opaque to 565.");
Leon Scroggins III1dc8ecb2017-08-17 13:42:48 -0400215 }
Leon Scroggins III0118e972018-03-13 11:14:33 -0400216 colorType = recommendedCT;
217
218 auto colorSpace = brd->computeOutputColorSpace(colorType, nullptr);
Leon Scroggins III1dc8ecb2017-08-17 13:42:48 -0400219
msaretta5783ae2015-09-08 15:35:32 -0700220 const uint32_t width = brd->width();
221 const uint32_t height = brd->height();
222 // Visually inspecting very small output images is not necessary.
223 if ((width / fSampleSize <= 10 || height / fSampleSize <= 10) && 1 != fSampleSize) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500224 return Result::Skip("Scaling very small images is uninteresting.");
msaretta5783ae2015-09-08 15:35:32 -0700225 }
226 switch (fMode) {
227 case kFullImage_Mode: {
msarett35e5d1b2015-10-27 12:50:25 -0700228 SkBitmap bitmap;
229 if (!brd->decodeRegion(&bitmap, nullptr, SkIRect::MakeXYWH(0, 0, width, height),
Leon Scroggins III0118e972018-03-13 11:14:33 -0400230 fSampleSize, colorType, false, colorSpace)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500231 return Result::Fatal("Cannot decode (full) region.");
msarett35e5d1b2015-10-27 12:50:25 -0700232 }
Matt Sarett334df3a2016-12-15 18:17:33 -0500233 alpha8_to_gray8(&bitmap);
Matt Sarett8db74f12017-06-14 13:02:05 +0000234
Mike Reed607a3822021-01-24 19:49:21 -0500235 canvas->drawImage(bitmap.asImage(), 0, 0);
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500236 return Result::Ok();
msaretta5783ae2015-09-08 15:35:32 -0700237 }
238 case kDivisor_Mode: {
239 const uint32_t divisor = 2;
240 if (width < divisor || height < divisor) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500241 return Result::Skip("Divisor is larger than image dimension.");
msaretta5783ae2015-09-08 15:35:32 -0700242 }
243
244 // Use a border to test subsets that extend outside the image.
245 // We will not allow the border to be larger than the image dimensions. Allowing
246 // these large borders causes off by one errors that indicate a problem with the
247 // test suite, not a problem with the implementation.
Brian Osman788b9162020-02-07 10:36:46 -0500248 const uint32_t maxBorder = std::min(width, height) / (fSampleSize * divisor);
249 const uint32_t scaledBorder = std::min(5u, maxBorder);
msaretta5783ae2015-09-08 15:35:32 -0700250 const uint32_t unscaledBorder = scaledBorder * fSampleSize;
251
252 // We may need to clear the canvas to avoid uninitialized memory.
253 // Assume we are scaling a 780x780 image with sampleSize = 8.
254 // The output image should be 97x97.
255 // Each subset will be 390x390.
256 // Each scaled subset be 48x48.
257 // Four scaled subsets will only fill a 96x96 image.
258 // The bottom row and last column will not be touched.
259 // This is an unfortunate result of our rounding rules when scaling.
260 // Maybe we need to consider testing scaled subsets without trying to
261 // combine them to match the full scaled image? Or maybe this is the
262 // best we can do?
263 canvas->clear(0);
264
265 for (uint32_t x = 0; x < divisor; x++) {
266 for (uint32_t y = 0; y < divisor; y++) {
267 // Calculate the subset dimensions
268 uint32_t subsetWidth = width / divisor;
269 uint32_t subsetHeight = height / divisor;
270 const int left = x * subsetWidth;
271 const int top = y * subsetHeight;
272
273 // Increase the size of the last subset in each row or column, when the
274 // divisor does not divide evenly into the image dimensions
275 subsetWidth += (x + 1 == divisor) ? (width % divisor) : 0;
276 subsetHeight += (y + 1 == divisor) ? (height % divisor) : 0;
277
278 // Increase the size of the subset in order to have a border on each side
279 const int decodeLeft = left - unscaledBorder;
280 const int decodeTop = top - unscaledBorder;
281 const uint32_t decodeWidth = subsetWidth + unscaledBorder * 2;
282 const uint32_t decodeHeight = subsetHeight + unscaledBorder * 2;
msarett35e5d1b2015-10-27 12:50:25 -0700283 SkBitmap bitmap;
284 if (!brd->decodeRegion(&bitmap, nullptr, SkIRect::MakeXYWH(decodeLeft,
Matt Sarett68feef42017-04-11 09:51:32 -0400285 decodeTop, decodeWidth, decodeHeight), fSampleSize, colorType, false,
Leon Scroggins III0118e972018-03-13 11:14:33 -0400286 colorSpace)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500287 return Result::Fatal("Cannot decode region.");
msarett35e5d1b2015-10-27 12:50:25 -0700288 }
msaretta5783ae2015-09-08 15:35:32 -0700289
Matt Sarett334df3a2016-12-15 18:17:33 -0500290 alpha8_to_gray8(&bitmap);
Mike Reed607a3822021-01-24 19:49:21 -0500291 canvas->drawImageRect(bitmap.asImage().get(),
msaretta5783ae2015-09-08 15:35:32 -0700292 SkRect::MakeXYWH((SkScalar) scaledBorder, (SkScalar) scaledBorder,
293 (SkScalar) (subsetWidth / fSampleSize),
294 (SkScalar) (subsetHeight / fSampleSize)),
295 SkRect::MakeXYWH((SkScalar) (left / fSampleSize),
296 (SkScalar) (top / fSampleSize),
297 (SkScalar) (subsetWidth / fSampleSize),
298 (SkScalar) (subsetHeight / fSampleSize)),
Mike Reed607a3822021-01-24 19:49:21 -0500299 SkSamplingOptions(), nullptr,
300 SkCanvas::kStrict_SrcRectConstraint);
msaretta5783ae2015-09-08 15:35:32 -0700301 }
302 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500303 return Result::Ok();
msaretta5783ae2015-09-08 15:35:32 -0700304 }
305 default:
306 SkASSERT(false);
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500307 return Result::Fatal("Error: Should not be reached.");
msaretta5783ae2015-09-08 15:35:32 -0700308 }
309}
310
311SkISize BRDSrc::size() const {
Leon Scroggins III87caae62020-05-04 10:02:45 -0400312 auto brd = create_brd(fPath);
msaretta5783ae2015-09-08 15:35:32 -0700313 if (brd) {
Brian Osman788b9162020-02-07 10:36:46 -0500314 return {std::max(1, brd->width() / (int)fSampleSize),
315 std::max(1, brd->height() / (int)fSampleSize)};
msaretta5783ae2015-09-08 15:35:32 -0700316 }
Hal Canaryfafe1352017-04-11 12:12:02 -0400317 return {0, 0};
msaretta5783ae2015-09-08 15:35:32 -0700318}
319
msaretta5783ae2015-09-08 15:35:32 -0700320Name BRDSrc::name() const {
321 // We will replicate the names used by CodecSrc so that images can
322 // be compared in Gold.
323 if (1 == fSampleSize) {
324 return SkOSPath::Basename(fPath.c_str());
325 }
msarett4b0778e2015-11-13 09:59:11 -0800326 return get_scaled_name(fPath, 1.0f / (float) fSampleSize);
msaretta5783ae2015-09-08 15:35:32 -0700327}
328
Leon Scroggins III87caae62020-05-04 10:02:45 -0400329#endif // SK_ENABLE_ANDROID_UTILS
330
msaretta5783ae2015-09-08 15:35:32 -0700331/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
332
scroggo3ac66e92016-02-08 15:09:48 -0800333static bool serial_from_path_name(const SkString& path) {
334 if (!FLAGS_RAW_threading) {
335 static const char* const exts[] = {
336 "arw", "cr2", "dng", "nef", "nrw", "orf", "raf", "rw2", "pef", "srw",
337 "ARW", "CR2", "DNG", "NEF", "NRW", "ORF", "RAF", "RW2", "PEF", "SRW",
338 };
339 const char* actualExt = strrchr(path.c_str(), '.');
340 if (actualExt) {
341 actualExt++;
342 for (auto* ext : exts) {
343 if (0 == strcmp(ext, actualExt)) {
344 return true;
345 }
346 }
347 }
348 }
349 return false;
350}
351
scroggoc5560be2016-02-03 09:42:42 -0800352CodecSrc::CodecSrc(Path path, Mode mode, DstColorType dstColorType, SkAlphaType dstAlphaType,
353 float scale)
msarett438b2ad2015-04-09 12:43:10 -0700354 : fPath(path)
355 , fMode(mode)
356 , fDstColorType(dstColorType)
scroggoc5560be2016-02-03 09:42:42 -0800357 , fDstAlphaType(dstAlphaType)
msarett0a242972015-06-11 14:27:27 -0700358 , fScale(scale)
scroggo3ac66e92016-02-08 15:09:48 -0800359 , fRunSerially(serial_from_path_name(path))
msarett438b2ad2015-04-09 12:43:10 -0700360{}
mtklein748ca3b2015-01-15 10:56:12 -0800361
mtklein99cab4e2015-07-31 06:43:04 -0700362bool CodecSrc::veto(SinkFlags flags) const {
msarett18976312016-03-09 14:20:58 -0800363 // Test to direct raster backends (8888 and 565).
msarettb714fb02016-01-22 14:46:42 -0800364 return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
mtkleine0effd62015-07-29 06:37:28 -0700365}
scroggo9b77ddd2015-03-19 06:03:39 -0700366
msarett34e0ec42016-04-22 16:27:24 -0700367// Allows us to test decodes to non-native 8888.
msarettb1be46b2016-05-17 08:52:11 -0700368static void swap_rb_if_necessary(SkBitmap& bitmap, CodecSrc::DstColorType dstColorType) {
msarett34e0ec42016-04-22 16:27:24 -0700369 if (CodecSrc::kNonNative8888_Always_DstColorType != dstColorType) {
370 return;
371 }
372
373 for (int y = 0; y < bitmap.height(); y++) {
374 uint32_t* row = (uint32_t*) bitmap.getAddr(0, y);
375 SkOpts::RGBA_to_BGRA(row, row, bitmap.width());
376 }
377}
378
msarettb1be46b2016-05-17 08:52:11 -0700379static bool get_decode_info(SkImageInfo* decodeInfo, SkColorType canvasColorType,
scroggoba584892016-05-20 13:56:13 -0700380 CodecSrc::DstColorType dstColorType, SkAlphaType dstAlphaType) {
msarett3d9d7a72015-10-21 10:27:10 -0700381 switch (dstColorType) {
msarett3d9d7a72015-10-21 10:27:10 -0700382 case CodecSrc::kGrayscale_Always_DstColorType:
scroggo1a361922016-05-20 14:27:16 -0700383 if (kRGB_565_SkColorType == canvasColorType) {
msarett3d9d7a72015-10-21 10:27:10 -0700384 return false;
385 }
scroggoc5560be2016-02-03 09:42:42 -0800386 *decodeInfo = decodeInfo->makeColorType(kGray_8_SkColorType);
msarett3d9d7a72015-10-21 10:27:10 -0700387 break;
msarett34e0ec42016-04-22 16:27:24 -0700388 case CodecSrc::kNonNative8888_Always_DstColorType:
Leon Scroggins III557fbbe2017-05-23 09:37:21 -0400389 if (kRGB_565_SkColorType == canvasColorType
390 || kRGBA_F16_SkColorType == canvasColorType) {
msarett34e0ec42016-04-22 16:27:24 -0700391 return false;
392 }
393#ifdef SK_PMCOLOR_IS_RGBA
394 *decodeInfo = decodeInfo->makeColorType(kBGRA_8888_SkColorType);
395#else
396 *decodeInfo = decodeInfo->makeColorType(kRGBA_8888_SkColorType);
397#endif
398 break;
msarett3d9d7a72015-10-21 10:27:10 -0700399 default:
msarett55f7bdd2016-02-16 13:24:54 -0800400 if (kRGB_565_SkColorType == canvasColorType &&
401 kOpaque_SkAlphaType != decodeInfo->alphaType()) {
402 return false;
403 }
Matt Sarett09a1c082017-02-01 15:34:22 -0800404
scroggoc5560be2016-02-03 09:42:42 -0800405 *decodeInfo = decodeInfo->makeColorType(canvasColorType);
msarett3d9d7a72015-10-21 10:27:10 -0700406 break;
407 }
408
scroggoba584892016-05-20 13:56:13 -0700409 *decodeInfo = decodeInfo->makeAlphaType(dstAlphaType);
msarett3d9d7a72015-10-21 10:27:10 -0700410 return true;
411}
412
msarettb1be46b2016-05-17 08:52:11 -0700413static void draw_to_canvas(SkCanvas* canvas, const SkImageInfo& info, void* pixels, size_t rowBytes,
Leon Scroggins571b30f2017-07-11 17:35:31 +0000414 CodecSrc::DstColorType dstColorType,
msarettb1be46b2016-05-17 08:52:11 -0700415 SkScalar left = 0, SkScalar top = 0) {
msarettb1be46b2016-05-17 08:52:11 -0700416 SkBitmap bitmap;
Leon Scroggins571b30f2017-07-11 17:35:31 +0000417 bitmap.installPixels(info, pixels, rowBytes);
msarettb1be46b2016-05-17 08:52:11 -0700418 swap_rb_if_necessary(bitmap, dstColorType);
Mike Reed34a0c972021-01-25 17:49:32 -0500419 canvas->drawImage(bitmap.asImage(), left, top);
msarettb1be46b2016-05-17 08:52:11 -0700420}
421
Matt Sarettf485cf92017-05-26 08:58:03 -0400422// For codec srcs, we want the "draw" step to be a memcpy. Any interesting color space or
423// color format conversions should be performed by the codec. Sometimes the output of the
424// decode will be in an interesting color space. On our srgb and f16 backends, we need to
425// "pretend" that the color space is standard sRGB to avoid triggering color conversion
426// at draw time.
427static void set_bitmap_color_space(SkImageInfo* info) {
Brian Osman6b622962018-08-27 19:16:02 +0000428 *info = info->makeColorSpace(SkColorSpace::MakeSRGB());
Matt Sarettf485cf92017-05-26 08:58:03 -0400429}
430
Robert Phillipsb87b39b2020-07-01 14:45:24 -0400431Result CodecSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
bungeman38d909e2016-08-02 14:40:46 -0700432 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
mtklein748ca3b2015-01-15 10:56:12 -0800433 if (!encoded) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500434 return Result::Fatal("Couldn't read %s.", fPath.c_str());
mtklein748ca3b2015-01-15 10:56:12 -0800435 }
msarettb714fb02016-01-22 14:46:42 -0800436
Mike Reedede7bac2017-07-23 15:30:02 -0400437 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(encoded));
John Stilesa008b0f2020-08-16 08:48:02 -0400438 if (nullptr == codec) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500439 return Result::Fatal("Couldn't create codec for %s.", fPath.c_str());
scroggo9b77ddd2015-03-19 06:03:39 -0700440 }
441
scroggoba584892016-05-20 13:56:13 -0700442 SkImageInfo decodeInfo = codec->getInfo();
443 if (!get_decode_info(&decodeInfo, canvas->imageInfo().colorType(), fDstColorType,
444 fDstAlphaType)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500445 return Result::Skip("Skipping uninteresting test.");
msarett438b2ad2015-04-09 12:43:10 -0700446 }
447
msarett0a242972015-06-11 14:27:27 -0700448 // Try to scale the image if it is desired
449 SkISize size = codec->getScaledDimensions(fScale);
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500450
451 std::unique_ptr<SkAndroidCodec> androidCodec;
452 if (1.0f != fScale && fMode == kAnimated_Mode) {
453 androidCodec = SkAndroidCodec::MakeFromData(encoded);
454 size = androidCodec->getSampledDimensions(1 / fScale);
455 }
456
msarett0a242972015-06-11 14:27:27 -0700457 if (size == decodeInfo.dimensions() && 1.0f != fScale) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500458 return Result::Skip("Test without scaling is uninteresting.");
msarett0a242972015-06-11 14:27:27 -0700459 }
msarettb32758a2015-08-18 13:22:46 -0700460
461 // Visually inspecting very small output images is not necessary. We will
462 // cover these cases in unit testing.
463 if ((size.width() <= 10 || size.height() <= 10) && 1.0f != fScale) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500464 return Result::Skip("Scaling very small images is uninteresting.");
msarettb32758a2015-08-18 13:22:46 -0700465 }
Brian Salomon9241a6d2019-10-03 13:26:54 -0400466 decodeInfo = decodeInfo.makeDimensions(size);
msarett0a242972015-06-11 14:27:27 -0700467
Mike Reed7fcfb622018-02-09 13:26:46 -0500468 const int bpp = decodeInfo.bytesPerPixel();
msarettb1be46b2016-05-17 08:52:11 -0700469 const size_t rowBytes = size.width() * bpp;
Mike Reedf0ffb892017-10-03 14:47:21 -0400470 const size_t safeSize = decodeInfo.computeByteSize(rowBytes);
scroggo19b91532016-10-24 09:03:26 -0700471 SkAutoMalloc pixels(safeSize);
msarett438b2ad2015-04-09 12:43:10 -0700472
msarettbb25b532016-01-13 09:31:39 -0800473 SkCodec::Options options;
474 if (kCodecZeroInit_Mode == fMode) {
msarettb1be46b2016-05-17 08:52:11 -0700475 memset(pixels.get(), 0, size.height() * rowBytes);
msarettbb25b532016-01-13 09:31:39 -0800476 options.fZeroInitialized = SkCodec::kYes_ZeroInitialized;
477 }
msarett34e0ec42016-04-22 16:27:24 -0700478
479 SkImageInfo bitmapInfo = decodeInfo;
Matt Sarettf485cf92017-05-26 08:58:03 -0400480 set_bitmap_color_space(&bitmapInfo);
msarett34e0ec42016-04-22 16:27:24 -0700481 if (kRGBA_8888_SkColorType == decodeInfo.colorType() ||
482 kBGRA_8888_SkColorType == decodeInfo.colorType()) {
483 bitmapInfo = bitmapInfo.makeColorType(kN32_SkColorType);
484 }
scroggo9b77ddd2015-03-19 06:03:39 -0700485
scroggo9c59ebc2015-03-25 13:48:49 -0700486 switch (fMode) {
scroggo19b91532016-10-24 09:03:26 -0700487 case kAnimated_Mode: {
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500488 SkAndroidCodec::AndroidOptions androidOptions;
489 if (fScale != 1.0f) {
490 SkASSERT(androidCodec);
491 androidOptions.fSampleSize = 1 / fScale;
492 auto dims = androidCodec->getSampledDimensions(androidOptions.fSampleSize);
493 decodeInfo = decodeInfo.makeDimensions(dims);
494 }
495
496 std::vector<SkCodec::FrameInfo> frameInfos = androidCodec
497 ? androidCodec->codec()->getFrameInfo() : codec->getFrameInfo();
scroggo19b91532016-10-24 09:03:26 -0700498 if (frameInfos.size() <= 1) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500499 return Result::Fatal("%s is not an animated image.", fPath.c_str());
scroggo19b91532016-10-24 09:03:26 -0700500 }
501
scroggo90e971a2016-10-25 06:11:01 -0700502 // As in CodecSrc::size(), compute a roughly square grid to draw the frames
503 // into. "factor" is the number of frames to draw on one row. There will be
504 // up to "factor" rows as well.
505 const float root = sqrt((float) frameInfos.size());
506 const int factor = sk_float_ceil2int(root);
507
scroggo19b91532016-10-24 09:03:26 -0700508 // Used to cache a frame that future frames will depend on.
509 SkAutoMalloc priorFramePixels;
Nigel Tao66bc5242018-08-22 10:56:03 +1000510 int cachedFrame = SkCodec::kNoFrame;
Leon Scroggins III249b8e32017-04-17 12:46:33 -0400511 for (int i = 0; static_cast<size_t>(i) < frameInfos.size(); i++) {
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500512 androidOptions.fFrameIndex = i;
scroggo19b91532016-10-24 09:03:26 -0700513 // Check for a prior frame
Leon Scroggins III249b8e32017-04-17 12:46:33 -0400514 const int reqFrame = frameInfos[i].fRequiredFrame;
Nigel Tao66bc5242018-08-22 10:56:03 +1000515 if (reqFrame != SkCodec::kNoFrame && reqFrame == cachedFrame
scroggo19b91532016-10-24 09:03:26 -0700516 && priorFramePixels.get()) {
517 // Copy into pixels
518 memcpy(pixels.get(), priorFramePixels.get(), safeSize);
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500519 androidOptions.fPriorFrame = reqFrame;
scroggo19b91532016-10-24 09:03:26 -0700520 } else {
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500521 androidOptions.fPriorFrame = SkCodec::kNoFrame;
scroggo19b91532016-10-24 09:03:26 -0700522 }
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500523 SkCodec::Result result = androidCodec
524 ? androidCodec->getAndroidPixels(decodeInfo, pixels.get(), rowBytes,
525 &androidOptions)
526 : codec->getPixels(decodeInfo, pixels.get(), rowBytes, &androidOptions);
Leon Scroggins III3fc97d72016-12-09 16:39:33 -0500527 if (SkCodec::kInvalidInput == result && i > 0) {
528 // Some of our test images have truncated later frames. Treat that
529 // the same as incomplete.
530 result = SkCodec::kIncompleteInput;
531 }
scroggo19b91532016-10-24 09:03:26 -0700532 switch (result) {
533 case SkCodec::kSuccess:
Leon Scroggins III674a1842017-07-06 12:26:09 -0400534 case SkCodec::kErrorInInput:
scroggo90e971a2016-10-25 06:11:01 -0700535 case SkCodec::kIncompleteInput: {
Leon Scroggins III557fbbe2017-05-23 09:37:21 -0400536 // If the next frame depends on this one, store it in priorFrame.
537 // It is possible that we may discard a frame that future frames depend on,
538 // but the codec will simply redecode the discarded frame.
539 // Do this before calling draw_to_canvas, which premultiplies in place. If
540 // we're decoding to unpremul, we want to pass the unmodified frame to the
541 // codec for decoding the next frame.
542 if (static_cast<size_t>(i+1) < frameInfos.size()
543 && frameInfos[i+1].fRequiredFrame == i) {
544 memcpy(priorFramePixels.reset(safeSize), pixels.get(), safeSize);
545 cachedFrame = i;
546 }
547
scroggo90e971a2016-10-25 06:11:01 -0700548 SkAutoCanvasRestore acr(canvas, true);
549 const int xTranslate = (i % factor) * decodeInfo.width();
550 const int yTranslate = (i / factor) * decodeInfo.height();
551 canvas->translate(SkIntToScalar(xTranslate), SkIntToScalar(yTranslate));
Leon Scroggins571b30f2017-07-11 17:35:31 +0000552 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, fDstColorType);
Leon Scroggins III674a1842017-07-06 12:26:09 -0400553 if (result != SkCodec::kSuccess) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500554 return Result::Ok();
scroggo19b91532016-10-24 09:03:26 -0700555 }
556 break;
scroggo90e971a2016-10-25 06:11:01 -0700557 }
scroggo53f63b62016-10-27 08:29:13 -0700558 case SkCodec::kInvalidConversion:
Mike Reed304a07c2017-07-12 15:10:28 -0400559 if (i > 0 && (decodeInfo.colorType() == kRGB_565_SkColorType)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500560 return Result::Skip(
561 "Cannot decode frame %i to 565 (%s).", i, fPath.c_str());
scroggo53f63b62016-10-27 08:29:13 -0700562 }
John Stiles30212b72020-06-11 17:55:07 -0400563 [[fallthrough]];
scroggo19b91532016-10-24 09:03:26 -0700564 default:
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500565 return Result::Fatal(
566 "Couldn't getPixels for frame %i in %s.", i, fPath.c_str());
scroggo19b91532016-10-24 09:03:26 -0700567 }
scroggo19b91532016-10-24 09:03:26 -0700568 }
569 break;
570 }
msarettbb25b532016-01-13 09:31:39 -0800571 case kCodecZeroInit_Mode:
msarett9e707a02015-09-01 14:57:57 -0700572 case kCodec_Mode: {
Leon Scroggins571b30f2017-07-11 17:35:31 +0000573 switch (codec->getPixels(decodeInfo, pixels.get(), rowBytes, &options)) {
scroggoeb602a52015-07-09 08:16:03 -0700574 case SkCodec::kSuccess:
Leon Scroggins III674a1842017-07-06 12:26:09 -0400575 // We consider these to be valid, since we should still decode what is
scroggo9c59ebc2015-03-25 13:48:49 -0700576 // available.
Leon Scroggins III674a1842017-07-06 12:26:09 -0400577 case SkCodec::kErrorInInput:
scroggoeb602a52015-07-09 08:16:03 -0700578 case SkCodec::kIncompleteInput:
scroggo9c59ebc2015-03-25 13:48:49 -0700579 break;
scroggo9c59ebc2015-03-25 13:48:49 -0700580 default:
581 // Everything else is considered a failure.
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500582 return Result::Fatal("Couldn't getPixels %s.", fPath.c_str());
scroggo9c59ebc2015-03-25 13:48:49 -0700583 }
msarettb1be46b2016-05-17 08:52:11 -0700584
Leon Scroggins571b30f2017-07-11 17:35:31 +0000585 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, fDstColorType);
scroggo9c59ebc2015-03-25 13:48:49 -0700586 break;
emmaleer0a4c3cb2015-06-22 10:40:21 -0700587 }
scroggo9c59ebc2015-03-25 13:48:49 -0700588 case kScanline_Mode: {
scroggod8d68552016-06-06 11:26:17 -0700589 void* dst = pixels.get();
590 uint32_t height = decodeInfo.height();
scroggo19b91532016-10-24 09:03:26 -0700591 const bool useIncremental = [this]() {
592 auto exts = { "png", "PNG", "gif", "GIF" };
593 for (auto ext : exts) {
594 if (fPath.endsWith(ext)) {
595 return true;
596 }
597 }
598 return false;
599 }();
600 // ico may use the old scanline method or the new one, depending on whether it
601 // internally holds a bmp or a png.
scroggo8e6c7ad2016-09-16 08:20:38 -0700602 const bool ico = fPath.endsWith("ico");
scroggo19b91532016-10-24 09:03:26 -0700603 bool useOldScanlineMethod = !useIncremental && !ico;
604 if (useIncremental || ico) {
scroggo8e6c7ad2016-09-16 08:20:38 -0700605 if (SkCodec::kSuccess == codec->startIncrementalDecode(decodeInfo, dst,
Leon Scroggins571b30f2017-07-11 17:35:31 +0000606 rowBytes, &options)) {
scroggo8e6c7ad2016-09-16 08:20:38 -0700607 int rowsDecoded;
Leon Scroggins III674a1842017-07-06 12:26:09 -0400608 auto result = codec->incrementalDecode(&rowsDecoded);
609 if (SkCodec::kIncompleteInput == result || SkCodec::kErrorInInput == result) {
scroggo8e6c7ad2016-09-16 08:20:38 -0700610 codec->fillIncompleteImage(decodeInfo, dst, rowBytes,
611 SkCodec::kNo_ZeroInitialized, height,
612 rowsDecoded);
scroggo6fb23912016-06-02 14:16:43 -0700613 }
scroggo8e6c7ad2016-09-16 08:20:38 -0700614 } else {
scroggo19b91532016-10-24 09:03:26 -0700615 if (useIncremental) {
616 // Error: These should support incremental decode.
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500617 return Result::Fatal("Could not start incremental decode");
scroggo8e6c7ad2016-09-16 08:20:38 -0700618 }
619 // Otherwise, this is an ICO. Since incremental failed, it must contain a BMP,
620 // which should work via startScanlineDecode
621 useOldScanlineMethod = true;
622 }
623 }
624
625 if (useOldScanlineMethod) {
Leon Scroggins571b30f2017-07-11 17:35:31 +0000626 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500627 return Result::Fatal("Could not start scanline decoder");
scroggo8e6c7ad2016-09-16 08:20:38 -0700628 }
629
Brian Osmanea176c62018-04-06 15:28:23 -0400630 // We do not need to check the return value. On an incomplete
631 // image, memory will be filled with a default value.
632 codec->getScanlines(dst, height, rowBytes);
msarett10522ff2015-09-07 08:54:01 -0700633 }
634
Leon Scroggins571b30f2017-07-11 17:35:31 +0000635 draw_to_canvas(canvas, bitmapInfo, dst, rowBytes, fDstColorType);
emmaleer97002062015-05-27 12:36:10 -0700636 break;
637 }
msarett0a242972015-06-11 14:27:27 -0700638 case kStripe_Mode: {
639 const int height = decodeInfo.height();
640 // This value is chosen arbitrarily. We exercise more cases by choosing a value that
641 // does not align with image blocks.
642 const int stripeHeight = 37;
643 const int numStripes = (height + stripeHeight - 1) / stripeHeight;
msarettb1be46b2016-05-17 08:52:11 -0700644 void* dst = pixels.get();
msarett0a242972015-06-11 14:27:27 -0700645
646 // Decode odd stripes
Leon Scroggins571b30f2017-07-11 17:35:31 +0000647 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo, &options)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500648 return Result::Fatal("Could not start scanline decoder");
msarettb65e6042016-02-23 05:37:25 -0800649 }
650
651 // This mode was designed to test the new skip scanlines API in libjpeg-turbo.
652 // Jpegs have kTopDown_SkScanlineOrder, and at this time, it is not interesting
653 // to run this test for image types that do not have this scanline ordering.
scroggo12e2f502016-05-16 09:04:18 -0700654 // We only run this on Jpeg, which is always kTopDown.
655 SkASSERT(SkCodec::kTopDown_SkScanlineOrder == codec->getScanlineOrder());
msarette6dd0042015-10-09 11:07:34 -0700656
msarett0a242972015-06-11 14:27:27 -0700657 for (int i = 0; i < numStripes; i += 2) {
658 // Skip a stripe
Brian Osman788b9162020-02-07 10:36:46 -0500659 const int linesToSkip = std::min(stripeHeight, height - i * stripeHeight);
msarette6dd0042015-10-09 11:07:34 -0700660 codec->skipScanlines(linesToSkip);
msarett0a242972015-06-11 14:27:27 -0700661
662 // Read a stripe
663 const int startY = (i + 1) * stripeHeight;
Brian Osman788b9162020-02-07 10:36:46 -0500664 const int linesToRead = std::min(stripeHeight, height - startY);
msarett0a242972015-06-11 14:27:27 -0700665 if (linesToRead > 0) {
msarettb1be46b2016-05-17 08:52:11 -0700666 codec->getScanlines(SkTAddOffset<void>(dst, rowBytes * startY), linesToRead,
667 rowBytes);
msarett0a242972015-06-11 14:27:27 -0700668 }
669 }
670
671 // Decode even stripes
Leon Scroggins571b30f2017-07-11 17:35:31 +0000672 const SkCodec::Result startResult = codec->startScanlineDecode(decodeInfo);
scroggo1c005e42015-08-04 09:24:45 -0700673 if (SkCodec::kSuccess != startResult) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500674 return Result::Fatal("Failed to restart scanline decoder with same parameters.");
msarett0a242972015-06-11 14:27:27 -0700675 }
676 for (int i = 0; i < numStripes; i += 2) {
677 // Read a stripe
678 const int startY = i * stripeHeight;
Brian Osman788b9162020-02-07 10:36:46 -0500679 const int linesToRead = std::min(stripeHeight, height - startY);
msarettb1be46b2016-05-17 08:52:11 -0700680 codec->getScanlines(SkTAddOffset<void>(dst, rowBytes * startY), linesToRead,
681 rowBytes);
msarett0a242972015-06-11 14:27:27 -0700682
683 // Skip a stripe
Brian Osman788b9162020-02-07 10:36:46 -0500684 const int linesToSkip = std::min(stripeHeight, height - (i + 1) * stripeHeight);
msarettf6db27e2015-06-12 09:34:04 -0700685 if (linesToSkip > 0) {
msarette6dd0042015-10-09 11:07:34 -0700686 codec->skipScanlines(linesToSkip);
msarett0a242972015-06-11 14:27:27 -0700687 }
688 }
msarettb1be46b2016-05-17 08:52:11 -0700689
Leon Scroggins571b30f2017-07-11 17:35:31 +0000690 draw_to_canvas(canvas, bitmapInfo, dst, rowBytes, fDstColorType);
emmaleer0a4c3cb2015-06-22 10:40:21 -0700691 break;
msarett0a242972015-06-11 14:27:27 -0700692 }
msarett91c22b22016-02-22 12:27:46 -0800693 case kCroppedScanline_Mode: {
694 const int width = decodeInfo.width();
695 const int height = decodeInfo.height();
696 // This value is chosen because, as we move across the image, it will sometimes
697 // align with the jpeg block sizes and it will sometimes not. This allows us
698 // to test interestingly different code paths in the implementation.
699 const int tileSize = 36;
msarett91c22b22016-02-22 12:27:46 -0800700 SkIRect subset;
701 for (int x = 0; x < width; x += tileSize) {
Brian Osman788b9162020-02-07 10:36:46 -0500702 subset = SkIRect::MakeXYWH(x, 0, std::min(tileSize, width - x), height);
Matt Saretta40d9c82017-05-19 15:21:05 -0400703 options.fSubset = &subset;
Leon Scroggins571b30f2017-07-11 17:35:31 +0000704 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo, &options)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500705 return Result::Fatal("Could not start scanline decoder.");
msarett91c22b22016-02-22 12:27:46 -0800706 }
707
msarettb1be46b2016-05-17 08:52:11 -0700708 codec->getScanlines(SkTAddOffset<void>(pixels.get(), x * bpp), height, rowBytes);
msarett91c22b22016-02-22 12:27:46 -0800709 }
710
Leon Scroggins571b30f2017-07-11 17:35:31 +0000711 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, fDstColorType);
msarett91c22b22016-02-22 12:27:46 -0800712 break;
713 }
scroggob636b452015-07-22 07:16:20 -0700714 case kSubset_Mode: {
715 // Arbitrarily choose a divisor.
716 int divisor = 2;
717 // Total width/height of the image.
718 const int W = codec->getInfo().width();
719 const int H = codec->getInfo().height();
720 if (divisor > W || divisor > H) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500721 return Result::Skip("Cannot codec subset: divisor %d is too big "
722 "for %s with dimensions (%d x %d)", divisor,
723 fPath.c_str(), W, H);
scroggob636b452015-07-22 07:16:20 -0700724 }
725 // subset dimensions
726 // SkWebpCodec, the only one that supports subsets, requires even top/left boundaries.
727 const int w = SkAlign2(W / divisor);
728 const int h = SkAlign2(H / divisor);
729 SkIRect subset;
Matt Saretta40d9c82017-05-19 15:21:05 -0400730 options.fSubset = &subset;
scroggob636b452015-07-22 07:16:20 -0700731 SkBitmap subsetBm;
732 // We will reuse pixel memory from bitmap.
msarettb1be46b2016-05-17 08:52:11 -0700733 void* dst = pixels.get();
scroggob636b452015-07-22 07:16:20 -0700734 // Keep track of left and top (for drawing subsetBm into canvas). We could use
735 // fScale * x and fScale * y, but we want integers such that the next subset will start
736 // where the last one ended. So we'll add decodeInfo.width() and height().
737 int left = 0;
738 for (int x = 0; x < W; x += w) {
739 int top = 0;
740 for (int y = 0; y < H; y+= h) {
741 // Do not make the subset go off the edge of the image.
Brian Osman788b9162020-02-07 10:36:46 -0500742 const int preScaleW = std::min(w, W - x);
743 const int preScaleH = std::min(h, H - y);
scroggob636b452015-07-22 07:16:20 -0700744 subset.setXYWH(x, y, preScaleW, preScaleH);
745 // And scale
746 // FIXME: Should we have a version of getScaledDimensions that takes a subset
747 // into account?
Brian Osman788b9162020-02-07 10:36:46 -0500748 const int scaledW = std::max(1, SkScalarRoundToInt(preScaleW * fScale));
749 const int scaledH = std::max(1, SkScalarRoundToInt(preScaleH * fScale));
msarettc7eb4902016-04-25 07:04:58 -0700750 decodeInfo = decodeInfo.makeWH(scaledW, scaledH);
751 SkImageInfo subsetBitmapInfo = bitmapInfo.makeWH(scaledW, scaledH);
msarettb1be46b2016-05-17 08:52:11 -0700752 size_t subsetRowBytes = subsetBitmapInfo.minRowBytes();
753 const SkCodec::Result result = codec->getPixels(decodeInfo, dst, subsetRowBytes,
Leon Scroggins571b30f2017-07-11 17:35:31 +0000754 &options);
scroggob636b452015-07-22 07:16:20 -0700755 switch (result) {
756 case SkCodec::kSuccess:
Leon Scroggins III674a1842017-07-06 12:26:09 -0400757 case SkCodec::kErrorInInput:
scroggob636b452015-07-22 07:16:20 -0700758 case SkCodec::kIncompleteInput:
759 break;
scroggob636b452015-07-22 07:16:20 -0700760 default:
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500761 return Result::Fatal("subset codec failed to decode (%d, %d, %d, %d) "
762 "from %s with dimensions (%d x %d)\t error %d",
763 x, y, decodeInfo.width(), decodeInfo.height(),
764 fPath.c_str(), W, H, result);
scroggob636b452015-07-22 07:16:20 -0700765 }
Leon Scroggins571b30f2017-07-11 17:35:31 +0000766 draw_to_canvas(canvas, subsetBitmapInfo, dst, subsetRowBytes, fDstColorType,
767 SkIntToScalar(left), SkIntToScalar(top));
msarettb1be46b2016-05-17 08:52:11 -0700768
scroggob636b452015-07-22 07:16:20 -0700769 // translate by the scaled height.
770 top += decodeInfo.height();
771 }
772 // translate by the scaled width.
773 left += decodeInfo.width();
774 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500775 return Result::Ok();
scroggob636b452015-07-22 07:16:20 -0700776 }
msarettb714fb02016-01-22 14:46:42 -0800777 default:
778 SkASSERT(false);
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500779 return Result::Fatal("Invalid fMode");
scroggo9b77ddd2015-03-19 06:03:39 -0700780 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500781 return Result::Ok();
scroggo9b77ddd2015-03-19 06:03:39 -0700782}
783
784SkISize CodecSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -0700785 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
Mike Reedede7bac2017-07-23 15:30:02 -0400786 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(encoded));
scroggo7fac5af2015-09-30 11:33:12 -0700787 if (nullptr == codec) {
Hal Canaryfafe1352017-04-11 12:12:02 -0400788 return {0, 0};
scroggo7fac5af2015-09-30 11:33:12 -0700789 }
scroggo19b91532016-10-24 09:03:26 -0700790
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500791 if (fMode != kAnimated_Mode) {
792 return codec->getScaledDimensions(fScale);
scroggo19b91532016-10-24 09:03:26 -0700793 }
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500794
795 // We'll draw one of each frame, so make it big enough to hold them all
796 // in a grid. The grid will be roughly square, with "factor" frames per
797 // row and up to "factor" rows.
798 const size_t count = codec->getFrameInfo().size();
799 const float root = sqrt((float) count);
800 const int factor = sk_float_ceil2int(root);
801
802 auto androidCodec = SkAndroidCodec::MakeFromCodec(std::move(codec));
803 auto imageSize = androidCodec->getSampledDimensions(1 / fScale);
804 imageSize.fWidth = imageSize.fWidth * factor;
805 imageSize.fHeight = imageSize.fHeight * sk_float_ceil2int((float) count / (float) factor);
scroggo19b91532016-10-24 09:03:26 -0700806 return imageSize;
scroggo9b77ddd2015-03-19 06:03:39 -0700807}
808
809Name CodecSrc::name() const {
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500810 Name name = SkOSPath::Basename(fPath.c_str());
811 if (fMode == kAnimated_Mode) {
812 name.append("_animated");
813 }
msarett0a242972015-06-11 14:27:27 -0700814 if (1.0f == fScale) {
scroggo6e8c68e2016-10-24 13:48:49 -0700815 return name;
msarett0a242972015-06-11 14:27:27 -0700816 }
Leon Scroggins1340dbd2020-11-09 14:18:12 -0500817 return get_scaled_name(name.c_str(), fScale);
scroggo9b77ddd2015-03-19 06:03:39 -0700818}
819
820/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
821
scroggof8dc9df2016-05-16 09:04:13 -0700822AndroidCodecSrc::AndroidCodecSrc(Path path, CodecSrc::DstColorType dstColorType,
scroggoc5560be2016-02-03 09:42:42 -0800823 SkAlphaType dstAlphaType, int sampleSize)
msarett3d9d7a72015-10-21 10:27:10 -0700824 : fPath(path)
msarett3d9d7a72015-10-21 10:27:10 -0700825 , fDstColorType(dstColorType)
scroggoc5560be2016-02-03 09:42:42 -0800826 , fDstAlphaType(dstAlphaType)
msarett3d9d7a72015-10-21 10:27:10 -0700827 , fSampleSize(sampleSize)
scroggo3ac66e92016-02-08 15:09:48 -0800828 , fRunSerially(serial_from_path_name(path))
msarett3d9d7a72015-10-21 10:27:10 -0700829{}
830
831bool AndroidCodecSrc::veto(SinkFlags flags) const {
832 // No need to test decoding to non-raster or indirect backend.
msarett3d9d7a72015-10-21 10:27:10 -0700833 return flags.type != SinkFlags::kRaster
834 || flags.approach != SinkFlags::kDirect;
835}
836
Robert Phillipsb87b39b2020-07-01 14:45:24 -0400837Result AndroidCodecSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
bungeman38d909e2016-08-02 14:40:46 -0700838 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
msarett3d9d7a72015-10-21 10:27:10 -0700839 if (!encoded) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500840 return Result::Fatal("Couldn't read %s.", fPath.c_str());
msarett3d9d7a72015-10-21 10:27:10 -0700841 }
Mike Reedede7bac2017-07-23 15:30:02 -0400842 std::unique_ptr<SkAndroidCodec> codec(SkAndroidCodec::MakeFromData(encoded));
843 if (nullptr == codec) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500844 return Result::Fatal("Couldn't create android codec for %s.", fPath.c_str());
msarett3d9d7a72015-10-21 10:27:10 -0700845 }
846
scroggoba584892016-05-20 13:56:13 -0700847 SkImageInfo decodeInfo = codec->getInfo();
848 if (!get_decode_info(&decodeInfo, canvas->imageInfo().colorType(), fDstColorType,
849 fDstAlphaType)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500850 return Result::Skip("Skipping uninteresting test.");
msarett3d9d7a72015-10-21 10:27:10 -0700851 }
852
853 // Scale the image if it is desired.
854 SkISize size = codec->getSampledDimensions(fSampleSize);
855
856 // Visually inspecting very small output images is not necessary. We will
857 // cover these cases in unit testing.
858 if ((size.width() <= 10 || size.height() <= 10) && 1 != fSampleSize) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500859 return Result::Skip("Scaling very small images is uninteresting.");
msarett3d9d7a72015-10-21 10:27:10 -0700860 }
Brian Salomon9241a6d2019-10-03 13:26:54 -0400861 decodeInfo = decodeInfo.makeDimensions(size);
msarett3d9d7a72015-10-21 10:27:10 -0700862
Mike Reed7fcfb622018-02-09 13:26:46 -0500863 int bpp = decodeInfo.bytesPerPixel();
msarettb1be46b2016-05-17 08:52:11 -0700864 size_t rowBytes = size.width() * bpp;
865 SkAutoMalloc pixels(size.height() * rowBytes);
msarett3d9d7a72015-10-21 10:27:10 -0700866
867 SkBitmap bitmap;
msarett34e0ec42016-04-22 16:27:24 -0700868 SkImageInfo bitmapInfo = decodeInfo;
Matt Sarettf485cf92017-05-26 08:58:03 -0400869 set_bitmap_color_space(&bitmapInfo);
msarett34e0ec42016-04-22 16:27:24 -0700870 if (kRGBA_8888_SkColorType == decodeInfo.colorType() ||
871 kBGRA_8888_SkColorType == decodeInfo.colorType()) {
872 bitmapInfo = bitmapInfo.makeColorType(kN32_SkColorType);
873 }
msarett3d9d7a72015-10-21 10:27:10 -0700874
875 // Create options for the codec.
876 SkAndroidCodec::AndroidOptions options;
msarett3d9d7a72015-10-21 10:27:10 -0700877 options.fSampleSize = fSampleSize;
878
msarettb1be46b2016-05-17 08:52:11 -0700879 switch (codec->getAndroidPixels(decodeInfo, pixels.get(), rowBytes, &options)) {
scroggof8dc9df2016-05-16 09:04:13 -0700880 case SkCodec::kSuccess:
Leon Scroggins III674a1842017-07-06 12:26:09 -0400881 case SkCodec::kErrorInInput:
scroggof8dc9df2016-05-16 09:04:13 -0700882 case SkCodec::kIncompleteInput:
883 break;
msarett3d9d7a72015-10-21 10:27:10 -0700884 default:
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500885 return Result::Fatal("Couldn't getPixels %s.", fPath.c_str());
msarett3d9d7a72015-10-21 10:27:10 -0700886 }
Leon Scroggins571b30f2017-07-11 17:35:31 +0000887 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, fDstColorType);
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500888 return Result::Ok();
msarett3d9d7a72015-10-21 10:27:10 -0700889}
890
891SkISize AndroidCodecSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -0700892 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
Mike Reedede7bac2017-07-23 15:30:02 -0400893 std::unique_ptr<SkAndroidCodec> codec(SkAndroidCodec::MakeFromData(encoded));
msarett3d9d7a72015-10-21 10:27:10 -0700894 if (nullptr == codec) {
Hal Canaryfafe1352017-04-11 12:12:02 -0400895 return {0, 0};
msarett3d9d7a72015-10-21 10:27:10 -0700896 }
897 return codec->getSampledDimensions(fSampleSize);
898}
899
900Name AndroidCodecSrc::name() const {
901 // We will replicate the names used by CodecSrc so that images can
902 // be compared in Gold.
903 if (1 == fSampleSize) {
904 return SkOSPath::Basename(fPath.c_str());
905 }
msarett4b0778e2015-11-13 09:59:11 -0800906 return get_scaled_name(fPath, 1.0f / (float) fSampleSize);
msarett3d9d7a72015-10-21 10:27:10 -0700907}
908
909/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
910
msarett18976312016-03-09 14:20:58 -0800911ImageGenSrc::ImageGenSrc(Path path, Mode mode, SkAlphaType alphaType, bool isGpu)
912 : fPath(path)
913 , fMode(mode)
914 , fDstAlphaType(alphaType)
915 , fIsGpu(isGpu)
916 , fRunSerially(serial_from_path_name(path))
917{}
918
919bool ImageGenSrc::veto(SinkFlags flags) const {
920 if (fIsGpu) {
Brian Salomonbd7c5512017-03-07 09:08:36 -0500921 // MSAA runs tend to run out of memory and tests the same code paths as regular gpu configs.
922 return flags.type != SinkFlags::kGPU || flags.approach != SinkFlags::kDirect ||
923 flags.multisampled == SinkFlags::kMultisampled;
msarett18976312016-03-09 14:20:58 -0800924 }
925
926 return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
927}
928
Robert Phillipsb87b39b2020-07-01 14:45:24 -0400929Result ImageGenSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
msarett18976312016-03-09 14:20:58 -0800930 if (kRGB_565_SkColorType == canvas->imageInfo().colorType()) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500931 return Result::Skip("Uninteresting to test image generator to 565.");
msarett18976312016-03-09 14:20:58 -0800932 }
933
bungeman38d909e2016-08-02 14:40:46 -0700934 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
msarett18976312016-03-09 14:20:58 -0800935 if (!encoded) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500936 return Result::Fatal("Couldn't read %s.", fPath.c_str());
msarett18976312016-03-09 14:20:58 -0800937 }
938
msarettfc0b6d12016-03-17 13:50:17 -0700939#if defined(SK_BUILD_FOR_WIN)
940 // Initialize COM in order to test with WIC.
941 SkAutoCoInitialize com;
942 if (!com.succeeded()) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500943 return Result::Fatal("Could not initialize COM.");
msarettfc0b6d12016-03-17 13:50:17 -0700944 }
945#endif
946
Ben Wagner145dbcd2016-11-03 14:40:50 -0400947 std::unique_ptr<SkImageGenerator> gen(nullptr);
msarett18976312016-03-09 14:20:58 -0800948 switch (fMode) {
949 case kCodec_Mode:
Mike Reed185130c2017-02-15 15:14:16 -0500950 gen = SkCodecImageGenerator::MakeFromEncodedCodec(encoded);
msarett18976312016-03-09 14:20:58 -0800951 if (!gen) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500952 return Result::Fatal("Could not create codec image generator.");
msarett18976312016-03-09 14:20:58 -0800953 }
954 break;
msarettfc0b6d12016-03-17 13:50:17 -0700955 case kPlatform_Mode: {
msarett18976312016-03-09 14:20:58 -0800956#if defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS)
Leon Scroggins III0cbc10f2017-10-30 09:07:53 -0400957 gen = SkImageGeneratorCG::MakeFromEncodedCG(encoded);
msarettfc0b6d12016-03-17 13:50:17 -0700958#elif defined(SK_BUILD_FOR_WIN)
Leon Scroggins IIIa57488a2018-06-25 14:01:29 -0400959 gen = SkImageGeneratorWIC::MakeFromEncodedWIC(encoded);
Leon Scroggins III326b9892020-08-05 16:51:10 -0400960#elif defined(SK_ENABLE_NDK_IMAGES)
Leon Scroggins IIIf21d6b92020-08-05 10:44:17 -0400961 gen = SkImageGeneratorNDK::MakeFromEncodedNDK(encoded);
msarettfc0b6d12016-03-17 13:50:17 -0700962#endif
msarett18976312016-03-09 14:20:58 -0800963 if (!gen) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500964 return Result::Fatal("Could not create platform image generator.");
msarett18976312016-03-09 14:20:58 -0800965 }
966 break;
msarettfc0b6d12016-03-17 13:50:17 -0700967 }
msarett18976312016-03-09 14:20:58 -0800968 default:
969 SkASSERT(false);
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500970 return Result::Fatal("Invalid image generator mode");
msarett18976312016-03-09 14:20:58 -0800971 }
972
973 // Test deferred decoding path on GPU
974 if (fIsGpu) {
Mike Reed564d49e2020-07-28 12:52:31 -0400975 sk_sp<SkImage> image(SkImage::MakeFromGenerator(std::move(gen)));
msarett18976312016-03-09 14:20:58 -0800976 if (!image) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500977 return Result::Fatal("Could not create image from codec image generator.");
msarett18976312016-03-09 14:20:58 -0800978 }
979 canvas->drawImage(image, 0, 0);
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500980 return Result::Ok();
msarett18976312016-03-09 14:20:58 -0800981 }
mtklein343a63d2016-03-22 11:46:53 -0700982
msarett18976312016-03-09 14:20:58 -0800983 // Test various color and alpha types on CPU
984 SkImageInfo decodeInfo = gen->getInfo().makeAlphaType(fDstAlphaType);
mtklein343a63d2016-03-22 11:46:53 -0700985
Mike Reed7fcfb622018-02-09 13:26:46 -0500986 int bpp = decodeInfo.bytesPerPixel();
msarettb1be46b2016-05-17 08:52:11 -0700987 size_t rowBytes = decodeInfo.width() * bpp;
988 SkAutoMalloc pixels(decodeInfo.height() * rowBytes);
Brian Osmanc87cfb62018-07-11 09:08:46 -0400989 if (!gen->getPixels(decodeInfo, pixels.get(), rowBytes)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500990 Result::Status status = Result::Status::Fatal;
Matt Sarett05cb4c32017-03-02 12:07:46 -0500991#if defined(SK_BUILD_FOR_WIN)
992 if (kPlatform_Mode == fMode) {
993 // Do not issue a fatal error for WIC flakiness.
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500994 status = Result::Status::Skip;
Matt Sarett05cb4c32017-03-02 12:07:46 -0500995 }
996#endif
Ben Wagnerea25fcf2020-02-12 11:18:46 -0500997 return Result(status, "Image generator could not getPixels() for %s\n", fPath.c_str());
msarett18976312016-03-09 14:20:58 -0800998 }
999
Matt Sarettf485cf92017-05-26 08:58:03 -04001000 set_bitmap_color_space(&decodeInfo);
Leon Scroggins571b30f2017-07-11 17:35:31 +00001001 draw_to_canvas(canvas, decodeInfo, pixels.get(), rowBytes,
msarettb1be46b2016-05-17 08:52:11 -07001002 CodecSrc::kGetFromCanvas_DstColorType);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001003 return Result::Ok();
msarett18976312016-03-09 14:20:58 -08001004}
1005
1006SkISize ImageGenSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -07001007 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
Mike Reedede7bac2017-07-23 15:30:02 -04001008 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(encoded));
msarett18976312016-03-09 14:20:58 -08001009 if (nullptr == codec) {
Hal Canaryfafe1352017-04-11 12:12:02 -04001010 return {0, 0};
msarett18976312016-03-09 14:20:58 -08001011 }
1012 return codec->getInfo().dimensions();
1013}
1014
1015Name ImageGenSrc::name() const {
1016 return SkOSPath::Basename(fPath.c_str());
1017}
1018
1019/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1020
Mike Klein0d5d1422019-03-06 10:56:04 -06001021ColorCodecSrc::ColorCodecSrc(Path path, bool decode_to_dst) : fPath(path)
1022 , fDecodeToDst(decode_to_dst) {}
msarett69deca82016-04-29 09:38:40 -07001023
1024bool ColorCodecSrc::veto(SinkFlags flags) const {
1025 // Test to direct raster backends (8888 and 565).
1026 return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
1027}
1028
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001029Result ColorCodecSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
bungeman38d909e2016-08-02 14:40:46 -07001030 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
msarett69deca82016-04-29 09:38:40 -07001031 if (!encoded) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001032 return Result::Fatal("Couldn't read %s.", fPath.c_str());
msarett69deca82016-04-29 09:38:40 -07001033 }
1034
Mike Reedede7bac2017-07-23 15:30:02 -04001035 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(encoded));
1036 if (nullptr == codec) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001037 return Result::Fatal("Couldn't create codec for %s.", fPath.c_str());
msarett69deca82016-04-29 09:38:40 -07001038 }
1039
Mike Klein0d5d1422019-03-06 10:56:04 -06001040 SkImageInfo info = codec->getInfo();
1041 if (fDecodeToDst) {
Leon Scroggins III42a604f2020-02-06 15:47:58 -05001042 SkImageInfo canvasInfo = canvas->imageInfo();
1043 if (!canvasInfo.colorSpace()) {
1044 // This will skip color conversion, and the resulting images will
1045 // look different from images they are compared against in Gold, but
1046 // that doesn't mean they are wrong. We have a test verifying that
1047 // passing a null SkColorSpace skips conversion, so skip this
1048 // misleading test.
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001049 return Result::Skip("Skipping decoding without color transform.");
Leon Scroggins III42a604f2020-02-06 15:47:58 -05001050 }
1051 info = canvasInfo.makeDimensions(info.dimensions());
msarett50ce1f22016-07-29 06:23:33 -07001052 }
1053
Mike Reed844beb52021-01-25 15:36:09 -05001054 auto [image, result] = codec->getImage(info);
1055 switch (result) {
Leon Scroggins III674a1842017-07-06 12:26:09 -04001056 case SkCodec::kSuccess:
1057 case SkCodec::kErrorInInput:
1058 case SkCodec::kIncompleteInput:
Mike Reed844beb52021-01-25 15:36:09 -05001059 canvas->drawImage(image, 0,0);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001060 return Result::Ok();
Mike Klein0d5d1422019-03-06 10:56:04 -06001061 case SkCodec::kInvalidConversion:
1062 // TODO(mtklein): why are there formats we can't decode to?
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001063 return Result::Skip("SkCodec can't decode to this format.");
Leon Scroggins III674a1842017-07-06 12:26:09 -04001064 default:
Mike Reed844beb52021-01-25 15:36:09 -05001065 return Result::Fatal("Couldn't getPixels %s. Error code %d", fPath.c_str(), result);
msarett50ce1f22016-07-29 06:23:33 -07001066 }
msarett69deca82016-04-29 09:38:40 -07001067}
1068
1069SkISize ColorCodecSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -07001070 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
Mike Reedede7bac2017-07-23 15:30:02 -04001071 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(encoded));
msarett69deca82016-04-29 09:38:40 -07001072 if (nullptr == codec) {
Hal Canaryfafe1352017-04-11 12:12:02 -04001073 return {0, 0};
msarett69deca82016-04-29 09:38:40 -07001074 }
Hal Canaryfafe1352017-04-11 12:12:02 -04001075 return {codec->getInfo().width(), codec->getInfo().height()};
msarett69deca82016-04-29 09:38:40 -07001076}
1077
1078Name ColorCodecSrc::name() const {
1079 return SkOSPath::Basename(fPath.c_str());
1080}
1081
1082/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1083
Mike Klein19fb3972019-03-21 13:08:08 -05001084static DEFINE_int(skpViewportSize, 1000,
1085 "Width & height of the viewport used to crop skp rendering.");
1086
Robert Phillipsad8a43f2017-08-30 12:06:35 -04001087SKPSrc::SKPSrc(Path path) : fPath(path) { }
mtklein748ca3b2015-01-15 10:56:12 -08001088
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001089Result SKPSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
Hal Canary72b91cb2019-01-14 12:47:59 -05001090 std::unique_ptr<SkStream> stream = SkStream::MakeFromFile(fPath.c_str());
1091 if (!stream) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001092 return Result::Fatal("Couldn't read %s.", fPath.c_str());
Robert Phillipsad8a43f2017-08-30 12:06:35 -04001093 }
Hal Canary72b91cb2019-01-14 12:47:59 -05001094 sk_sp<SkPicture> pic(SkPicture::MakeFromStream(stream.get()));
1095 if (!pic) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001096 return Result::Fatal("Couldn't parse file %s.", fPath.c_str());
Hal Canary72b91cb2019-01-14 12:47:59 -05001097 }
1098 stream = nullptr; // Might as well drop this when we're done with it.
Robert Phillips33f02ed2018-03-27 08:06:57 -04001099 canvas->clipRect(SkRect::MakeWH(FLAGS_skpViewportSize, FLAGS_skpViewportSize));
mtklein748ca3b2015-01-15 10:56:12 -08001100 canvas->drawPicture(pic);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001101 return Result::Ok();
mtklein748ca3b2015-01-15 10:56:12 -08001102}
1103
Robert Phillipsad8a43f2017-08-30 12:06:35 -04001104static SkRect get_cull_rect_for_skp(const char* path) {
1105 std::unique_ptr<SkStream> stream = SkStream::MakeFromFile(path);
mtkleinffa901a2015-03-16 10:38:07 -07001106 if (!stream) {
Robert Phillipsad8a43f2017-08-30 12:06:35 -04001107 return SkRect::MakeEmpty();
mtkleinffa901a2015-03-16 10:38:07 -07001108 }
1109 SkPictInfo info;
Mike Reede7a58322017-12-20 14:09:20 -05001110 if (!SkPicture_StreamIsSKP(stream.get(), &info)) {
Robert Phillipsad8a43f2017-08-30 12:06:35 -04001111 return SkRect::MakeEmpty();
mtkleinffa901a2015-03-16 10:38:07 -07001112 }
Robert Phillipsad8a43f2017-08-30 12:06:35 -04001113
1114 return info.fCullRect;
1115}
1116
1117SkISize SKPSrc::size() const {
1118 SkRect viewport = get_cull_rect_for_skp(fPath.c_str());
Robert Phillips33f02ed2018-03-27 08:06:57 -04001119 if (!viewport.intersect((SkRect::MakeWH(FLAGS_skpViewportSize, FLAGS_skpViewportSize)))) {
Hal Canaryfafe1352017-04-11 12:12:02 -04001120 return {0, 0};
mtkleinffa901a2015-03-16 10:38:07 -07001121 }
1122 return viewport.roundOut().size();
mtklein748ca3b2015-01-15 10:56:12 -08001123}
1124
1125Name SKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
1126
Florin Malitafc043dc2017-12-31 11:08:42 -05001127/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1128
Chris Dalton184c37e2018-09-28 11:27:39 -06001129BisectSrc::BisectSrc(Path path, const char* trail) : INHERITED(path), fTrail(trail) {}
1130
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001131Result BisectSrc::draw(GrDirectContext* context, SkCanvas* canvas) const {
Chris Dalton184c37e2018-09-28 11:27:39 -06001132 struct FoundPath {
1133 SkPath fPath;
1134 SkPaint fPaint;
1135 SkMatrix fViewMatrix;
1136 };
1137
1138 // This subclass of SkCanvas just extracts all the SkPaths (drawn via drawPath) from an SKP.
1139 class PathFindingCanvas : public SkCanvas {
1140 public:
1141 PathFindingCanvas(int width, int height) : SkCanvas(width, height, nullptr) {}
1142 const SkTArray<FoundPath>& foundPaths() const { return fFoundPaths; }
1143
1144 private:
1145 void onDrawPath(const SkPath& path, const SkPaint& paint) override {
1146 fFoundPaths.push_back() = {path, paint, this->getTotalMatrix()};
1147 }
1148
1149 SkTArray<FoundPath> fFoundPaths;
1150 };
1151
1152 PathFindingCanvas pathFinder(canvas->getBaseLayerSize().width(),
1153 canvas->getBaseLayerSize().height());
Robert Phillips889d6132020-06-16 11:11:33 -04001154 Result result = this->INHERITED::draw(context, &pathFinder);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001155 if (!result.isOk()) {
1156 return result;
Chris Dalton184c37e2018-09-28 11:27:39 -06001157 }
1158
1159 int start = 0, end = pathFinder.foundPaths().count();
1160 for (const char* ch = fTrail.c_str(); *ch; ++ch) {
1161 int midpt = (start + end) / 2;
1162 if ('l' == *ch) {
1163 start = midpt;
1164 } else if ('r' == *ch) {
1165 end = midpt;
1166 }
1167 }
1168
1169 for (int i = start; i < end; ++i) {
1170 const FoundPath& path = pathFinder.foundPaths()[i];
1171 SkAutoCanvasRestore acr(canvas, true);
1172 canvas->concat(path.fViewMatrix);
1173 canvas->drawPath(path.fPath, path.fPaint);
1174 }
1175
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001176 return Result::Ok();
Chris Dalton184c37e2018-09-28 11:27:39 -06001177}
1178
1179/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1180
Florin Malita87ccf332018-05-04 12:23:24 -04001181#if defined(SK_ENABLE_SKOTTIE)
Florin Malita67ff5412020-05-20 17:04:21 -04001182static DEFINE_bool(useLottieGlyphPaths, false,
1183 "Prioritize embedded glyph paths over native fonts.");
1184
Florin Malita9f7d4cd2018-07-30 15:49:20 -04001185SkottieSrc::SkottieSrc(Path path) : fPath(std::move(path)) {}
Florin Malitafc043dc2017-12-31 11:08:42 -05001186
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001187Result SkottieSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
Florin Malitafbddfbb2020-05-06 15:55:18 -04001188 auto resource_provider =
1189 skresources::DataURIResourceProviderProxy::Make(
1190 skresources::FileResourceProvider::Make(SkOSPath::Dirname(fPath.c_str()),
1191 /*predecode=*/true),
1192 /*predecode=*/true);
1193
1194 static constexpr char kInterceptPrefix[] = "__";
1195 auto precomp_interceptor =
1196 sk_make_sp<skottie_utils::ExternalAnimationPrecompInterceptor>(resource_provider,
1197 kInterceptPrefix);
Florin Malita67ff5412020-05-20 17:04:21 -04001198 uint32_t flags = 0;
1199 if (FLAGS_useLottieGlyphPaths) {
1200 flags |= skottie::Animation::Builder::kPreferEmbeddedFonts;
1201 }
1202
1203 auto animation = skottie::Animation::Builder(flags)
Florin Malitafbddfbb2020-05-06 15:55:18 -04001204 .setResourceProvider(std::move(resource_provider))
1205 .setPrecompInterceptor(std::move(precomp_interceptor))
Florin Malitaa8316552018-11-09 16:19:44 -05001206 .makeFromFile(fPath.c_str());
Florin Malita9f7d4cd2018-07-30 15:49:20 -04001207 if (!animation) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001208 return Result::Fatal("Unable to parse file: %s", fPath.c_str());
Florin Malitafc043dc2017-12-31 11:08:42 -05001209 }
1210
1211 canvas->drawColor(SK_ColorWHITE);
1212
Florin Malitaa33447d2018-05-29 13:46:54 -04001213 const auto t_rate = 1.0f / (kTileCount * kTileCount - 1);
Florin Malitafc043dc2017-12-31 11:08:42 -05001214
Kevin Lubickc5fe15d2018-08-29 09:52:52 -04001215 // Draw the frames in a shuffled order to exercise non-linear
1216 // frame progression. The film strip will still be in order left-to-right,
1217 // top-down, just not drawn in that order.
1218 static constexpr int frameOrder[] = { 4, 0, 3, 1, 2 };
1219 static_assert(SK_ARRAY_COUNT(frameOrder) == kTileCount, "");
Florin Malitad3c1b842018-01-27 12:43:24 -05001220
Florin Malitafc043dc2017-12-31 11:08:42 -05001221 for (int i = 0; i < kTileCount; ++i) {
Kevin Lubickc5fe15d2018-08-29 09:52:52 -04001222 const SkScalar y = frameOrder[i] * kTileSize;
Florin Malitafc043dc2017-12-31 11:08:42 -05001223
1224 for (int j = 0; j < kTileCount; ++j) {
Kevin Lubickc5fe15d2018-08-29 09:52:52 -04001225 const SkScalar x = frameOrder[j] * kTileSize;
Florin Malita9f7d4cd2018-07-30 15:49:20 -04001226 SkRect dest = SkRect::MakeXYWH(x, y, kTileSize, kTileSize);
Florin Malitafc043dc2017-12-31 11:08:42 -05001227
Kevin Lubickc5fe15d2018-08-29 09:52:52 -04001228 const auto t = t_rate * (frameOrder[i] * kTileCount + frameOrder[j]);
Florin Malitadf2713c2018-01-09 15:51:21 -05001229 {
1230 SkAutoCanvasRestore acr(canvas, true);
Florin Malitad3c1b842018-01-27 12:43:24 -05001231 canvas->clipRect(dest, true);
Mike Reed2ac6ce82021-01-15 12:26:22 -05001232 canvas->concat(SkMatrix::RectToRect(SkRect::MakeSize(animation->size()), dest,
1233 SkMatrix::kCenter_ScaleToFit));
Florin Malita9f7d4cd2018-07-30 15:49:20 -04001234 animation->seek(t);
1235 animation->render(canvas);
Florin Malitadf2713c2018-01-09 15:51:21 -05001236 }
Florin Malitafc043dc2017-12-31 11:08:42 -05001237 }
1238 }
1239
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001240 return Result::Ok();
Florin Malitafc043dc2017-12-31 11:08:42 -05001241}
1242
Florin Malita54f65c42018-01-16 17:04:30 -05001243SkISize SkottieSrc::size() const {
Florin Malita9f7d4cd2018-07-30 15:49:20 -04001244 return SkISize::Make(kTargetSize, kTargetSize);
Florin Malitafc043dc2017-12-31 11:08:42 -05001245}
1246
Florin Malita9f7d4cd2018-07-30 15:49:20 -04001247Name SkottieSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
Florin Malitafc043dc2017-12-31 11:08:42 -05001248
Florin Malita54f65c42018-01-16 17:04:30 -05001249bool SkottieSrc::veto(SinkFlags flags) const {
Florin Malitafc043dc2017-12-31 11:08:42 -05001250 // No need to test to non-(raster||gpu||vector) or indirect backends.
1251 bool type_ok = flags.type == SinkFlags::kRaster
1252 || flags.type == SinkFlags::kGPU
1253 || flags.type == SinkFlags::kVector;
1254
1255 return !type_ok || flags.approach != SinkFlags::kDirect;
1256}
Florin Malita124d5af2017-12-31 17:02:26 -05001257#endif
Florin Malitafc043dc2017-12-31 11:08:42 -05001258
1259/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
Florin Malita02616ea2020-06-25 13:33:17 -04001260#if defined(SK_ENABLE_SKRIVE)
1261SkRiveSrc::SkRiveSrc(Path path) : fPath(std::move(path)) {}
1262
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001263Result SkRiveSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
John Stilesfb355542020-06-26 13:55:28 -04001264 auto fileStream = SkFILEStream::Make(fPath.c_str());
1265 if (!fileStream) {
1266 return Result::Fatal("Unable to open file: %s", fPath.c_str());
1267 }
1268
1269 const auto skrive = skrive::SkRive::Builder().make(std::move(fileStream));
Florin Malita02616ea2020-06-25 13:33:17 -04001270 if (!skrive) {
1271 return Result::Fatal("Unable to parse file: %s", fPath.c_str());
1272 }
1273
1274 auto bounds = SkRect::MakeEmpty();
1275
1276 for (const auto& ab : skrive->artboards()) {
1277 const auto& pos = ab->getTranslation();
1278 const auto& size = ab->getSize();
1279
1280 bounds.join(SkRect::MakeXYWH(pos.x, pos.y, size.x, size.y));
1281 }
1282
1283 canvas->drawColor(SK_ColorWHITE);
1284
1285 if (!bounds.isEmpty()) {
1286 // TODO: tiled frames when we add animation support
1287 SkAutoCanvasRestore acr(canvas, true);
Mike Reed2ac6ce82021-01-15 12:26:22 -05001288 canvas->concat(SkMatrix::RectToRect(bounds, SkRect::MakeWH(kTargetSize, kTargetSize),
1289 SkMatrix::kCenter_ScaleToFit));
Florin Malita02616ea2020-06-25 13:33:17 -04001290 for (const auto& ab : skrive->artboards()) {
1291 ab->render(canvas);
1292 }
1293 }
1294
1295 return Result::Ok();
1296}
1297
1298SkISize SkRiveSrc::size() const {
1299 return SkISize::Make(kTargetSize, kTargetSize);
1300}
1301
1302Name SkRiveSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
1303
1304bool SkRiveSrc::veto(SinkFlags flags) const {
1305 // No need to test to non-(raster||gpu||vector) or indirect backends.
1306 bool type_ok = flags.type == SinkFlags::kRaster
1307 || flags.type == SinkFlags::kGPU
1308 || flags.type == SinkFlags::kVector;
1309
1310 return !type_ok || flags.approach != SinkFlags::kDirect;
1311}
1312#endif
1313
1314/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
Robert Phillips2af13c12021-09-01 16:47:01 +00001315#if defined(SK_ENABLE_SVG)
fmalitabdf3e5c2016-09-17 07:26:26 -07001316// Used when the image doesn't have an intrinsic size.
Hal Canaryfafe1352017-04-11 12:12:02 -04001317static const SkSize kDefaultSVGSize = {1000, 1000};
fmalitaa2b9fdf2016-08-03 19:53:36 -07001318
fmalitabdf3e5c2016-09-17 07:26:26 -07001319// Used to force-scale tiny fixed-size images.
Hal Canaryfafe1352017-04-11 12:12:02 -04001320static const SkSize kMinimumSVGSize = {128, 128};
fmalitaa2b9fdf2016-08-03 19:53:36 -07001321
fmalitaacd2f5c2016-11-08 07:13:45 -08001322SVGSrc::SVGSrc(Path path)
1323 : fName(SkOSPath::Basename(path.c_str()))
1324 , fScale(1) {
fmalitabdf3e5c2016-09-17 07:26:26 -07001325
Florin Malita24df67d2021-01-26 18:45:34 -05001326 auto stream = SkStream::MakeFromFile(path.c_str());
1327 if (!stream) {
Brian Osman133823d2018-09-19 14:14:15 -04001328 return;
1329 }
fmalitaacd2f5c2016-11-08 07:13:45 -08001330
Florin Malita24df67d2021-01-26 18:45:34 -05001331 auto rp = skresources::DataURIResourceProviderProxy::Make(
1332 skresources::FileResourceProvider::Make(SkOSPath::Dirname(path.c_str()),
1333 /*predecode=*/true),
1334 /*predecode=*/true);
1335 fDom = SkSVGDOM::Builder().setResourceProvider(std::move(rp))
1336 .make(*stream);
Brian Osman133823d2018-09-19 14:14:15 -04001337 if (!fDom) {
1338 return;
1339 }
1340
1341 const SkSize& sz = fDom->containerSize();
1342 if (sz.isEmpty()) {
1343 // no intrinsic size
1344 fDom->setContainerSize(kDefaultSVGSize);
1345 } else {
Brian Osman788b9162020-02-07 10:36:46 -05001346 fScale = std::max(1.f, std::max(kMinimumSVGSize.width() / sz.width(),
Brian Osman133823d2018-09-19 14:14:15 -04001347 kMinimumSVGSize.height() / sz.height()));
1348 }
fmalitaacd2f5c2016-11-08 07:13:45 -08001349}
1350
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001351Result SVGSrc::draw(GrDirectContext*, SkCanvas* canvas) const {
fmalitabdf3e5c2016-09-17 07:26:26 -07001352 if (!fDom) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001353 return Result::Fatal("Unable to parse file: %s", fName.c_str());
fmalitaa2b9fdf2016-08-03 19:53:36 -07001354 }
1355
fmalitaacd2f5c2016-11-08 07:13:45 -08001356 SkAutoCanvasRestore acr(canvas, true);
1357 canvas->scale(fScale, fScale);
Florin Malita385e7442020-10-21 16:55:46 -04001358 canvas->drawColor(SK_ColorWHITE);
fmalitaacd2f5c2016-11-08 07:13:45 -08001359 fDom->render(canvas);
1360
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001361 return Result::Ok();
fmalitaa2b9fdf2016-08-03 19:53:36 -07001362}
1363
1364SkISize SVGSrc::size() const {
fmalitaacd2f5c2016-11-08 07:13:45 -08001365 if (!fDom) {
Hal Canaryfafe1352017-04-11 12:12:02 -04001366 return {0, 0};
fmalitabdf3e5c2016-09-17 07:26:26 -07001367 }
1368
Hal Canaryfafe1352017-04-11 12:12:02 -04001369 return SkSize{fDom->containerSize().width() * fScale, fDom->containerSize().height() * fScale}
1370 .toRound();
fmalitaa2b9fdf2016-08-03 19:53:36 -07001371}
1372
fmalitaacd2f5c2016-11-08 07:13:45 -08001373Name SVGSrc::name() const { return fName; }
fmalitaa2b9fdf2016-08-03 19:53:36 -07001374
fmalita179d8852016-08-16 14:23:29 -07001375bool SVGSrc::veto(SinkFlags flags) const {
Florin Malita93323eb2017-06-16 15:31:17 -04001376 // No need to test to non-(raster||gpu||vector) or indirect backends.
fmalita179d8852016-08-16 14:23:29 -07001377 bool type_ok = flags.type == SinkFlags::kRaster
Florin Malita93323eb2017-06-16 15:31:17 -04001378 || flags.type == SinkFlags::kGPU
1379 || flags.type == SinkFlags::kVector;
fmalita179d8852016-08-16 14:23:29 -07001380
1381 return !type_ok || flags.approach != SinkFlags::kDirect;
1382}
1383
Robert Phillips2af13c12021-09-01 16:47:01 +00001384#endif // defined(SK_ENABLE_SVG)
fmalitaa2b9fdf2016-08-03 19:53:36 -07001385/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
mtklein748ca3b2015-01-15 10:56:12 -08001386
halcanary45420a92016-06-02 12:41:14 -07001387MSKPSrc::MSKPSrc(Path path) : fPath(path) {
bungemanf93d7112016-09-16 06:24:20 -07001388 std::unique_ptr<SkStreamAsset> stream = SkStream::MakeFromFile(fPath.c_str());
Hal Canary45cde312017-04-03 16:06:42 -04001389 int count = SkMultiPictureDocumentReadPageCount(stream.get());
1390 if (count > 0) {
1391 fPages.reset(count);
1392 (void)SkMultiPictureDocumentReadPageSizes(stream.get(), &fPages[0], fPages.count());
1393 }
halcanary45420a92016-06-02 12:41:14 -07001394}
1395
Hal Canary45cde312017-04-03 16:06:42 -04001396int MSKPSrc::pageCount() const { return fPages.count(); }
halcanary45420a92016-06-02 12:41:14 -07001397
1398SkISize MSKPSrc::size() const { return this->size(0); }
Hal Canary45cde312017-04-03 16:06:42 -04001399SkISize MSKPSrc::size(int i) const {
Hal Canaryfafe1352017-04-11 12:12:02 -04001400 return i >= 0 && i < fPages.count() ? fPages[i].fSize.toCeil() : SkISize{0, 0};
Hal Canary45cde312017-04-03 16:06:42 -04001401}
halcanary45420a92016-06-02 12:41:14 -07001402
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001403Result MSKPSrc::draw(GrDirectContext* context, SkCanvas* c) const {
Robert Phillips889d6132020-06-16 11:11:33 -04001404 return this->draw(0, context, c);
Robert Phillips762cb4e2020-06-15 13:12:32 -04001405}
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001406Result MSKPSrc::draw(int i, GrDirectContext*, SkCanvas* canvas) const {
Hal Canary45cde312017-04-03 16:06:42 -04001407 if (this->pageCount() == 0) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001408 return Result::Fatal("Unable to parse MultiPictureDocument file: %s", fPath.c_str());
halcanary45420a92016-06-02 12:41:14 -07001409 }
Hal Canary45cde312017-04-03 16:06:42 -04001410 if (i >= fPages.count() || i < 0) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001411 return Result::Fatal("MultiPictureDocument page number out of range: %d", i);
halcanary45420a92016-06-02 12:41:14 -07001412 }
Hal Canary45cde312017-04-03 16:06:42 -04001413 SkPicture* page = fPages[i].fPicture.get();
halcanary45420a92016-06-02 12:41:14 -07001414 if (!page) {
Hal Canary45cde312017-04-03 16:06:42 -04001415 std::unique_ptr<SkStreamAsset> stream = SkStream::MakeFromFile(fPath.c_str());
1416 if (!stream) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001417 return Result::Fatal("Unable to open file: %s", fPath.c_str());
Hal Canary45cde312017-04-03 16:06:42 -04001418 }
1419 if (!SkMultiPictureDocumentRead(stream.get(), &fPages[0], fPages.count())) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001420 return Result::Fatal("SkMultiPictureDocument reader failed on page %d: %s", i,
1421 fPath.c_str());
Hal Canary45cde312017-04-03 16:06:42 -04001422 }
1423 page = fPages[i].fPicture.get();
halcanary45420a92016-06-02 12:41:14 -07001424 }
1425 canvas->drawPicture(page);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001426 return Result::Ok();
halcanary45420a92016-06-02 12:41:14 -07001427}
1428
1429Name MSKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
1430
1431/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1432
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001433Result NullSink::draw(const Src& src, SkBitmap*, SkWStream*, SkString*) const {
Robert Phillips889d6132020-06-16 11:11:33 -04001434 return src.draw(nullptr, SkMakeNullCanvas().get());
mtkleinad66f9b2015-02-13 15:11:10 -08001435}
1436
1437/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1438
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001439static Result compare_bitmaps(const SkBitmap& reference, const SkBitmap& bitmap) {
Brian Osmane5756ec2017-09-06 17:08:30 -04001440 // The dimensions are a property of the Src only, and so should be identical.
Mike Reedf0ffb892017-10-03 14:47:21 -04001441 SkASSERT(reference.computeByteSize() == bitmap.computeByteSize());
1442 if (reference.computeByteSize() != bitmap.computeByteSize()) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001443 return Result::Fatal("Dimensions don't match reference");
Brian Osmane5756ec2017-09-06 17:08:30 -04001444 }
1445 // All SkBitmaps in DM are tight, so this comparison is easy.
Mike Reedf0ffb892017-10-03 14:47:21 -04001446 if (0 != memcmp(reference.getPixels(), bitmap.getPixels(), reference.computeByteSize())) {
Brian Osmane5756ec2017-09-06 17:08:30 -04001447 SkString encoded;
1448 SkString errString("Pixels don't match reference");
Brian Salomon28a8f282019-10-24 20:07:39 -04001449 if (BipmapToBase64DataURI(reference, &encoded)) {
Michael Ludwige8e10752018-10-01 12:42:53 -04001450 errString.append("\nExpected: ");
Brian Osmane5756ec2017-09-06 17:08:30 -04001451 errString.append(encoded);
1452 } else {
1453 errString.append("\nExpected image failed to encode: ");
1454 errString.append(encoded);
1455 }
Brian Salomon28a8f282019-10-24 20:07:39 -04001456 if (BipmapToBase64DataURI(bitmap, &encoded)) {
Michael Ludwige8e10752018-10-01 12:42:53 -04001457 errString.append("\nActual: ");
Brian Osmane5756ec2017-09-06 17:08:30 -04001458 errString.append(encoded);
1459 } else {
1460 errString.append("\nActual image failed to encode: ");
1461 errString.append(encoded);
1462 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001463 return Result::Fatal(errString);
Brian Osmane5756ec2017-09-06 17:08:30 -04001464 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001465 return Result::Ok();
Brian Osmane5756ec2017-09-06 17:08:30 -04001466}
1467
1468/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1469
Mike Klein84836b72019-03-21 11:31:36 -05001470static DEFINE_bool(gpuStats, false, "Append GPU stats to the log for each GPU task?");
Mike Klein19fb3972019-03-21 13:08:08 -05001471static DEFINE_bool(preAbandonGpuContext, false,
1472 "Test abandoning the GrContext before running the test.");
1473static DEFINE_bool(abandonGpuContext, false,
1474 "Test abandoning the GrContext after running each test.");
1475static DEFINE_bool(releaseAndAbandonGpuContext, false,
1476 "Test releasing all gpu resources and abandoning the GrContext "
1477 "after running each test");
1478static DEFINE_bool(drawOpClip, false, "Clip each GrDrawOp to its device bounds for testing.");
Brian Osman9c310472019-06-27 16:43:27 -04001479static DEFINE_bool(programBinaryCache, true, "Use in-memory program binary cache");
mtkleinb9eb4ac2015-02-02 18:26:03 -08001480
Brian Osman3fdfe282019-09-09 13:46:52 -04001481GPUSink::GPUSink(const SkCommandLineConfigGpu* config,
Brian Osmanf21aa042017-08-21 16:48:46 -04001482 const GrContextOptions& grCtxOptions)
Brian Osman3fdfe282019-09-09 13:46:52 -04001483 : fContextType(config->getContextType())
1484 , fContextOverrides(config->getContextOverrides())
1485 , fSurfType(config->getSurfType())
1486 , fSampleCount(config->getSamples())
Chris Dalton180b4a12021-03-16 20:49:15 -06001487 , fSurfaceFlags(config->getSurfaceFlags())
Brian Osman3fdfe282019-09-09 13:46:52 -04001488 , fColorType(config->getColorType())
1489 , fAlphaType(config->getAlphaType())
Brian Osman9c310472019-06-27 16:43:27 -04001490 , fBaseContextOptions(grCtxOptions) {
1491 if (FLAGS_programBinaryCache) {
1492 fBaseContextOptions.fPersistentCache = &fMemoryCache;
1493 }
1494}
mtklein748ca3b2015-01-15 10:56:12 -08001495
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001496Result GPUSink::draw(const Src& src, SkBitmap* dst, SkWStream* dstStream, SkString* log) const {
Brian Osmanf9810662017-08-30 10:02:10 -04001497 return this->onDraw(src, dst, dstStream, log, fBaseContextOptions);
1498}
1499
Brian Salomonf9b00422020-10-08 16:00:14 -04001500sk_sp<SkSurface> GPUSink::createDstSurface(GrDirectContext* context, SkISize size) const {
Robert Phillips3c1efd42020-02-13 15:51:59 -05001501 sk_sp<SkSurface> surface;
1502
1503 SkImageInfo info = SkImageInfo::Make(size, fColorType, fAlphaType, fColorSpace);
Chris Dalton180b4a12021-03-16 20:49:15 -06001504 SkSurfaceProps props(fSurfaceFlags, kRGB_H_SkPixelGeometry);
Robert Phillips3c1efd42020-02-13 15:51:59 -05001505
1506 switch (fSurfType) {
1507 case SkCommandLineConfigGpu::SurfType::kDefault:
1508 surface = SkSurface::MakeRenderTarget(context, SkBudgeted::kNo, info, fSampleCount,
1509 &props);
1510 break;
1511 case SkCommandLineConfigGpu::SurfType::kBackendTexture:
Brian Salomonf9b00422020-10-08 16:00:14 -04001512 surface = sk_gpu_test::MakeBackendTextureSurface(context,
Brian Salomon72050802020-10-12 20:45:06 -04001513 info,
Brian Salomonf9b00422020-10-08 16:00:14 -04001514 kTopLeft_GrSurfaceOrigin,
1515 fSampleCount,
Brian Salomonf9b00422020-10-08 16:00:14 -04001516 GrMipmapped::kNo,
1517 GrProtected::kNo,
1518 &props);
Robert Phillips3c1efd42020-02-13 15:51:59 -05001519 break;
1520 case SkCommandLineConfigGpu::SurfType::kBackendRenderTarget:
Brian Salomonf9b00422020-10-08 16:00:14 -04001521 surface = sk_gpu_test::MakeBackendRenderTargetSurface(context,
Brian Salomon72050802020-10-12 20:45:06 -04001522 info,
Brian Salomonf9b00422020-10-08 16:00:14 -04001523 kBottomLeft_GrSurfaceOrigin,
1524 fSampleCount,
Brian Salomonf9b00422020-10-08 16:00:14 -04001525 GrProtected::kNo,
1526 &props);
Robert Phillips3c1efd42020-02-13 15:51:59 -05001527 break;
1528 }
1529
1530 return surface;
1531}
1532
1533bool GPUSink::readBack(SkSurface* surface, SkBitmap* dst) const {
1534 SkCanvas* canvas = surface->getCanvas();
1535 SkISize size = surface->imageInfo().dimensions();
1536
1537 SkImageInfo info = SkImageInfo::Make(size, fColorType, fAlphaType, fColorSpace);
Robert Phillips3c1efd42020-02-13 15:51:59 -05001538 dst->allocPixels(info);
1539 return canvas->readPixels(*dst, 0, 0);
1540}
1541
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001542Result GPUSink::onDraw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log,
1543 const GrContextOptions& baseOptions,
Robert Phillipse219c7e2020-08-13 15:58:29 -04001544 std::function<void(GrDirectContext*)> initContext) const {
Brian Osmanf9810662017-08-30 10:02:10 -04001545 GrContextOptions grOptions = baseOptions;
kkinnunen64492c42015-12-08 01:24:40 -08001546
Brian Salomon00a5eb82018-07-11 15:32:05 -04001547 // We don't expect the src to mess with the persistent cache or the executor.
1548 SkDEBUGCODE(auto cache = grOptions.fPersistentCache);
1549 SkDEBUGCODE(auto exec = grOptions.fExecutor);
kkinnunen5219fd92015-12-10 06:28:13 -08001550 src.modifyGrContextOptions(&grOptions);
Brian Salomon00a5eb82018-07-11 15:32:05 -04001551 SkASSERT(cache == grOptions.fPersistentCache);
1552 SkASSERT(exec == grOptions.fExecutor);
kkinnunen5219fd92015-12-10 06:28:13 -08001553
1554 GrContextFactory factory(grOptions);
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001555 auto direct = factory.getContextInfo(fContextType, fContextOverrides).directContext();
Brian Osmaned58e002019-09-06 14:42:43 -04001556 if (initContext) {
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001557 initContext(direct);
Brian Osmaned58e002019-09-06 14:42:43 -04001558 }
Robert Phillips3c1efd42020-02-13 15:51:59 -05001559
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001560 const int maxDimension = direct->priv().caps()->maxTextureSize();
Robert Phillips3c1efd42020-02-13 15:51:59 -05001561 if (maxDimension < std::max(src.size().width(), src.size().height())) {
Robert Phillips45f37c32020-02-13 20:26:58 +00001562 return Result::Skip("Src too large to create a texture.\n");
1563 }
Robert Phillips3c1efd42020-02-13 15:51:59 -05001564
Brian Salomonf9b00422020-10-08 16:00:14 -04001565 sk_sp<SkSurface> surface = this->createDstSurface(direct, src.size());
mtklein748ca3b2015-01-15 10:56:12 -08001566 if (!surface) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001567 return Result::Fatal("Could not create a surface.");
mtklein748ca3b2015-01-15 10:56:12 -08001568 }
joshualitt5f5a8d72015-02-25 14:09:45 -08001569 if (FLAGS_preAbandonGpuContext) {
joshualitt5f5a8d72015-02-25 14:09:45 -08001570 factory.abandonContexts();
1571 }
John Stiles65ec1982021-07-20 12:53:54 -04001572
1573 Result result = src.draw(direct, surface->getCanvas());
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001574 if (!result.isOk()) {
1575 return result;
mtklein748ca3b2015-01-15 10:56:12 -08001576 }
Greg Daniel0a2464f2020-05-14 15:45:44 -04001577 surface->flushAndSubmit();
mtkleinb9eb4ac2015-02-02 18:26:03 -08001578 if (FLAGS_gpuStats) {
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001579 direct->priv().dumpCacheStats(log);
1580 direct->priv().dumpGpuStats(log);
1581 direct->priv().dumpContextStats(log);
mtkleinb9eb4ac2015-02-02 18:26:03 -08001582 }
Robert Phillips3c1efd42020-02-13 15:51:59 -05001583
1584 this->readBack(surface.get(), dst);
1585
mtklein55e88b22015-01-21 15:50:13 -08001586 if (FLAGS_abandonGpuContext) {
1587 factory.abandonContexts();
bsalomon6e2aad42016-04-01 11:54:31 -07001588 } else if (FLAGS_releaseAndAbandonGpuContext) {
1589 factory.releaseResourcesAndAbandonContexts();
mtklein55e88b22015-01-21 15:50:13 -08001590 }
Brian Salomonf9b00422020-10-08 16:00:14 -04001591
Greg Daniela870b462019-01-08 15:49:46 -05001592 if (grOptions.fPersistentCache) {
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001593 direct->storeVkPipelineCacheData();
Greg Daniela870b462019-01-08 15:49:46 -05001594 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001595 return Result::Ok();
mtklein748ca3b2015-01-15 10:56:12 -08001596}
1597
1598/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1599
Brian Osman3fdfe282019-09-09 13:46:52 -04001600GPUThreadTestingSink::GPUThreadTestingSink(const SkCommandLineConfigGpu* config,
Brian Osmanf9810662017-08-30 10:02:10 -04001601 const GrContextOptions& grCtxOptions)
Brian Osman3fdfe282019-09-09 13:46:52 -04001602 : INHERITED(config, grCtxOptions)
Mike Klein022cfa22017-09-01 11:53:16 -04001603 , fExecutor(SkExecutor::MakeFIFOThreadPool(FLAGS_gpuThreads)) {
Brian Osmanf9810662017-08-30 10:02:10 -04001604 SkASSERT(fExecutor);
1605}
1606
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001607Result GPUThreadTestingSink::draw(const Src& src, SkBitmap* dst, SkWStream* wStream,
Brian Osmanf9810662017-08-30 10:02:10 -04001608 SkString* log) const {
1609 // Draw twice, once with worker threads, and once without. Verify that we get the same result.
1610 // Also, force us to only use the software path renderer, so we really stress-test the threaded
1611 // version of that code.
1612 GrContextOptions contextOptions = this->baseContextOptions();
Brian Osman195c05b2017-08-30 15:14:04 -04001613 contextOptions.fGpuPathRenderers = GpuPathRenderers::kNone;
Greg Daniel024615e2018-05-10 17:25:52 +00001614 contextOptions.fExecutor = fExecutor.get();
Adrienne Walkerab7181d2018-05-14 14:02:03 -07001615
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001616 Result result = this->onDraw(src, dst, wStream, log, contextOptions);
1617 if (!result.isOk() || !dst) {
1618 return result;
Brian Osmanf9810662017-08-30 10:02:10 -04001619 }
1620
1621 SkBitmap reference;
1622 SkString refLog;
1623 SkDynamicMemoryWStream refStream;
1624 contextOptions.fExecutor = nullptr;
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001625 Result refResult = this->onDraw(src, &reference, &refStream, &refLog, contextOptions);
1626 if (!refResult.isOk()) {
1627 return refResult;
Brian Osmanf9810662017-08-30 10:02:10 -04001628 }
1629
Brian Osmane5756ec2017-09-06 17:08:30 -04001630 return compare_bitmaps(reference, *dst);
Brian Osmanf9810662017-08-30 10:02:10 -04001631}
1632
1633/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1634
Brian Osman3fdfe282019-09-09 13:46:52 -04001635GPUPersistentCacheTestingSink::GPUPersistentCacheTestingSink(const SkCommandLineConfigGpu* config,
1636 const GrContextOptions& grCtxOptions)
1637 : INHERITED(config, grCtxOptions)
1638 , fCacheType(config->getTestPersistentCache()) {}
Brian Salomon00a5eb82018-07-11 15:32:05 -04001639
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001640Result GPUPersistentCacheTestingSink::draw(const Src& src, SkBitmap* dst, SkWStream* wStream,
1641 SkString* log) const {
Brian Salomon00a5eb82018-07-11 15:32:05 -04001642 // Draw twice, once with a cold cache, and again with a warm cache. Verify that we get the same
1643 // result.
1644 sk_gpu_test::MemoryCache memoryCache;
1645 GrContextOptions contextOptions = this->baseContextOptions();
1646 contextOptions.fPersistentCache = &memoryCache;
Brian Osmana66081d2019-09-03 14:59:26 -04001647 if (fCacheType == 2) {
1648 contextOptions.fShaderCacheStrategy = GrContextOptions::ShaderCacheStrategy::kBackendSource;
1649 }
Brian Salomon00a5eb82018-07-11 15:32:05 -04001650
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001651 Result result = this->onDraw(src, dst, wStream, log, contextOptions);
1652 if (!result.isOk() || !dst) {
1653 return result;
Brian Salomon00a5eb82018-07-11 15:32:05 -04001654 }
1655
1656 SkBitmap reference;
1657 SkString refLog;
1658 SkDynamicMemoryWStream refStream;
Brian Osman43f443f2020-06-05 11:11:36 -04001659 memoryCache.resetCacheStats();
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001660 Result refResult = this->onDraw(src, &reference, &refStream, &refLog, contextOptions);
1661 if (!refResult.isOk()) {
1662 return refResult;
Brian Salomon00a5eb82018-07-11 15:32:05 -04001663 }
Brian Salomon7fc52992018-07-12 16:20:23 -04001664 SkASSERT(!memoryCache.numCacheMisses());
Brian Osman43f443f2020-06-05 11:11:36 -04001665 SkASSERT(!memoryCache.numCacheStores());
Brian Salomon00a5eb82018-07-11 15:32:05 -04001666
1667 return compare_bitmaps(reference, *dst);
1668}
1669
Brian Osmaned58e002019-09-06 14:42:43 -04001670
1671/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1672
Brian Osman3fdfe282019-09-09 13:46:52 -04001673GPUPrecompileTestingSink::GPUPrecompileTestingSink(const SkCommandLineConfigGpu* config,
1674 const GrContextOptions& grCtxOptions)
1675 : INHERITED(config, grCtxOptions) {}
Brian Osmaned58e002019-09-06 14:42:43 -04001676
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001677Result GPUPrecompileTestingSink::draw(const Src& src, SkBitmap* dst, SkWStream* wStream,
1678 SkString* log) const {
Brian Osmaned58e002019-09-06 14:42:43 -04001679 // Three step process:
1680 // 1) Draw once with an SkSL cache, and store off the shader blobs.
1681 // 2) For the second context, pre-compile the shaders to warm the cache.
1682 // 3) Draw with the second context, ensuring that we get the same result, and no cache misses.
1683 sk_gpu_test::MemoryCache memoryCache;
1684 GrContextOptions contextOptions = this->baseContextOptions();
1685 contextOptions.fPersistentCache = &memoryCache;
1686 contextOptions.fShaderCacheStrategy = GrContextOptions::ShaderCacheStrategy::kSkSL;
Brian Osmaned58e002019-09-06 14:42:43 -04001687
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001688 Result result = this->onDraw(src, dst, wStream, log, contextOptions);
1689 if (!result.isOk() || !dst) {
1690 return result;
Brian Osmaned58e002019-09-06 14:42:43 -04001691 }
1692
Robert Phillipse219c7e2020-08-13 15:58:29 -04001693 auto precompileShaders = [&memoryCache](GrDirectContext* dContext) {
1694 memoryCache.foreach([dContext](sk_sp<const SkData> key,
1695 sk_sp<SkData> data,
Brian Osmanf0de96f2021-02-26 13:54:11 -05001696 const SkString& /*description*/,
Robert Phillipse219c7e2020-08-13 15:58:29 -04001697 int /*count*/) {
1698 SkAssertResult(dContext->precompileShader(*key, *data));
Brian Osmaned58e002019-09-06 14:42:43 -04001699 });
1700 };
1701
1702 sk_gpu_test::MemoryCache replayCache;
1703 GrContextOptions replayOptions = this->baseContextOptions();
1704 // Ensure that the runtime cache is large enough to hold all of the shaders we pre-compile
1705 replayOptions.fRuntimeProgramCacheSize = memoryCache.numCacheMisses();
1706 replayOptions.fPersistentCache = &replayCache;
Brian Osmaned58e002019-09-06 14:42:43 -04001707
1708 SkBitmap reference;
1709 SkString refLog;
1710 SkDynamicMemoryWStream refStream;
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001711 Result refResult = this->onDraw(src, &reference, &refStream, &refLog, replayOptions,
1712 precompileShaders);
1713 if (!refResult.isOk()) {
1714 return refResult;
Brian Osmaned58e002019-09-06 14:42:43 -04001715 }
1716 SkASSERT(!replayCache.numCacheMisses());
1717
1718 return compare_bitmaps(reference, *dst);
1719}
1720
Robert Phillips62768192020-04-22 08:28:58 -04001721/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
Robert Phillips762cb4e2020-06-15 13:12:32 -04001722GPUOOPRSink::GPUOOPRSink(const SkCommandLineConfigGpu* config, const GrContextOptions& ctxOptions)
1723 : INHERITED(config, ctxOptions) {
1724}
1725
1726Result GPUOOPRSink::ooprDraw(const Src& src,
1727 sk_sp<SkSurface> dstSurface,
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001728 GrDirectContext* context) const {
Robert Phillips762cb4e2020-06-15 13:12:32 -04001729 SkSurfaceCharacterization dstCharacterization;
1730 SkAssertResult(dstSurface->characterize(&dstCharacterization));
1731
1732 SkDeferredDisplayListRecorder recorder(dstCharacterization);
1733
Robert Phillips889d6132020-06-16 11:11:33 -04001734 Result result = src.draw(context, recorder.getCanvas());
Robert Phillips762cb4e2020-06-15 13:12:32 -04001735 if (!result.isOk()) {
1736 return result;
1737 }
1738
Adlai Holler7580ad42020-06-24 13:45:25 -04001739 auto ddl = recorder.detach();
Robert Phillips762cb4e2020-06-15 13:12:32 -04001740
1741 SkDeferredDisplayList::ProgramIterator iter(context, ddl.get());
1742 for (; !iter.done(); iter.next()) {
1743 iter.compile();
1744 }
1745
Adlai Hollerf19bbb52020-06-29 10:00:08 -04001746 SkAssertResult(dstSurface->draw(std::move(ddl)));
Robert Phillips762cb4e2020-06-15 13:12:32 -04001747
1748 return Result::Ok();
1749}
1750
1751Result GPUOOPRSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log) const {
1752 GrContextOptions contextOptions = this->baseContextOptions();
1753 src.modifyGrContextOptions(&contextOptions);
1754 contextOptions.fPersistentCache = nullptr;
1755 contextOptions.fExecutor = nullptr;
1756
1757 GrContextFactory factory(contextOptions);
1758
1759 ContextInfo ctxInfo = factory.getContextInfo(this->contextType(), this->contextOverrides());
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001760 auto context = ctxInfo.directContext();
Robert Phillips762cb4e2020-06-15 13:12:32 -04001761 if (!context) {
1762 return Result::Fatal("Could not create context.");
1763 }
1764
1765 SkASSERT(context->priv().getGpu());
1766
Brian Salomonf9b00422020-10-08 16:00:14 -04001767 sk_sp<SkSurface> surface = this->createDstSurface(context, src.size());
Robert Phillips762cb4e2020-06-15 13:12:32 -04001768 if (!surface) {
1769 return Result::Fatal("Could not create a surface.");
1770 }
1771
1772 Result result = this->ooprDraw(src, surface, context);
1773 if (!result.isOk()) {
1774 return result;
1775 }
1776
1777 if (FLAGS_gpuStats) {
1778 context->priv().dumpCacheStats(log);
1779 context->priv().dumpGpuStats(log);
1780 context->priv().dumpContextStats(log);
1781 }
1782
1783 if (!this->readBack(surface.get(), dst)) {
1784 return Result::Fatal("Could not readback from surface.");
1785 }
1786
Robert Phillips762cb4e2020-06-15 13:12:32 -04001787 return Result::Ok();
1788}
1789
1790/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1791GPUDDLSink::GPUDDLSink(const SkCommandLineConfigGpu* config, const GrContextOptions& ctxOptions)
1792 : INHERITED(config, ctxOptions)
1793 , fRecordingExecutor(SkExecutor::MakeLIFOThreadPool(1))
1794 , fGPUExecutor(SkExecutor::MakeFIFOThreadPool(1, false)) {
Robert Phillips291f3402020-02-19 14:14:47 -05001795}
1796
1797Result GPUDDLSink::ddlDraw(const Src& src,
1798 sk_sp<SkSurface> dstSurface,
1799 SkTaskGroup* recordingTaskGroup,
1800 SkTaskGroup* gpuTaskGroup,
Robert Phillips19f466d2020-02-26 10:27:07 -05001801 sk_gpu_test::TestContext* gpuTestCtx,
Adlai Holler55aaefe2021-03-03 16:12:56 -07001802 GrDirectContext* dContext) const {
Robert Phillips19f466d2020-02-26 10:27:07 -05001803
1804 // We have to do this here bc characterization can hit the SkGpuDevice's thread guard (i.e.,
1805 // leaving it until the DDLTileHelper ctor will result in multiple threads trying to use the
1806 // same context (this thread and the gpuThread - which will be uploading textures)).
1807 SkSurfaceCharacterization dstCharacterization;
1808 SkAssertResult(dstSurface->characterize(&dstCharacterization));
1809
Robert Phillips291f3402020-02-19 14:14:47 -05001810 auto size = src.size();
1811 SkPictureRecorder recorder;
Adlai Holler55aaefe2021-03-03 16:12:56 -07001812 Result result = src.draw(dContext, recorder.beginRecording(SkIntToScalar(size.width()),
1813 SkIntToScalar(size.height())));
Robert Phillips291f3402020-02-19 14:14:47 -05001814 if (!result.isOk()) {
1815 return result;
1816 }
1817 sk_sp<SkPicture> inputPicture(recorder.finishRecordingAsPicture());
1818
1819 // this is our ultimate final drawing area/rect
1820 SkIRect viewport = SkIRect::MakeWH(size.fWidth, size.fHeight);
1821
Adlai Holler55aaefe2021-03-03 16:12:56 -07001822 SkYUVAPixmapInfo::SupportedDataTypes supportedYUVADataTypes(*dContext);
Brian Salomon59c60b02020-09-01 15:01:15 -04001823 DDLPromiseImageHelper promiseImageHelper(supportedYUVADataTypes);
Robert Phillips0d8722c2021-03-29 13:29:40 -04001824 sk_sp<SkPicture> newSKP = promiseImageHelper.recreateSKP(dContext, inputPicture.get());
1825 if (!newSKP) {
1826 return Result::Fatal("GPUDDLSink: Couldn't recreate the SKP");
Robert Phillips291f3402020-02-19 14:14:47 -05001827 }
1828
Robert Phillipsc869ff72020-06-19 09:50:33 -04001829 // 'gpuTestCtx/gpuThreadCtx' is being shifted to the gpuThread. Leave the main (this)
1830 // thread w/o a context.
1831 gpuTestCtx->makeNotCurrent();
1832
1833 // Job one for the GPU thread is to make 'gpuTestCtx' current!
1834 gpuTaskGroup->add([gpuTestCtx] { gpuTestCtx->makeCurrent(); });
1835
Robert Phillips291f3402020-02-19 14:14:47 -05001836 // TODO: move the image upload to the utility thread
Adlai Holler55aaefe2021-03-03 16:12:56 -07001837 promiseImageHelper.uploadAllToGPU(gpuTaskGroup, dContext);
Robert Phillips291f3402020-02-19 14:14:47 -05001838
Robert Phillips11c67672020-04-23 15:10:03 -04001839 // Care must be taken when using 'gpuThreadCtx' bc it moves between the gpu-thread and this
1840 // one. About all it can be consistently used for is GrCaps access and 'defaultBackendFormat'
1841 // calls.
Robert Phillips291f3402020-02-19 14:14:47 -05001842 constexpr int kNumDivisions = 3;
Adlai Holler55aaefe2021-03-03 16:12:56 -07001843 DDLTileHelper tiles(dContext, dstCharacterization, viewport,
Robert Phillips96f6d9a2021-02-26 10:41:06 -05001844 kNumDivisions, kNumDivisions,
Robert Phillips0c088492020-11-10 08:30:50 -05001845 /* addRandomPaddingToDst */ false);
Robert Phillips291f3402020-02-19 14:14:47 -05001846
Adlai Holler55aaefe2021-03-03 16:12:56 -07001847 tiles.createBackendTextures(gpuTaskGroup, dContext);
Robert Phillips8472a3d2020-04-16 16:27:45 -04001848
Robert Phillips0d8722c2021-03-29 13:29:40 -04001849 tiles.kickOffThreadedWork(recordingTaskGroup, gpuTaskGroup, dContext, newSKP.get());
Robert Phillips11c67672020-04-23 15:10:03 -04001850
1851 // We have to wait for the recording threads to schedule all their work on the gpu thread
1852 // before we can schedule the composition draw and the flush. Note that the gpu thread
1853 // is not blocked at this point and this thread is borrowing recording work.
Robert Phillips19f466d2020-02-26 10:27:07 -05001854 recordingTaskGroup->wait();
Robert Phillips291f3402020-02-19 14:14:47 -05001855
Robert Phillips11c67672020-04-23 15:10:03 -04001856 // Note: at this point the recording thread(s) are stalled out w/ nothing to do.
1857
1858 // The recording threads have already scheduled the drawing of each tile's DDL on the gpu
1859 // thread. The composition DDL must be scheduled last bc it relies on the result of all
1860 // the tiles' rendering. Additionally, bc we're aliasing the tiles' backend textures,
1861 // there is nothing in the DAG to automatically force the required order.
1862 gpuTaskGroup->add([dstSurface, ddl = tiles.composeDDL()]() {
1863 dstSurface->draw(ddl);
1864 });
1865
Robert Phillips5dbcca52020-05-29 10:41:33 -04001866 // This should be the only explicit flush for the entire DDL draw.
Adlai Holler55aaefe2021-03-03 16:12:56 -07001867 gpuTaskGroup->add([dContext]() {
Robert Phillips4e3ceb92020-04-16 15:42:50 -04001868 // We need to ensure all the GPU work is finished so
1869 // the following 'deleteAllFromGPU' call will work
1870 // on Vulkan.
1871 // TODO: switch over to using the promiseImage callbacks
1872 // to free the backendTextures. This is complicated a
1873 // bit by which thread possesses the direct context.
Adlai Holler55aaefe2021-03-03 16:12:56 -07001874 dContext->flush();
1875 dContext->submit(true);
Robert Phillips4e3ceb92020-04-16 15:42:50 -04001876 });
Robert Phillips291f3402020-02-19 14:14:47 -05001877
Robert Phillips19f466d2020-02-26 10:27:07 -05001878 // The backend textures are created on the gpuThread by the 'uploadAllToGPU' call.
1879 // It is simpler to also delete them at this point on the gpuThread.
Adlai Holler55aaefe2021-03-03 16:12:56 -07001880 promiseImageHelper.deleteAllFromGPU(gpuTaskGroup, dContext);
Robert Phillips19f466d2020-02-26 10:27:07 -05001881
Adlai Holler55aaefe2021-03-03 16:12:56 -07001882 tiles.deleteBackendTextures(gpuTaskGroup, dContext);
Robert Phillips8472a3d2020-04-16 16:27:45 -04001883
Robert Phillips19f466d2020-02-26 10:27:07 -05001884 // A flush has already been scheduled on the gpu thread along with the clean up of the backend
Robert Phillips8472a3d2020-04-16 16:27:45 -04001885 // textures so it is safe to schedule making 'gpuTestCtx' not current on the gpuThread.
Robert Phillips19f466d2020-02-26 10:27:07 -05001886 gpuTaskGroup->add([gpuTestCtx] { gpuTestCtx->makeNotCurrent(); });
1887
1888 // All the work is scheduled on the gpu thread, we just need to wait
Robert Phillips291f3402020-02-19 14:14:47 -05001889 gpuTaskGroup->wait();
1890
1891 return Result::Ok();
1892}
1893
Robert Phillips762cb4e2020-06-15 13:12:32 -04001894Result GPUDDLSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log) const {
Robert Phillips291f3402020-02-19 14:14:47 -05001895 GrContextOptions contextOptions = this->baseContextOptions();
1896 src.modifyGrContextOptions(&contextOptions);
1897 contextOptions.fPersistentCache = nullptr;
1898 contextOptions.fExecutor = nullptr;
1899
1900 GrContextFactory factory(contextOptions);
1901
1902 // This captures the context destined to be the main gpu context
1903 ContextInfo mainCtxInfo = factory.getContextInfo(this->contextType(), this->contextOverrides());
1904 sk_gpu_test::TestContext* mainTestCtx = mainCtxInfo.testContext();
Robert Phillipsb87b39b2020-07-01 14:45:24 -04001905 auto mainCtx = mainCtxInfo.directContext();
Robert Phillips291f3402020-02-19 14:14:47 -05001906 if (!mainCtx) {
1907 return Result::Fatal("Could not create context.");
1908 }
1909
1910 SkASSERT(mainCtx->priv().getGpu());
1911
1912 // TODO: make use of 'otherCtx' for uploads & compilation
1913#if 0
1914 // This captures the context destined to be the utility context. It is in a share group
1915 // with the main context
1916 ContextInfo otherCtxInfo = factory.getSharedContextInfo(mainCtx);
1917 sk_gpu_test::TestContext* otherTestCtx = otherCtxInfo.testContext();
Robert Phillipse219c7e2020-08-13 15:58:29 -04001918 auto otherCtx = otherCtxInfo.directContext();
Robert Phillips291f3402020-02-19 14:14:47 -05001919 if (!otherCtx) {
1920 return Result::Fatal("Cound not create shared context.");
1921 }
1922
1923 SkASSERT(otherCtx->priv().getGpu());
1924#endif
1925
Robert Phillips762cb4e2020-06-15 13:12:32 -04001926 SkTaskGroup recordingTaskGroup(*fRecordingExecutor);
1927 SkTaskGroup gpuTaskGroup(*fGPUExecutor);
Robert Phillips291f3402020-02-19 14:14:47 -05001928
1929 // Make sure 'mainCtx' is current
1930 mainTestCtx->makeCurrent();
1931
Brian Salomonf9b00422020-10-08 16:00:14 -04001932 sk_sp<SkSurface> surface = this->createDstSurface(mainCtx, src.size());
Robert Phillips291f3402020-02-19 14:14:47 -05001933 if (!surface) {
1934 return Result::Fatal("Could not create a surface.");
1935 }
1936
Robert Phillips19f466d2020-02-26 10:27:07 -05001937 Result result = this->ddlDraw(src, surface, &recordingTaskGroup, &gpuTaskGroup,
1938 mainTestCtx, mainCtx);
Robert Phillips291f3402020-02-19 14:14:47 -05001939 if (!result.isOk()) {
1940 return result;
1941 }
1942
Robert Phillips19f466d2020-02-26 10:27:07 -05001943 // 'ddlDraw' will have made 'mainCtx' not current on the gpuThread
Robert Phillips291f3402020-02-19 14:14:47 -05001944 mainTestCtx->makeCurrent();
1945
1946 if (FLAGS_gpuStats) {
1947 mainCtx->priv().dumpCacheStats(log);
1948 mainCtx->priv().dumpGpuStats(log);
Robert Phillips273f1072020-05-05 13:03:07 -04001949 mainCtx->priv().dumpContextStats(log);
Robert Phillips291f3402020-02-19 14:14:47 -05001950
1951#if 0
1952 otherCtx->priv().dumpCacheStats(log);
1953 otherCtx->priv().dumpGpuStats(log);
Robert Phillips273f1072020-05-05 13:03:07 -04001954 otherCtx->priv().dumpContextStats(log);
Robert Phillips291f3402020-02-19 14:14:47 -05001955#endif
1956 }
1957
1958 if (!this->readBack(surface.get(), dst)) {
1959 return Result::Fatal("Could not readback from surface.");
1960 }
1961
Robert Phillips291f3402020-02-19 14:14:47 -05001962 return Result::Ok();
1963}
1964
Brian Salomon00a5eb82018-07-11 15:32:05 -04001965/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001966static Result draw_skdocument(const Src& src, SkDocument* doc, SkWStream* dst) {
halcanary4ba051c2016-03-10 10:31:53 -08001967 if (src.size().isEmpty()) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001968 return Result::Fatal("Source has empty dimensions");
halcanary4ba051c2016-03-10 10:31:53 -08001969 }
halcanary47ef4d52015-03-03 09:13:09 -08001970 SkASSERT(doc);
halcanary45420a92016-06-02 12:41:14 -07001971 int pageCount = src.pageCount();
1972 for (int i = 0; i < pageCount; ++i) {
1973 int width = src.size(i).width(), height = src.size(i).height();
halcanary7e798182015-04-14 14:06:18 -07001974 SkCanvas* canvas =
1975 doc->beginPage(SkIntToScalar(width), SkIntToScalar(height));
1976 if (!canvas) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001977 return Result::Fatal("SkDocument::beginPage(w,h) returned nullptr");
halcanary7e798182015-04-14 14:06:18 -07001978 }
Robert Phillips889d6132020-06-16 11:11:33 -04001979 Result result = src.draw(i, nullptr, canvas);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001980 if (!result.isOk()) {
1981 return result;
halcanary7e798182015-04-14 14:06:18 -07001982 }
1983 doc->endPage();
mtklein748ca3b2015-01-15 10:56:12 -08001984 }
reedd14df7c2016-09-22 14:12:46 -07001985 doc->close();
halcanaryfd4a9932015-01-28 11:45:58 -08001986 dst->flush();
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001987 return Result::Ok();
mtklein748ca3b2015-01-15 10:56:12 -08001988}
1989
Ben Wagnerea25fcf2020-02-12 11:18:46 -05001990Result PDFSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
Hal Canary23564b92018-09-07 14:33:14 -04001991 SkPDF::Metadata metadata;
halcanary4b656662016-04-27 07:45:18 -07001992 metadata.fTitle = src.name();
1993 metadata.fSubject = "rendering correctness test";
1994 metadata.fCreator = "Skia/DM";
Mike Reeda4daf192017-12-14 13:25:04 -05001995 metadata.fRasterDPI = fRasterDpi;
1996 metadata.fPDFA = fPDFA;
Hal Canaryf34131a2018-12-18 15:11:03 -05001997#if SK_PDF_TEST_EXECUTOR
Hal Canary9a3f5542018-12-10 19:59:07 -05001998 std::unique_ptr<SkExecutor> executor = SkExecutor::MakeFIFOThreadPool();
1999 metadata.fExecutor = executor.get();
2000#endif
Hal Canary3026d4b2019-01-07 10:00:48 -05002001 auto doc = SkPDF::MakeDocument(dst, metadata);
halcanary47ef4d52015-03-03 09:13:09 -08002002 if (!doc) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002003 return Result::Fatal("SkPDF::MakeDocument() returned nullptr");
halcanary47ef4d52015-03-03 09:13:09 -08002004 }
2005 return draw_skdocument(src, doc.get(), dst);
2006}
2007
2008/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2009
2010XPSSink::XPSSink() {}
2011
Robert Phillips9b00f1f2020-03-24 12:29:44 -04002012#if defined(SK_SUPPORT_XPS)
Hal Canary5e221e72017-02-06 09:51:42 -05002013static SkTScopedComPtr<IXpsOMObjectFactory> make_xps_factory() {
2014 IXpsOMObjectFactory* factory;
2015 HRN(CoCreateInstance(CLSID_XpsOMObjectFactory,
2016 nullptr,
2017 CLSCTX_INPROC_SERVER,
2018 IID_PPV_ARGS(&factory)));
2019 return SkTScopedComPtr<IXpsOMObjectFactory>(factory);
2020}
2021
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002022Result XPSSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
Hal Canaryabc88d22017-02-06 09:26:49 -05002023 SkAutoCoInitialize com;
2024 if (!com.succeeded()) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002025 return Result::Fatal("Could not initialize COM.");
Hal Canaryabc88d22017-02-06 09:26:49 -05002026 }
Hal Canary5e221e72017-02-06 09:51:42 -05002027 SkTScopedComPtr<IXpsOMObjectFactory> factory = make_xps_factory();
2028 if (!factory) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002029 return Result::Fatal("Failed to create XPS Factory.");
Hal Canary5e221e72017-02-06 09:51:42 -05002030 }
Hal Canary3026d4b2019-01-07 10:00:48 -05002031 auto doc = SkXPS::MakeDocument(dst, factory.get());
halcanary47ef4d52015-03-03 09:13:09 -08002032 if (!doc) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002033 return Result::Fatal("SkXPS::MakeDocument() returned nullptr");
halcanary47ef4d52015-03-03 09:13:09 -08002034 }
2035 return draw_skdocument(src, doc.get(), dst);
2036}
Hal Canary5e221e72017-02-06 09:51:42 -05002037#else
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002038Result XPSSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
2039 return Result::Fatal("XPS not supported on this platform.");
Hal Canary5e221e72017-02-06 09:51:42 -05002040}
2041#endif
reed54dc4872016-09-13 08:09:45 -07002042
2043/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2044
mtklein9c3f17d2015-01-28 11:35:18 -08002045SKPSink::SKPSink() {}
2046
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002047Result SKPSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
Brian Salomon4e1066f2019-12-04 10:33:52 -05002048 auto size = SkSize::Make(src.size());
mtklein9c3f17d2015-01-28 11:35:18 -08002049 SkPictureRecorder recorder;
Robert Phillips889d6132020-06-16 11:11:33 -04002050 Result result = src.draw(nullptr, recorder.beginRecording(size.width(), size.height()));
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002051 if (!result.isOk()) {
2052 return result;
mtklein9c3f17d2015-01-28 11:35:18 -08002053 }
reedca2622b2016-03-18 07:25:55 -07002054 recorder.finishRecordingAsPicture()->serialize(dst);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002055 return Result::Ok();
mtklein9c3f17d2015-01-28 11:35:18 -08002056}
2057
2058/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2059
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002060Result DebugSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
Mike Klein8f4e2242019-03-20 11:59:00 -05002061 DebugCanvas debugCanvas(src.size().width(), src.size().height());
Robert Phillips889d6132020-06-16 11:11:33 -04002062 Result result = src.draw(nullptr, &debugCanvas);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002063 if (!result.isOk()) {
2064 return result;
Hal Canary85c7fe82016-10-25 10:33:27 -04002065 }
Mike Reed5df49342016-11-12 08:06:55 -06002066 std::unique_ptr<SkCanvas> nullCanvas = SkMakeNullCanvas();
Hal Canary85c7fe82016-10-25 10:33:27 -04002067 UrlDataManager dataManager(SkString("data"));
Brian Osmand8a90f92019-01-28 13:41:19 -05002068 SkJSONWriter writer(dst, SkJSONWriter::Mode::kPretty);
2069 writer.beginObject(); // root
Nathaniel Nifonga072b7b2019-12-13 13:51:14 -05002070 debugCanvas.toJSON(writer, dataManager, nullCanvas.get());
Brian Osmand8a90f92019-01-28 13:41:19 -05002071 writer.endObject(); // root
2072 writer.flush();
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002073 return Result::Ok();
Hal Canary85c7fe82016-10-25 10:33:27 -04002074}
2075
2076/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2077
Bryce Thomas95a7b762018-03-02 13:54:21 -08002078SVGSink::SVGSink(int pageIndex) : fPageIndex(pageIndex) {}
mtklein8a4527e2015-01-31 20:00:58 -08002079
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002080Result SVGSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
Robert Phillips2af13c12021-09-01 16:47:01 +00002081#if defined(SK_ENABLE_SVG)
Bryce Thomas95a7b762018-03-02 13:54:21 -08002082 if (src.pageCount() > 1) {
2083 int pageCount = src.pageCount();
2084 if (fPageIndex > pageCount - 1) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002085 return Result::Fatal("Page index %d too high for document with only %d pages.",
2086 fPageIndex, pageCount);
Bryce Thomas95a7b762018-03-02 13:54:21 -08002087 }
2088 }
Robert Phillips889d6132020-06-16 11:11:33 -04002089 return src.draw(fPageIndex, nullptr,
Bryce Thomas95a7b762018-03-02 13:54:21 -08002090 SkSVGCanvas::Make(SkRect::MakeWH(SkIntToScalar(src.size().width()),
Mike Reed5df49342016-11-12 08:06:55 -06002091 SkIntToScalar(src.size().height())),
Florin Malita562017b2019-02-14 13:42:15 -05002092 dst)
Bryce Thomas95a7b762018-03-02 13:54:21 -08002093 .get());
fmalita718df0a2016-07-15 10:33:29 -07002094#else
Hal Canary327ef032018-03-22 13:10:51 -04002095 (void)fPageIndex;
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002096 return Result::Fatal("SVG sink is disabled.");
Robert Phillips2af13c12021-09-01 16:47:01 +00002097#endif // SK_ENABLE_SVG
mtklein8a4527e2015-01-31 20:00:58 -08002098}
2099
2100/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2101
Brian Osmanfcb50232021-08-12 10:51:48 -04002102RasterSink::RasterSink(SkColorType colorType)
2103 : fColorType(colorType) {}
mtklein748ca3b2015-01-15 10:56:12 -08002104
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002105Result RasterSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
mtkleinf4ba3212015-01-28 15:32:24 -08002106 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -08002107 // If there's an appropriate alpha type for this color type, use it, otherwise use premul.
2108 SkAlphaType alphaType = kPremul_SkAlphaType;
2109 (void)SkColorTypeValidateAlphaType(fColorType, alphaType, &alphaType);
2110
Brian Salomon4bc0c1f2019-09-30 15:12:27 -04002111 dst->allocPixelsFlags(SkImageInfo::Make(size, fColorType, alphaType, fColorSpace),
Mike Reed086a4272017-07-18 10:53:11 -04002112 SkBitmap::kZeroPixels_AllocFlag);
Yuqian Lib8b62532018-02-23 14:13:36 +08002113
Ben Wagnerae4bb982020-09-24 14:49:00 -04002114 SkCanvas canvas(*dst, SkSurfaceProps(0, kRGB_H_SkPixelGeometry));
Robert Phillips889d6132020-06-16 11:11:33 -04002115 return src.draw(nullptr, &canvas);
mtklein748ca3b2015-01-15 10:56:12 -08002116}
2117
2118/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2119
Robert Phillips755e00e2021-09-21 11:59:57 -04002120GraphiteSink::GraphiteSink() {}
Robert Phillips1a4080a2021-09-21 09:57:43 -04002121
Robert Phillips755e00e2021-09-21 11:59:57 -04002122#ifdef SK_GRAPHITE_ENABLED
Robert Phillips1a4080a2021-09-21 09:57:43 -04002123
Robert Phillips755e00e2021-09-21 11:59:57 -04002124Result GraphiteSink::draw(const Src& src,
2125 SkBitmap* dst,
2126 SkWStream* dstStream,
2127 SkString* log) const {
Robert Phillips091694f2021-09-30 11:41:16 -04002128 using ContextType = sk_graphite_test::ContextFactory::ContextType;
2129
Robert Phillips1a4080a2021-09-21 09:57:43 -04002130 SkImageInfo ii = SkImageInfo::Make(src.size(), kRGBA_8888_SkColorType, kPremul_SkAlphaType);
2131
Robert Phillips091694f2021-09-30 11:41:16 -04002132 sk_graphite_test::ContextFactory factory;
2133 auto [_, context] = factory.getContextInfo(ContextType::kMetal);
2134
2135 sk_sp<SkSurface> surface = MakeGraphite(std::move(context), ii);
Robert Phillips1a4080a2021-09-21 09:57:43 -04002136 if (!surface) {
2137 return Result::Fatal("Could not create a surface.");
2138 }
2139 Result result = src.draw(/* dContext */ nullptr, surface->getCanvas());
2140 if (!result.isOk()) {
2141 return result;
2142 }
2143 surface->flushAndSubmit();
2144
Robert Phillipsd6bf3742021-09-22 16:07:30 -04002145 dst->allocPixels(surface->imageInfo());
2146 if (!surface->readPixels(*dst, 0, 0)) {
2147 return Result::Fatal("Could not readback from surface.");
2148 }
Robert Phillips1a4080a2021-09-21 09:57:43 -04002149
2150 return Result::Ok();
2151}
2152#else
Robert Phillips755e00e2021-09-21 11:59:57 -04002153Result GraphiteSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
2154 return Result::Fatal("Graphite not enabled.");
Robert Phillips1a4080a2021-09-21 09:57:43 -04002155}
2156#endif
2157
2158/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2159
mtkleina16e69e2015-05-05 11:38:45 -07002160// Handy for front-patching a Src. Do whatever up-front work you need, then call draw_to_canvas(),
mtkleine44b5082015-05-07 10:53:34 -07002161// passing the Sink draw() arguments, a size, and a function draws into an SkCanvas.
mtkleina16e69e2015-05-05 11:38:45 -07002162// Several examples below.
2163
John Stilesdfd1a702021-07-20 12:50:40 -04002164using DrawToCanvasFn = std::function<DM::Result(GrDirectContext*, SkCanvas*)>;
2165
2166static Result draw_to_canvas(Sink* sink, SkBitmap* bitmap, SkWStream* stream,
2167 SkString* log, SkISize size, const DrawToCanvasFn& draw) {
mtkleina16e69e2015-05-05 11:38:45 -07002168 class ProxySrc : public Src {
2169 public:
John Stilesdfd1a702021-07-20 12:50:40 -04002170 ProxySrc(SkISize size, const DrawToCanvasFn& draw) : fSize(size), fDraw(draw) {}
2171 Result draw(GrDirectContext* context, SkCanvas* canvas) const override {
2172 return fDraw(context, canvas);
2173 }
halcanaryb4a7f142016-03-30 08:31:27 -07002174 Name name() const override { return "ProxySrc"; }
2175 SkISize size() const override { return fSize; }
mtkleina16e69e2015-05-05 11:38:45 -07002176 private:
John Stilesdfd1a702021-07-20 12:50:40 -04002177 SkISize fSize;
2178 const DrawToCanvasFn& fDraw;
mtkleina16e69e2015-05-05 11:38:45 -07002179 };
msarett62d3b102015-12-10 15:14:27 -08002180 return sink->draw(ProxySrc(size, draw), bitmap, stream, log);
mtkleina16e69e2015-05-05 11:38:45 -07002181}
2182
2183/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2184
Mike Klein84836b72019-03-21 11:31:36 -05002185static DEFINE_bool(check, true, "If true, have most Via- modes fail if they affect the output.");
mtklein4a34ecb2016-01-08 10:19:35 -08002186
2187// Is *bitmap identical to what you get drawing src into sink?
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002188static Result check_against_reference(const SkBitmap* bitmap, const Src& src, Sink* sink) {
mtklein4a34ecb2016-01-08 10:19:35 -08002189 // We can only check raster outputs.
2190 // (Non-raster outputs like .pdf, .skp, .svg may differ but still draw identically.)
2191 if (FLAGS_check && bitmap) {
2192 SkBitmap reference;
2193 SkString log;
halcanaryb4a7f142016-03-30 08:31:27 -07002194 SkDynamicMemoryWStream wStream;
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002195 Result result = sink->draw(src, &reference, &wStream, &log);
mtklein4a34ecb2016-01-08 10:19:35 -08002196 // If we can draw into this Sink via some pipeline, we should be able to draw directly.
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002197 SkASSERT(result.isOk());
2198 if (!result.isOk()) {
2199 return result;
mtklein4a34ecb2016-01-08 10:19:35 -08002200 }
Brian Osmane5756ec2017-09-06 17:08:30 -04002201 return compare_bitmaps(reference, *bitmap);
mtklein4a34ecb2016-01-08 10:19:35 -08002202 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002203 return Result::Ok();
mtklein4a34ecb2016-01-08 10:19:35 -08002204}
2205
2206/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2207
mtkleind603b222015-02-17 11:13:33 -08002208static SkISize auto_compute_translate(SkMatrix* matrix, int srcW, int srcH) {
2209 SkRect bounds = SkRect::MakeIWH(srcW, srcH);
2210 matrix->mapRect(&bounds);
2211 matrix->postTranslate(-bounds.x(), -bounds.y());
Hal Canaryfafe1352017-04-11 12:12:02 -04002212 return {SkScalarRoundToInt(bounds.width()), SkScalarRoundToInt(bounds.height())};
mtkleind603b222015-02-17 11:13:33 -08002213}
2214
msarett62d3b102015-12-10 15:14:27 -08002215ViaMatrix::ViaMatrix(SkMatrix matrix, Sink* sink) : Via(sink), fMatrix(matrix) {}
mtklein748ca3b2015-01-15 10:56:12 -08002216
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002217Result ViaMatrix::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtkleina16e69e2015-05-05 11:38:45 -07002218 SkMatrix matrix = fMatrix;
2219 SkISize size = auto_compute_translate(&matrix, src.size().width(), src.size().height());
John Stilesdfd1a702021-07-20 12:50:40 -04002220 return draw_to_canvas(fSink.get(), bitmap, stream, log, size,
2221 [&](GrDirectContext* context, SkCanvas* canvas) {
2222 canvas->concat(matrix);
2223 return src.draw(context, canvas);
2224 });
mtklein748ca3b2015-01-15 10:56:12 -08002225}
2226
mtkleind603b222015-02-17 11:13:33 -08002227// Undoes any flip or 90 degree rotate without changing the scale of the bitmap.
2228// This should be pixel-preserving.
msarett62d3b102015-12-10 15:14:27 -08002229ViaUpright::ViaUpright(SkMatrix matrix, Sink* sink) : Via(sink), fMatrix(matrix) {}
mtkleind603b222015-02-17 11:13:33 -08002230
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002231Result ViaUpright::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
2232 Result result = fSink->draw(src, bitmap, stream, log);
2233 if (!result.isOk()) {
2234 return result;
mtkleind603b222015-02-17 11:13:33 -08002235 }
2236
2237 SkMatrix inverse;
2238 if (!fMatrix.rectStaysRect() || !fMatrix.invert(&inverse)) {
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002239 return Result::Fatal("Cannot upright --matrix.");
mtkleind603b222015-02-17 11:13:33 -08002240 }
2241 SkMatrix upright = SkMatrix::I();
2242 upright.setScaleX(SkScalarSignAsScalar(inverse.getScaleX()));
2243 upright.setScaleY(SkScalarSignAsScalar(inverse.getScaleY()));
2244 upright.setSkewX(SkScalarSignAsScalar(inverse.getSkewX()));
2245 upright.setSkewY(SkScalarSignAsScalar(inverse.getSkewY()));
2246
2247 SkBitmap uprighted;
2248 SkISize size = auto_compute_translate(&upright, bitmap->width(), bitmap->height());
Brian Salomon9241a6d2019-10-03 13:26:54 -04002249 uprighted.allocPixels(bitmap->info().makeDimensions(size));
mtkleind603b222015-02-17 11:13:33 -08002250
2251 SkCanvas canvas(uprighted);
2252 canvas.concat(upright);
2253 SkPaint paint;
reed374772b2016-10-05 17:33:02 -07002254 paint.setBlendMode(SkBlendMode::kSrc);
Mike Reed34a0c972021-01-25 17:49:32 -05002255 canvas.drawImage(bitmap->asImage(), 0, 0, SkSamplingOptions(), &paint);
mtkleind603b222015-02-17 11:13:33 -08002256
2257 *bitmap = uprighted;
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002258 return Result::Ok();
mtkleind603b222015-02-17 11:13:33 -08002259}
2260
mtklein748ca3b2015-01-15 10:56:12 -08002261/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2262
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002263Result ViaSerialization::draw(
mtkleina16e69e2015-05-05 11:38:45 -07002264 const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -08002265 // Record our Src into a picture.
mtkleina16e69e2015-05-05 11:38:45 -07002266 auto size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -08002267 SkPictureRecorder recorder;
Robert Phillips889d6132020-06-16 11:11:33 -04002268 Result result = src.draw(nullptr, recorder.beginRecording(SkIntToScalar(size.width()),
2269 SkIntToScalar(size.height())));
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002270 if (!result.isOk()) {
2271 return result;
mtklein748ca3b2015-01-15 10:56:12 -08002272 }
reedca2622b2016-03-18 07:25:55 -07002273 sk_sp<SkPicture> pic(recorder.finishRecordingAsPicture());
mtklein748ca3b2015-01-15 10:56:12 -08002274
2275 // Serialize it and then deserialize it.
reed39eaf5f2016-09-15 07:19:35 -07002276 sk_sp<SkPicture> deserialized(SkPicture::MakeFromData(pic->serialize().get()));
mtklein748ca3b2015-01-15 10:56:12 -08002277
John Stilesdfd1a702021-07-20 12:50:40 -04002278 result = draw_to_canvas(fSink.get(), bitmap, stream, log, size,
2279 [&](GrDirectContext*, SkCanvas* canvas) {
2280 canvas->drawPicture(deserialized);
2281 return Result::Ok();
2282 });
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002283 if (!result.isOk()) {
2284 return result;
Ben Wagner1861e882018-04-04 17:40:46 -04002285 }
2286
2287 return check_against_reference(bitmap, src, fSink.get());
mtklein748ca3b2015-01-15 10:56:12 -08002288}
2289
2290/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2291
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002292Result ViaPicture::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein4a34ecb2016-01-08 10:19:35 -08002293 auto size = src.size();
John Stilesdfd1a702021-07-20 12:50:40 -04002294 Result result = draw_to_canvas(fSink.get(), bitmap, stream, log, size,
2295 [&](GrDirectContext* context, SkCanvas* canvas) {
mtklein4a34ecb2016-01-08 10:19:35 -08002296 SkPictureRecorder recorder;
reedca2622b2016-03-18 07:25:55 -07002297 sk_sp<SkPicture> pic;
John Stilesdfd1a702021-07-20 12:50:40 -04002298 Result result = src.draw(context, recorder.beginRecording(SkIntToScalar(size.width()),
Robert Phillips889d6132020-06-16 11:11:33 -04002299 SkIntToScalar(size.height())));
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002300 if (!result.isOk()) {
2301 return result;
mtklein4a34ecb2016-01-08 10:19:35 -08002302 }
reedca2622b2016-03-18 07:25:55 -07002303 pic = recorder.finishRecordingAsPicture();
mtklein4a34ecb2016-01-08 10:19:35 -08002304 canvas->drawPicture(pic);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002305 return result;
mtklein4a34ecb2016-01-08 10:19:35 -08002306 });
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002307 if (!result.isOk()) {
2308 return result;
Ben Wagner1861e882018-04-04 17:40:46 -04002309 }
2310
2311 return check_against_reference(bitmap, src, fSink.get());
mtklein4a34ecb2016-01-08 10:19:35 -08002312}
2313
2314/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2315
John Stiles65ec1982021-07-20 12:53:54 -04002316Result ViaRuntimeBlend::draw(const Src& src,
2317 SkBitmap* bitmap,
2318 SkWStream* stream,
2319 SkString* log) const {
2320 class RuntimeBlendFilterCanvas : public SkPaintFilterCanvas {
2321 public:
2322 RuntimeBlendFilterCanvas(SkCanvas* canvas) : INHERITED(canvas) { }
2323
2324 protected:
2325 bool onFilter(SkPaint& paint) const override {
2326 if (skstd::optional<SkBlendMode> mode = paint.asBlendMode()) {
2327 paint.setBlender(GetRuntimeBlendForBlendMode(*mode));
2328 }
2329 return true;
2330 }
2331
2332 private:
2333 using INHERITED = SkPaintFilterCanvas;
2334 };
2335
2336 return draw_to_canvas(fSink.get(), bitmap, stream, log, src.size(),
2337 [&](GrDirectContext* context, SkCanvas* canvas) {
2338 RuntimeBlendFilterCanvas runtimeBlendCanvas{canvas};
2339 return src.draw(context, &runtimeBlendCanvas);
2340 });
2341}
2342
2343/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
2344
Mike Reedbae888e2017-02-18 16:50:45 -05002345#ifdef TEST_VIA_SVG
Mike Kleinc0bd9f92019-04-23 12:05:21 -05002346#include "include/svg/SkSVGCanvas.h"
Florin Malitab3418102020-10-15 18:10:29 -04002347#include "modules/svg/include/SkSVGDOM.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -05002348#include "src/xml/SkXMLWriter.h"
Mike Reedf67c4592017-02-17 17:06:11 -05002349
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002350Result ViaSVG::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
Mike Reedf67c4592017-02-17 17:06:11 -05002351 auto size = src.size();
John Stilesdfd1a702021-07-20 12:50:40 -04002352 return draw_to_canvas(fSink.get(), bitmap, stream, log, size,
2353 [&](GrDirectContext*, SkCanvas* canvas) -> Result {
Mike Reedf67c4592017-02-17 17:06:11 -05002354 SkDynamicMemoryWStream wstream;
2355 SkXMLStreamWriter writer(&wstream);
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002356 Result result = src.draw(SkSVGCanvas::Make(SkRect::Make(size), &writer).get());
2357 if (!result.isOk()) {
2358 return result;
Mike Reedf67c4592017-02-17 17:06:11 -05002359 }
2360 std::unique_ptr<SkStream> rstream(wstream.detachAsStream());
2361 auto dom = SkSVGDOM::MakeFromStream(*rstream);
2362 if (dom) {
2363 dom->setContainerSize(SkSize::Make(size));
2364 dom->render(canvas);
2365 }
Ben Wagnerea25fcf2020-02-12 11:18:46 -05002366 return Result::Ok();
Mike Reedf67c4592017-02-17 17:06:11 -05002367 });
2368}
Mike Reedbae888e2017-02-18 16:50:45 -05002369#endif
Mike Reedf67c4592017-02-17 17:06:11 -05002370
mtklein748ca3b2015-01-15 10:56:12 -08002371} // namespace DM