skia2/tests/CachedDecodingPixelRefTest.cpp
halcanary@google.com ad04eb49f5 Add SkImageGenerator Interface
-   Add SkDiscardablePixelRef class that uses SkDiscardableMemory and
    a SkImageGenerator.

-   Add SkDecodingImageGenerator class as an example of a
    SkImageGenerator.

-   Add DecodingImageGenerator unit test.

-   Add SkBasicDiscardableMemory implmentation for unit tests only.

R=reed@google.com, scroggo@google.com

Review URL: https://codereview.chromium.org/74793011

git-svn-id: http://skia.googlecode.com/svn/trunk@12341 2bbb7eff-a529-9590-31e7-b0007b416f81
2013-11-21 15:32:08 +00:00

280 lines
9.5 KiB
C++

/*
* Copyright 2013 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "SkBitmap.h"
#include "SkCanvas.h"
#include "SkData.h"
#include "SkDecodingImageGenerator.h"
#include "SkForceLinking.h"
#include "SkImageDecoder.h"
#include "SkImagePriv.h"
#include "SkLazyCachingPixelRef.h"
#include "SkLazyPixelRef.h"
#include "SkScaledImageCache.h"
#include "SkStream.h"
#include "Test.h"
#include "TestClassDef.h"
__SK_FORCE_IMAGE_DECODER_LINKING;
/**
* Fill this bitmap with some color.
*/
static void make_test_image(SkBitmap* bm) {
static const int W = 50, H = 50;
static const SkBitmap::Config config = SkBitmap::kARGB_8888_Config;
bm->setConfig(config, W, H);
bm->allocPixels();
bm->eraseColor(SK_ColorBLACK);
SkCanvas canvas(*bm);
SkPaint paint;
paint.setColor(SK_ColorBLUE);
canvas.drawRectCoords(0, 0, SkIntToScalar(W/2),
SkIntToScalar(H/2), paint);
paint.setColor(SK_ColorWHITE);
canvas.drawRectCoords(SkIntToScalar(W/2), SkIntToScalar(H/2),
SkIntToScalar(W), SkIntToScalar(H), paint);
}
/**
* encode this bitmap into some data via SkImageEncoder
*/
static SkData* create_data_from_bitmap(const SkBitmap& bm,
SkImageEncoder::Type type) {
SkDynamicMemoryWStream stream;
if (SkImageEncoder::EncodeStream(&stream, bm, type, 100)) {
return stream.copyToData();
}
return NULL;
}
/**
* A simplified version of SkBitmapFactory
*/
static bool simple_bitmap_factory(SkBitmapFactory::DecodeProc proc,
SkData* data,
SkBitmap* dst) {
SkImageInfo info;
if (!proc(data->data(), data->size(), &info, NULL)) {
return false;
}
dst->setConfig(SkImageInfoToBitmapConfig(info), info.fWidth,
info.fHeight, 0, info.fAlphaType);
SkAutoTUnref<SkLazyPixelRef> ref(SkNEW_ARGS(SkLazyPixelRef,
(data, proc, NULL)));
dst->setPixelRef(ref);
return true;
}
static void compare_bitmaps(skiatest::Reporter* reporter,
const SkBitmap& b1, const SkBitmap& b2,
bool pixelPerfect = true) {
REPORTER_ASSERT(reporter, b1.empty() == b2.empty());
REPORTER_ASSERT(reporter, b1.width() == b2.width());
REPORTER_ASSERT(reporter, b1.height() == b2.height());
REPORTER_ASSERT(reporter, b1.isNull() == b2.isNull());
SkAutoLockPixels autoLockPixels1(b1);
SkAutoLockPixels autoLockPixels2(b2);
REPORTER_ASSERT(reporter, b1.isNull() == b2.isNull());
if (b1.isNull() || b1.empty()) {
return;
}
REPORTER_ASSERT(reporter, NULL != b1.getPixels());
REPORTER_ASSERT(reporter, NULL != b2.getPixels());
if ((!(b1.getPixels())) || (!(b2.getPixels()))) {
return;
}
if ((b1.width() != b2.width()) ||
(b1.height() != b2.height())) {
return;
}
if (!pixelPerfect) {
return;
}
int pixelErrors = 0;
for (int y = 0; y < b2.height(); ++y) {
for (int x = 0; x < b2.width(); ++x) {
if (b1.getColor(x, y) != b2.getColor(x, y)) {
++pixelErrors;
}
}
}
REPORTER_ASSERT(reporter, 0 == pixelErrors);
}
typedef void(*CompareEncodedToOriginal)(skiatest::Reporter* reporter,
SkData* encoded,
const SkBitmap& original,
bool pixelPerfect);
/**
this function tests three differently encoded images against the
original bitmap */
static void test_three_encodings(skiatest::Reporter* reporter,
CompareEncodedToOriginal comp) {
SkBitmap original;
make_test_image(&original);
REPORTER_ASSERT(reporter, !original.empty());
REPORTER_ASSERT(reporter, !original.isNull());
if (original.empty() || original.isNull()) {
return;
}
static const SkImageEncoder::Type types[] = {
SkImageEncoder::kPNG_Type,
SkImageEncoder::kJPEG_Type,
SkImageEncoder::kWEBP_Type
};
for (size_t i = 0; i < SK_ARRAY_COUNT(types); i++) {
SkImageEncoder::Type type = types[i];
SkAutoDataUnref encoded(create_data_from_bitmap(original, type));
REPORTER_ASSERT(reporter, encoded.get() != NULL);
if (NULL != encoded.get()) {
bool comparePixels = (SkImageEncoder::kPNG_Type == type);
comp(reporter, encoded, original, comparePixels);
}
}
}
/**
* This checks to see that a SkLazyPixelRef works as advertised.
*/
static void compare_with_skLazyPixelRef(skiatest::Reporter* reporter,
SkData* encoded,
const SkBitmap& original,
bool comparePixels) {
SkBitmap lazy;
static const SkBitmapFactory::DecodeProc decoder =
&(SkImageDecoder::DecodeMemoryToTarget);
bool success = simple_bitmap_factory(decoder, encoded, &lazy);
REPORTER_ASSERT(reporter, success);
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
{
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
REPORTER_ASSERT(reporter, NULL != lazy.getPixels());
}
// pixels should be gone!
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
{
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
REPORTER_ASSERT(reporter, NULL != lazy.getPixels());
}
compare_bitmaps(reporter, original, lazy, comparePixels);
}
DEF_TEST(LazyPixelRef, reporter) {
test_three_encodings(reporter, compare_with_skLazyPixelRef);
}
/**
* This checks to see that a SkLazyCachedPixelRef works as advertised.
*/
static void compare_with_skLazyCachedPixelRef(skiatest::Reporter* reporter,
SkData* encoded,
const SkBitmap& original,
bool comparePixels) {
SkBitmap lazy;
static const SkBitmapFactory::DecodeProc decoder =
&(SkImageDecoder::DecodeMemoryToTarget);
bool success = SkLazyCachingPixelRef::Install(decoder, encoded, &lazy);
REPORTER_ASSERT(reporter, success);
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
{
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
REPORTER_ASSERT(reporter, NULL != lazy.getPixels());
}
// pixels should be gone!
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
{
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
REPORTER_ASSERT(reporter, NULL != lazy.getPixels());
}
compare_bitmaps(reporter, original, lazy, comparePixels);
}
DEF_TEST(LazyCachedPixelRef, reporter) {
test_three_encodings(reporter, compare_with_skLazyCachedPixelRef);
}
class TestPixelRef : public SkCachingPixelRef {
public:
TestPixelRef(int x) : fX(x) { }
virtual ~TestPixelRef() { }
static bool Install(SkBitmap* destination, int x) {
SkAutoTUnref<TestPixelRef> ref(SkNEW_ARGS(TestPixelRef, (x)));
return ref->configure(destination) && destination->setPixelRef(ref);
}
SK_DECLARE_UNFLATTENABLE_OBJECT()
protected:
virtual bool onDecodeInfo(SkImageInfo* info) SK_OVERRIDE {
if (fX == 0) {
return false;
}
SkASSERT(info);
info->fWidth = 10;
info->fHeight = 10;
info->fColorType = kRGBA_8888_SkColorType;
info->fAlphaType = kOpaque_SkAlphaType;
return true;
}
virtual bool onDecodePixels(const SkImageInfo& info,
void* pixels,
size_t rowBytes) SK_OVERRIDE {
return false;
}
private:
int fX; // controls where the failure happens
typedef SkCachingPixelRef INHERITED;
};
DEF_TEST(CachingPixelRef, reporter) {
SkBitmap lazy;
// test the error handling
REPORTER_ASSERT(reporter, !TestPixelRef::Install(&lazy, 0));
// onDecodeInfo should succeed, allowing installation
REPORTER_ASSERT(reporter, TestPixelRef::Install(&lazy, 1));
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
// onDecodePixels should fail, so getting pixels will fail.
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
}
static void compare_with_SkDecodingImageGenerator(skiatest::Reporter* reporter,
SkData* encoded,
const SkBitmap& original,
bool comparePixels) {
SkBitmap lazy;
bool success = SkDecodingImageGenerator::Install(encoded, &lazy);
REPORTER_ASSERT(reporter, success);
if (!success) {
return;
}
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
{
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
REPORTER_ASSERT(reporter, NULL != lazy.getPixels());
if (NULL == lazy.getPixels()) {
return;
}
}
// pixels should be gone!
REPORTER_ASSERT(reporter, NULL == lazy.getPixels());
{
SkAutoLockPixels autoLockPixels(lazy); // now pixels are good.
REPORTER_ASSERT(reporter, NULL != lazy.getPixels());
}
compare_bitmaps(reporter, original, lazy, comparePixels);
}
DEF_TEST(DecodingImageGenerator, reporter) {
test_three_encodings(reporter, compare_with_SkDecodingImageGenerator);
}