blob: 06ca7507349b7e5cf05f34246c48a22a4b5eaec5 [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
mtklein748ca3b2015-01-15 10:56:12 -08008#include "DMSrcSink.h"
msarett9876ac52016-06-01 14:47:18 -07009#include "Resources.h"
msarett3d9d7a72015-10-21 10:27:10 -070010#include "SkAndroidCodec.h"
scroggof24f2242015-03-03 08:59:20 -080011#include "SkCodec.h"
msarettb714fb02016-01-22 14:46:42 -080012#include "SkCodecImageGenerator.h"
msarett888dc162016-05-23 10:21:17 -070013#include "SkColorSpace.h"
raftias94888332016-10-18 10:02:51 -070014#include "SkColorSpace_XYZ.h"
msarett9876ac52016-06-01 14:47:18 -070015#include "SkColorSpaceXform.h"
mtkleina16e69e2015-05-05 11:38:45 -070016#include "SkCommonFlags.h"
mtkleinb3e5e4d2015-03-25 13:13:43 -070017#include "SkData.h"
reedbabc3de2016-07-08 08:43:27 -070018#include "SkDeferredCanvas.h"
mtklein748ca3b2015-01-15 10:56:12 -080019#include "SkDocument.h"
mtkleinb3e5e4d2015-03-25 13:13:43 -070020#include "SkImageGenerator.h"
msarett18976312016-03-09 14:20:58 -080021#include "SkImageGeneratorCG.h"
msarettfc0b6d12016-03-17 13:50:17 -070022#include "SkImageGeneratorWIC.h"
mtklein8bbbb692016-08-15 12:56:00 -070023#include "SkLiteDL.h"
24#include "SkLiteRecorder.h"
mtkleinc8be09a2016-01-04 18:56:57 -080025#include "SkMallocPixelRef.h"
mtklein748ca3b2015-01-15 10:56:12 -080026#include "SkMultiPictureDraw.h"
mtkleinad66f9b2015-02-13 15:11:10 -080027#include "SkNullCanvas.h"
mtklein748ca3b2015-01-15 10:56:12 -080028#include "SkOSFile.h"
msarett9e9444c2016-02-03 12:39:10 -080029#include "SkOpts.h"
mtkleinffa901a2015-03-16 10:38:07 -070030#include "SkPictureData.h"
mtklein748ca3b2015-01-15 10:56:12 -080031#include "SkPictureRecorder.h"
reed54dc4872016-09-13 08:09:45 -070032#include "SkPipe.h"
mtklein748ca3b2015-01-15 10:56:12 -080033#include "SkRandom.h"
mtkleind31c13d2015-05-05 12:59:56 -070034#include "SkRecordDraw.h"
35#include "SkRecorder.h"
fmalita2aafe6f2015-02-06 12:51:10 -080036#include "SkSVGCanvas.h"
scroggoa1193e42015-01-21 12:09:53 -080037#include "SkStream.h"
mtklein449d9b72015-09-28 10:33:02 -070038#include "SkTLogic.h"
msarette6dd0042015-10-09 11:07:34 -070039#include "SkSwizzler.h"
mtklein64593522015-11-12 10:41:05 -080040#include <functional>
mtklein748ca3b2015-01-15 10:56:12 -080041
msarettfc0b6d12016-03-17 13:50:17 -070042#if defined(SK_BUILD_FOR_WIN)
43 #include "SkAutoCoInitialize.h"
44#endif
45
msarett469f1c52016-06-06 08:20:37 -070046#if defined(SK_TEST_QCMS)
msarett9876ac52016-06-01 14:47:18 -070047 #include "qcms.h"
48#endif
49
fmalita718df0a2016-07-15 10:33:29 -070050#if defined(SK_XML)
fmalitaa48f0e32016-08-04 06:26:05 -070051 #include "SkSVGDOM.h"
fmalita718df0a2016-07-15 10:33:29 -070052 #include "SkXMLWriter.h"
53#endif
54
halcanary7e798182015-04-14 14:06:18 -070055DEFINE_bool(multiPage, false, "For document-type backends, render the source"
56 " into multiple pages");
scroggo3ac66e92016-02-08 15:09:48 -080057DEFINE_bool(RAW_threading, true, "Allow RAW decodes to run on multiple threads?");
halcanary7e798182015-04-14 14:06:18 -070058
bsalomon3724e572016-03-30 18:56:19 -070059using sk_gpu_test::GrContextFactory;
60
mtklein748ca3b2015-01-15 10:56:12 -080061namespace DM {
62
mtklein748ca3b2015-01-15 10:56:12 -080063GMSrc::GMSrc(skiagm::GMRegistry::Factory factory) : fFactory(factory) {}
64
65Error GMSrc::draw(SkCanvas* canvas) const {
halcanary96fcdcc2015-08-27 07:41:13 -070066 SkAutoTDelete<skiagm::GM> gm(fFactory(nullptr));
mtklein748ca3b2015-01-15 10:56:12 -080067 canvas->concat(gm->getInitialTransform());
68 gm->draw(canvas);
69 return "";
70}
71
72SkISize GMSrc::size() const {
halcanary96fcdcc2015-08-27 07:41:13 -070073 SkAutoTDelete<skiagm::GM> gm(fFactory(nullptr));
mtklein748ca3b2015-01-15 10:56:12 -080074 return gm->getISize();
75}
76
77Name GMSrc::name() const {
halcanary96fcdcc2015-08-27 07:41:13 -070078 SkAutoTDelete<skiagm::GM> gm(fFactory(nullptr));
mtklein748ca3b2015-01-15 10:56:12 -080079 return gm->getName();
80}
81
bsalomon4ee6bd82015-05-27 13:23:23 -070082void GMSrc::modifyGrContextOptions(GrContextOptions* options) const {
halcanary96fcdcc2015-08-27 07:41:13 -070083 SkAutoTDelete<skiagm::GM> gm(fFactory(nullptr));
bsalomon4ee6bd82015-05-27 13:23:23 -070084 gm->modifyGrContextOptions(options);
85}
86
mtklein748ca3b2015-01-15 10:56:12 -080087/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
88
msarettd1227a72016-05-18 06:23:57 -070089BRDSrc::BRDSrc(Path path, Mode mode, CodecSrc::DstColorType dstColorType, uint32_t sampleSize)
msaretta5783ae2015-09-08 15:35:32 -070090 : fPath(path)
msaretta5783ae2015-09-08 15:35:32 -070091 , fMode(mode)
92 , fDstColorType(dstColorType)
93 , fSampleSize(sampleSize)
94{}
95
96bool BRDSrc::veto(SinkFlags flags) const {
97 // No need to test to non-raster or indirect backends.
98 return flags.type != SinkFlags::kRaster
99 || flags.approach != SinkFlags::kDirect;
100}
101
msarettd1227a72016-05-18 06:23:57 -0700102static SkBitmapRegionDecoder* create_brd(Path path) {
bungeman38d909e2016-08-02 14:40:46 -0700103 sk_sp<SkData> encoded(SkData::MakeFromFileName(path.c_str()));
msaretta5783ae2015-09-08 15:35:32 -0700104 if (!encoded) {
105 return NULL;
106 }
reed42943c82016-09-12 12:01:44 -0700107 return SkBitmapRegionDecoder::Create(encoded, SkBitmapRegionDecoder::kAndroidCodec_Strategy);
msaretta5783ae2015-09-08 15:35:32 -0700108}
109
110Error BRDSrc::draw(SkCanvas* canvas) const {
111 SkColorType colorType = canvas->imageInfo().colorType();
112 if (kRGB_565_SkColorType == colorType &&
113 CodecSrc::kGetFromCanvas_DstColorType != fDstColorType) {
114 return Error::Nonfatal("Testing non-565 to 565 is uninteresting.");
115 }
116 switch (fDstColorType) {
117 case CodecSrc::kGetFromCanvas_DstColorType:
118 break;
119 case CodecSrc::kIndex8_Always_DstColorType:
120 colorType = kIndex_8_SkColorType;
121 break;
122 case CodecSrc::kGrayscale_Always_DstColorType:
123 colorType = kGray_8_SkColorType;
124 break;
msarett34e0ec42016-04-22 16:27:24 -0700125 default:
126 SkASSERT(false);
127 break;
msaretta5783ae2015-09-08 15:35:32 -0700128 }
129
msarettd1227a72016-05-18 06:23:57 -0700130 SkAutoTDelete<SkBitmapRegionDecoder> brd(create_brd(fPath));
msaretta5783ae2015-09-08 15:35:32 -0700131 if (nullptr == brd.get()) {
132 return Error::Nonfatal(SkStringPrintf("Could not create brd for %s.", fPath.c_str()));
133 }
134
msarett04965c62015-10-12 10:24:38 -0700135 if (!brd->conversionSupported(colorType)) {
mtklein9b439152015-12-09 13:02:26 -0800136 return Error::Nonfatal("Cannot convert to color type.");
msarett04965c62015-10-12 10:24:38 -0700137 }
138
msaretta5783ae2015-09-08 15:35:32 -0700139 const uint32_t width = brd->width();
140 const uint32_t height = brd->height();
141 // Visually inspecting very small output images is not necessary.
142 if ((width / fSampleSize <= 10 || height / fSampleSize <= 10) && 1 != fSampleSize) {
143 return Error::Nonfatal("Scaling very small images is uninteresting.");
144 }
145 switch (fMode) {
146 case kFullImage_Mode: {
msarett35e5d1b2015-10-27 12:50:25 -0700147 SkBitmap bitmap;
148 if (!brd->decodeRegion(&bitmap, nullptr, SkIRect::MakeXYWH(0, 0, width, height),
149 fSampleSize, colorType, false)) {
mtklein9b439152015-12-09 13:02:26 -0800150 return "Cannot decode (full) region.";
msarett35e5d1b2015-10-27 12:50:25 -0700151 }
152 if (colorType != bitmap.colorType()) {
mtklein9b439152015-12-09 13:02:26 -0800153 return Error::Nonfatal("Cannot convert to color type.");
msaretta5783ae2015-09-08 15:35:32 -0700154 }
msarett35e5d1b2015-10-27 12:50:25 -0700155 canvas->drawBitmap(bitmap, 0, 0);
msaretta5783ae2015-09-08 15:35:32 -0700156 return "";
157 }
158 case kDivisor_Mode: {
159 const uint32_t divisor = 2;
160 if (width < divisor || height < divisor) {
mtklein9b439152015-12-09 13:02:26 -0800161 return Error::Nonfatal("Divisor is larger than image dimension.");
msaretta5783ae2015-09-08 15:35:32 -0700162 }
163
164 // Use a border to test subsets that extend outside the image.
165 // We will not allow the border to be larger than the image dimensions. Allowing
166 // these large borders causes off by one errors that indicate a problem with the
167 // test suite, not a problem with the implementation.
168 const uint32_t maxBorder = SkTMin(width, height) / (fSampleSize * divisor);
169 const uint32_t scaledBorder = SkTMin(5u, maxBorder);
170 const uint32_t unscaledBorder = scaledBorder * fSampleSize;
171
172 // We may need to clear the canvas to avoid uninitialized memory.
173 // Assume we are scaling a 780x780 image with sampleSize = 8.
174 // The output image should be 97x97.
175 // Each subset will be 390x390.
176 // Each scaled subset be 48x48.
177 // Four scaled subsets will only fill a 96x96 image.
178 // The bottom row and last column will not be touched.
179 // This is an unfortunate result of our rounding rules when scaling.
180 // Maybe we need to consider testing scaled subsets without trying to
181 // combine them to match the full scaled image? Or maybe this is the
182 // best we can do?
183 canvas->clear(0);
184
185 for (uint32_t x = 0; x < divisor; x++) {
186 for (uint32_t y = 0; y < divisor; y++) {
187 // Calculate the subset dimensions
188 uint32_t subsetWidth = width / divisor;
189 uint32_t subsetHeight = height / divisor;
190 const int left = x * subsetWidth;
191 const int top = y * subsetHeight;
192
193 // Increase the size of the last subset in each row or column, when the
194 // divisor does not divide evenly into the image dimensions
195 subsetWidth += (x + 1 == divisor) ? (width % divisor) : 0;
196 subsetHeight += (y + 1 == divisor) ? (height % divisor) : 0;
197
198 // Increase the size of the subset in order to have a border on each side
199 const int decodeLeft = left - unscaledBorder;
200 const int decodeTop = top - unscaledBorder;
201 const uint32_t decodeWidth = subsetWidth + unscaledBorder * 2;
202 const uint32_t decodeHeight = subsetHeight + unscaledBorder * 2;
msarett35e5d1b2015-10-27 12:50:25 -0700203 SkBitmap bitmap;
204 if (!brd->decodeRegion(&bitmap, nullptr, SkIRect::MakeXYWH(decodeLeft,
205 decodeTop, decodeWidth, decodeHeight), fSampleSize, colorType, false)) {
mtklein9b439152015-12-09 13:02:26 -0800206 return "Cannot decode region.";
msarett35e5d1b2015-10-27 12:50:25 -0700207 }
208 if (colorType != bitmap.colorType()) {
mtklein9b439152015-12-09 13:02:26 -0800209 return Error::Nonfatal("Cannot convert to color type.");
msaretta5783ae2015-09-08 15:35:32 -0700210 }
211
msarett35e5d1b2015-10-27 12:50:25 -0700212 canvas->drawBitmapRect(bitmap,
msaretta5783ae2015-09-08 15:35:32 -0700213 SkRect::MakeXYWH((SkScalar) scaledBorder, (SkScalar) scaledBorder,
214 (SkScalar) (subsetWidth / fSampleSize),
215 (SkScalar) (subsetHeight / fSampleSize)),
216 SkRect::MakeXYWH((SkScalar) (left / fSampleSize),
217 (SkScalar) (top / fSampleSize),
218 (SkScalar) (subsetWidth / fSampleSize),
219 (SkScalar) (subsetHeight / fSampleSize)),
220 nullptr);
221 }
222 }
223 return "";
224 }
225 default:
226 SkASSERT(false);
mtklein9b439152015-12-09 13:02:26 -0800227 return "Error: Should not be reached.";
msaretta5783ae2015-09-08 15:35:32 -0700228 }
229}
230
231SkISize BRDSrc::size() const {
msarettd1227a72016-05-18 06:23:57 -0700232 SkAutoTDelete<SkBitmapRegionDecoder> brd(create_brd(fPath));
msaretta5783ae2015-09-08 15:35:32 -0700233 if (brd) {
234 return SkISize::Make(SkTMax(1, brd->width() / (int) fSampleSize),
235 SkTMax(1, brd->height() / (int) fSampleSize));
236 }
237 return SkISize::Make(0, 0);
238}
239
240static SkString get_scaled_name(const Path& path, float scale) {
241 return SkStringPrintf("%s_%.3f", SkOSPath::Basename(path.c_str()).c_str(), scale);
242}
243
244Name BRDSrc::name() const {
245 // We will replicate the names used by CodecSrc so that images can
246 // be compared in Gold.
247 if (1 == fSampleSize) {
248 return SkOSPath::Basename(fPath.c_str());
249 }
msarett4b0778e2015-11-13 09:59:11 -0800250 return get_scaled_name(fPath, 1.0f / (float) fSampleSize);
msaretta5783ae2015-09-08 15:35:32 -0700251}
252
253/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
254
scroggo3ac66e92016-02-08 15:09:48 -0800255static bool serial_from_path_name(const SkString& path) {
256 if (!FLAGS_RAW_threading) {
257 static const char* const exts[] = {
258 "arw", "cr2", "dng", "nef", "nrw", "orf", "raf", "rw2", "pef", "srw",
259 "ARW", "CR2", "DNG", "NEF", "NRW", "ORF", "RAF", "RW2", "PEF", "SRW",
260 };
261 const char* actualExt = strrchr(path.c_str(), '.');
262 if (actualExt) {
263 actualExt++;
264 for (auto* ext : exts) {
265 if (0 == strcmp(ext, actualExt)) {
266 return true;
267 }
268 }
269 }
270 }
271 return false;
272}
273
scroggoc5560be2016-02-03 09:42:42 -0800274CodecSrc::CodecSrc(Path path, Mode mode, DstColorType dstColorType, SkAlphaType dstAlphaType,
275 float scale)
msarett438b2ad2015-04-09 12:43:10 -0700276 : fPath(path)
277 , fMode(mode)
278 , fDstColorType(dstColorType)
scroggoc5560be2016-02-03 09:42:42 -0800279 , fDstAlphaType(dstAlphaType)
msarett0a242972015-06-11 14:27:27 -0700280 , fScale(scale)
scroggo3ac66e92016-02-08 15:09:48 -0800281 , fRunSerially(serial_from_path_name(path))
msarett438b2ad2015-04-09 12:43:10 -0700282{}
mtklein748ca3b2015-01-15 10:56:12 -0800283
mtklein99cab4e2015-07-31 06:43:04 -0700284bool CodecSrc::veto(SinkFlags flags) const {
msarett18976312016-03-09 14:20:58 -0800285 // Test to direct raster backends (8888 and 565).
msarettb714fb02016-01-22 14:46:42 -0800286 return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
mtkleine0effd62015-07-29 06:37:28 -0700287}
scroggo9b77ddd2015-03-19 06:03:39 -0700288
msarett34e0ec42016-04-22 16:27:24 -0700289// Allows us to test decodes to non-native 8888.
msarettb1be46b2016-05-17 08:52:11 -0700290static void swap_rb_if_necessary(SkBitmap& bitmap, CodecSrc::DstColorType dstColorType) {
msarett34e0ec42016-04-22 16:27:24 -0700291 if (CodecSrc::kNonNative8888_Always_DstColorType != dstColorType) {
292 return;
293 }
294
295 for (int y = 0; y < bitmap.height(); y++) {
296 uint32_t* row = (uint32_t*) bitmap.getAddr(0, y);
297 SkOpts::RGBA_to_BGRA(row, row, bitmap.width());
298 }
299}
300
msarett9e9444c2016-02-03 12:39:10 -0800301// FIXME: Currently we cannot draw unpremultiplied sources. skbug.com/3338 and skbug.com/3339.
302// This allows us to still test unpremultiplied decodes.
msarettb1be46b2016-05-17 08:52:11 -0700303static void premultiply_if_necessary(SkBitmap& bitmap) {
msarett9e9444c2016-02-03 12:39:10 -0800304 if (kUnpremul_SkAlphaType != bitmap.alphaType()) {
305 return;
306 }
307
308 switch (bitmap.colorType()) {
309 case kN32_SkColorType:
310 for (int y = 0; y < bitmap.height(); y++) {
311 uint32_t* row = (uint32_t*) bitmap.getAddr(0, y);
312 SkOpts::RGBA_to_rgbA(row, row, bitmap.width());
313 }
314 break;
315 case kIndex_8_SkColorType: {
316 SkColorTable* colorTable = bitmap.getColorTable();
317 SkPMColor* colorPtr = const_cast<SkPMColor*>(colorTable->readColors());
318 SkOpts::RGBA_to_rgbA(colorPtr, colorPtr, colorTable->count());
319 break;
320 }
321 default:
322 // No need to premultiply kGray or k565 outputs.
323 break;
324 }
msarette1daa482016-02-03 15:31:18 -0800325
326 // In the kIndex_8 case, the canvas won't even try to draw unless we mark the
327 // bitmap as kPremul.
328 bitmap.setAlphaType(kPremul_SkAlphaType);
msarett9e9444c2016-02-03 12:39:10 -0800329}
330
msarettb1be46b2016-05-17 08:52:11 -0700331static bool get_decode_info(SkImageInfo* decodeInfo, SkColorType canvasColorType,
scroggoba584892016-05-20 13:56:13 -0700332 CodecSrc::DstColorType dstColorType, SkAlphaType dstAlphaType) {
msarett3d9d7a72015-10-21 10:27:10 -0700333 switch (dstColorType) {
334 case CodecSrc::kIndex8_Always_DstColorType:
335 if (kRGB_565_SkColorType == canvasColorType) {
336 return false;
337 }
scroggoc5560be2016-02-03 09:42:42 -0800338 *decodeInfo = decodeInfo->makeColorType(kIndex_8_SkColorType);
msarett3d9d7a72015-10-21 10:27:10 -0700339 break;
340 case CodecSrc::kGrayscale_Always_DstColorType:
scroggo1a361922016-05-20 14:27:16 -0700341 if (kRGB_565_SkColorType == canvasColorType) {
msarett3d9d7a72015-10-21 10:27:10 -0700342 return false;
343 }
scroggoc5560be2016-02-03 09:42:42 -0800344 *decodeInfo = decodeInfo->makeColorType(kGray_8_SkColorType);
msarett3d9d7a72015-10-21 10:27:10 -0700345 break;
msarett34e0ec42016-04-22 16:27:24 -0700346 case CodecSrc::kNonNative8888_Always_DstColorType:
347 if (kRGB_565_SkColorType == canvasColorType) {
348 return false;
349 }
350#ifdef SK_PMCOLOR_IS_RGBA
351 *decodeInfo = decodeInfo->makeColorType(kBGRA_8888_SkColorType);
352#else
353 *decodeInfo = decodeInfo->makeColorType(kRGBA_8888_SkColorType);
354#endif
355 break;
msarett3d9d7a72015-10-21 10:27:10 -0700356 default:
msarett55f7bdd2016-02-16 13:24:54 -0800357 if (kRGB_565_SkColorType == canvasColorType &&
358 kOpaque_SkAlphaType != decodeInfo->alphaType()) {
359 return false;
360 }
scroggoc5560be2016-02-03 09:42:42 -0800361 *decodeInfo = decodeInfo->makeColorType(canvasColorType);
msarett3d9d7a72015-10-21 10:27:10 -0700362 break;
363 }
364
scroggoba584892016-05-20 13:56:13 -0700365 *decodeInfo = decodeInfo->makeAlphaType(dstAlphaType);
msarett3d9d7a72015-10-21 10:27:10 -0700366 return true;
367}
368
msarettb1be46b2016-05-17 08:52:11 -0700369static void draw_to_canvas(SkCanvas* canvas, const SkImageInfo& info, void* pixels, size_t rowBytes,
370 SkPMColor* colorPtr, int colorCount, CodecSrc::DstColorType dstColorType,
371 SkScalar left = 0, SkScalar top = 0) {
372 SkAutoTUnref<SkColorTable> colorTable(new SkColorTable(colorPtr, colorCount));
373 SkBitmap bitmap;
374 bitmap.installPixels(info, pixels, rowBytes, colorTable.get(), nullptr, nullptr);
375 premultiply_if_necessary(bitmap);
376 swap_rb_if_necessary(bitmap, dstColorType);
377 canvas->drawBitmap(bitmap, left, top);
378}
379
mtkleine0effd62015-07-29 06:37:28 -0700380Error CodecSrc::draw(SkCanvas* canvas) const {
bungeman38d909e2016-08-02 14:40:46 -0700381 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
mtklein748ca3b2015-01-15 10:56:12 -0800382 if (!encoded) {
383 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
384 }
msarettb714fb02016-01-22 14:46:42 -0800385
reed42943c82016-09-12 12:01:44 -0700386 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
msarett9e707a02015-09-01 14:57:57 -0700387 if (nullptr == codec.get()) {
388 return SkStringPrintf("Couldn't create codec for %s.", fPath.c_str());
scroggo9b77ddd2015-03-19 06:03:39 -0700389 }
390
scroggoba584892016-05-20 13:56:13 -0700391 SkImageInfo decodeInfo = codec->getInfo();
392 if (!get_decode_info(&decodeInfo, canvas->imageInfo().colorType(), fDstColorType,
393 fDstAlphaType)) {
msarett3d9d7a72015-10-21 10:27:10 -0700394 return Error::Nonfatal("Testing non-565 to 565 is uninteresting.");
msarett438b2ad2015-04-09 12:43:10 -0700395 }
396
msarett0a242972015-06-11 14:27:27 -0700397 // Try to scale the image if it is desired
398 SkISize size = codec->getScaledDimensions(fScale);
399 if (size == decodeInfo.dimensions() && 1.0f != fScale) {
400 return Error::Nonfatal("Test without scaling is uninteresting.");
401 }
msarettb32758a2015-08-18 13:22:46 -0700402
403 // Visually inspecting very small output images is not necessary. We will
404 // cover these cases in unit testing.
405 if ((size.width() <= 10 || size.height() <= 10) && 1.0f != fScale) {
406 return Error::Nonfatal("Scaling very small images is uninteresting.");
407 }
msarett0a242972015-06-11 14:27:27 -0700408 decodeInfo = decodeInfo.makeWH(size.width(), size.height());
409
msarettb1be46b2016-05-17 08:52:11 -0700410 const int bpp = SkColorTypeBytesPerPixel(decodeInfo.colorType());
411 const size_t rowBytes = size.width() * bpp;
scroggo19b91532016-10-24 09:03:26 -0700412 const size_t safeSize = decodeInfo.getSafeSize(rowBytes);
413 SkAutoMalloc pixels(safeSize);
msarettb1be46b2016-05-17 08:52:11 -0700414 SkPMColor colorPtr[256];
415 int colorCount = 256;
msarett438b2ad2015-04-09 12:43:10 -0700416
msarettbb25b532016-01-13 09:31:39 -0800417 SkCodec::Options options;
418 if (kCodecZeroInit_Mode == fMode) {
msarettb1be46b2016-05-17 08:52:11 -0700419 memset(pixels.get(), 0, size.height() * rowBytes);
msarettbb25b532016-01-13 09:31:39 -0800420 options.fZeroInitialized = SkCodec::kYes_ZeroInitialized;
421 }
msarett34e0ec42016-04-22 16:27:24 -0700422
423 SkImageInfo bitmapInfo = decodeInfo;
424 if (kRGBA_8888_SkColorType == decodeInfo.colorType() ||
425 kBGRA_8888_SkColorType == decodeInfo.colorType()) {
426 bitmapInfo = bitmapInfo.makeColorType(kN32_SkColorType);
427 }
scroggo9b77ddd2015-03-19 06:03:39 -0700428
scroggo9c59ebc2015-03-25 13:48:49 -0700429 switch (fMode) {
scroggo19b91532016-10-24 09:03:26 -0700430 case kAnimated_Mode: {
431 std::vector<SkCodec::FrameInfo> frameInfos = codec->getFrameInfo();
432 if (frameInfos.size() <= 1) {
433 return SkStringPrintf("%s is not an animated image.", fPath.c_str());
434 }
435
436 SkAutoCanvasRestore acr(canvas, true);
437 // Used to cache a frame that future frames will depend on.
438 SkAutoMalloc priorFramePixels;
439 size_t cachedFrame = SkCodec::kNone;
440 for (size_t i = 0; i < frameInfos.size(); i++) {
441 options.fFrameIndex = i;
442 // Check for a prior frame
443 const size_t reqFrame = frameInfos[i].fRequiredFrame;
444 if (reqFrame != SkCodec::kNone && reqFrame == cachedFrame
445 && priorFramePixels.get()) {
446 // Copy into pixels
447 memcpy(pixels.get(), priorFramePixels.get(), safeSize);
448 options.fHasPriorFrame = true;
449 } else {
450 options.fHasPriorFrame = false;
451 }
452 const SkCodec::Result result = codec->getPixels(decodeInfo, pixels.get(),
453 rowBytes, &options,
454 colorPtr, &colorCount);
455 switch (result) {
456 case SkCodec::kSuccess:
457 case SkCodec::kIncompleteInput:
458 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes,
459 colorPtr, colorCount, fDstColorType);
460 if (result == SkCodec::kIncompleteInput) {
461 return "";
462 }
463 break;
464 default:
465 return SkStringPrintf("Couldn't getPixels for frame %i in %s.",
466 i, fPath.c_str());
467 }
468
469 // If a future frame depends on this one, store it in priorFrame.
470 // (Note that if i+1 does *not* depend on i, then no future frame can.)
471 if (i+1 < frameInfos.size() && frameInfos[i+1].fRequiredFrame == i) {
472 memcpy(priorFramePixels.reset(safeSize), pixels.get(), safeSize);
473 cachedFrame = i;
474 }
475
476 canvas->translate(SkIntToScalar(0), SkIntToScalar(decodeInfo.height()));
477 }
478 break;
479 }
msarettbb25b532016-01-13 09:31:39 -0800480 case kCodecZeroInit_Mode:
msarett9e707a02015-09-01 14:57:57 -0700481 case kCodec_Mode: {
msarettb1be46b2016-05-17 08:52:11 -0700482 switch (codec->getPixels(decodeInfo, pixels.get(), rowBytes, &options,
483 colorPtr, &colorCount)) {
scroggoeb602a52015-07-09 08:16:03 -0700484 case SkCodec::kSuccess:
scroggo9c59ebc2015-03-25 13:48:49 -0700485 // We consider incomplete to be valid, since we should still decode what is
486 // available.
scroggoeb602a52015-07-09 08:16:03 -0700487 case SkCodec::kIncompleteInput:
scroggo9c59ebc2015-03-25 13:48:49 -0700488 break;
scroggo9c59ebc2015-03-25 13:48:49 -0700489 default:
490 // Everything else is considered a failure.
491 return SkStringPrintf("Couldn't getPixels %s.", fPath.c_str());
492 }
msarettb1be46b2016-05-17 08:52:11 -0700493
494 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, colorPtr, colorCount,
495 fDstColorType);
scroggo9c59ebc2015-03-25 13:48:49 -0700496 break;
emmaleer0a4c3cb2015-06-22 10:40:21 -0700497 }
scroggo9c59ebc2015-03-25 13:48:49 -0700498 case kScanline_Mode: {
scroggod8d68552016-06-06 11:26:17 -0700499 void* dst = pixels.get();
500 uint32_t height = decodeInfo.height();
scroggo19b91532016-10-24 09:03:26 -0700501 const bool useIncremental = [this]() {
502 auto exts = { "png", "PNG", "gif", "GIF" };
503 for (auto ext : exts) {
504 if (fPath.endsWith(ext)) {
505 return true;
506 }
507 }
508 return false;
509 }();
510 // ico may use the old scanline method or the new one, depending on whether it
511 // internally holds a bmp or a png.
scroggo8e6c7ad2016-09-16 08:20:38 -0700512 const bool ico = fPath.endsWith("ico");
scroggo19b91532016-10-24 09:03:26 -0700513 bool useOldScanlineMethod = !useIncremental && !ico;
514 if (useIncremental || ico) {
scroggo8e6c7ad2016-09-16 08:20:38 -0700515 if (SkCodec::kSuccess == codec->startIncrementalDecode(decodeInfo, dst,
516 rowBytes, nullptr, colorPtr, &colorCount)) {
517 int rowsDecoded;
518 if (SkCodec::kIncompleteInput == codec->incrementalDecode(&rowsDecoded)) {
519 codec->fillIncompleteImage(decodeInfo, dst, rowBytes,
520 SkCodec::kNo_ZeroInitialized, height,
521 rowsDecoded);
scroggo6fb23912016-06-02 14:16:43 -0700522 }
scroggo8e6c7ad2016-09-16 08:20:38 -0700523 } else {
scroggo19b91532016-10-24 09:03:26 -0700524 if (useIncremental) {
525 // Error: These should support incremental decode.
scroggo8e6c7ad2016-09-16 08:20:38 -0700526 return "Could not start incremental decode";
527 }
528 // Otherwise, this is an ICO. Since incremental failed, it must contain a BMP,
529 // which should work via startScanlineDecode
530 useOldScanlineMethod = true;
531 }
532 }
533
534 if (useOldScanlineMethod) {
535 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo, NULL, colorPtr,
536 &colorCount)) {
537 return "Could not start scanline decoder";
538 }
539
540 switch (codec->getScanlineOrder()) {
541 case SkCodec::kTopDown_SkScanlineOrder:
542 case SkCodec::kBottomUp_SkScanlineOrder:
543 // We do not need to check the return value. On an incomplete
544 // image, memory will be filled with a default value.
545 codec->getScanlines(dst, height, rowBytes);
546 break;
msarett10522ff2015-09-07 08:54:01 -0700547 }
548 }
549
msarettb1be46b2016-05-17 08:52:11 -0700550 draw_to_canvas(canvas, bitmapInfo, dst, rowBytes, colorPtr, colorCount, fDstColorType);
emmaleer97002062015-05-27 12:36:10 -0700551 break;
552 }
msarett0a242972015-06-11 14:27:27 -0700553 case kStripe_Mode: {
554 const int height = decodeInfo.height();
555 // This value is chosen arbitrarily. We exercise more cases by choosing a value that
556 // does not align with image blocks.
557 const int stripeHeight = 37;
558 const int numStripes = (height + stripeHeight - 1) / stripeHeight;
msarettb1be46b2016-05-17 08:52:11 -0700559 void* dst = pixels.get();
msarett0a242972015-06-11 14:27:27 -0700560
561 // Decode odd stripes
msarettb1be46b2016-05-17 08:52:11 -0700562 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo, nullptr, colorPtr,
563 &colorCount)) {
msarettb65e6042016-02-23 05:37:25 -0800564 return "Could not start scanline decoder";
565 }
566
567 // This mode was designed to test the new skip scanlines API in libjpeg-turbo.
568 // Jpegs have kTopDown_SkScanlineOrder, and at this time, it is not interesting
569 // to run this test for image types that do not have this scanline ordering.
scroggo12e2f502016-05-16 09:04:18 -0700570 // We only run this on Jpeg, which is always kTopDown.
571 SkASSERT(SkCodec::kTopDown_SkScanlineOrder == codec->getScanlineOrder());
msarette6dd0042015-10-09 11:07:34 -0700572
msarett0a242972015-06-11 14:27:27 -0700573 for (int i = 0; i < numStripes; i += 2) {
574 // Skip a stripe
575 const int linesToSkip = SkTMin(stripeHeight, height - i * stripeHeight);
msarette6dd0042015-10-09 11:07:34 -0700576 codec->skipScanlines(linesToSkip);
msarett0a242972015-06-11 14:27:27 -0700577
578 // Read a stripe
579 const int startY = (i + 1) * stripeHeight;
580 const int linesToRead = SkTMin(stripeHeight, height - startY);
581 if (linesToRead > 0) {
msarettb1be46b2016-05-17 08:52:11 -0700582 codec->getScanlines(SkTAddOffset<void>(dst, rowBytes * startY), linesToRead,
583 rowBytes);
msarett0a242972015-06-11 14:27:27 -0700584 }
585 }
586
587 // Decode even stripes
scroggo46c57472015-09-30 08:57:13 -0700588 const SkCodec::Result startResult = codec->startScanlineDecode(decodeInfo, nullptr,
msarettb1be46b2016-05-17 08:52:11 -0700589 colorPtr, &colorCount);
scroggo1c005e42015-08-04 09:24:45 -0700590 if (SkCodec::kSuccess != startResult) {
591 return "Failed to restart scanline decoder with same parameters.";
msarett0a242972015-06-11 14:27:27 -0700592 }
593 for (int i = 0; i < numStripes; i += 2) {
594 // Read a stripe
595 const int startY = i * stripeHeight;
596 const int linesToRead = SkTMin(stripeHeight, height - startY);
msarettb1be46b2016-05-17 08:52:11 -0700597 codec->getScanlines(SkTAddOffset<void>(dst, rowBytes * startY), linesToRead,
598 rowBytes);
msarett0a242972015-06-11 14:27:27 -0700599
600 // Skip a stripe
msarettf6db27e2015-06-12 09:34:04 -0700601 const int linesToSkip = SkTMin(stripeHeight, height - (i + 1) * stripeHeight);
602 if (linesToSkip > 0) {
msarette6dd0042015-10-09 11:07:34 -0700603 codec->skipScanlines(linesToSkip);
msarett0a242972015-06-11 14:27:27 -0700604 }
605 }
msarettb1be46b2016-05-17 08:52:11 -0700606
607 draw_to_canvas(canvas, bitmapInfo, dst, rowBytes, colorPtr, colorCount, fDstColorType);
emmaleer0a4c3cb2015-06-22 10:40:21 -0700608 break;
msarett0a242972015-06-11 14:27:27 -0700609 }
msarett91c22b22016-02-22 12:27:46 -0800610 case kCroppedScanline_Mode: {
611 const int width = decodeInfo.width();
612 const int height = decodeInfo.height();
613 // This value is chosen because, as we move across the image, it will sometimes
614 // align with the jpeg block sizes and it will sometimes not. This allows us
615 // to test interestingly different code paths in the implementation.
616 const int tileSize = 36;
617
618 SkCodec::Options opts;
619 SkIRect subset;
620 for (int x = 0; x < width; x += tileSize) {
621 subset = SkIRect::MakeXYWH(x, 0, SkTMin(tileSize, width - x), height);
622 opts.fSubset = &subset;
623 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo, &opts,
msarettb1be46b2016-05-17 08:52:11 -0700624 colorPtr, &colorCount)) {
msarett91c22b22016-02-22 12:27:46 -0800625 return "Could not start scanline decoder.";
626 }
627
msarettb1be46b2016-05-17 08:52:11 -0700628 codec->getScanlines(SkTAddOffset<void>(pixels.get(), x * bpp), height, rowBytes);
msarett91c22b22016-02-22 12:27:46 -0800629 }
630
msarettb1be46b2016-05-17 08:52:11 -0700631 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, colorPtr, colorCount,
632 fDstColorType);
msarett91c22b22016-02-22 12:27:46 -0800633 break;
634 }
scroggob636b452015-07-22 07:16:20 -0700635 case kSubset_Mode: {
636 // Arbitrarily choose a divisor.
637 int divisor = 2;
638 // Total width/height of the image.
639 const int W = codec->getInfo().width();
640 const int H = codec->getInfo().height();
641 if (divisor > W || divisor > H) {
642 return Error::Nonfatal(SkStringPrintf("Cannot codec subset: divisor %d is too big "
643 "for %s with dimensions (%d x %d)", divisor,
644 fPath.c_str(), W, H));
645 }
646 // subset dimensions
647 // SkWebpCodec, the only one that supports subsets, requires even top/left boundaries.
648 const int w = SkAlign2(W / divisor);
649 const int h = SkAlign2(H / divisor);
650 SkIRect subset;
651 SkCodec::Options opts;
652 opts.fSubset = &subset;
653 SkBitmap subsetBm;
654 // We will reuse pixel memory from bitmap.
msarettb1be46b2016-05-17 08:52:11 -0700655 void* dst = pixels.get();
scroggob636b452015-07-22 07:16:20 -0700656 // Keep track of left and top (for drawing subsetBm into canvas). We could use
657 // fScale * x and fScale * y, but we want integers such that the next subset will start
658 // where the last one ended. So we'll add decodeInfo.width() and height().
659 int left = 0;
660 for (int x = 0; x < W; x += w) {
661 int top = 0;
662 for (int y = 0; y < H; y+= h) {
663 // Do not make the subset go off the edge of the image.
664 const int preScaleW = SkTMin(w, W - x);
665 const int preScaleH = SkTMin(h, H - y);
666 subset.setXYWH(x, y, preScaleW, preScaleH);
667 // And scale
668 // FIXME: Should we have a version of getScaledDimensions that takes a subset
669 // into account?
msarettc7eb4902016-04-25 07:04:58 -0700670 const int scaledW = SkTMax(1, SkScalarRoundToInt(preScaleW * fScale));
671 const int scaledH = SkTMax(1, SkScalarRoundToInt(preScaleH * fScale));
672 decodeInfo = decodeInfo.makeWH(scaledW, scaledH);
673 SkImageInfo subsetBitmapInfo = bitmapInfo.makeWH(scaledW, scaledH);
msarettb1be46b2016-05-17 08:52:11 -0700674 size_t subsetRowBytes = subsetBitmapInfo.minRowBytes();
675 const SkCodec::Result result = codec->getPixels(decodeInfo, dst, subsetRowBytes,
676 &opts, colorPtr, &colorCount);
scroggob636b452015-07-22 07:16:20 -0700677 switch (result) {
678 case SkCodec::kSuccess:
679 case SkCodec::kIncompleteInput:
680 break;
scroggob636b452015-07-22 07:16:20 -0700681 default:
682 return SkStringPrintf("subset codec failed to decode (%d, %d, %d, %d) "
683 "from %s with dimensions (%d x %d)\t error %d",
684 x, y, decodeInfo.width(), decodeInfo.height(),
685 fPath.c_str(), W, H, result);
686 }
msarettb1be46b2016-05-17 08:52:11 -0700687 draw_to_canvas(canvas, subsetBitmapInfo, dst, subsetRowBytes, colorPtr,
688 colorCount, fDstColorType, SkIntToScalar(left),
689 SkIntToScalar(top));
690
scroggob636b452015-07-22 07:16:20 -0700691 // translate by the scaled height.
692 top += decodeInfo.height();
693 }
694 // translate by the scaled width.
695 left += decodeInfo.width();
696 }
697 return "";
698 }
msarettb714fb02016-01-22 14:46:42 -0800699 default:
700 SkASSERT(false);
701 return "Invalid fMode";
scroggo9b77ddd2015-03-19 06:03:39 -0700702 }
scroggo9c59ebc2015-03-25 13:48:49 -0700703 return "";
scroggo9b77ddd2015-03-19 06:03:39 -0700704}
705
706SkISize CodecSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -0700707 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
reed42943c82016-09-12 12:01:44 -0700708 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
scroggo7fac5af2015-09-30 11:33:12 -0700709 if (nullptr == codec) {
710 return SkISize::Make(0, 0);
711 }
scroggo19b91532016-10-24 09:03:26 -0700712
713 auto imageSize = codec->getScaledDimensions(fScale);
714 if (fMode == kAnimated_Mode) {
715 // We'll draw one of each frame, so make it big enough to hold them all.
716 const size_t count = codec->getFrameInfo().size();
717 imageSize.fHeight = imageSize.fHeight * count;
718 }
719 return imageSize;
scroggo9b77ddd2015-03-19 06:03:39 -0700720}
721
722Name CodecSrc::name() const {
msarett0a242972015-06-11 14:27:27 -0700723 if (1.0f == fScale) {
scroggo6e8c68e2016-10-24 13:48:49 -0700724 Name name = SkOSPath::Basename(fPath.c_str());
725 if (fMode == kAnimated_Mode) {
726 name.append("_animated");
727 }
728 return name;
msarett0a242972015-06-11 14:27:27 -0700729 }
scroggo6e8c68e2016-10-24 13:48:49 -0700730 SkASSERT(fMode != kAnimated_Mode);
msaretta5783ae2015-09-08 15:35:32 -0700731 return get_scaled_name(fPath, fScale);
scroggo9b77ddd2015-03-19 06:03:39 -0700732}
733
734/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
735
scroggof8dc9df2016-05-16 09:04:13 -0700736AndroidCodecSrc::AndroidCodecSrc(Path path, CodecSrc::DstColorType dstColorType,
scroggoc5560be2016-02-03 09:42:42 -0800737 SkAlphaType dstAlphaType, int sampleSize)
msarett3d9d7a72015-10-21 10:27:10 -0700738 : fPath(path)
msarett3d9d7a72015-10-21 10:27:10 -0700739 , fDstColorType(dstColorType)
scroggoc5560be2016-02-03 09:42:42 -0800740 , fDstAlphaType(dstAlphaType)
msarett3d9d7a72015-10-21 10:27:10 -0700741 , fSampleSize(sampleSize)
scroggo3ac66e92016-02-08 15:09:48 -0800742 , fRunSerially(serial_from_path_name(path))
msarett3d9d7a72015-10-21 10:27:10 -0700743{}
744
745bool AndroidCodecSrc::veto(SinkFlags flags) const {
746 // No need to test decoding to non-raster or indirect backend.
msarett3d9d7a72015-10-21 10:27:10 -0700747 return flags.type != SinkFlags::kRaster
748 || flags.approach != SinkFlags::kDirect;
749}
750
751Error AndroidCodecSrc::draw(SkCanvas* canvas) const {
bungeman38d909e2016-08-02 14:40:46 -0700752 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
msarett3d9d7a72015-10-21 10:27:10 -0700753 if (!encoded) {
754 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
755 }
reed42943c82016-09-12 12:01:44 -0700756 SkAutoTDelete<SkAndroidCodec> codec(SkAndroidCodec::NewFromData(encoded));
msarett3d9d7a72015-10-21 10:27:10 -0700757 if (nullptr == codec.get()) {
758 return SkStringPrintf("Couldn't create android codec for %s.", fPath.c_str());
759 }
760
scroggoba584892016-05-20 13:56:13 -0700761 SkImageInfo decodeInfo = codec->getInfo();
762 if (!get_decode_info(&decodeInfo, canvas->imageInfo().colorType(), fDstColorType,
763 fDstAlphaType)) {
msarett3d9d7a72015-10-21 10:27:10 -0700764 return Error::Nonfatal("Testing non-565 to 565 is uninteresting.");
765 }
766
767 // Scale the image if it is desired.
768 SkISize size = codec->getSampledDimensions(fSampleSize);
769
770 // Visually inspecting very small output images is not necessary. We will
771 // cover these cases in unit testing.
772 if ((size.width() <= 10 || size.height() <= 10) && 1 != fSampleSize) {
773 return Error::Nonfatal("Scaling very small images is uninteresting.");
774 }
775 decodeInfo = decodeInfo.makeWH(size.width(), size.height());
776
msarettb1be46b2016-05-17 08:52:11 -0700777 int bpp = SkColorTypeBytesPerPixel(decodeInfo.colorType());
778 size_t rowBytes = size.width() * bpp;
779 SkAutoMalloc pixels(size.height() * rowBytes);
780 SkPMColor colorPtr[256];
781 int colorCount = 256;
msarett3d9d7a72015-10-21 10:27:10 -0700782
783 SkBitmap bitmap;
msarett34e0ec42016-04-22 16:27:24 -0700784 SkImageInfo bitmapInfo = decodeInfo;
785 if (kRGBA_8888_SkColorType == decodeInfo.colorType() ||
786 kBGRA_8888_SkColorType == decodeInfo.colorType()) {
787 bitmapInfo = bitmapInfo.makeColorType(kN32_SkColorType);
788 }
msarett3d9d7a72015-10-21 10:27:10 -0700789
790 // Create options for the codec.
791 SkAndroidCodec::AndroidOptions options;
792 options.fColorPtr = colorPtr;
msarettb1be46b2016-05-17 08:52:11 -0700793 options.fColorCount = &colorCount;
msarett3d9d7a72015-10-21 10:27:10 -0700794 options.fSampleSize = fSampleSize;
795
msarettb1be46b2016-05-17 08:52:11 -0700796 switch (codec->getAndroidPixels(decodeInfo, pixels.get(), rowBytes, &options)) {
scroggof8dc9df2016-05-16 09:04:13 -0700797 case SkCodec::kSuccess:
798 case SkCodec::kIncompleteInput:
799 break;
msarett3d9d7a72015-10-21 10:27:10 -0700800 default:
scroggof8dc9df2016-05-16 09:04:13 -0700801 return SkStringPrintf("Couldn't getPixels %s.", fPath.c_str());
msarett3d9d7a72015-10-21 10:27:10 -0700802 }
msarettb1be46b2016-05-17 08:52:11 -0700803 draw_to_canvas(canvas, bitmapInfo, pixels.get(), rowBytes, colorPtr, colorCount, fDstColorType);
scroggof8dc9df2016-05-16 09:04:13 -0700804 return "";
msarett3d9d7a72015-10-21 10:27:10 -0700805}
806
807SkISize AndroidCodecSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -0700808 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
reed42943c82016-09-12 12:01:44 -0700809 SkAutoTDelete<SkAndroidCodec> codec(SkAndroidCodec::NewFromData(encoded));
msarett3d9d7a72015-10-21 10:27:10 -0700810 if (nullptr == codec) {
811 return SkISize::Make(0, 0);
812 }
813 return codec->getSampledDimensions(fSampleSize);
814}
815
816Name AndroidCodecSrc::name() const {
817 // We will replicate the names used by CodecSrc so that images can
818 // be compared in Gold.
819 if (1 == fSampleSize) {
820 return SkOSPath::Basename(fPath.c_str());
821 }
msarett4b0778e2015-11-13 09:59:11 -0800822 return get_scaled_name(fPath, 1.0f / (float) fSampleSize);
msarett3d9d7a72015-10-21 10:27:10 -0700823}
824
825/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
826
msarett18976312016-03-09 14:20:58 -0800827ImageGenSrc::ImageGenSrc(Path path, Mode mode, SkAlphaType alphaType, bool isGpu)
828 : fPath(path)
829 , fMode(mode)
830 , fDstAlphaType(alphaType)
831 , fIsGpu(isGpu)
832 , fRunSerially(serial_from_path_name(path))
833{}
834
835bool ImageGenSrc::veto(SinkFlags flags) const {
836 if (fIsGpu) {
837 return flags.type != SinkFlags::kGPU || flags.approach != SinkFlags::kDirect;
838 }
839
840 return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
841}
842
843Error ImageGenSrc::draw(SkCanvas* canvas) const {
844 if (kRGB_565_SkColorType == canvas->imageInfo().colorType()) {
845 return Error::Nonfatal("Uninteresting to test image generator to 565.");
846 }
847
bungeman38d909e2016-08-02 14:40:46 -0700848 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
msarett18976312016-03-09 14:20:58 -0800849 if (!encoded) {
850 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
851 }
852
msarettfc0b6d12016-03-17 13:50:17 -0700853#if defined(SK_BUILD_FOR_WIN)
854 // Initialize COM in order to test with WIC.
855 SkAutoCoInitialize com;
856 if (!com.succeeded()) {
857 return "Could not initialize COM.";
858 }
859#endif
860
msarett18976312016-03-09 14:20:58 -0800861 SkAutoTDelete<SkImageGenerator> gen(nullptr);
862 switch (fMode) {
863 case kCodec_Mode:
bungeman38d909e2016-08-02 14:40:46 -0700864 gen.reset(SkCodecImageGenerator::NewFromEncodedCodec(encoded.get()));
msarett18976312016-03-09 14:20:58 -0800865 if (!gen) {
866 return "Could not create codec image generator.";
867 }
868 break;
msarettfc0b6d12016-03-17 13:50:17 -0700869 case kPlatform_Mode: {
msarett18976312016-03-09 14:20:58 -0800870#if defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS)
bungeman38d909e2016-08-02 14:40:46 -0700871 gen.reset(SkImageGeneratorCG::NewFromEncodedCG(encoded.get()));
msarettfc0b6d12016-03-17 13:50:17 -0700872#elif defined(SK_BUILD_FOR_WIN)
bungeman38d909e2016-08-02 14:40:46 -0700873 gen.reset(SkImageGeneratorWIC::NewFromEncodedWIC(encoded.get()));
msarettfc0b6d12016-03-17 13:50:17 -0700874#endif
875
msarett18976312016-03-09 14:20:58 -0800876 if (!gen) {
msarettfc0b6d12016-03-17 13:50:17 -0700877 return "Could not create platform image generator.";
msarett18976312016-03-09 14:20:58 -0800878 }
879 break;
msarettfc0b6d12016-03-17 13:50:17 -0700880 }
msarett18976312016-03-09 14:20:58 -0800881 default:
882 SkASSERT(false);
883 return "Invalid image generator mode";
884 }
885
886 // Test deferred decoding path on GPU
887 if (fIsGpu) {
reed9ce9d672016-03-17 10:51:11 -0700888 sk_sp<SkImage> image(SkImage::MakeFromGenerator(gen.release(), nullptr));
msarett18976312016-03-09 14:20:58 -0800889 if (!image) {
890 return "Could not create image from codec image generator.";
891 }
892 canvas->drawImage(image, 0, 0);
893 return "";
894 }
mtklein343a63d2016-03-22 11:46:53 -0700895
msarett18976312016-03-09 14:20:58 -0800896 // Test various color and alpha types on CPU
897 SkImageInfo decodeInfo = gen->getInfo().makeAlphaType(fDstAlphaType);
mtklein343a63d2016-03-22 11:46:53 -0700898
msarettb1be46b2016-05-17 08:52:11 -0700899 int bpp = SkColorTypeBytesPerPixel(decodeInfo.colorType());
900 size_t rowBytes = decodeInfo.width() * bpp;
901 SkAutoMalloc pixels(decodeInfo.height() * rowBytes);
902 SkPMColor colorPtr[256];
903 int colorCount = 256;
msarett18976312016-03-09 14:20:58 -0800904
msarettb1be46b2016-05-17 08:52:11 -0700905 if (!gen->getPixels(decodeInfo, pixels.get(), rowBytes, colorPtr, &colorCount)) {
msarett18976312016-03-09 14:20:58 -0800906 return SkStringPrintf("Image generator could not getPixels() for %s\n", fPath.c_str());
907 }
908
msarettb1be46b2016-05-17 08:52:11 -0700909 draw_to_canvas(canvas, decodeInfo, pixels.get(), rowBytes, colorPtr, colorCount,
910 CodecSrc::kGetFromCanvas_DstColorType);
msarett18976312016-03-09 14:20:58 -0800911 return "";
912}
913
914SkISize ImageGenSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -0700915 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
reed42943c82016-09-12 12:01:44 -0700916 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
msarett18976312016-03-09 14:20:58 -0800917 if (nullptr == codec) {
918 return SkISize::Make(0, 0);
919 }
920 return codec->getInfo().dimensions();
921}
922
923Name ImageGenSrc::name() const {
924 return SkOSPath::Basename(fPath.c_str());
925}
926
927/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
928
msarett9ce3a542016-07-15 13:54:38 -0700929ColorCodecSrc::ColorCodecSrc(Path path, Mode mode, SkColorType colorType)
msarett69deca82016-04-29 09:38:40 -0700930 : fPath(path)
931 , fMode(mode)
msarett9ce3a542016-07-15 13:54:38 -0700932 , fColorType(colorType)
msarett69deca82016-04-29 09:38:40 -0700933{}
934
935bool ColorCodecSrc::veto(SinkFlags flags) const {
936 // Test to direct raster backends (8888 and 565).
937 return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
938}
939
940Error ColorCodecSrc::draw(SkCanvas* canvas) const {
941 if (kRGB_565_SkColorType == canvas->imageInfo().colorType()) {
942 return Error::Nonfatal("No need to test color correction to 565 backend.");
943 }
944
msarettd1ec89b2016-08-03 12:59:27 -0700945 bool runInLegacyMode = kBaseline_Mode == fMode;
946#if defined(SK_TEST_QCMS)
947 runInLegacyMode = runInLegacyMode || kQCMS_HPZR30w_Mode == fMode;
948#endif
949
950 if (runInLegacyMode && canvas->imageInfo().colorSpace()) {
951 return Error::Nonfatal("Skipping tests that are only interesting in legacy mode.");
952 } else if (!runInLegacyMode && !canvas->imageInfo().colorSpace()) {
953 return Error::Nonfatal("Skipping tests that are only interesting in srgb mode.");
msarett9ce3a542016-07-15 13:54:38 -0700954 }
955
bungeman38d909e2016-08-02 14:40:46 -0700956 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
msarett69deca82016-04-29 09:38:40 -0700957 if (!encoded) {
958 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
959 }
960
reed42943c82016-09-12 12:01:44 -0700961 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
msarett69deca82016-04-29 09:38:40 -0700962 if (nullptr == codec.get()) {
963 return SkStringPrintf("Couldn't create codec for %s.", fPath.c_str());
964 }
965
msarett9876ac52016-06-01 14:47:18 -0700966 // Load the dst ICC profile. This particular dst is fairly similar to Adobe RGB.
967 sk_sp<SkData> dstData = SkData::MakeFromFileName(
msaretta714bc32016-07-29 08:58:33 -0700968 GetResourcePath("icc_profiles/HP_ZR30w.icc").c_str());
msarett9876ac52016-06-01 14:47:18 -0700969 if (!dstData) {
970 return "Cannot read monitor profile. Is the resource path set correctly?";
971 }
972
msarett50ce1f22016-07-29 06:23:33 -0700973 sk_sp<SkColorSpace> dstSpace = nullptr;
974 if (kDst_sRGB_Mode == fMode) {
Brian Osman526972e2016-10-24 09:24:02 -0400975 dstSpace = SkColorSpace::MakeNamed(SkColorSpace::kSRGB_Named);
msarett50ce1f22016-07-29 06:23:33 -0700976 } else if (kDst_HPZR30w_Mode == fMode) {
Brian Osman526972e2016-10-24 09:24:02 -0400977 dstSpace = SkColorSpace::MakeICC(dstData->data(), dstData->size());
msarett50ce1f22016-07-29 06:23:33 -0700978 }
979
980 SkImageInfo decodeInfo = codec->getInfo().makeColorType(fColorType).makeColorSpace(dstSpace);
msarettd1ec89b2016-08-03 12:59:27 -0700981 if (kUnpremul_SkAlphaType == decodeInfo.alphaType()) {
982 decodeInfo = decodeInfo.makeAlphaType(kPremul_SkAlphaType);
983 }
msarett2ecc35f2016-09-08 11:55:16 -0700984 if (kRGBA_F16_SkColorType == fColorType) {
raftias94888332016-10-18 10:02:51 -0700985 SkASSERT(SkColorSpace_Base::Type::kXYZ == as_CSB(decodeInfo.colorSpace())->type());
986 SkColorSpace_XYZ* csXYZ = static_cast<SkColorSpace_XYZ*>(decodeInfo.colorSpace());
987 decodeInfo = decodeInfo.makeColorSpace(csXYZ->makeLinearGamma());
msarett2ecc35f2016-09-08 11:55:16 -0700988 }
msarettd1ec89b2016-08-03 12:59:27 -0700989
msarett50ce1f22016-07-29 06:23:33 -0700990 SkImageInfo bitmapInfo = decodeInfo;
991 if (kRGBA_8888_SkColorType == decodeInfo.colorType() ||
992 kBGRA_8888_SkColorType == decodeInfo.colorType())
993 {
994 bitmapInfo = bitmapInfo.makeColorType(kN32_SkColorType);
995 }
996
997 SkBitmap bitmap;
998 if (!bitmap.tryAllocPixels(bitmapInfo)) {
999 return SkStringPrintf("Image(%s) is too large (%d x %d)", fPath.c_str(),
1000 bitmapInfo.width(), bitmapInfo.height());
1001 }
1002
1003 size_t rowBytes = bitmap.rowBytes();
1004 SkCodec::Result r = codec->getPixels(decodeInfo, bitmap.getPixels(), rowBytes);
msarettf7eb6fc2016-09-13 09:04:11 -07001005 if (SkCodec::kSuccess != r && SkCodec::kIncompleteInput != r) {
Matt Sarett313c4632016-10-20 12:35:23 -04001006 // FIXME (raftias):
1007 // This should be a fatal error. We need to add support for
1008 // A2B images in SkColorSpaceXform.
1009 return Error::Nonfatal(SkStringPrintf("Couldn't getPixels %s. Error code %d",
1010 fPath.c_str(), r));
msarett50ce1f22016-07-29 06:23:33 -07001011 }
1012
msarett69deca82016-04-29 09:38:40 -07001013 switch (fMode) {
1014 case kBaseline_Mode:
msarett39979d82016-07-28 17:11:18 -07001015 case kDst_sRGB_Mode:
msarett50ce1f22016-07-29 06:23:33 -07001016 case kDst_HPZR30w_Mode:
msarett39979d82016-07-28 17:11:18 -07001017 canvas->drawBitmap(bitmap, 0, 0);
1018 break;
msarett469f1c52016-06-06 08:20:37 -07001019#if defined(SK_TEST_QCMS)
msarett9876ac52016-06-01 14:47:18 -07001020 case kQCMS_HPZR30w_Mode: {
1021 sk_sp<SkData> srcData = codec->getICCData();
msarettd1ec89b2016-08-03 12:59:27 -07001022 if (!srcData) {
1023 return Error::Nonfatal("No ICC profile data. Cannot test with QCMS.\n");
1024 }
1025
msarett9876ac52016-06-01 14:47:18 -07001026 SkAutoTCallVProc<qcms_profile, qcms_profile_release>
1027 srcSpace(qcms_profile_from_memory(srcData->data(), srcData->size()));
1028 if (!srcSpace) {
1029 return Error::Nonfatal(SkStringPrintf("QCMS cannot create profile for %s.\n",
1030 fPath.c_str()));
1031 }
1032
1033 SkAutoTCallVProc<qcms_profile, qcms_profile_release>
1034 dstSpace(qcms_profile_from_memory(dstData->data(), dstData->size()));
1035 SkASSERT(dstSpace);
msarette8bd9ab2016-06-08 10:02:32 -07001036
1037 // Optimizes conversion by precomputing the inverse transformation to dst. Also
1038 // causes QCMS to use a completely different codepath. This is how Chrome uses QCMS.
1039 qcms_profile_precache_output_transform(dstSpace);
msarett9876ac52016-06-01 14:47:18 -07001040 SkAutoTCallVProc<qcms_transform, qcms_transform_release>
1041 transform (qcms_transform_create(srcSpace, QCMS_DATA_RGBA_8, dstSpace,
1042 QCMS_DATA_RGBA_8, QCMS_INTENT_PERCEPTUAL));
1043 if (!transform) {
1044 return SkStringPrintf("QCMS cannot create transform for %s.\n", fPath.c_str());
1045 }
1046
1047#ifdef SK_PMCOLOR_IS_RGBA
1048 qcms_output_type outType = QCMS_OUTPUT_RGBX;
1049#else
1050 qcms_output_type outType = QCMS_OUTPUT_BGRX;
1051#endif
1052
1053 // Perform color correction.
msarett50ce1f22016-07-29 06:23:33 -07001054 uint32_t* row = (uint32_t*) bitmap.getPixels();
1055 for (int y = 0; y < decodeInfo.height(); y++) {
1056 qcms_transform_data_type(transform, row, row, decodeInfo.width(), outType);
1057 row = SkTAddOffset<uint32_t>(row, rowBytes);
msarett9876ac52016-06-01 14:47:18 -07001058 }
1059
1060 canvas->drawBitmap(bitmap, 0, 0);
1061 break;
1062 }
1063#endif
msarett69deca82016-04-29 09:38:40 -07001064 default:
1065 SkASSERT(false);
1066 return "Invalid fMode";
1067 }
1068 return "";
1069}
1070
1071SkISize ColorCodecSrc::size() const {
bungeman38d909e2016-08-02 14:40:46 -07001072 sk_sp<SkData> encoded(SkData::MakeFromFileName(fPath.c_str()));
reed42943c82016-09-12 12:01:44 -07001073 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
msarett69deca82016-04-29 09:38:40 -07001074 if (nullptr == codec) {
1075 return SkISize::Make(0, 0);
1076 }
1077 return SkISize::Make(codec->getInfo().width(), codec->getInfo().height());
1078}
1079
1080Name ColorCodecSrc::name() const {
1081 return SkOSPath::Basename(fPath.c_str());
1082}
1083
1084/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1085
mtkleinf4ba3212015-01-28 15:32:24 -08001086static const SkRect kSKPViewport = {0,0, 1000,1000};
1087
mtklein8d17a132015-01-30 11:42:31 -08001088SKPSrc::SKPSrc(Path path) : fPath(path) {}
mtklein748ca3b2015-01-15 10:56:12 -08001089
1090Error SKPSrc::draw(SkCanvas* canvas) const {
bungemanf93d7112016-09-16 06:24:20 -07001091 std::unique_ptr<SkStream> stream = SkStream::MakeFromFile(fPath.c_str());
mtklein75d98fd2015-01-18 07:05:01 -08001092 if (!stream) {
mtklein748ca3b2015-01-15 10:56:12 -08001093 return SkStringPrintf("Couldn't read %s.", fPath.c_str());
1094 }
bungemanf93d7112016-09-16 06:24:20 -07001095 sk_sp<SkPicture> pic(SkPicture::MakeFromStream(stream.get()));
mtklein75d98fd2015-01-18 07:05:01 -08001096 if (!pic) {
1097 return SkStringPrintf("Couldn't decode %s as a picture.", fPath.c_str());
1098 }
bungemanf93d7112016-09-16 06:24:20 -07001099 stream = nullptr; // Might as well drop this when we're done with it.
joshualitt7c3a2f82015-03-31 13:32:05 -07001100
mtkleinf4ba3212015-01-28 15:32:24 -08001101 canvas->clipRect(kSKPViewport);
mtklein748ca3b2015-01-15 10:56:12 -08001102 canvas->drawPicture(pic);
1103 return "";
1104}
1105
1106SkISize SKPSrc::size() const {
bungemanf93d7112016-09-16 06:24:20 -07001107 std::unique_ptr<SkStream> stream = SkStream::MakeFromFile(fPath.c_str());
mtkleinffa901a2015-03-16 10:38:07 -07001108 if (!stream) {
1109 return SkISize::Make(0,0);
1110 }
1111 SkPictInfo info;
bungemanf93d7112016-09-16 06:24:20 -07001112 if (!SkPicture::InternalOnly_StreamIsSKP(stream.get(), &info)) {
mtkleinffa901a2015-03-16 10:38:07 -07001113 return SkISize::Make(0,0);
1114 }
1115 SkRect viewport = kSKPViewport;
1116 if (!viewport.intersect(info.fCullRect)) {
1117 return SkISize::Make(0,0);
1118 }
1119 return viewport.roundOut().size();
mtklein748ca3b2015-01-15 10:56:12 -08001120}
1121
1122Name SKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
1123
1124/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
fmalitaa2b9fdf2016-08-03 19:53:36 -07001125#if defined(SK_XML)
fmalitabdf3e5c2016-09-17 07:26:26 -07001126// Used when the image doesn't have an intrinsic size.
1127static const SkSize kDefaultSVGSize = SkSize::Make(1000, 1000);
fmalitaa2b9fdf2016-08-03 19:53:36 -07001128
fmalitabdf3e5c2016-09-17 07:26:26 -07001129// Used to force-scale tiny fixed-size images.
1130static const SkSize kMinimumSVGSize = SkSize::Make(128, 128);
fmalitaa2b9fdf2016-08-03 19:53:36 -07001131
fmalitabdf3e5c2016-09-17 07:26:26 -07001132SVGSrc::SVGSrc(Path path) : fPath(path), fScale(1) {}
1133
1134Error SVGSrc::ensureDom() const {
1135 if (!fDom) {
1136 SkFILEStream stream(fPath.c_str());
1137 if (!stream.isValid()) {
1138 return SkStringPrintf("Unable to open file: %s", fPath.c_str());
1139 }
1140 fDom = SkSVGDOM::MakeFromStream(stream);
1141 if (!fDom) {
1142 return SkStringPrintf("Unable to parse file: %s", fPath.c_str());
1143 }
1144
1145 const SkSize& sz = fDom->containerSize();
1146 if (sz.isEmpty()) {
1147 // no intrinsic size
1148 fDom->setContainerSize(kDefaultSVGSize);
1149 } else {
1150 fScale = SkTMax(1.f, SkTMax(kMinimumSVGSize.width() / sz.width(),
1151 kMinimumSVGSize.height() / sz.height()));
1152 }
fmalitaa2b9fdf2016-08-03 19:53:36 -07001153 }
1154
fmalitaa2b9fdf2016-08-03 19:53:36 -07001155 return "";
1156}
1157
fmalitabdf3e5c2016-09-17 07:26:26 -07001158Error SVGSrc::draw(SkCanvas* canvas) const {
1159 Error err = this->ensureDom();
1160 if (err.isEmpty()) {
1161 SkAutoCanvasRestore acr(canvas, true);
1162 canvas->scale(fScale, fScale);
1163 fDom->render(canvas);
1164 }
1165
1166 return err;
1167}
1168
fmalitaa2b9fdf2016-08-03 19:53:36 -07001169SkISize SVGSrc::size() const {
fmalitabdf3e5c2016-09-17 07:26:26 -07001170 Error err = this->ensureDom();
1171 if (!err.isEmpty()) {
1172 return SkISize::Make(0, 0);
1173 }
1174
1175 return SkSize::Make(fDom->containerSize().width() * fScale,
1176 fDom->containerSize().height() * fScale).toRound();
fmalitaa2b9fdf2016-08-03 19:53:36 -07001177}
1178
1179Name SVGSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
1180
fmalita179d8852016-08-16 14:23:29 -07001181bool SVGSrc::veto(SinkFlags flags) const {
1182 // No need to test to non-(raster||gpu) or indirect backends.
1183 bool type_ok = flags.type == SinkFlags::kRaster
1184 || flags.type == SinkFlags::kGPU;
1185
1186 return !type_ok || flags.approach != SinkFlags::kDirect;
1187}
1188
fmalitaa2b9fdf2016-08-03 19:53:36 -07001189#endif // defined(SK_XML)
1190/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
mtklein748ca3b2015-01-15 10:56:12 -08001191
halcanary45420a92016-06-02 12:41:14 -07001192MSKPSrc::MSKPSrc(Path path) : fPath(path) {
bungemanf93d7112016-09-16 06:24:20 -07001193 std::unique_ptr<SkStreamAsset> stream = SkStream::MakeFromFile(fPath.c_str());
halcanary45420a92016-06-02 12:41:14 -07001194 (void)fReader.init(stream.get());
1195}
1196
1197int MSKPSrc::pageCount() const { return fReader.pageCount(); }
1198
1199SkISize MSKPSrc::size() const { return this->size(0); }
1200SkISize MSKPSrc::size(int i) const { return fReader.pageSize(i).toCeil(); }
1201
1202Error MSKPSrc::draw(SkCanvas* c) const { return this->draw(0, c); }
1203Error MSKPSrc::draw(int i, SkCanvas* canvas) const {
bungemanf93d7112016-09-16 06:24:20 -07001204 std::unique_ptr<SkStreamAsset> stream = SkStream::MakeFromFile(fPath.c_str());
halcanary45420a92016-06-02 12:41:14 -07001205 if (!stream) {
1206 return SkStringPrintf("Unable to open file: %s", fPath.c_str());
1207 }
1208 if (fReader.pageCount() == 0) {
1209 return SkStringPrintf("Unable to parse MultiPictureDocument file: %s", fPath.c_str());
1210 }
1211 if (i >= fReader.pageCount()) {
1212 return SkStringPrintf("MultiPictureDocument page number out of range: %d", i);
1213 }
1214 sk_sp<SkPicture> page = fReader.readPage(stream.get(), i);
1215 if (!page) {
1216 return SkStringPrintf("SkMultiPictureDocumentReader failed on page %d: %s",
1217 i, fPath.c_str());
1218 }
1219 canvas->drawPicture(page);
1220 return "";
1221}
1222
1223Name MSKPSrc::name() const { return SkOSPath::Basename(fPath.c_str()); }
1224
1225/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1226
mtkleinad66f9b2015-02-13 15:11:10 -08001227Error NullSink::draw(const Src& src, SkBitmap*, SkWStream*, SkString*) const {
1228 SkAutoTDelete<SkCanvas> canvas(SkCreateNullCanvas());
1229 return src.draw(canvas);
1230}
1231
1232/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1233
mtkleinb9eb4ac2015-02-02 18:26:03 -08001234DEFINE_bool(gpuStats, false, "Append GPU stats to the log for each GPU task?");
1235
bsalomon85b4b532016-04-05 11:06:27 -07001236GPUSink::GPUSink(GrContextFactory::ContextType ct,
1237 GrContextFactory::ContextOptions options,
mtklein82d28432015-01-15 12:46:02 -08001238 int samples,
bsalomonafcd7cd2015-08-31 12:39:41 -07001239 bool diText,
brianosmand93c1202016-03-10 07:49:08 -08001240 SkColorType colorType,
brianosmanb109b8c2016-06-16 13:03:24 -07001241 sk_sp<SkColorSpace> colorSpace,
mtklein82d28432015-01-15 12:46:02 -08001242 bool threaded)
mtklein748ca3b2015-01-15 10:56:12 -08001243 : fContextType(ct)
kkinnunen5219fd92015-12-10 06:28:13 -08001244 , fContextOptions(options)
mtklein748ca3b2015-01-15 10:56:12 -08001245 , fSampleCount(samples)
bsalomonafcd7cd2015-08-31 12:39:41 -07001246 , fUseDIText(diText)
brianosmand93c1202016-03-10 07:49:08 -08001247 , fColorType(colorType)
brianosmanb109b8c2016-06-16 13:03:24 -07001248 , fColorSpace(std::move(colorSpace))
mtklein82d28432015-01-15 12:46:02 -08001249 , fThreaded(threaded) {}
mtklein748ca3b2015-01-15 10:56:12 -08001250
bsalomon648c6962015-10-23 09:06:59 -07001251DEFINE_bool(imm, false, "Run gpu configs in immediate mode.");
bsalomon69cfe952015-11-30 13:27:47 -08001252DEFINE_bool(batchClip, false, "Clip each GrBatch to its device bounds for testing.");
bsalomon6dea83f2015-12-03 12:58:06 -08001253DEFINE_bool(batchBounds, false, "Draw a wireframe bounds of each GrBatch.");
bsalomon489147c2015-12-14 12:13:09 -08001254DEFINE_int32(batchLookback, -1, "Maximum GrBatch lookback for combining, negative means default.");
bsalomonaecc0182016-03-07 11:50:44 -08001255DEFINE_int32(batchLookahead, -1, "Maximum GrBatch lookahead for combining, negative means "
1256 "default.");
bsalomon648c6962015-10-23 09:06:59 -07001257
mtkleinb9eb4ac2015-02-02 18:26:03 -08001258Error GPUSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString* log) const {
kkinnunen5219fd92015-12-10 06:28:13 -08001259 GrContextOptions grOptions;
bsalomon489147c2015-12-14 12:13:09 -08001260 grOptions.fImmediateMode = FLAGS_imm;
1261 grOptions.fClipBatchToBounds = FLAGS_batchClip;
1262 grOptions.fDrawBatchBounds = FLAGS_batchBounds;
1263 grOptions.fMaxBatchLookback = FLAGS_batchLookback;
bsalomonaecc0182016-03-07 11:50:44 -08001264 grOptions.fMaxBatchLookahead = FLAGS_batchLookahead;
kkinnunen64492c42015-12-08 01:24:40 -08001265
kkinnunen5219fd92015-12-10 06:28:13 -08001266 src.modifyGrContextOptions(&grOptions);
1267
1268 GrContextFactory factory(grOptions);
mtkleinf4ba3212015-01-28 15:32:24 -08001269 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -08001270 const SkImageInfo info =
brianosmand93c1202016-03-10 07:49:08 -08001271 SkImageInfo::Make(size.width(), size.height(), fColorType,
brianosmanb109b8c2016-06-16 13:03:24 -07001272 kPremul_SkAlphaType, fColorSpace);
msarett13a036b2016-02-08 09:10:47 -08001273#if SK_SUPPORT_GPU
bsalomon8b7451a2016-05-11 06:33:06 -07001274 GrContext* context = factory.getContextInfo(fContextType, fContextOptions).grContext();
1275 const int maxDimension = context->caps()->maxTextureSize();
msarett13a036b2016-02-08 09:10:47 -08001276 if (maxDimension < SkTMax(size.width(), size.height())) {
1277 return Error::Nonfatal("Src too large to create a texture.\n");
1278 }
1279#endif
1280
reede8f30622016-03-23 18:59:25 -07001281 auto surface(
kkinnunen3e980c32015-12-23 01:33:00 -08001282 NewGpuSurface(&factory, fContextType, fContextOptions, info, fSampleCount, fUseDIText));
mtklein748ca3b2015-01-15 10:56:12 -08001283 if (!surface) {
1284 return "Could not create a surface.";
1285 }
joshualitt5f5a8d72015-02-25 14:09:45 -08001286 if (FLAGS_preAbandonGpuContext) {
joshualitt5f5a8d72015-02-25 14:09:45 -08001287 factory.abandonContexts();
1288 }
mtklein748ca3b2015-01-15 10:56:12 -08001289 SkCanvas* canvas = surface->getCanvas();
1290 Error err = src.draw(canvas);
1291 if (!err.isEmpty()) {
1292 return err;
1293 }
1294 canvas->flush();
mtkleinb9eb4ac2015-02-02 18:26:03 -08001295 if (FLAGS_gpuStats) {
1296 canvas->getGrContext()->dumpCacheStats(log);
1297 canvas->getGrContext()->dumpGpuStats(log);
1298 }
mtklein748ca3b2015-01-15 10:56:12 -08001299 dst->allocPixels(info);
joshualitt5f5a8d72015-02-25 14:09:45 -08001300 canvas->readPixels(dst, 0, 0);
mtklein55e88b22015-01-21 15:50:13 -08001301 if (FLAGS_abandonGpuContext) {
1302 factory.abandonContexts();
bsalomon6e2aad42016-04-01 11:54:31 -07001303 } else if (FLAGS_releaseAndAbandonGpuContext) {
1304 factory.releaseResourcesAndAbandonContexts();
mtklein55e88b22015-01-21 15:50:13 -08001305 }
mtklein748ca3b2015-01-15 10:56:12 -08001306 return "";
1307}
1308
1309/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1310
halcanary47ef4d52015-03-03 09:13:09 -08001311static Error draw_skdocument(const Src& src, SkDocument* doc, SkWStream* dst) {
halcanary4ba051c2016-03-10 10:31:53 -08001312 if (src.size().isEmpty()) {
1313 return "Source has empty dimensions";
1314 }
halcanary47ef4d52015-03-03 09:13:09 -08001315 SkASSERT(doc);
halcanary45420a92016-06-02 12:41:14 -07001316 int pageCount = src.pageCount();
1317 for (int i = 0; i < pageCount; ++i) {
1318 int width = src.size(i).width(), height = src.size(i).height();
halcanary7e798182015-04-14 14:06:18 -07001319 SkCanvas* canvas =
1320 doc->beginPage(SkIntToScalar(width), SkIntToScalar(height));
1321 if (!canvas) {
halcanary96fcdcc2015-08-27 07:41:13 -07001322 return "SkDocument::beginPage(w,h) returned nullptr";
halcanary7e798182015-04-14 14:06:18 -07001323 }
halcanary45420a92016-06-02 12:41:14 -07001324 Error err = src.draw(i, canvas);
halcanary7e798182015-04-14 14:06:18 -07001325 if (!err.isEmpty()) {
1326 return err;
1327 }
1328 doc->endPage();
mtklein748ca3b2015-01-15 10:56:12 -08001329 }
reedd14df7c2016-09-22 14:12:46 -07001330 doc->close();
halcanaryfd4a9932015-01-28 11:45:58 -08001331 dst->flush();
mtklein748ca3b2015-01-15 10:56:12 -08001332 return "";
1333}
1334
halcanary47ef4d52015-03-03 09:13:09 -08001335Error PDFSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
halcanary4b656662016-04-27 07:45:18 -07001336 SkDocument::PDFMetadata metadata;
1337 metadata.fTitle = src.name();
1338 metadata.fSubject = "rendering correctness test";
1339 metadata.fCreator = "Skia/DM";
1340 sk_sp<SkDocument> doc = SkDocument::MakePDF(dst, SK_ScalarDefaultRasterDPI,
1341 metadata, nullptr, fPDFA);
halcanary47ef4d52015-03-03 09:13:09 -08001342 if (!doc) {
halcanary4b656662016-04-27 07:45:18 -07001343 return "SkDocument::MakePDF() returned nullptr";
halcanary47ef4d52015-03-03 09:13:09 -08001344 }
1345 return draw_skdocument(src, doc.get(), dst);
1346}
1347
1348/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1349
1350XPSSink::XPSSink() {}
1351
1352Error XPSSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
halcanary4b656662016-04-27 07:45:18 -07001353 sk_sp<SkDocument> doc(SkDocument::MakeXPS(dst));
halcanary47ef4d52015-03-03 09:13:09 -08001354 if (!doc) {
halcanary4b656662016-04-27 07:45:18 -07001355 return "SkDocument::MakeXPS() returned nullptr";
halcanary47ef4d52015-03-03 09:13:09 -08001356 }
1357 return draw_skdocument(src, doc.get(), dst);
1358}
reed54dc4872016-09-13 08:09:45 -07001359
1360/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1361
1362PipeSink::PipeSink() {}
1363
1364Error PipeSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
1365 return src.draw(SkPipeSerializer().beginWrite(SkRect::Make(src.size()), dst));
1366}
1367
mtklein748ca3b2015-01-15 10:56:12 -08001368/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1369
mtklein9c3f17d2015-01-28 11:35:18 -08001370SKPSink::SKPSink() {}
1371
mtkleinb9eb4ac2015-02-02 18:26:03 -08001372Error SKPSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
mtklein9c3f17d2015-01-28 11:35:18 -08001373 SkSize size;
1374 size = src.size();
1375 SkPictureRecorder recorder;
1376 Error err = src.draw(recorder.beginRecording(size.width(), size.height()));
1377 if (!err.isEmpty()) {
1378 return err;
1379 }
reedca2622b2016-03-18 07:25:55 -07001380 recorder.finishRecordingAsPicture()->serialize(dst);
mtklein9c3f17d2015-01-28 11:35:18 -08001381 return "";
1382}
1383
1384/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1385
mtklein8a4527e2015-01-31 20:00:58 -08001386SVGSink::SVGSink() {}
1387
mtkleinb9eb4ac2015-02-02 18:26:03 -08001388Error SVGSink::draw(const Src& src, SkBitmap*, SkWStream* dst, SkString*) const {
fmalita718df0a2016-07-15 10:33:29 -07001389#if defined(SK_XML)
halcanary385fe4d2015-08-26 13:07:48 -07001390 SkAutoTDelete<SkXMLWriter> xmlWriter(new SkXMLStreamWriter(dst));
fmalita2aafe6f2015-02-06 12:51:10 -08001391 SkAutoTUnref<SkCanvas> canvas(SkSVGCanvas::Create(
1392 SkRect::MakeWH(SkIntToScalar(src.size().width()), SkIntToScalar(src.size().height())),
1393 xmlWriter));
1394 return src.draw(canvas);
fmalita718df0a2016-07-15 10:33:29 -07001395#else
1396 return Error("SVG sink is disabled.");
1397#endif // SK_XML
mtklein8a4527e2015-01-31 20:00:58 -08001398}
1399
1400/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1401
brianosmanb109b8c2016-06-16 13:03:24 -07001402RasterSink::RasterSink(SkColorType colorType, sk_sp<SkColorSpace> colorSpace)
mtklein27c3fdd2016-02-26 14:43:21 -08001403 : fColorType(colorType)
brianosmanb109b8c2016-06-16 13:03:24 -07001404 , fColorSpace(std::move(colorSpace)) {}
mtklein748ca3b2015-01-15 10:56:12 -08001405
mtkleinb9eb4ac2015-02-02 18:26:03 -08001406Error RasterSink::draw(const Src& src, SkBitmap* dst, SkWStream*, SkString*) const {
mtkleinf4ba3212015-01-28 15:32:24 -08001407 const SkISize size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -08001408 // If there's an appropriate alpha type for this color type, use it, otherwise use premul.
1409 SkAlphaType alphaType = kPremul_SkAlphaType;
1410 (void)SkColorTypeValidateAlphaType(fColorType, alphaType, &alphaType);
1411
mtkleinc8be09a2016-01-04 18:56:57 -08001412 SkMallocPixelRef::ZeroedPRFactory factory;
mtklein27c3fdd2016-02-26 14:43:21 -08001413 dst->allocPixels(SkImageInfo::Make(size.width(), size.height(),
brianosmanb109b8c2016-06-16 13:03:24 -07001414 fColorType, alphaType, fColorSpace),
mtkleinc8be09a2016-01-04 18:56:57 -08001415 &factory,
1416 nullptr/*colortable*/);
mtklein748ca3b2015-01-15 10:56:12 -08001417 SkCanvas canvas(*dst);
1418 return src.draw(&canvas);
1419}
1420
1421/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1422
mtkleina16e69e2015-05-05 11:38:45 -07001423// Handy for front-patching a Src. Do whatever up-front work you need, then call draw_to_canvas(),
mtkleine44b5082015-05-07 10:53:34 -07001424// passing the Sink draw() arguments, a size, and a function draws into an SkCanvas.
mtkleina16e69e2015-05-05 11:38:45 -07001425// Several examples below.
1426
mtkleincbf89782016-02-19 14:27:14 -08001427template <typename Fn>
msarett62d3b102015-12-10 15:14:27 -08001428static Error draw_to_canvas(Sink* sink, SkBitmap* bitmap, SkWStream* stream, SkString* log,
mtkleincbf89782016-02-19 14:27:14 -08001429 SkISize size, const Fn& draw) {
mtkleina16e69e2015-05-05 11:38:45 -07001430 class ProxySrc : public Src {
1431 public:
mtkleincbf89782016-02-19 14:27:14 -08001432 ProxySrc(SkISize size, const Fn& draw) : fSize(size), fDraw(draw) {}
mtkleina16e69e2015-05-05 11:38:45 -07001433 Error draw(SkCanvas* canvas) const override { return fDraw(canvas); }
halcanaryb4a7f142016-03-30 08:31:27 -07001434 Name name() const override { return "ProxySrc"; }
1435 SkISize size() const override { return fSize; }
mtkleina16e69e2015-05-05 11:38:45 -07001436 private:
mtkleincbf89782016-02-19 14:27:14 -08001437 SkISize fSize;
1438 const Fn& fDraw;
mtkleina16e69e2015-05-05 11:38:45 -07001439 };
msarett62d3b102015-12-10 15:14:27 -08001440 return sink->draw(ProxySrc(size, draw), bitmap, stream, log);
mtkleina16e69e2015-05-05 11:38:45 -07001441}
1442
1443/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1444
mtklein4a34ecb2016-01-08 10:19:35 -08001445DEFINE_bool(check, true, "If true, have most Via- modes fail if they affect the output.");
1446
1447// Is *bitmap identical to what you get drawing src into sink?
1448static Error check_against_reference(const SkBitmap* bitmap, const Src& src, Sink* sink) {
1449 // We can only check raster outputs.
1450 // (Non-raster outputs like .pdf, .skp, .svg may differ but still draw identically.)
1451 if (FLAGS_check && bitmap) {
1452 SkBitmap reference;
1453 SkString log;
halcanaryb4a7f142016-03-30 08:31:27 -07001454 SkDynamicMemoryWStream wStream;
1455 Error err = sink->draw(src, &reference, &wStream, &log);
mtklein4a34ecb2016-01-08 10:19:35 -08001456 // If we can draw into this Sink via some pipeline, we should be able to draw directly.
1457 SkASSERT(err.isEmpty());
1458 if (!err.isEmpty()) {
1459 return err;
1460 }
1461 // The dimensions are a property of the Src only, and so should be identical.
1462 SkASSERT(reference.getSize() == bitmap->getSize());
1463 if (reference.getSize() != bitmap->getSize()) {
1464 return "Dimensions don't match reference";
1465 }
1466 // All SkBitmaps in DM are pre-locked and tight, so this comparison is easy.
1467 if (0 != memcmp(reference.getPixels(), bitmap->getPixels(), reference.getSize())) {
1468 return "Pixels don't match reference";
1469 }
1470 }
1471 return "";
1472}
1473
1474/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1475
mtkleind603b222015-02-17 11:13:33 -08001476static SkISize auto_compute_translate(SkMatrix* matrix, int srcW, int srcH) {
1477 SkRect bounds = SkRect::MakeIWH(srcW, srcH);
1478 matrix->mapRect(&bounds);
1479 matrix->postTranslate(-bounds.x(), -bounds.y());
1480 return SkISize::Make(SkScalarRoundToInt(bounds.width()), SkScalarRoundToInt(bounds.height()));
1481}
1482
msarett62d3b102015-12-10 15:14:27 -08001483ViaMatrix::ViaMatrix(SkMatrix matrix, Sink* sink) : Via(sink), fMatrix(matrix) {}
mtklein748ca3b2015-01-15 10:56:12 -08001484
mtkleinb9eb4ac2015-02-02 18:26:03 -08001485Error ViaMatrix::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtkleina16e69e2015-05-05 11:38:45 -07001486 SkMatrix matrix = fMatrix;
1487 SkISize size = auto_compute_translate(&matrix, src.size().width(), src.size().height());
msarett62d3b102015-12-10 15:14:27 -08001488 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) {
mtkleina16e69e2015-05-05 11:38:45 -07001489 canvas->concat(matrix);
1490 return src.draw(canvas);
1491 });
mtklein748ca3b2015-01-15 10:56:12 -08001492}
1493
mtkleind603b222015-02-17 11:13:33 -08001494// Undoes any flip or 90 degree rotate without changing the scale of the bitmap.
1495// This should be pixel-preserving.
msarett62d3b102015-12-10 15:14:27 -08001496ViaUpright::ViaUpright(SkMatrix matrix, Sink* sink) : Via(sink), fMatrix(matrix) {}
mtkleind603b222015-02-17 11:13:33 -08001497
1498Error ViaUpright::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
1499 Error err = fSink->draw(src, bitmap, stream, log);
1500 if (!err.isEmpty()) {
1501 return err;
1502 }
1503
1504 SkMatrix inverse;
1505 if (!fMatrix.rectStaysRect() || !fMatrix.invert(&inverse)) {
1506 return "Cannot upright --matrix.";
1507 }
1508 SkMatrix upright = SkMatrix::I();
1509 upright.setScaleX(SkScalarSignAsScalar(inverse.getScaleX()));
1510 upright.setScaleY(SkScalarSignAsScalar(inverse.getScaleY()));
1511 upright.setSkewX(SkScalarSignAsScalar(inverse.getSkewX()));
1512 upright.setSkewY(SkScalarSignAsScalar(inverse.getSkewY()));
1513
1514 SkBitmap uprighted;
1515 SkISize size = auto_compute_translate(&upright, bitmap->width(), bitmap->height());
1516 uprighted.allocPixels(bitmap->info().makeWH(size.width(), size.height()));
1517
1518 SkCanvas canvas(uprighted);
1519 canvas.concat(upright);
1520 SkPaint paint;
reed374772b2016-10-05 17:33:02 -07001521 paint.setBlendMode(SkBlendMode::kSrc);
mtkleind603b222015-02-17 11:13:33 -08001522 canvas.drawBitmap(*bitmap, 0, 0, &paint);
1523
1524 *bitmap = uprighted;
1525 bitmap->lockPixels();
1526 return "";
1527}
1528
mtklein748ca3b2015-01-15 10:56:12 -08001529/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1530
mtkleina16e69e2015-05-05 11:38:45 -07001531Error ViaSerialization::draw(
1532 const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtklein748ca3b2015-01-15 10:56:12 -08001533 // Record our Src into a picture.
mtkleina16e69e2015-05-05 11:38:45 -07001534 auto size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -08001535 SkPictureRecorder recorder;
mtkleina16e69e2015-05-05 11:38:45 -07001536 Error err = src.draw(recorder.beginRecording(SkIntToScalar(size.width()),
1537 SkIntToScalar(size.height())));
mtklein748ca3b2015-01-15 10:56:12 -08001538 if (!err.isEmpty()) {
1539 return err;
1540 }
reedca2622b2016-03-18 07:25:55 -07001541 sk_sp<SkPicture> pic(recorder.finishRecordingAsPicture());
mtklein748ca3b2015-01-15 10:56:12 -08001542
1543 // Serialize it and then deserialize it.
reed39eaf5f2016-09-15 07:19:35 -07001544 sk_sp<SkPicture> deserialized(SkPicture::MakeFromData(pic->serialize().get()));
mtklein748ca3b2015-01-15 10:56:12 -08001545
msarett62d3b102015-12-10 15:14:27 -08001546 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) {
mtkleina16e69e2015-05-05 11:38:45 -07001547 canvas->drawPicture(deserialized);
mtklein4a34ecb2016-01-08 10:19:35 -08001548 return check_against_reference(bitmap, src, fSink);
mtkleina16e69e2015-05-05 11:38:45 -07001549 });
mtklein748ca3b2015-01-15 10:56:12 -08001550}
1551
1552/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1553
msarett62d3b102015-12-10 15:14:27 -08001554ViaTiles::ViaTiles(int w, int h, SkBBHFactory* factory, Sink* sink)
1555 : Via(sink)
mtklein78829242015-05-06 07:54:07 -07001556 , fW(w)
mtklein748ca3b2015-01-15 10:56:12 -08001557 , fH(h)
mtklein78829242015-05-06 07:54:07 -07001558 , fFactory(factory) {}
mtklein748ca3b2015-01-15 10:56:12 -08001559
mtkleinb9eb4ac2015-02-02 18:26:03 -08001560Error ViaTiles::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtkleina16e69e2015-05-05 11:38:45 -07001561 auto size = src.size();
mtklein748ca3b2015-01-15 10:56:12 -08001562 SkPictureRecorder recorder;
mtkleina16e69e2015-05-05 11:38:45 -07001563 Error err = src.draw(recorder.beginRecording(SkIntToScalar(size.width()),
1564 SkIntToScalar(size.height()),
1565 fFactory.get()));
mtklein748ca3b2015-01-15 10:56:12 -08001566 if (!err.isEmpty()) {
1567 return err;
1568 }
reedca2622b2016-03-18 07:25:55 -07001569 sk_sp<SkPicture> pic(recorder.finishRecordingAsPicture());
mtklein748ca3b2015-01-15 10:56:12 -08001570
msarett62d3b102015-12-10 15:14:27 -08001571 return draw_to_canvas(fSink, bitmap, stream, log, src.size(), [&](SkCanvas* canvas) {
mtkleina16e69e2015-05-05 11:38:45 -07001572 const int xTiles = (size.width() + fW - 1) / fW,
1573 yTiles = (size.height() + fH - 1) / fH;
1574 SkMultiPictureDraw mpd(xTiles*yTiles);
reede8f30622016-03-23 18:59:25 -07001575 SkTArray<sk_sp<SkSurface>> surfaces;
1576// surfaces.setReserve(xTiles*yTiles);
mtklein748ca3b2015-01-15 10:56:12 -08001577
mtkleina16e69e2015-05-05 11:38:45 -07001578 SkImageInfo info = canvas->imageInfo().makeWH(fW, fH);
1579 for (int j = 0; j < yTiles; j++) {
1580 for (int i = 0; i < xTiles; i++) {
1581 // This lets our ultimate Sink determine the best kind of surface.
1582 // E.g., if it's a GpuSink, the surfaces and images are textures.
reede8f30622016-03-23 18:59:25 -07001583 auto s = canvas->makeSurface(info);
mtkleina16e69e2015-05-05 11:38:45 -07001584 if (!s) {
reede8f30622016-03-23 18:59:25 -07001585 s = SkSurface::MakeRaster(info); // Some canvases can't create surfaces.
mtklein748ca3b2015-01-15 10:56:12 -08001586 }
reede8f30622016-03-23 18:59:25 -07001587 surfaces.push_back(s);
mtkleina16e69e2015-05-05 11:38:45 -07001588 SkCanvas* c = s->getCanvas();
1589 c->translate(SkIntToScalar(-i * fW),
1590 SkIntToScalar(-j * fH)); // Line up the canvas with this tile.
reedca2622b2016-03-18 07:25:55 -07001591 mpd.add(c, pic.get());
mtklein748ca3b2015-01-15 10:56:12 -08001592 }
mtklein748ca3b2015-01-15 10:56:12 -08001593 }
mtkleina16e69e2015-05-05 11:38:45 -07001594 mpd.draw();
1595 for (int j = 0; j < yTiles; j++) {
1596 for (int i = 0; i < xTiles; i++) {
reed9ce9d672016-03-17 10:51:11 -07001597 sk_sp<SkImage> image(surfaces[i+xTiles*j]->makeImageSnapshot());
mtkleina16e69e2015-05-05 11:38:45 -07001598 canvas->drawImage(image, SkIntToScalar(i*fW), SkIntToScalar(j*fH));
1599 }
1600 }
mtkleina16e69e2015-05-05 11:38:45 -07001601 return "";
1602 });
mtklein748ca3b2015-01-15 10:56:12 -08001603}
1604
mtkleinb7e8d692015-04-07 08:30:32 -07001605/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1606
mtklein4a34ecb2016-01-08 10:19:35 -08001607Error ViaPicture::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
1608 auto size = src.size();
1609 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) -> Error {
1610 SkPictureRecorder recorder;
reedca2622b2016-03-18 07:25:55 -07001611 sk_sp<SkPicture> pic;
mtklein4a34ecb2016-01-08 10:19:35 -08001612 Error err = src.draw(recorder.beginRecording(SkIntToScalar(size.width()),
1613 SkIntToScalar(size.height())));
1614 if (!err.isEmpty()) {
1615 return err;
1616 }
reedca2622b2016-03-18 07:25:55 -07001617 pic = recorder.finishRecordingAsPicture();
mtklein4a34ecb2016-01-08 10:19:35 -08001618 canvas->drawPicture(pic);
1619 return check_against_reference(bitmap, src, fSink);
1620 });
1621}
1622
1623/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1624
reedbabc3de2016-07-08 08:43:27 -07001625Error ViaDefer::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
1626 auto size = src.size();
1627 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) -> Error {
1628 SkDeferredCanvas deferred(canvas);
1629 return src.draw(&deferred);
1630 });
1631}
1632
1633/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1634
reed54dc4872016-09-13 08:09:45 -07001635Error ViaPipe::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
1636 auto size = src.size();
1637 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) -> Error {
1638 SkDynamicMemoryWStream tmpStream;
1639 Error err = src.draw(SkPipeSerializer().beginWrite(SkRect::Make(size), &tmpStream));
1640 if (!err.isEmpty()) {
1641 return err;
1642 }
1643 sk_sp<SkData> data = tmpStream.detachAsData();
1644 SkPipeDeserializer().playback(data->data(), data->size(), canvas);
1645 return check_against_reference(bitmap, src, fSink);
1646 });
1647}
1648
1649/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1650
mtkleinb7e8d692015-04-07 08:30:32 -07001651// Draw the Src into two pictures, then draw the second picture into the wrapped Sink.
1652// This tests that any shortcuts we may take while recording that second picture are legal.
1653Error ViaSecondPicture::draw(
1654 const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
mtkleina16e69e2015-05-05 11:38:45 -07001655 auto size = src.size();
msarett62d3b102015-12-10 15:14:27 -08001656 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) -> Error {
mtkleina16e69e2015-05-05 11:38:45 -07001657 SkPictureRecorder recorder;
reedca2622b2016-03-18 07:25:55 -07001658 sk_sp<SkPicture> pic;
mtkleina16e69e2015-05-05 11:38:45 -07001659 for (int i = 0; i < 2; i++) {
1660 Error err = src.draw(recorder.beginRecording(SkIntToScalar(size.width()),
1661 SkIntToScalar(size.height())));
1662 if (!err.isEmpty()) {
1663 return err;
mtkleinb7e8d692015-04-07 08:30:32 -07001664 }
reedca2622b2016-03-18 07:25:55 -07001665 pic = recorder.finishRecordingAsPicture();
mtkleinb7e8d692015-04-07 08:30:32 -07001666 }
mtkleina16e69e2015-05-05 11:38:45 -07001667 canvas->drawPicture(pic);
mtklein4a34ecb2016-01-08 10:19:35 -08001668 return check_against_reference(bitmap, src, fSink);
mtkleina16e69e2015-05-05 11:38:45 -07001669 });
mtkleinb7e8d692015-04-07 08:30:32 -07001670}
1671
mtkleind31c13d2015-05-05 12:59:56 -07001672/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1673
mtklein6fbf4b32015-05-06 11:35:40 -07001674// Draw the Src twice. This can help exercise caching.
1675Error ViaTwice::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
msarett62d3b102015-12-10 15:14:27 -08001676 return draw_to_canvas(fSink, bitmap, stream, log, src.size(), [&](SkCanvas* canvas) -> Error {
mtklein6fbf4b32015-05-06 11:35:40 -07001677 for (int i = 0; i < 2; i++) {
1678 SkAutoCanvasRestore acr(canvas, true/*save now*/);
1679 canvas->clear(SK_ColorTRANSPARENT);
1680 Error err = src.draw(canvas);
1681 if (err.isEmpty()) {
1682 return err;
1683 }
1684 }
mtklein4a34ecb2016-01-08 10:19:35 -08001685 return check_against_reference(bitmap, src, fSink);
mtklein6fbf4b32015-05-06 11:35:40 -07001686 });
1687}
1688
1689/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1690
mtkleind31c13d2015-05-05 12:59:56 -07001691// This is like SkRecords::Draw, in that it plays back SkRecords ops into a Canvas.
1692// Unlike SkRecords::Draw, it builds a single-op sub-picture out of each Draw-type op.
1693// This is an only-slightly-exaggerated simluation of Blink's Slimming Paint pictures.
1694struct DrawsAsSingletonPictures {
1695 SkCanvas* fCanvas;
mtkleind2baa902015-07-07 09:43:28 -07001696 const SkDrawableList& fDrawables;
mtklein11064df2016-04-06 15:01:57 -07001697 SkRect fBounds;
mtkleind31c13d2015-05-05 12:59:56 -07001698
mtkleind31c13d2015-05-05 12:59:56 -07001699 template <typename T>
1700 void draw(const T& op, SkCanvas* canvas) {
1701 // We must pass SkMatrix::I() as our initial matrix.
1702 // By default SkRecords::Draw() uses the canvas' matrix as its initial matrix,
1703 // which would have the funky effect of applying transforms over and over.
mtkleind2baa902015-07-07 09:43:28 -07001704 SkRecords::Draw d(canvas, nullptr, fDrawables.begin(), fDrawables.count(), &SkMatrix::I());
1705 d(op);
mtkleind31c13d2015-05-05 12:59:56 -07001706 }
1707
mtklein449d9b72015-09-28 10:33:02 -07001708 // Draws get their own picture.
mtkleind31c13d2015-05-05 12:59:56 -07001709 template <typename T>
mtklein449d9b72015-09-28 10:33:02 -07001710 SK_WHEN(T::kTags & SkRecords::kDraw_Tag, void) operator()(const T& op) {
mtkleind31c13d2015-05-05 12:59:56 -07001711 SkPictureRecorder rec;
mtklein11064df2016-04-06 15:01:57 -07001712 this->draw(op, rec.beginRecording(fBounds));
reedca2622b2016-03-18 07:25:55 -07001713 sk_sp<SkPicture> pic(rec.finishRecordingAsPicture());
mtkleind31c13d2015-05-05 12:59:56 -07001714 fCanvas->drawPicture(pic);
1715 }
1716
mtklein449d9b72015-09-28 10:33:02 -07001717 // We'll just issue non-draws directly.
mtkleind31c13d2015-05-05 12:59:56 -07001718 template <typename T>
mtklein449d9b72015-09-28 10:33:02 -07001719 skstd::enable_if_t<!(T::kTags & SkRecords::kDraw_Tag), void> operator()(const T& op) {
1720 this->draw(op, fCanvas);
1721 }
mtkleind31c13d2015-05-05 12:59:56 -07001722};
1723
mtkleind31c13d2015-05-05 12:59:56 -07001724// Record Src into a picture, then record it into a macro picture with a sub-picture for each draw.
1725// Then play back that macro picture into our wrapped sink.
1726Error ViaSingletonPictures::draw(
1727 const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
1728 auto size = src.size();
msarett62d3b102015-12-10 15:14:27 -08001729 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) -> Error {
mtkleind31c13d2015-05-05 12:59:56 -07001730 // Use low-level (Skia-private) recording APIs so we can read the SkRecord.
1731 SkRecord skr;
1732 SkRecorder recorder(&skr, size.width(), size.height());
1733 Error err = src.draw(&recorder);
1734 if (!err.isEmpty()) {
1735 return err;
1736 }
1737
1738 // Record our macro-picture, with each draw op as its own sub-picture.
1739 SkPictureRecorder macroRec;
1740 SkCanvas* macroCanvas = macroRec.beginRecording(SkIntToScalar(size.width()),
1741 SkIntToScalar(size.height()));
mtkleind2baa902015-07-07 09:43:28 -07001742
1743 SkAutoTDelete<SkDrawableList> drawables(recorder.detachDrawableList());
1744 const SkDrawableList empty;
1745
1746 DrawsAsSingletonPictures drawsAsSingletonPictures = {
1747 macroCanvas,
1748 drawables ? *drawables : empty,
mtklein11064df2016-04-06 15:01:57 -07001749 SkRect::MakeWH((SkScalar)size.width(), (SkScalar)size.height()),
mtkleind2baa902015-07-07 09:43:28 -07001750 };
mtkleinc6ad06a2015-08-19 09:51:00 -07001751 for (int i = 0; i < skr.count(); i++) {
mtklein343a63d2016-03-22 11:46:53 -07001752 skr.visit(i, drawsAsSingletonPictures);
mtkleind31c13d2015-05-05 12:59:56 -07001753 }
reedca2622b2016-03-18 07:25:55 -07001754 sk_sp<SkPicture> macroPic(macroRec.finishRecordingAsPicture());
mtkleind31c13d2015-05-05 12:59:56 -07001755
1756 canvas->drawPicture(macroPic);
mtklein4a34ecb2016-01-08 10:19:35 -08001757 return check_against_reference(bitmap, src, fSink);
mtkleind31c13d2015-05-05 12:59:56 -07001758 });
1759}
1760
mtklein9c5052f2016-08-06 12:51:51 -07001761/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
1762
1763Error ViaLite::draw(const Src& src, SkBitmap* bitmap, SkWStream* stream, SkString* log) const {
1764 auto size = src.size();
mtklein8bbbb692016-08-15 12:56:00 -07001765 SkRect bounds = {0,0, (SkScalar)size.width(), (SkScalar)size.height()};
mtklein9c5052f2016-08-06 12:51:51 -07001766 return draw_to_canvas(fSink, bitmap, stream, log, size, [&](SkCanvas* canvas) -> Error {
mtklein8bbbb692016-08-15 12:56:00 -07001767 sk_sp<SkLiteDL> dl = SkLiteDL::New(bounds);
1768
1769 SkLiteRecorder rec;
1770 rec.reset(dl.get());
1771
1772 Error err = src.draw(&rec);
mtklein9c5052f2016-08-06 12:51:51 -07001773 if (!err.isEmpty()) {
1774 return err;
1775 }
mtklein8bbbb692016-08-15 12:56:00 -07001776 dl->draw(canvas);
mtklein03591a72016-08-07 13:35:46 -07001777 return check_against_reference(bitmap, src, fSink);
mtklein9c5052f2016-08-06 12:51:51 -07001778 });
1779}
1780
mtklein748ca3b2015-01-15 10:56:12 -08001781} // namespace DM