1/* 2 * Copyright 2014 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 "SkData.h" 9#include "SkCanvas.h" 10#include "SkGraphics.h" 11#include "SkImageGenerator.h" 12#include "SkImageInfoPriv.h" 13#include "Test.h" 14 15static bool gMyFactoryWasCalled; 16 17static std::unique_ptr<SkImageGenerator> my_factory(sk_sp<SkData>) { 18 gMyFactoryWasCalled = true; 19 return nullptr; 20} 21 22static void test_imagegenerator_factory(skiatest::Reporter* reporter) { 23 // just need a non-empty data to test things 24 sk_sp<SkData> data(SkData::MakeWithCString("test_imagegenerator_factory")); 25 26 gMyFactoryWasCalled = false; 27 28 REPORTER_ASSERT(reporter, !gMyFactoryWasCalled); 29 30 std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(data); 31 REPORTER_ASSERT(reporter, nullptr == gen); 32 REPORTER_ASSERT(reporter, !gMyFactoryWasCalled); 33 34 // Test is racy, in that it hopes no other thread is changing this global... 35 auto prev = SkGraphics::SetImageGeneratorFromEncodedDataFactory(my_factory); 36 gen = SkImageGenerator::MakeFromEncoded(data); 37 REPORTER_ASSERT(reporter, nullptr == gen); 38 REPORTER_ASSERT(reporter, gMyFactoryWasCalled); 39 SkGraphics::SetImageGeneratorFromEncodedDataFactory(prev); 40} 41 42class MyImageGenerator : public SkImageGenerator { 43public: 44 MyImageGenerator() : SkImageGenerator(SkImageInfo::MakeN32Premul(0, 0)) {} 45}; 46 47DEF_TEST(ImageGenerator, reporter) { 48 MyImageGenerator ig; 49 SkYUVSizeInfo sizeInfo; 50 sizeInfo.fSizes[SkYUVSizeInfo::kY] = SkISize::Make(200, 200); 51 sizeInfo.fSizes[SkYUVSizeInfo::kU] = SkISize::Make(100, 100); 52 sizeInfo.fSizes[SkYUVSizeInfo::kV] = SkISize::Make( 50, 50); 53 sizeInfo.fWidthBytes[SkYUVSizeInfo::kY] = 0; 54 sizeInfo.fWidthBytes[SkYUVSizeInfo::kU] = 0; 55 sizeInfo.fWidthBytes[SkYUVSizeInfo::kV] = 0; 56 void* planes[3] = { nullptr }; 57 SkYUVColorSpace colorSpace; 58 59 // Check that the YUV decoding API does not cause any crashes 60 ig.queryYUV8(&sizeInfo, nullptr); 61 ig.queryYUV8(&sizeInfo, &colorSpace); 62 sizeInfo.fWidthBytes[SkYUVSizeInfo::kY] = 250; 63 sizeInfo.fWidthBytes[SkYUVSizeInfo::kU] = 250; 64 sizeInfo.fWidthBytes[SkYUVSizeInfo::kV] = 250; 65 int dummy; 66 planes[SkYUVSizeInfo::kY] = planes[SkYUVSizeInfo::kU] = planes[SkYUVSizeInfo::kV] = &dummy; 67 ig.getYUV8Planes(sizeInfo, planes); 68 69 // Suppressed due to https://code.google.com/p/skia/issues/detail?id=4339 70 if (false) { 71 test_imagegenerator_factory(reporter); 72 } 73} 74 75#include "SkAutoMalloc.h" 76#include "SkPictureRecorder.h" 77 78static sk_sp<SkPicture> make_picture() { 79 SkPictureRecorder recorder; 80 recorder.beginRecording(100, 100)->drawColor(SK_ColorRED); 81 return recorder.finishRecordingAsPicture(); 82} 83 84DEF_TEST(PictureImageGenerator, reporter) { 85 const struct { 86 SkColorType fColorType; 87 SkAlphaType fAlphaType; 88 bool fExpectSuccess; 89 } recs[] = { 90 { kRGBA_8888_SkColorType, kPremul_SkAlphaType, kRGBA_8888_SkColorType == kN32_SkColorType }, 91 { kBGRA_8888_SkColorType, kPremul_SkAlphaType, kBGRA_8888_SkColorType == kN32_SkColorType }, 92 { kRGBA_F16_SkColorType, kPremul_SkAlphaType, true }, 93 { kRGBA_1010102_SkColorType, kPremul_SkAlphaType, true }, 94 95 { kRGBA_8888_SkColorType, kUnpremul_SkAlphaType, false }, 96 { kBGRA_8888_SkColorType, kUnpremul_SkAlphaType, false }, 97 { kRGBA_F16_SkColorType, kUnpremul_SkAlphaType, false }, 98 { kRGBA_1010102_SkColorType, kUnpremul_SkAlphaType, false }, 99 }; 100 101 auto colorspace = SkColorSpace::MakeSRGB(); 102 auto picture = make_picture(); 103 auto gen = SkImageGenerator::MakeFromPicture({100, 100}, picture, nullptr, nullptr, 104 SkImage::BitDepth::kU8, colorspace); 105 106 // worst case for all requests 107 SkAutoMalloc storage(100 * 100 * SkColorTypeBytesPerPixel(kRGBA_F16_SkColorType)); 108 109 for (const auto& rec : recs) { 110 SkImageInfo info = SkImageInfo::Make(100, 100, rec.fColorType, rec.fAlphaType, colorspace); 111 bool success = gen->getPixels(info, storage.get(), info.minRowBytes()); 112 REPORTER_ASSERT(reporter, success == rec.fExpectSuccess); 113 } 114} 115 116