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#include "sk_tool_utils.h"
13
14static void test_peekpixels(skiatest::Reporter* reporter) {
15    const SkImageInfo info = SkImageInfo::MakeN32Premul(10, 10);
16
17    SkPixmap pmap;
18    SkBitmap bm;
19
20    // empty should return false
21    REPORTER_ASSERT(reporter, !bm.peekPixels(nullptr));
22    REPORTER_ASSERT(reporter, !bm.peekPixels(&pmap));
23
24    // no pixels should return false
25    bm.setInfo(SkImageInfo::MakeN32Premul(10, 10));
26    REPORTER_ASSERT(reporter, !bm.peekPixels(nullptr));
27    REPORTER_ASSERT(reporter, !bm.peekPixels(&pmap));
28
29    // real pixels should return true
30    bm.allocPixels(info);
31    REPORTER_ASSERT(reporter, bm.peekPixels(nullptr));
32    REPORTER_ASSERT(reporter, bm.peekPixels(&pmap));
33    REPORTER_ASSERT(reporter, pmap.info() == bm.info());
34    REPORTER_ASSERT(reporter, pmap.addr() == bm.getPixels());
35    REPORTER_ASSERT(reporter, pmap.rowBytes() == bm.rowBytes());
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    sk_sp<SkPixelRef> pr = SkMallocPixelRef::MakeAllocate(info, info.minRowBytes());
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 (!sk_tool_utils::copy_to(&copy, ct, source)) {
141            ERRORF(r, "SkBitmap::copy failed %d", (int)ct);
142            continue;
143        }
144        REPORTER_ASSERT(r, source.getColor(0, 0) == copy.getColor(0, 0));
145    }
146}
147
148static void test_erasecolor_premul(skiatest::Reporter* reporter, SkColorType ct, SkColor input,
149                                   SkColor expected) {
150  SkBitmap bm;
151  bm.allocPixels(SkImageInfo::Make(1, 1, ct, kPremul_SkAlphaType));
152  bm.eraseColor(input);
153  INFOF(reporter, "expected: %x actual: %x\n", expected, bm.getColor(0, 0));
154  REPORTER_ASSERT(reporter, bm.getColor(0, 0) == expected);
155}
156
157/**
158 *  This test checks that eraseColor premultiplies the color correctly.
159 */
160DEF_TEST(Bitmap_eraseColor_Premul, r) {
161    SkColor color = 0x80FF0080;
162    test_erasecolor_premul(r, kAlpha_8_SkColorType, color, 0x80000000);
163    test_erasecolor_premul(r, kRGB_565_SkColorType, color, 0xFF840042);
164    test_erasecolor_premul(r, kARGB_4444_SkColorType, color, 0x88FF0080);
165    test_erasecolor_premul(r, kRGBA_8888_SkColorType, color, color);
166    test_erasecolor_premul(r, kBGRA_8888_SkColorType, color, color);
167}
168
169// Test that SkBitmap::ComputeOpaque() is correct for various colortypes.
170DEF_TEST(Bitmap_compute_is_opaque, r) {
171    struct {
172        SkColorType fCT;
173        SkAlphaType fAT;
174    } types[] = {
175        { kGray_8_SkColorType,    kOpaque_SkAlphaType },
176        { kAlpha_8_SkColorType,   kPremul_SkAlphaType },
177        { kARGB_4444_SkColorType, kPremul_SkAlphaType },
178        { kRGB_565_SkColorType,   kOpaque_SkAlphaType },
179        { kBGRA_8888_SkColorType, kPremul_SkAlphaType },
180        { kRGBA_8888_SkColorType, kPremul_SkAlphaType },
181        { kRGBA_F16_SkColorType,  kPremul_SkAlphaType },
182    };
183    for (auto type : types) {
184        SkBitmap bm;
185        REPORTER_ASSERT(r, !SkBitmap::ComputeIsOpaque(bm));
186
187        bm.allocPixels(SkImageInfo::Make(13, 17, type.fCT, type.fAT));
188        bm.eraseColor(SkColorSetARGB(255, 10, 20, 30));
189        REPORTER_ASSERT(r, SkBitmap::ComputeIsOpaque(bm));
190
191        bm.eraseColor(SkColorSetARGB(128, 255, 255, 255));
192        bool isOpaque = SkBitmap::ComputeIsOpaque(bm);
193        bool shouldBeOpaque = (type.fAT == kOpaque_SkAlphaType);
194        REPORTER_ASSERT(r, isOpaque == shouldBeOpaque);
195    }
196}
197
198// Test that erase+getColor round trips with RGBA_F16 pixels.
199DEF_TEST(Bitmap_erase_f16_erase_getColor, r) {
200    SkRandom random;
201    SkPixmap pm;
202    SkBitmap bm;
203    bm.allocPixels(SkImageInfo::Make(1, 1, kRGBA_F16_SkColorType, kPremul_SkAlphaType));
204    REPORTER_ASSERT(r, bm.peekPixels(&pm));
205    for (unsigned i = 0; i < 0x100; ++i) {
206        // Test all possible values of blue component.
207        SkColor color1 = (SkColor)((random.nextU() & 0xFFFFFF00) | i);
208        // Test all possible values of alpha component.
209        SkColor color2 = (SkColor)((random.nextU() & 0x00FFFFFF) | (i << 24));
210        for (SkColor color : {color1, color2}) {
211            pm.erase(color);
212            if (SkColorGetA(color) != 0) {
213                REPORTER_ASSERT(r, color == pm.getColor(0, 0));
214            } else {
215                REPORTER_ASSERT(r, 0 == SkColorGetA(pm.getColor(0, 0)));
216            }
217        }
218    }
219}
220
221// Make sure that the bitmap remains valid when pixelref is removed.
222DEF_TEST(Bitmap_clear_pixelref_keep_info, r) {
223    SkBitmap bm;
224    bm.allocPixels(SkImageInfo::MakeN32Premul(100,100));
225    bm.setPixelRef(nullptr, 0, 0);
226    SkDEBUGCODE(bm.validate();)
227}
228
229