41a8ec060e
Bug: skia: Change-Id: I11ce7858306dfe05162aa84a5f862b1b01c2cee9 Reviewed-on: https://skia-review.googlesource.com/15637 Reviewed-by: Leon Scroggins <scroggo@google.com> Commit-Queue: Matt Sarett <msarett@google.com>
87 lines
2.5 KiB
C++
87 lines
2.5 KiB
C++
/*
|
|
* Copyright 2017 Google Inc.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#include "Resources.h"
|
|
#include "Test.h"
|
|
|
|
#include "SkBitmap.h"
|
|
#include "SkEncodedImageFormat.h"
|
|
#include "SkJpegEncoder.h"
|
|
#include "SkPngEncoder.h"
|
|
#include "SkStream.h"
|
|
|
|
static bool encode(SkEncodedImageFormat format, SkWStream* dst, const SkPixmap& src) {
|
|
switch (format) {
|
|
case SkEncodedImageFormat::kJPEG:
|
|
return SkJpegEncoder::Encode(dst, src, SkJpegEncoder::Options());
|
|
case SkEncodedImageFormat::kPNG:
|
|
return SkPngEncoder::Encode(dst, src, SkPngEncoder::Options());
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
static std::unique_ptr<SkEncoder> make(SkEncodedImageFormat format, SkWStream* dst,
|
|
const SkPixmap& src) {
|
|
switch (format) {
|
|
case SkEncodedImageFormat::kJPEG:
|
|
return SkJpegEncoder::Make(dst, src, SkJpegEncoder::Options());
|
|
case SkEncodedImageFormat::kPNG:
|
|
return SkPngEncoder::Make(dst, src, SkPngEncoder::Options());
|
|
default:
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
static void test_encode(skiatest::Reporter* r, SkEncodedImageFormat format) {
|
|
SkBitmap bitmap;
|
|
bool success = GetResourceAsBitmap("mandrill_128.png", &bitmap);
|
|
if (!success) {
|
|
return;
|
|
}
|
|
|
|
SkPixmap src;
|
|
success = bitmap.peekPixels(&src);
|
|
REPORTER_ASSERT(r, success);
|
|
if (!success) {
|
|
return;
|
|
}
|
|
|
|
SkDynamicMemoryWStream dst0, dst1, dst2, dst3;
|
|
success = encode(format, &dst0, src);
|
|
REPORTER_ASSERT(r, success);
|
|
|
|
auto encoder1 = make(format, &dst1, src);
|
|
for (int i = 0; i < src.height(); i++) {
|
|
success = encoder1->encodeRows(1);
|
|
REPORTER_ASSERT(r, success);
|
|
}
|
|
|
|
auto encoder2 = make(format, &dst2, src);
|
|
for (int i = 0; i < src.height(); i+=3) {
|
|
success = encoder2->encodeRows(3);
|
|
REPORTER_ASSERT(r, success);
|
|
}
|
|
|
|
auto encoder3 = make(format, &dst3, src);
|
|
success = encoder3->encodeRows(200);
|
|
REPORTER_ASSERT(r, success);
|
|
|
|
sk_sp<SkData> data0 = dst0.detachAsData();
|
|
sk_sp<SkData> data1 = dst1.detachAsData();
|
|
sk_sp<SkData> data2 = dst2.detachAsData();
|
|
sk_sp<SkData> data3 = dst3.detachAsData();
|
|
REPORTER_ASSERT(r, data0->equals(data1.get()));
|
|
REPORTER_ASSERT(r, data0->equals(data2.get()));
|
|
REPORTER_ASSERT(r, data0->equals(data3.get()));
|
|
}
|
|
|
|
DEF_TEST(Encoder, r) {
|
|
test_encode(r, SkEncodedImageFormat::kJPEG);
|
|
test_encode(r, SkEncodedImageFormat::kPNG);
|
|
}
|