aboutsummaryrefslogtreecommitdiffhomepage
path: root/tests/BitmapTransformerTest.cpp
blob: d9ce696b6678b1ea7bc0d80d458017484ac5d36b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

/*
 * Copyright 2012 Google Inc.
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

/**
 * Tests for SkBitmapTransformer.h and SkBitmapTransformer.cpp
 */

#include "Test.h"
#include "SkBitmap.h"
#include "SkBitmapTransformer.h"

namespace skiatest {
    class BitmapTransformerTestClass : public Test {
    public:
        static Test* Factory(void*) {return SkNEW(BitmapTransformerTestClass); }
    protected:
        virtual void onGetName(SkString* name) { name->set("BitmapTransformer"); }
        virtual void onRun(Reporter* reporter) {
            this->fReporter = reporter;
            RunTest();
        }
    private:
        void RunTest() {
            SkBitmap bitmap;
            SkBitmap::Config supportedConfig = SkBitmap::kARGB_8888_Config;
            SkBitmap::Config unsupportedConfig = SkBitmap::kARGB_4444_Config;
            SkBitmapTransformer::PixelFormat supportedPixelFormat =
                SkBitmapTransformer::kARGB_8888_Premul_PixelFormat;
            const int kWidth = 55;
            const int kHeight = 333;

            // Transformations that we know are unsupported:
            {
                bitmap.setConfig(unsupportedConfig, kWidth, kHeight);
                SkBitmapTransformer transformer = SkBitmapTransformer(bitmap, supportedPixelFormat);
                REPORTER_ASSERT(fReporter, !transformer.isValid());
            }

            // Valid transformations:
            {
                // Bytes we expect to get:
                const int kWidth = 3;
                const int kHeight = 5;
                const unsigned char comparisonBuffer[] = {
                    // kHeight rows, each with kWidth pixels, premultiplied ARGB for each pixel
                    0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, // red
                    0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, // green
                    0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue
                    0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue
                    0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue
                };

                // A bitmap that should generate the above bytes:
                bitmap.setConfig(supportedConfig, kWidth, kHeight);
                REPORTER_ASSERT(fReporter, bitmap.allocPixels());
                bitmap.setIsOpaque(true);
                bitmap.eraseColor(SK_ColorBLUE);
                bitmap.lockPixels();
                // Change rows [0,1] from blue to [red,green].
                SkColor oldColor = SK_ColorBLUE;
                SkColor newColors[] = {SK_ColorRED, SK_ColorGREEN};
                for (int y = 0; y <= 1; y++) {
                    for (int x = 0; x < kWidth; x++) {
                        REPORTER_ASSERT(fReporter, bitmap.getColor(x, y) == oldColor);
                        SkPMColor* pixel = static_cast<SkPMColor *>(bitmap.getAddr(x, y));
                        *pixel = SkPreMultiplyColor(newColors[y]);
                        REPORTER_ASSERT(fReporter, bitmap.getColor(x, y) == newColors[y]);
                    }
                }
                bitmap.unlockPixels();

                // Transform the bitmap and confirm we got the expected results.
                SkBitmapTransformer transformer = SkBitmapTransformer(bitmap, supportedPixelFormat);
                REPORTER_ASSERT(fReporter, transformer.isValid());
                REPORTER_ASSERT(fReporter, transformer.bytesNeededPerRow() == kWidth * 4);
                REPORTER_ASSERT(fReporter, transformer.bytesNeededTotal() == kWidth * kHeight * 4);
                int bufferSize = transformer.bytesNeededTotal();
                SkAutoMalloc pixelBufferManager(bufferSize);
                char *pixelBuffer = static_cast<char *>(pixelBufferManager.get());
                REPORTER_ASSERT(fReporter,
                                transformer.copyBitmapToPixelBuffer(pixelBuffer, bufferSize));
                REPORTER_ASSERT(fReporter, bufferSize == sizeof(comparisonBuffer));
                REPORTER_ASSERT(fReporter, memcmp(pixelBuffer, comparisonBuffer, bufferSize) == 0);
            }

        }

        Reporter* fReporter;
    };

    static TestRegistry gReg(BitmapTransformerTestClass::Factory);
}