1 /*
2 * Copyright 2017 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/gm.h"
9 #include "include/core/SkCanvas.h"
10 #include "include/core/SkColorSpace.h"
11 #include "include/core/SkData.h"
12 #include "include/core/SkImage.h"
13 #include "include/core/SkImageInfo.h"
14 #include "include/core/SkPixmap.h"
15 #include "include/core/SkRefCnt.h"
16 #include "include/core/SkSize.h"
17 #include "include/core/SkStream.h"
18 #include "include/core/SkString.h"
19 #include "include/core/SkTypes.h"
20 #include "include/encode/SkJpegEncoder.h"
21 #include "include/private/base/SkTemplates.h"
22 #include "tools/DecodeUtils.h"
23 #include "tools/Resources.h"
24
25 using namespace skia_private;
26
27 namespace skiagm {
28
read_into_pixmap(SkPixmap * dst,SkImageInfo dstInfo,void * dstPixels,sk_sp<SkImage> src)29 static inline void read_into_pixmap(SkPixmap* dst, SkImageInfo dstInfo, void* dstPixels,
30 sk_sp<SkImage> src) {
31 dst->reset(dstInfo, dstPixels, dstInfo.minRowBytes());
32 src->readPixels(nullptr, *dst, 0, 0, SkImage::CachingHint::kDisallow_CachingHint);
33 }
34
encode_pixmap_and_make_image(const SkPixmap & src,SkJpegEncoder::AlphaOption alphaOption)35 static inline sk_sp<SkImage> encode_pixmap_and_make_image(const SkPixmap& src,
36 SkJpegEncoder::AlphaOption alphaOption) {
37 SkDynamicMemoryWStream dst;
38 SkJpegEncoder::Options options;
39 options.fAlphaOption = alphaOption;
40 SkJpegEncoder::Encode(&dst, src, options);
41 return SkImages::DeferredFromEncodedData(dst.detachAsData());
42 }
43
44 class EncodeJpegAlphaOptsGM : public GM {
45 public:
EncodeJpegAlphaOptsGM()46 EncodeJpegAlphaOptsGM() {}
47
48 protected:
getName() const49 SkString getName() const override { return SkString("encode-alpha-jpeg"); }
50
getISize()51 SkISize getISize() override { return SkISize::Make(400, 200); }
52
onDraw(SkCanvas * canvas,SkString * errorMsg)53 DrawResult onDraw(SkCanvas* canvas, SkString* errorMsg) override {
54 sk_sp<SkImage> srcImg = ToolUtils::GetResourceAsImage("images/rainbow-gradient.png");
55 if (!srcImg) {
56 *errorMsg = "Could not load images/rainbow-gradient.png. "
57 "Did you forget to set the resourcePath?";
58 return DrawResult::kFail;
59 }
60 fStorage.reset(srcImg->width() * srcImg->height() *
61 SkColorTypeBytesPerPixel(kRGBA_F16_SkColorType));
62
63 SkPixmap src;
64 SkImageInfo info = SkImageInfo::MakeN32Premul(srcImg->width(), srcImg->height(),
65 canvas->imageInfo().colorSpace() ? SkColorSpace::MakeSRGB() : nullptr);
66 read_into_pixmap(&src, info, fStorage.get(), srcImg);
67
68 // Encode 8888 premul.
69 auto img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
70 auto img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
71 canvas->drawImage(img0, 0.0f, 0.0f);
72 canvas->drawImage(img1, 0.0f, 100.0f);
73
74 // Encode 8888 unpremul
75 info = info.makeAlphaType(kUnpremul_SkAlphaType);
76 read_into_pixmap(&src, info, fStorage.get(), srcImg);
77 img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
78 img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
79 canvas->drawImage(img0, 100.0f, 0.0f);
80 canvas->drawImage(img1, 100.0f, 100.0f);
81
82 // Encode F16 premul
83 info = SkImageInfo::Make(srcImg->width(), srcImg->height(), kRGBA_F16_SkColorType,
84 kPremul_SkAlphaType, SkColorSpace::MakeSRGB());
85 read_into_pixmap(&src, info, fStorage.get(), srcImg);
86 img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
87 img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
88 canvas->drawImage(img0, 200.0f, 0.0f);
89 canvas->drawImage(img1, 200.0f, 100.0f);
90
91 // Encode F16 unpremul
92 info = info.makeAlphaType(kUnpremul_SkAlphaType);
93 read_into_pixmap(&src, info, fStorage.get(), srcImg);
94 img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
95 img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
96 canvas->drawImage(img0, 300.0f, 0.0f);
97 canvas->drawImage(img1, 300.0f, 100.0f);
98
99 return DrawResult::kOk;
100 }
101
102 private:
103 AutoTMalloc<uint8_t> fStorage;
104
105 using INHERITED = GM;
106 };
107
108 DEF_GM( return new EncodeJpegAlphaOptsGM; )
109
110 } // namespace skiagm
111
