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