1 /*
2  * Copyright 2016 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 "gm.h"
9 
10 #include "Resources.h"
11 #include "SkCanvas.h"
12 #include "SkData.h"
13 #include "SkImage.h"
14 #include "SkImageEncoderPriv.h"
15 #include "SkJpegEncoder.h"
16 #include "SkPngEncoder.h"
17 #include "SkUnPreMultiply.h"
18 #include "SkWebpEncoder.h"
19 
20 namespace skiagm {
21 
22 static void make_opaque_256(SkBitmap* bitmap) {
23     GetResourceAsBitmap("images/mandrill_256.png", bitmap);
24 }
25 
26 static void make_premul_256(SkBitmap* bitmap) {
27     SkBitmap tmp;
28     GetResourceAsBitmap("images/yellow_rose.png", &tmp);
29     tmp.extractSubset(bitmap, SkIRect::MakeWH(256, 256));
30 }
31 
32 static void make_unpremul_256(SkBitmap* bitmap) {
33     make_premul_256(bitmap);
34     for (int y = 0; y < bitmap->height(); y++) {
35         for (int x = 0; x < bitmap->width(); x++) {
36             SkPMColor* pixel = bitmap->getAddr32(x, y);
37             *pixel = SkUnPreMultiply::UnPreMultiplyPreservingByteOrder(*pixel);
38         }
39     }
40     bitmap->setAlphaType(kUnpremul_SkAlphaType);
41 }
42 
43 #if defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS)
44 static SkEncodedImageFormat kTypes[] {
45         SkEncodedImageFormat::kPNG, SkEncodedImageFormat::kJPEG, SkEncodedImageFormat::kGIF,
46         SkEncodedImageFormat::kBMP, SkEncodedImageFormat::kICO,
47 };
48 #elif defined(SK_BUILD_FOR_WIN)
49 // Use PNG multiple times because our WIC encoder does not support GIF, BMP, or ICO.
50 static SkEncodedImageFormat kTypes[] {
51         SkEncodedImageFormat::kPNG, SkEncodedImageFormat::kJPEG, SkEncodedImageFormat::kPNG,
52         SkEncodedImageFormat::kPNG, SkEncodedImageFormat::kPNG,
53 };
54 #else
55 // Use WEBP in place of GIF.  Use PNG two extra times.  We don't support GIF, BMP, or ICO.
56 static SkEncodedImageFormat kTypes[] {
57         SkEncodedImageFormat::kPNG, SkEncodedImageFormat::kJPEG, SkEncodedImageFormat::kWEBP,
58         SkEncodedImageFormat::kPNG, SkEncodedImageFormat::kPNG,
59 };
60 #endif
61 
62 static sk_sp<SkData> encode_data(SkEncodedImageFormat type, const SkBitmap& bitmap) {
63     SkPixmap src;
64     if (!bitmap.peekPixels(&src)) {
65         return nullptr;
66     }
67     SkDynamicMemoryWStream buf;
68     #if defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS)
69         return SkEncodeImageWithCG(&buf, src, type) ? buf.detachAsData() : nullptr;
70     #elif defined(SK_BUILD_FOR_WIN)
71         return SkEncodeImageWithWIC(&buf, src, type, 100) ? buf.detachAsData() : nullptr;
72     #else
73         switch (type) {
74             case SkEncodedImageFormat::kPNG: {
75                 SkPngEncoder::Options options;
76                 options.fUnpremulBehavior = SkTransferFunctionBehavior::kIgnore;
77                 bool success = SkPngEncoder::Encode(&buf, src, options);
78                 return success ? buf.detachAsData() : nullptr;
79             }
80             case SkEncodedImageFormat::kJPEG: {
81                 bool success = SkJpegEncoder::Encode(&buf, src, SkJpegEncoder::Options());
82                 return success ? buf.detachAsData() : nullptr;
83             }
84             case SkEncodedImageFormat::kWEBP: {
85                 SkWebpEncoder::Options options;
86                 options.fUnpremulBehavior = SkTransferFunctionBehavior::kIgnore;
87                 bool success = SkWebpEncoder::Encode(&buf, src, options);
88                 return success ? buf.detachAsData() : nullptr;
89             }
90             default:
91                 SkASSERT(false);
92                 return nullptr;
93         }
94     #endif
95 }
96 
97 class EncodePlatformGM : public GM {
98 public:
99     EncodePlatformGM() {}
100 
101 protected:
102     SkString onShortName() override {
103         return SkString("encode-platform");
104     }
105 
106     SkISize onISize() override {
107         return SkISize::Make(256 * SK_ARRAY_COUNT(kTypes), 256 * 3);
108     }
109 
110     void onDraw(SkCanvas* canvas) override {
111         SkBitmap opaqueBm, premulBm, unpremulBm;
112         make_opaque_256(&opaqueBm);
113         make_premul_256(&premulBm);
114         make_unpremul_256(&unpremulBm);
115 
116         for (SkEncodedImageFormat type : kTypes) {
117             auto opaqueImage = SkImage::MakeFromEncoded(encode_data(type, opaqueBm));
118             auto premulImage = SkImage::MakeFromEncoded(encode_data(type, premulBm));
119             auto unpremulImage = SkImage::MakeFromEncoded(encode_data(type, unpremulBm));
120 
121             canvas->drawImage(opaqueImage.get(), 0.0f, 0.0f);
122             canvas->drawImage(premulImage.get(), 0.0f, 256.0f);
123             canvas->drawImage(unpremulImage.get(), 0.0f, 512.0f);
124 
125             canvas->translate(256.0f, 0.0f);
126         }
127     }
128 
129 private:
130     typedef GM INHERITED;
131 };
132 
133 DEF_GM( return new EncodePlatformGM; )
134 }
135