1/*
2 * Copyright 2013 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 "SkBitmap.h"
9#include "SkMallocPixelRef.h"
10#include "SkRandom.h"
11#include "Test.h"
12
13static void test_peekpixels(skiatest::Reporter* reporter) {
14    const SkImageInfo info = SkImageInfo::MakeN32Premul(10, 10);
15
16    SkPixmap pmap;
17    SkBitmap bm;
18
19    // empty should return false
20    REPORTER_ASSERT(reporter, !bm.peekPixels(nullptr));
21    REPORTER_ASSERT(reporter, !bm.peekPixels(&pmap));
22
23    // no pixels should return false
24    bm.setInfo(SkImageInfo::MakeN32Premul(10, 10));
25    REPORTER_ASSERT(reporter, !bm.peekPixels(nullptr));
26    REPORTER_ASSERT(reporter, !bm.peekPixels(&pmap));
27
28    // real pixels should return true
29    bm.allocPixels(info);
30    REPORTER_ASSERT(reporter, bm.peekPixels(nullptr));
31    REPORTER_ASSERT(reporter, bm.peekPixels(&pmap));
32    REPORTER_ASSERT(reporter, pmap.info() == bm.info());
33    REPORTER_ASSERT(reporter, pmap.addr() == bm.getPixels());
34    REPORTER_ASSERT(reporter, pmap.rowBytes() == bm.rowBytes());
35    REPORTER_ASSERT(reporter, pmap.ctable() == bm.getColorTable());
36}
37
38// https://code.google.com/p/chromium/issues/detail?id=446164
39static void test_bigalloc(skiatest::Reporter* reporter) {
40    const int width = 0x40000001;
41    const int height = 0x00000096;
42    const SkImageInfo info = SkImageInfo::MakeN32Premul(width, height);
43
44    SkBitmap bm;
45    REPORTER_ASSERT(reporter, !bm.tryAllocPixels(info));
46
47    SkPixelRef* pr = SkMallocPixelRef::NewAllocate(info, info.minRowBytes(), nullptr);
48    REPORTER_ASSERT(reporter, !pr);
49}
50
51static void test_allocpixels(skiatest::Reporter* reporter) {
52    const int width = 10;
53    const int height = 10;
54    const SkImageInfo info = SkImageInfo::MakeN32Premul(width, height);
55    const size_t explicitRowBytes = info.minRowBytes() + 24;
56
57    SkBitmap bm;
58    bm.setInfo(info);
59    REPORTER_ASSERT(reporter, info.minRowBytes() == bm.rowBytes());
60    bm.allocPixels();
61    REPORTER_ASSERT(reporter, info.minRowBytes() == bm.rowBytes());
62    bm.reset();
63    bm.allocPixels(info);
64    REPORTER_ASSERT(reporter, info.minRowBytes() == bm.rowBytes());
65
66    bm.setInfo(info, explicitRowBytes);
67    REPORTER_ASSERT(reporter, explicitRowBytes == bm.rowBytes());
68    bm.allocPixels();
69    REPORTER_ASSERT(reporter, explicitRowBytes == bm.rowBytes());
70    bm.reset();
71    bm.allocPixels(info, explicitRowBytes);
72    REPORTER_ASSERT(reporter, explicitRowBytes == bm.rowBytes());
73
74    bm.reset();
75    bm.setInfo(info, 0);
76    REPORTER_ASSERT(reporter, info.minRowBytes() == bm.rowBytes());
77    bm.reset();
78    bm.allocPixels(info, 0);
79    REPORTER_ASSERT(reporter, info.minRowBytes() == bm.rowBytes());
80
81    bm.reset();
82    bool success = bm.setInfo(info, info.minRowBytes() - 1);   // invalid for 32bit
83    REPORTER_ASSERT(reporter, !success);
84    REPORTER_ASSERT(reporter, bm.isNull());
85}
86
87static void test_bigwidth(skiatest::Reporter* reporter) {
88    SkBitmap bm;
89    int width = 1 << 29;    // *4 will be the high-bit of 32bit int
90
91    SkImageInfo info = SkImageInfo::MakeA8(width, 1);
92    REPORTER_ASSERT(reporter, bm.setInfo(info));
93    REPORTER_ASSERT(reporter, bm.setInfo(info.makeColorType(kRGB_565_SkColorType)));
94
95    // for a 4-byte config, this width will compute a rowbytes of 0x80000000,
96    // which does not fit in a int32_t. setConfig should detect this, and fail.
97
98    // TODO: perhaps skia can relax this, and only require that rowBytes fit
99    //       in a uint32_t (or larger), but for now this is the constraint.
100
101    REPORTER_ASSERT(reporter, !bm.setInfo(info.makeColorType(kN32_SkColorType)));
102}
103
104/**
105 *  This test contains basic sanity checks concerning bitmaps.
106 */
107DEF_TEST(Bitmap, reporter) {
108    // Zero-sized bitmaps are allowed
109    for (int width = 0; width < 2; ++width) {
110        for (int height = 0; height < 2; ++height) {
111            SkBitmap bm;
112            bool setConf = bm.setInfo(SkImageInfo::MakeN32Premul(width, height));
113            REPORTER_ASSERT(reporter, setConf);
114            if (setConf) {
115                bm.allocPixels();
116            }
117            REPORTER_ASSERT(reporter, SkToBool(width & height) != bm.empty());
118        }
119    }
120
121    test_bigwidth(reporter);
122    test_allocpixels(reporter);
123    test_bigalloc(reporter);
124    test_peekpixels(reporter);
125}
126
127/**
128 *  This test checks that getColor works for both swizzles.
129 */
130DEF_TEST(Bitmap_getColor_Swizzle, r) {
131    SkBitmap source;
132    source.allocN32Pixels(1,1);
133    source.eraseColor(SK_ColorRED);
134    SkColorType colorTypes[] = {
135        kRGBA_8888_SkColorType,
136        kBGRA_8888_SkColorType,
137    };
138    for (SkColorType ct : colorTypes) {
139        SkBitmap copy;
140        if (!source.copyTo(&copy, ct)) {
141            ERRORF(r, "SkBitmap::copy failed %d", (int)ct);
142            continue;
143        }
144        SkAutoLockPixels autoLockPixels1(copy);
145        SkAutoLockPixels autoLockPixels2(source);
146        REPORTER_ASSERT(r, source.getColor(0, 0) == copy.getColor(0, 0));
147    }
148}
149
150static void test_erasecolor_premul(skiatest::Reporter* reporter, SkColorType ct, SkColor input,
151                                   SkColor expected) {
152  SkBitmap bm;
153  bm.allocPixels(SkImageInfo::Make(1, 1, ct, kPremul_SkAlphaType));
154  bm.eraseColor(input);
155  INFOF(reporter, "expected: %x actual: %x\n", expected, bm.getColor(0, 0));
156  REPORTER_ASSERT(reporter, bm.getColor(0, 0) == expected);
157}
158
159/**
160 *  This test checks that eraseColor premultiplies the color correctly.
161 */
162DEF_TEST(Bitmap_eraseColor_Premul, r) {
163    SkColor color = 0x80FF0080;
164    test_erasecolor_premul(r, kAlpha_8_SkColorType, color, 0x80000000);
165    test_erasecolor_premul(r, kRGB_565_SkColorType, color, 0xFF840042);
166    test_erasecolor_premul(r, kARGB_4444_SkColorType, color, 0x88FF0080);
167    test_erasecolor_premul(r, kRGBA_8888_SkColorType, color, color);
168    test_erasecolor_premul(r, kBGRA_8888_SkColorType, color, color);
169}
170
171// Test that SkBitmap::ComputeOpaque() is correct for various colortypes.
172DEF_TEST(Bitmap_compute_is_opaque, r) {
173    struct {
174        SkColorType fCT;
175        SkAlphaType fAT;
176    } types[] = {
177        { kGray_8_SkColorType,    kOpaque_SkAlphaType },
178        { kAlpha_8_SkColorType,   kPremul_SkAlphaType },
179        { kARGB_4444_SkColorType, kPremul_SkAlphaType },
180        { kRGB_565_SkColorType,   kOpaque_SkAlphaType },
181        { kBGRA_8888_SkColorType, kPremul_SkAlphaType },
182        { kRGBA_8888_SkColorType, kPremul_SkAlphaType },
183        { kRGBA_F16_SkColorType,  kPremul_SkAlphaType },
184    };
185    for (auto type : types) {
186        SkBitmap bm;
187        REPORTER_ASSERT(r, !SkBitmap::ComputeIsOpaque(bm));
188
189        bm.allocPixels(SkImageInfo::Make(13, 17, type.fCT, type.fAT));
190        bm.eraseColor(SkColorSetARGB(255, 10, 20, 30));
191        REPORTER_ASSERT(r, SkBitmap::ComputeIsOpaque(bm));
192
193        bm.eraseColor(SkColorSetARGB(128, 255, 255, 255));
194        bool isOpaque = SkBitmap::ComputeIsOpaque(bm);
195        bool shouldBeOpaque = (type.fAT == kOpaque_SkAlphaType);
196        REPORTER_ASSERT(r, isOpaque == shouldBeOpaque);
197    }
198}
199
200// Test that erase+getColor round trips with RGBA_F16 pixels.
201DEF_TEST(Bitmap_erase_f16_erase_getColor, r) {
202    SkRandom random;
203    SkPixmap pm;
204    SkBitmap bm;
205    bm.allocPixels(SkImageInfo::Make(1, 1, kRGBA_F16_SkColorType, kPremul_SkAlphaType));
206    REPORTER_ASSERT(r, bm.peekPixels(&pm));
207    for (unsigned i = 0; i < 0x100; ++i) {
208        // Test all possible values of blue component.
209        SkColor color1 = (SkColor)((random.nextU() & 0xFFFFFF00) | i);
210        // Test all possible values of alpha component.
211        SkColor color2 = (SkColor)((random.nextU() & 0x00FFFFFF) | (i << 24));
212        for (SkColor color : {color1, color2}) {
213            pm.erase(color);
214            if (SkColorGetA(color) != 0) {
215                REPORTER_ASSERT(r, color == pm.getColor(0, 0));
216            } else {
217                REPORTER_ASSERT(r, 0 == SkColorGetA(pm.getColor(0, 0)));
218            }
219        }
220    }
221}
222
223