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