1/*
2 * Copyright 2012 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 "SkImageEncoder.h"
9
10#include "SkBitmap.h"
11#include "SkCanvas.h"
12#include "SkStream.h"
13#include "SkTemplates.h"
14#include "Test.h"
15
16static SkColorType gColorTypes[] = {
17    kRGB_565_SkColorType,
18    kN32_SkColorType,
19};
20
21DEF_TEST(ARGBImageEncoder, reporter) {
22    // Bytes we expect to get:
23    const int kWidth = 3;
24    const int kHeight = 5;
25    const unsigned char comparisonBuffer[] = {
26        // kHeight rows, each with kWidth pixels, premultiplied ARGB for each pixel
27        0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, // red
28        0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, // green
29        0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue
30        0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue
31        0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue
32    };
33
34    SkAutoTDelete<SkImageEncoder> enc(CreateARGBImageEncoder());
35    for (size_t ctIndex = 0; ctIndex < SK_ARRAY_COUNT(gColorTypes); ++ctIndex) {
36        // A bitmap that should generate the above bytes:
37        SkBitmap bitmap;
38        {
39            bitmap.allocPixels(SkImageInfo::Make(kWidth, kHeight, gColorTypes[ctIndex],
40                                                 kOpaque_SkAlphaType));
41            bitmap.eraseColor(SK_ColorBLUE);
42            // Change rows [0,1] from blue to [red,green].
43            SkCanvas canvas(bitmap);
44            SkPaint paint;
45            paint.setColor(SK_ColorRED);
46            canvas.drawIRect(SkIRect::MakeLTRB(0, 0, kWidth, 1), paint);
47            paint.setColor(SK_ColorGREEN);
48            canvas.drawIRect(SkIRect::MakeLTRB(0, 1, kWidth, 2), paint);
49        }
50
51        // Transform the bitmap.
52        int bufferSize = bitmap.width() * bitmap.height() * 4;
53        SkAutoTMalloc<char> pixelBufferManager(bufferSize);
54        char* pixelBuffer = pixelBufferManager.get();
55        SkMemoryWStream out(pixelBuffer, bufferSize);
56        REPORTER_ASSERT(reporter, enc->encodeStream(&out, bitmap, SkImageEncoder::kDefaultQuality));
57
58        // Confirm we got the expected results.
59        REPORTER_ASSERT(reporter, bufferSize == sizeof(comparisonBuffer));
60        REPORTER_ASSERT(reporter, memcmp(pixelBuffer, comparisonBuffer, bufferSize) == 0);
61    }
62}
63