| /* |
| * 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 "SkCachingPixelRef.h" |
| #include "SkCanvas.h" |
| #include "SkData.h" |
| #include "SkDecodingImageGenerator.h" |
| #include "SkDiscardableMemoryPool.h" |
| #include "SkImageDecoder.h" |
| #include "SkImageGeneratorPriv.h" |
| #include "SkResourceCache.h" |
| #include "SkStream.h" |
| #include "SkUtils.h" |
| |
| #include "Test.h" |
| |
| /** |
| * Fill this bitmap with some color. |
| */ |
| static void make_test_image(SkBitmap* bm) { |
| const int W = 50, H = 50; |
| bm->allocN32Pixels(W, H); |
| 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; |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| |
| 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, b1.getPixels()); |
| REPORTER_ASSERT(reporter, 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 bool (*InstallEncoded)(SkData* encoded, SkBitmap* dst); |
| |
| /** |
| This function tests three differently encoded images against the |
| original bitmap */ |
| static void test_three_encodings(skiatest::Reporter* reporter, |
| InstallEncoded install) { |
| 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()) { |
| continue; |
| } |
| SkBitmap lazy; |
| bool installSuccess = install(encoded.get(), &lazy); |
| REPORTER_ASSERT(reporter, installSuccess); |
| if (!installSuccess) { |
| continue; |
| } |
| REPORTER_ASSERT(reporter, NULL == lazy.getPixels()); |
| { |
| SkAutoLockPixels autoLockPixels(lazy); // now pixels are good. |
| REPORTER_ASSERT(reporter, lazy.getPixels()); |
| if (NULL == lazy.getPixels()) { |
| continue; |
| } |
| } |
| // pixels should be gone! |
| REPORTER_ASSERT(reporter, NULL == lazy.getPixels()); |
| { |
| SkAutoLockPixels autoLockPixels(lazy); // now pixels are good. |
| REPORTER_ASSERT(reporter, lazy.getPixels()); |
| if (NULL == lazy.getPixels()) { |
| continue; |
| } |
| } |
| bool comparePixels = (SkImageEncoder::kPNG_Type == type); |
| compare_bitmaps(reporter, original, lazy, comparePixels); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| static bool install_skCachingPixelRef(SkData* encoded, SkBitmap* dst) { |
| return SkCachingPixelRef::Install( |
| SkDecodingImageGenerator::Create( |
| encoded, SkDecodingImageGenerator::Options()), dst); |
| } |
| static bool install_skDiscardablePixelRef(SkData* encoded, SkBitmap* dst) { |
| // Use system-default discardable memory. |
| return SkInstallDiscardablePixelRef( |
| SkDecodingImageGenerator::Create( |
| encoded, SkDecodingImageGenerator::Options()), dst); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| /** |
| * This checks to see that a SkCachingPixelRef and a |
| * SkDiscardablePixelRef works as advertised with a |
| * SkDecodingImageGenerator. |
| */ |
| DEF_TEST(DecodingImageGenerator, reporter) { |
| test_three_encodings(reporter, install_skCachingPixelRef); |
| test_three_encodings(reporter, install_skDiscardablePixelRef); |
| } |
| |
| class TestImageGenerator : public SkImageGenerator { |
| public: |
| enum TestType { |
| kFailGetInfo_TestType, |
| kFailGetPixels_TestType, |
| kSucceedGetPixels_TestType, |
| kLast_TestType = kSucceedGetPixels_TestType |
| }; |
| static int Width() { return 10; } |
| static int Height() { return 10; } |
| static uint32_t Color() { return 0xff123456; } |
| TestImageGenerator(TestType type, skiatest::Reporter* reporter) |
| : fType(type), fReporter(reporter) { |
| SkASSERT((fType <= kLast_TestType) && (fType >= 0)); |
| } |
| virtual ~TestImageGenerator() { } |
| |
| protected: |
| virtual bool onGetInfo(SkImageInfo* info) SK_OVERRIDE { |
| REPORTER_ASSERT(fReporter, info); |
| if ((NULL == info) || (kFailGetInfo_TestType == fType)) { |
| return false; |
| } |
| *info = SkImageInfo::MakeN32(TestImageGenerator::Width(), |
| TestImageGenerator::Height(), |
| kOpaque_SkAlphaType); |
| return true; |
| } |
| |
| virtual bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, |
| SkPMColor ctable[], int* ctableCount) SK_OVERRIDE { |
| REPORTER_ASSERT(fReporter, pixels != NULL); |
| size_t minRowBytes = static_cast<size_t>(info.width() * info.bytesPerPixel()); |
| REPORTER_ASSERT(fReporter, rowBytes >= minRowBytes); |
| if ((NULL == pixels) |
| || (fType != kSucceedGetPixels_TestType) |
| || (info.colorType() != kN32_SkColorType)) { |
| return false; |
| } |
| char* bytePtr = static_cast<char*>(pixels); |
| for (int y = 0; y < info.height(); ++y) { |
| sk_memset32(reinterpret_cast<SkColor*>(bytePtr), |
| TestImageGenerator::Color(), info.width()); |
| bytePtr += rowBytes; |
| } |
| return true; |
| } |
| |
| private: |
| const TestType fType; |
| skiatest::Reporter* const fReporter; |
| }; |
| |
| static void check_test_image_generator_bitmap(skiatest::Reporter* reporter, |
| const SkBitmap& bm) { |
| REPORTER_ASSERT(reporter, TestImageGenerator::Width() == bm.width()); |
| REPORTER_ASSERT(reporter, TestImageGenerator::Height() == bm.height()); |
| SkAutoLockPixels autoLockPixels(bm); |
| REPORTER_ASSERT(reporter, bm.getPixels()); |
| if (NULL == bm.getPixels()) { |
| return; |
| } |
| int errors = 0; |
| for (int y = 0; y < bm.height(); ++y) { |
| for (int x = 0; x < bm.width(); ++x) { |
| if (TestImageGenerator::Color() != *bm.getAddr32(x, y)) { |
| ++errors; |
| } |
| } |
| } |
| REPORTER_ASSERT(reporter, 0 == errors); |
| } |
| |
| enum PixelRefType { |
| kSkCaching_PixelRefType, |
| kSkDiscardable_PixelRefType, |
| kLast_PixelRefType = kSkDiscardable_PixelRefType |
| }; |
| |
| static void check_pixelref(TestImageGenerator::TestType type, |
| skiatest::Reporter* reporter, |
| PixelRefType pixelRefType, |
| SkDiscardableMemory::Factory* factory) { |
| SkASSERT((pixelRefType >= 0) && (pixelRefType <= kLast_PixelRefType)); |
| SkAutoTDelete<SkImageGenerator> gen(SkNEW_ARGS(TestImageGenerator, |
| (type, reporter))); |
| REPORTER_ASSERT(reporter, gen.get() != NULL); |
| SkBitmap lazy; |
| bool success; |
| if (kSkCaching_PixelRefType == pixelRefType) { |
| // Ignore factory; use global cache. |
| success = SkCachingPixelRef::Install(gen.detach(), &lazy); |
| } else { |
| success = SkInstallDiscardablePixelRef(gen.detach(), &lazy, factory); |
| } |
| REPORTER_ASSERT(reporter, success |
| == (TestImageGenerator::kFailGetInfo_TestType != type)); |
| if (TestImageGenerator::kSucceedGetPixels_TestType == type) { |
| check_test_image_generator_bitmap(reporter, lazy); |
| } else if (TestImageGenerator::kFailGetPixels_TestType == type) { |
| SkAutoLockPixels autoLockPixels(lazy); |
| REPORTER_ASSERT(reporter, NULL == lazy.getPixels()); |
| } |
| } |
| |
| // new/lock/delete is an odd pattern for a pixelref, but it needs to not assert |
| static void test_newlockdelete(skiatest::Reporter* reporter) { |
| SkBitmap bm; |
| SkImageGenerator* ig = new TestImageGenerator( |
| TestImageGenerator::kSucceedGetPixels_TestType, reporter); |
| SkInstallDiscardablePixelRef(ig, &bm); |
| bm.pixelRef()->lockPixels(); |
| } |
| |
| /** |
| * This tests the basic functionality of SkDiscardablePixelRef with a |
| * basic SkImageGenerator implementation and several |
| * SkDiscardableMemory::Factory choices. |
| */ |
| DEF_TEST(DiscardableAndCachingPixelRef, reporter) { |
| test_newlockdelete(reporter); |
| |
| check_pixelref(TestImageGenerator::kFailGetInfo_TestType, |
| reporter, kSkCaching_PixelRefType, NULL); |
| check_pixelref(TestImageGenerator::kFailGetPixels_TestType, |
| reporter, kSkCaching_PixelRefType, NULL); |
| check_pixelref(TestImageGenerator::kSucceedGetPixels_TestType, |
| reporter, kSkCaching_PixelRefType, NULL); |
| |
| check_pixelref(TestImageGenerator::kFailGetInfo_TestType, |
| reporter, kSkDiscardable_PixelRefType, NULL); |
| check_pixelref(TestImageGenerator::kFailGetPixels_TestType, |
| reporter, kSkDiscardable_PixelRefType, NULL); |
| check_pixelref(TestImageGenerator::kSucceedGetPixels_TestType, |
| reporter, kSkDiscardable_PixelRefType, NULL); |
| |
| SkAutoTUnref<SkDiscardableMemoryPool> pool( |
| SkDiscardableMemoryPool::Create(1, NULL)); |
| REPORTER_ASSERT(reporter, 0 == pool->getRAMUsed()); |
| check_pixelref(TestImageGenerator::kFailGetPixels_TestType, |
| reporter, kSkDiscardable_PixelRefType, pool); |
| REPORTER_ASSERT(reporter, 0 == pool->getRAMUsed()); |
| check_pixelref(TestImageGenerator::kSucceedGetPixels_TestType, |
| reporter, kSkDiscardable_PixelRefType, pool); |
| REPORTER_ASSERT(reporter, 0 == pool->getRAMUsed()); |
| |
| SkDiscardableMemoryPool* globalPool = SkGetGlobalDiscardableMemoryPool(); |
| // Only acts differently from NULL on a platform that has a |
| // default discardable memory implementation that differs from the |
| // global DM pool. |
| check_pixelref(TestImageGenerator::kFailGetPixels_TestType, |
| reporter, kSkDiscardable_PixelRefType, globalPool); |
| check_pixelref(TestImageGenerator::kSucceedGetPixels_TestType, |
| reporter, kSkDiscardable_PixelRefType, globalPool); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| |
| DEF_TEST(Image_NewFromGenerator, r) { |
| TestImageGenerator::TestType testTypes[] = { |
| TestImageGenerator::kFailGetInfo_TestType, |
| TestImageGenerator::kFailGetPixels_TestType, |
| TestImageGenerator::kSucceedGetPixels_TestType, |
| }; |
| for (size_t i = 0; i < SK_ARRAY_COUNT(testTypes); ++i) { |
| TestImageGenerator::TestType test = testTypes[i]; |
| SkImageGenerator* gen = SkNEW_ARGS(TestImageGenerator, (test, r)); |
| SkAutoTUnref<SkImage> image(SkImage::NewFromGenerator(gen)); |
| if (TestImageGenerator::kFailGetInfo_TestType == test) { |
| REPORTER_ASSERT(r, NULL == image.get()); |
| continue; |
| } |
| if (NULL == image.get()) { |
| ERRORF(r, "SkImage::NewFromGenerator unexpecedly failed [" |
| SK_SIZE_T_SPECIFIER "]", i); |
| continue; |
| } |
| REPORTER_ASSERT(r, TestImageGenerator::Width() == image->width()); |
| REPORTER_ASSERT(r, TestImageGenerator::Height() == image->height()); |
| |
| SkBitmap bitmap; |
| bitmap.allocN32Pixels(TestImageGenerator::Width(), TestImageGenerator::Height()); |
| SkCanvas canvas(bitmap); |
| const SkColor kDefaultColor = 0xffabcdef; |
| canvas.clear(kDefaultColor); |
| image->draw(&canvas, 0, 0, NULL); |
| if (TestImageGenerator::kSucceedGetPixels_TestType == test) { |
| REPORTER_ASSERT( |
| r, TestImageGenerator::Color() == *bitmap.getAddr32(0, 0)); |
| } else { |
| REPORTER_ASSERT(r, kDefaultColor == bitmap.getColor(0,0)); |
| } |
| } |
| } |