blob: 91fb6897b5875fb6d2db3c5fe636105c97a48b16 [file] [log] [blame]
halcanarya096d7a2015-03-27 12:16:53 -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
8#include "Resources.h"
9#include "SkBitmap.h"
10#include "SkCodec.h"
msarette6dd0042015-10-09 11:07:34 -070011#include "SkData.h"
halcanarya096d7a2015-03-27 12:16:53 -070012#include "SkMD5.h"
scroggob636b452015-07-22 07:16:20 -070013#include "SkRandom.h"
msarettb32758a2015-08-18 13:22:46 -070014#include "SkScaledCodec.h"
halcanarya096d7a2015-03-27 12:16:53 -070015#include "Test.h"
16
17static SkStreamAsset* resource(const char path[]) {
18 SkString fullPath = GetResourcePath(path);
19 return SkStream::NewFromFile(fullPath.c_str());
20}
21
22static void md5(const SkBitmap& bm, SkMD5::Digest* digest) {
23 SkAutoLockPixels autoLockPixels(bm);
24 SkASSERT(bm.getPixels());
25 SkMD5 md5;
26 size_t rowLen = bm.info().bytesPerPixel() * bm.width();
27 for (int y = 0; y < bm.height(); ++y) {
28 md5.update(static_cast<uint8_t*>(bm.getAddr(0, y)), rowLen);
29 }
30 md5.finish(*digest);
31}
32
scroggo9b2cdbf42015-07-10 12:07:02 -070033/**
34 * Compute the digest for bm and compare it to a known good digest.
35 * @param r Reporter to assert that bm's digest matches goodDigest.
36 * @param goodDigest The known good digest to compare to.
37 * @param bm The bitmap to test.
38 */
39static void compare_to_good_digest(skiatest::Reporter* r, const SkMD5::Digest& goodDigest,
40 const SkBitmap& bm) {
41 SkMD5::Digest digest;
42 md5(bm, &digest);
43 REPORTER_ASSERT(r, digest == goodDigest);
44}
45
scroggod1bc5742015-08-12 08:31:44 -070046/**
47 * Test decoding an SkCodec to a particular SkImageInfo.
48 *
halcanary96fcdcc2015-08-27 07:41:13 -070049 * Calling getPixels(info) should return expectedResult, and if goodDigest is non nullptr,
scroggod1bc5742015-08-12 08:31:44 -070050 * the resulting decode should match.
51 */
52static void test_info(skiatest::Reporter* r, SkCodec* codec, const SkImageInfo& info,
53 SkCodec::Result expectedResult, const SkMD5::Digest* goodDigest) {
54 SkBitmap bm;
55 bm.allocPixels(info);
56 SkAutoLockPixels autoLockPixels(bm);
57
58 SkCodec::Result result = codec->getPixels(info, bm.getPixels(), bm.rowBytes());
59 REPORTER_ASSERT(r, result == expectedResult);
60
61 if (goodDigest) {
62 compare_to_good_digest(r, *goodDigest, bm);
63 }
64}
65
scroggob636b452015-07-22 07:16:20 -070066SkIRect generate_random_subset(SkRandom* rand, int w, int h) {
67 SkIRect rect;
68 do {
69 rect.fLeft = rand->nextRangeU(0, w);
70 rect.fTop = rand->nextRangeU(0, h);
71 rect.fRight = rand->nextRangeU(0, w);
72 rect.fBottom = rand->nextRangeU(0, h);
73 rect.sort();
74 } while (rect.isEmpty());
75 return rect;
76}
77
msarettcc7f3052015-10-05 14:20:27 -070078static void test_codec(skiatest::Reporter* r, SkCodec* codec, SkBitmap& bm, const SkImageInfo& info,
msarette6dd0042015-10-09 11:07:34 -070079 const SkISize& size, bool supports565, SkCodec::Result expectedResult,
80 SkMD5::Digest* digest, const SkMD5::Digest* goodDigest) {
81
halcanarya096d7a2015-03-27 12:16:53 -070082 REPORTER_ASSERT(r, info.dimensions() == size);
halcanarya096d7a2015-03-27 12:16:53 -070083 bm.allocPixels(info);
84 SkAutoLockPixels autoLockPixels(bm);
msarettcc7f3052015-10-05 14:20:27 -070085
86 SkCodec::Result result = codec->getPixels(info, bm.getPixels(), bm.rowBytes());
msarette6dd0042015-10-09 11:07:34 -070087 REPORTER_ASSERT(r, result == expectedResult);
halcanarya096d7a2015-03-27 12:16:53 -070088
msarettcc7f3052015-10-05 14:20:27 -070089 md5(bm, digest);
90 if (goodDigest) {
91 REPORTER_ASSERT(r, *digest == *goodDigest);
92 }
halcanarya096d7a2015-03-27 12:16:53 -070093
msarett8ff6ca62015-09-18 12:06:04 -070094 {
95 // Test decoding to 565
96 SkImageInfo info565 = info.makeColorType(kRGB_565_SkColorType);
msarette6dd0042015-10-09 11:07:34 -070097 SkCodec::Result expected565 = (supports565 && info.alphaType() == kOpaque_SkAlphaType) ?
98 expectedResult : SkCodec::kInvalidConversion;
99 test_info(r, codec, info565, expected565, nullptr);
msarett8ff6ca62015-09-18 12:06:04 -0700100 }
101
102 // Verify that re-decoding gives the same result. It is interesting to check this after
103 // a decode to 565, since choosing to decode to 565 may result in some of the decode
104 // options being modified. These options should return to their defaults on another
105 // decode to kN32, so the new digest should match the old digest.
msarette6dd0042015-10-09 11:07:34 -0700106 test_info(r, codec, info, expectedResult, digest);
scroggod1bc5742015-08-12 08:31:44 -0700107
108 {
109 // Check alpha type conversions
110 if (info.alphaType() == kOpaque_SkAlphaType) {
111 test_info(r, codec, info.makeAlphaType(kUnpremul_SkAlphaType),
halcanary96fcdcc2015-08-27 07:41:13 -0700112 SkCodec::kInvalidConversion, nullptr);
scroggod1bc5742015-08-12 08:31:44 -0700113 test_info(r, codec, info.makeAlphaType(kPremul_SkAlphaType),
halcanary96fcdcc2015-08-27 07:41:13 -0700114 SkCodec::kInvalidConversion, nullptr);
scroggod1bc5742015-08-12 08:31:44 -0700115 } else {
116 // Decoding to opaque should fail
117 test_info(r, codec, info.makeAlphaType(kOpaque_SkAlphaType),
halcanary96fcdcc2015-08-27 07:41:13 -0700118 SkCodec::kInvalidConversion, nullptr);
scroggod1bc5742015-08-12 08:31:44 -0700119 SkAlphaType otherAt = info.alphaType();
120 if (kPremul_SkAlphaType == otherAt) {
121 otherAt = kUnpremul_SkAlphaType;
122 } else {
123 otherAt = kPremul_SkAlphaType;
124 }
125 // The other non-opaque alpha type should always succeed, but not match.
msarette6dd0042015-10-09 11:07:34 -0700126 test_info(r, codec, info.makeAlphaType(otherAt), expectedResult, nullptr);
scroggod1bc5742015-08-12 08:31:44 -0700127 }
128 }
msarettcc7f3052015-10-05 14:20:27 -0700129}
130
scroggo2c3b2182015-10-09 08:40:59 -0700131// FIXME: SkScaledCodec is currently only supported for types used by BRD
132// skbug.com/4428
133static bool supports_scaled_codec(const char path[]) {
134 static const char* const exts[] = {
135 "jpg", "jpeg", "png", "webp"
136 "JPG", "JPEG", "PNG", "WEBP"
137 };
138
139 for (uint32_t i = 0; i < SK_ARRAY_COUNT(exts); i++) {
140 if (SkStrEndsWith(path, exts[i])) {
141 return true;
142 }
143 }
144 return false;
145}
146
msarettcc7f3052015-10-05 14:20:27 -0700147static void check(skiatest::Reporter* r,
148 const char path[],
149 SkISize size,
150 bool supportsScanlineDecoding,
151 bool supportsSubsetDecoding,
msarette6dd0042015-10-09 11:07:34 -0700152 bool supports565 = true,
153 bool supportsIncomplete = true) {
msarettcc7f3052015-10-05 14:20:27 -0700154
155 SkAutoTDelete<SkStream> stream(resource(path));
156 if (!stream) {
157 SkDebugf("Missing resource '%s'\n", path);
158 return;
159 }
msarette6dd0042015-10-09 11:07:34 -0700160
161 SkAutoTDelete<SkCodec> codec(nullptr);
162 bool isIncomplete = supportsIncomplete;
163 if (isIncomplete) {
164 size_t size = stream->getLength();
165 SkAutoTUnref<SkData> data((SkData::NewFromStream(stream, 2 * size / 3)));
166 codec.reset(SkCodec::NewFromData(data));
167 } else {
168 codec.reset(SkCodec::NewFromStream(stream.detach()));
169 }
msarettcc7f3052015-10-05 14:20:27 -0700170 if (!codec) {
171 ERRORF(r, "Unable to decode '%s'", path);
172 return;
173 }
174
175 // Test full image decodes with SkCodec
176 SkMD5::Digest codecDigest;
177 SkImageInfo info = codec->getInfo().makeColorType(kN32_SkColorType);
178 SkBitmap bm;
msarette6dd0042015-10-09 11:07:34 -0700179 SkCodec::Result expectedResult = isIncomplete ? SkCodec::kIncompleteInput : SkCodec::kSuccess;
180 test_codec(r, codec, bm, info, size, supports565, expectedResult, &codecDigest, nullptr);
scroggod1bc5742015-08-12 08:31:44 -0700181
182 // Scanline decoding follows.
msarettcc7f3052015-10-05 14:20:27 -0700183 // Need to call startScanlineDecode() first.
scroggo46c57472015-09-30 08:57:13 -0700184 REPORTER_ASSERT(r, codec->getScanlines(bm.getAddr(0, 0), 1, 0)
msarette6dd0042015-10-09 11:07:34 -0700185 == 0);
scroggo46c57472015-09-30 08:57:13 -0700186 REPORTER_ASSERT(r, codec->skipScanlines(1)
msarette6dd0042015-10-09 11:07:34 -0700187 == 0);
scroggo46c57472015-09-30 08:57:13 -0700188
189 const SkCodec::Result startResult = codec->startScanlineDecode(info);
scroggo58421542015-04-01 11:25:20 -0700190 if (supportsScanlineDecoding) {
191 bm.eraseColor(SK_ColorYELLOW);
msarettc0e80c12015-07-01 06:50:35 -0700192
scroggo46c57472015-09-30 08:57:13 -0700193 REPORTER_ASSERT(r, startResult == SkCodec::kSuccess);
scroggo9b2cdbf42015-07-10 12:07:02 -0700194
scroggo58421542015-04-01 11:25:20 -0700195 for (int y = 0; y < info.height(); y++) {
msarette6dd0042015-10-09 11:07:34 -0700196 const int lines = codec->getScanlines(bm.getAddr(0, y), 1, 0);
197 if (!isIncomplete) {
198 REPORTER_ASSERT(r, 1 == lines);
199 }
scroggo58421542015-04-01 11:25:20 -0700200 }
201 // verify that scanline decoding gives the same result.
scroggo46c57472015-09-30 08:57:13 -0700202 if (SkCodec::kTopDown_SkScanlineOrder == codec->getScanlineOrder()) {
msarettcc7f3052015-10-05 14:20:27 -0700203 compare_to_good_digest(r, codecDigest, bm);
msarett5406d6f2015-08-31 06:55:13 -0700204 }
scroggo46c57472015-09-30 08:57:13 -0700205
206 // Cannot continue to decode scanlines beyond the end
207 REPORTER_ASSERT(r, codec->getScanlines(bm.getAddr(0, 0), 1, 0)
msarette6dd0042015-10-09 11:07:34 -0700208 == 0);
scroggo46c57472015-09-30 08:57:13 -0700209
210 // Interrupting a scanline decode with a full decode starts from
211 // scratch
212 REPORTER_ASSERT(r, codec->startScanlineDecode(info) == SkCodec::kSuccess);
msarette6dd0042015-10-09 11:07:34 -0700213 const int lines = codec->getScanlines(bm.getAddr(0, 0), 1, 0);
214 if (!isIncomplete) {
215 REPORTER_ASSERT(r, lines == 1);
216 }
scroggo46c57472015-09-30 08:57:13 -0700217 REPORTER_ASSERT(r, codec->getPixels(bm.info(), bm.getPixels(), bm.rowBytes())
msarette6dd0042015-10-09 11:07:34 -0700218 == expectedResult);
scroggo46c57472015-09-30 08:57:13 -0700219 REPORTER_ASSERT(r, codec->getScanlines(bm.getAddr(0, 0), 1, 0)
msarette6dd0042015-10-09 11:07:34 -0700220 == 0);
scroggo46c57472015-09-30 08:57:13 -0700221 REPORTER_ASSERT(r, codec->skipScanlines(1)
msarette6dd0042015-10-09 11:07:34 -0700222 == 0);
scroggo58421542015-04-01 11:25:20 -0700223 } else {
scroggo46c57472015-09-30 08:57:13 -0700224 REPORTER_ASSERT(r, startResult == SkCodec::kUnimplemented);
halcanarya096d7a2015-03-27 12:16:53 -0700225 }
scroggob636b452015-07-22 07:16:20 -0700226
227 // The rest of this function tests decoding subsets, and will decode an arbitrary number of
228 // random subsets.
229 // Do not attempt to decode subsets of an image of only once pixel, since there is no
230 // meaningful subset.
231 if (size.width() * size.height() == 1) {
232 return;
233 }
234
235 SkRandom rand;
236 SkIRect subset;
237 SkCodec::Options opts;
238 opts.fSubset = &subset;
239 for (int i = 0; i < 5; i++) {
240 subset = generate_random_subset(&rand, size.width(), size.height());
241 SkASSERT(!subset.isEmpty());
242 const bool supported = codec->getValidSubset(&subset);
243 REPORTER_ASSERT(r, supported == supportsSubsetDecoding);
244
245 SkImageInfo subsetInfo = info.makeWH(subset.width(), subset.height());
246 SkBitmap bm;
247 bm.allocPixels(subsetInfo);
248 const SkCodec::Result result = codec->getPixels(bm.info(), bm.getPixels(), bm.rowBytes(),
halcanary96fcdcc2015-08-27 07:41:13 -0700249 &opts, nullptr, nullptr);
scroggob636b452015-07-22 07:16:20 -0700250
251 if (supportsSubsetDecoding) {
msarette6dd0042015-10-09 11:07:34 -0700252 REPORTER_ASSERT(r, result == expectedResult);
scroggob636b452015-07-22 07:16:20 -0700253 // Webp is the only codec that supports subsets, and it will have modified the subset
254 // to have even left/top.
255 REPORTER_ASSERT(r, SkIsAlign2(subset.fLeft) && SkIsAlign2(subset.fTop));
256 } else {
257 // No subsets will work.
258 REPORTER_ASSERT(r, result == SkCodec::kUnimplemented);
259 }
260 }
msarettcc7f3052015-10-05 14:20:27 -0700261
262 // SkScaledCodec tests
scroggo2c3b2182015-10-09 08:40:59 -0700263 if ((supportsScanlineDecoding || supportsSubsetDecoding) && supports_scaled_codec(path)) {
264
msarettcc7f3052015-10-05 14:20:27 -0700265 SkAutoTDelete<SkStream> stream(resource(path));
266 if (!stream) {
267 SkDebugf("Missing resource '%s'\n", path);
268 return;
269 }
msarette6dd0042015-10-09 11:07:34 -0700270
271 SkAutoTDelete<SkCodec> codec(nullptr);
272 if (isIncomplete) {
273 size_t size = stream->getLength();
274 SkAutoTUnref<SkData> data((SkData::NewFromStream(stream, 2 * size / 3)));
275 codec.reset(SkScaledCodec::NewFromData(data));
276 } else {
277 codec.reset(SkScaledCodec::NewFromStream(stream.detach()));
278 }
msarettcc7f3052015-10-05 14:20:27 -0700279 if (!codec) {
280 ERRORF(r, "Unable to decode '%s'", path);
281 return;
282 }
283
284 SkBitmap bm;
285 SkMD5::Digest scaledCodecDigest;
msarette6dd0042015-10-09 11:07:34 -0700286 test_codec(r, codec, bm, info, size, supports565, expectedResult, &scaledCodecDigest,
287 &codecDigest);
288 }
289
290 // If we've just tested incomplete decodes, let's run the same test again on full decodes.
291 if (isIncomplete) {
292 check(r, path, size, supportsScanlineDecoding, supportsSubsetDecoding, supports565, false);
msarettcc7f3052015-10-05 14:20:27 -0700293 }
halcanarya096d7a2015-03-27 12:16:53 -0700294}
295
296DEF_TEST(Codec, r) {
297 // WBMP
msarett99f567e2015-08-05 12:58:26 -0700298 check(r, "mandrill.wbmp", SkISize::Make(512, 512), true, false);
halcanarya096d7a2015-03-27 12:16:53 -0700299
scroggo6f5e6192015-06-18 12:53:43 -0700300 // WEBP
scroggob636b452015-07-22 07:16:20 -0700301 check(r, "baby_tux.webp", SkISize::Make(386, 395), false, true);
302 check(r, "color_wheel.webp", SkISize::Make(128, 128), false, true);
303 check(r, "yellow_rose.webp", SkISize::Make(400, 301), false, true);
scroggo6f5e6192015-06-18 12:53:43 -0700304
halcanarya096d7a2015-03-27 12:16:53 -0700305 // BMP
msarett5406d6f2015-08-31 06:55:13 -0700306 check(r, "randPixels.bmp", SkISize::Make(8, 8), true, false);
halcanarya096d7a2015-03-27 12:16:53 -0700307
308 // ICO
msarette6dd0042015-10-09 11:07:34 -0700309 // FIXME: We are not ready to test incomplete ICOs
msarett68b204e2015-04-01 12:09:21 -0700310 // These two tests examine interestingly different behavior:
311 // Decodes an embedded BMP image
msarette6dd0042015-10-09 11:07:34 -0700312 check(r, "color_wheel.ico", SkISize::Make(128, 128), false, false, true, false);
msarett68b204e2015-04-01 12:09:21 -0700313 // Decodes an embedded PNG image
msarette6dd0042015-10-09 11:07:34 -0700314 check(r, "google_chrome.ico", SkISize::Make(256, 256), false, false, true, false);
halcanarya096d7a2015-03-27 12:16:53 -0700315
msarett438b2ad2015-04-09 12:43:10 -0700316 // GIF
msarette6dd0042015-10-09 11:07:34 -0700317 // FIXME: We are not ready to test incomplete GIFs
318 check(r, "box.gif", SkISize::Make(200, 55), true, false, true, false);
319 check(r, "color_wheel.gif", SkISize::Make(128, 128), true, false, true, false);
320 // randPixels.gif is too small to test incomplete
321 check(r, "randPixels.gif", SkISize::Make(8, 8), true, false, true, false);
msarett438b2ad2015-04-09 12:43:10 -0700322
msarette16b04a2015-04-15 07:32:19 -0700323 // JPG
scroggocc2feb12015-08-14 08:32:46 -0700324 check(r, "CMYK.jpg", SkISize::Make(642, 516), true, false, false);
scroggob636b452015-07-22 07:16:20 -0700325 check(r, "color_wheel.jpg", SkISize::Make(128, 128), true, false);
msarette6dd0042015-10-09 11:07:34 -0700326 // grayscale.jpg is too small to test incomplete
327 check(r, "grayscale.jpg", SkISize::Make(128, 128), true, false, true, false);
scroggob636b452015-07-22 07:16:20 -0700328 check(r, "mandrill_512_q075.jpg", SkISize::Make(512, 512), true, false);
msarette6dd0042015-10-09 11:07:34 -0700329 // randPixels.jpg is too small to test incomplete
330 check(r, "randPixels.jpg", SkISize::Make(8, 8), true, false, true, false);
msarette16b04a2015-04-15 07:32:19 -0700331
halcanarya096d7a2015-03-27 12:16:53 -0700332 // PNG
msarette6dd0042015-10-09 11:07:34 -0700333 check(r, "arrow.png", SkISize::Make(187, 312), true, false, true, false);
334 check(r, "baby_tux.png", SkISize::Make(240, 246), true, false, true, false);
335 check(r, "color_wheel.png", SkISize::Make(128, 128), true, false, true, false);
336 check(r, "half-transparent-white-pixel.png", SkISize::Make(1, 1), true, false, true, false);
337 check(r, "mandrill_128.png", SkISize::Make(128, 128), true, false, true, false);
338 check(r, "mandrill_16.png", SkISize::Make(16, 16), true, false, true, false);
339 check(r, "mandrill_256.png", SkISize::Make(256, 256), true, false, true, false);
340 check(r, "mandrill_32.png", SkISize::Make(32, 32), true, false, true, false);
341 check(r, "mandrill_512.png", SkISize::Make(512, 512), true, false, true, false);
342 check(r, "mandrill_64.png", SkISize::Make(64, 64), true, false, true, false);
343 check(r, "plane.png", SkISize::Make(250, 126), true, false, true, false);
344 // FIXME: We are not ready to test incomplete interlaced pngs
345 check(r, "plane_interlaced.png", SkISize::Make(250, 126), true, false, true, false);
346 check(r, "randPixels.png", SkISize::Make(8, 8), true, false, true, false);
347 check(r, "yellow_rose.png", SkISize::Make(400, 301), true, false, true, false);
halcanarya096d7a2015-03-27 12:16:53 -0700348}
scroggo0a7e69c2015-04-03 07:22:22 -0700349
scroggo46c57472015-09-30 08:57:13 -0700350// Test interlaced PNG in stripes, similar to DM's kStripe_Mode
351DEF_TEST(Codec_stripes, r) {
352 const char * path = "plane_interlaced.png";
353 SkAutoTDelete<SkStream> stream(resource(path));
354 if (!stream) {
355 SkDebugf("Missing resource '%s'\n", path);
356 }
357
358 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromStream(stream.detach()));
359 REPORTER_ASSERT(r, codec);
360
361 if (!codec) {
362 return;
363 }
364
365 switch (codec->getScanlineOrder()) {
366 case SkCodec::kBottomUp_SkScanlineOrder:
367 case SkCodec::kOutOfOrder_SkScanlineOrder:
368 ERRORF(r, "This scanline order will not match the original.");
369 return;
370 default:
371 break;
372 }
373
374 // Baseline for what the image should look like, using N32.
375 const SkImageInfo info = codec->getInfo().makeColorType(kN32_SkColorType);
376
377 SkBitmap bm;
378 bm.allocPixels(info);
379 SkAutoLockPixels autoLockPixels(bm);
380 SkCodec::Result result = codec->getPixels(info, bm.getPixels(), bm.rowBytes());
381 REPORTER_ASSERT(r, result == SkCodec::kSuccess);
382
383 SkMD5::Digest digest;
384 md5(bm, &digest);
385
386 // Now decode in stripes
387 const int height = info.height();
388 const int numStripes = 4;
389 int stripeHeight;
390 int remainingLines;
391 SkTDivMod(height, numStripes, &stripeHeight, &remainingLines);
392
393 bm.eraseColor(SK_ColorYELLOW);
394
395 result = codec->startScanlineDecode(info);
396 REPORTER_ASSERT(r, result == SkCodec::kSuccess);
397
398 // Odd stripes
399 for (int i = 1; i < numStripes; i += 2) {
400 // Skip the even stripes
msarette6dd0042015-10-09 11:07:34 -0700401 bool skipResult = codec->skipScanlines(stripeHeight);
402 REPORTER_ASSERT(r, skipResult);
scroggo46c57472015-09-30 08:57:13 -0700403
msarette6dd0042015-10-09 11:07:34 -0700404 int linesDecoded = codec->getScanlines(bm.getAddr(0, i * stripeHeight), stripeHeight,
scroggo46c57472015-09-30 08:57:13 -0700405 bm.rowBytes());
msarette6dd0042015-10-09 11:07:34 -0700406 REPORTER_ASSERT(r, linesDecoded == stripeHeight);
scroggo46c57472015-09-30 08:57:13 -0700407 }
408
409 // Even stripes
410 result = codec->startScanlineDecode(info);
411 REPORTER_ASSERT(r, result == SkCodec::kSuccess);
412
413 for (int i = 0; i < numStripes; i += 2) {
msarette6dd0042015-10-09 11:07:34 -0700414 int linesDecoded = codec->getScanlines(bm.getAddr(0, i * stripeHeight), stripeHeight,
scroggo46c57472015-09-30 08:57:13 -0700415 bm.rowBytes());
msarette6dd0042015-10-09 11:07:34 -0700416 REPORTER_ASSERT(r, linesDecoded == stripeHeight);
scroggo46c57472015-09-30 08:57:13 -0700417
418 // Skip the odd stripes
419 if (i + 1 < numStripes) {
msarette6dd0042015-10-09 11:07:34 -0700420 bool skipResult = codec->skipScanlines(stripeHeight);
421 REPORTER_ASSERT(r, skipResult);
scroggo46c57472015-09-30 08:57:13 -0700422 }
423 }
424
425 // Remainder at the end
426 if (remainingLines > 0) {
427 result = codec->startScanlineDecode(info);
428 REPORTER_ASSERT(r, result == SkCodec::kSuccess);
429
msarette6dd0042015-10-09 11:07:34 -0700430 bool skipResult = codec->skipScanlines(height - remainingLines);
431 REPORTER_ASSERT(r, skipResult);
scroggo46c57472015-09-30 08:57:13 -0700432
msarette6dd0042015-10-09 11:07:34 -0700433 int linesDecoded = codec->getScanlines(bm.getAddr(0, height - remainingLines),
scroggo46c57472015-09-30 08:57:13 -0700434 remainingLines, bm.rowBytes());
msarette6dd0042015-10-09 11:07:34 -0700435 REPORTER_ASSERT(r, linesDecoded == remainingLines);
scroggo46c57472015-09-30 08:57:13 -0700436 }
437
438 compare_to_good_digest(r, digest, bm);
439}
440
scroggo0a7e69c2015-04-03 07:22:22 -0700441static void test_invalid_stream(skiatest::Reporter* r, const void* stream, size_t len) {
scroggo2c3b2182015-10-09 08:40:59 -0700442 // Neither of these calls should return a codec. Bots should catch us if we leaked anything.
scroggo0a7e69c2015-04-03 07:22:22 -0700443 SkCodec* codec = SkCodec::NewFromStream(new SkMemoryStream(stream, len, false));
scroggo2c3b2182015-10-09 08:40:59 -0700444 REPORTER_ASSERT(r, !codec);
445
446 codec = SkScaledCodec::NewFromStream(new SkMemoryStream(stream, len, false));
scroggo0a7e69c2015-04-03 07:22:22 -0700447 REPORTER_ASSERT(r, !codec);
448}
449
450// Ensure that SkCodec::NewFromStream handles freeing the passed in SkStream,
451// even on failure. Test some bad streams.
452DEF_TEST(Codec_leaks, r) {
453 // No codec should claim this as their format, so this tests SkCodec::NewFromStream.
454 const char nonSupportedStream[] = "hello world";
455 // The other strings should look like the beginning of a file type, so we'll call some
456 // internal version of NewFromStream, which must also delete the stream on failure.
457 const unsigned char emptyPng[] = { 0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a };
458 const unsigned char emptyJpeg[] = { 0xFF, 0xD8, 0xFF };
459 const char emptyWebp[] = "RIFF1234WEBPVP";
460 const char emptyBmp[] = { 'B', 'M' };
461 const char emptyIco[] = { '\x00', '\x00', '\x01', '\x00' };
462 const char emptyGif[] = "GIFVER";
463
464 test_invalid_stream(r, nonSupportedStream, sizeof(nonSupportedStream));
465 test_invalid_stream(r, emptyPng, sizeof(emptyPng));
466 test_invalid_stream(r, emptyJpeg, sizeof(emptyJpeg));
467 test_invalid_stream(r, emptyWebp, sizeof(emptyWebp));
468 test_invalid_stream(r, emptyBmp, sizeof(emptyBmp));
469 test_invalid_stream(r, emptyIco, sizeof(emptyIco));
470 test_invalid_stream(r, emptyGif, sizeof(emptyGif));
471}
msarette16b04a2015-04-15 07:32:19 -0700472
scroggo2c3b2182015-10-09 08:40:59 -0700473DEF_TEST(Codec_null, r) {
474 // Attempting to create an SkCodec or an SkScaledCodec with null should not
475 // crash.
476 SkCodec* codec = SkCodec::NewFromStream(nullptr);
477 REPORTER_ASSERT(r, !codec);
478
479 codec = SkScaledCodec::NewFromStream(nullptr);
480 REPORTER_ASSERT(r, !codec);
481}
482
msarette16b04a2015-04-15 07:32:19 -0700483static void test_dimensions(skiatest::Reporter* r, const char path[]) {
484 // Create the codec from the resource file
485 SkAutoTDelete<SkStream> stream(resource(path));
486 if (!stream) {
487 SkDebugf("Missing resource '%s'\n", path);
488 return;
489 }
msarettb32758a2015-08-18 13:22:46 -0700490 SkAutoTDelete<SkCodec> codec(SkScaledCodec::NewFromStream(stream.detach()));
msarette16b04a2015-04-15 07:32:19 -0700491 if (!codec) {
492 ERRORF(r, "Unable to create codec '%s'", path);
493 return;
494 }
495
496 // Check that the decode is successful for a variety of scales
msarettb32758a2015-08-18 13:22:46 -0700497 for (float scale = 0.05f; scale < 2.0f; scale += 0.05f) {
msarette16b04a2015-04-15 07:32:19 -0700498 // Scale the output dimensions
499 SkISize scaledDims = codec->getScaledDimensions(scale);
msarettb32758a2015-08-18 13:22:46 -0700500 SkImageInfo scaledInfo = codec->getInfo()
501 .makeWH(scaledDims.width(), scaledDims.height())
502 .makeColorType(kN32_SkColorType);
msarette16b04a2015-04-15 07:32:19 -0700503
504 // Set up for the decode
505 size_t rowBytes = scaledDims.width() * sizeof(SkPMColor);
506 size_t totalBytes = scaledInfo.getSafeSize(rowBytes);
507 SkAutoTMalloc<SkPMColor> pixels(totalBytes);
508
scroggoeb602a52015-07-09 08:16:03 -0700509 SkCodec::Result result =
halcanary96fcdcc2015-08-27 07:41:13 -0700510 codec->getPixels(scaledInfo, pixels.get(), rowBytes, nullptr, nullptr, nullptr);
scroggoeb602a52015-07-09 08:16:03 -0700511 REPORTER_ASSERT(r, SkCodec::kSuccess == result);
msarette16b04a2015-04-15 07:32:19 -0700512 }
513}
514
515// Ensure that onGetScaledDimensions returns valid image dimensions to use for decodes
516DEF_TEST(Codec_Dimensions, r) {
517 // JPG
518 test_dimensions(r, "CMYK.jpg");
519 test_dimensions(r, "color_wheel.jpg");
520 test_dimensions(r, "grayscale.jpg");
521 test_dimensions(r, "mandrill_512_q075.jpg");
522 test_dimensions(r, "randPixels.jpg");
msarettb32758a2015-08-18 13:22:46 -0700523
524 // Decoding small images with very large scaling factors is a potential
525 // source of bugs and crashes. We disable these tests in Gold because
526 // tiny images are not very useful to look at.
527 // Here we make sure that we do not crash or access illegal memory when
528 // performing scaled decodes on small images.
529 test_dimensions(r, "1x1.png");
530 test_dimensions(r, "2x2.png");
531 test_dimensions(r, "3x3.png");
532 test_dimensions(r, "3x1.png");
533 test_dimensions(r, "1x1.png");
534 test_dimensions(r, "16x1.png");
535 test_dimensions(r, "1x16.png");
536 test_dimensions(r, "mandrill_16.png");
537
msarette16b04a2015-04-15 07:32:19 -0700538}
539
msarettd0375bc2015-08-12 08:08:56 -0700540static void test_invalid(skiatest::Reporter* r, const char path[]) {
msarett4b17fa32015-04-23 08:53:39 -0700541 SkAutoTDelete<SkStream> stream(resource(path));
542 if (!stream) {
543 SkDebugf("Missing resource '%s'\n", path);
544 return;
545 }
546 SkAutoTDelete<SkCodec> codec(SkCodec::NewFromStream(stream.detach()));
halcanary96fcdcc2015-08-27 07:41:13 -0700547 REPORTER_ASSERT(r, nullptr == codec);
msarett4b17fa32015-04-23 08:53:39 -0700548}
msarette16b04a2015-04-15 07:32:19 -0700549
msarett4b17fa32015-04-23 08:53:39 -0700550DEF_TEST(Codec_Empty, r) {
551 // Test images that should not be able to create a codec
msarettd0375bc2015-08-12 08:08:56 -0700552 test_invalid(r, "empty_images/zero-dims.gif");
553 test_invalid(r, "empty_images/zero-embedded.ico");
554 test_invalid(r, "empty_images/zero-width.bmp");
555 test_invalid(r, "empty_images/zero-height.bmp");
556 test_invalid(r, "empty_images/zero-width.jpg");
557 test_invalid(r, "empty_images/zero-height.jpg");
558 test_invalid(r, "empty_images/zero-width.png");
559 test_invalid(r, "empty_images/zero-height.png");
560 test_invalid(r, "empty_images/zero-width.wbmp");
561 test_invalid(r, "empty_images/zero-height.wbmp");
562 // This image is an ico with an embedded mask-bmp. This is illegal.
563 test_invalid(r, "invalid_images/mask-bmp-ico.ico");
msarett4b17fa32015-04-23 08:53:39 -0700564}
msarett99f567e2015-08-05 12:58:26 -0700565
566static void test_invalid_parameters(skiatest::Reporter* r, const char path[]) {
567 SkAutoTDelete<SkStream> stream(resource(path));
568 if (!stream) {
569 SkDebugf("Missing resource '%s'\n", path);
570 return;
571 }
scroggo46c57472015-09-30 08:57:13 -0700572 SkAutoTDelete<SkCodec> decoder(SkCodec::NewFromStream(stream.detach()));
msarett99f567e2015-08-05 12:58:26 -0700573
574 // This should return kSuccess because kIndex8 is supported.
575 SkPMColor colorStorage[256];
576 int colorCount;
scroggo46c57472015-09-30 08:57:13 -0700577 SkCodec::Result result = decoder->startScanlineDecode(
halcanary96fcdcc2015-08-27 07:41:13 -0700578 decoder->getInfo().makeColorType(kIndex_8_SkColorType), nullptr, colorStorage, &colorCount);
msarett99f567e2015-08-05 12:58:26 -0700579 REPORTER_ASSERT(r, SkCodec::kSuccess == result);
580 // The rest of the test is uninteresting if kIndex8 is not supported
581 if (SkCodec::kSuccess != result) {
582 return;
583 }
584
585 // This should return kInvalidParameters because, in kIndex_8 mode, we must pass in a valid
586 // colorPtr and a valid colorCountPtr.
scroggo46c57472015-09-30 08:57:13 -0700587 result = decoder->startScanlineDecode(
halcanary96fcdcc2015-08-27 07:41:13 -0700588 decoder->getInfo().makeColorType(kIndex_8_SkColorType), nullptr, nullptr, nullptr);
msarett99f567e2015-08-05 12:58:26 -0700589 REPORTER_ASSERT(r, SkCodec::kInvalidParameters == result);
scroggo46c57472015-09-30 08:57:13 -0700590 result = decoder->startScanlineDecode(
msarett99f567e2015-08-05 12:58:26 -0700591 decoder->getInfo().makeColorType(kIndex_8_SkColorType));
592 REPORTER_ASSERT(r, SkCodec::kInvalidParameters == result);
593}
594
595DEF_TEST(Codec_Params, r) {
596 test_invalid_parameters(r, "index8.png");
597 test_invalid_parameters(r, "mandrill.wbmp");
598}