d7a5a1d27e
Bug: skia:8731 Change-Id: If73216bd427a1ce773fa41044a45c1bbd7ea08e9 Reviewed-on: https://skia-review.googlesource.com/c/189124 Reviewed-by: Mike Klein <mtklein@google.com> Commit-Queue: Chris Dalton <csmartdalton@google.com>
103 lines
3.7 KiB
C++
103 lines
3.7 KiB
C++
/*
|
|
* Copyright 2017 Google Inc.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#include "gm.h"
|
|
#include "SkImage.h"
|
|
#include "SkImageInfoPriv.h"
|
|
#include "SkJpegEncoder.h"
|
|
|
|
#include "Resources.h"
|
|
|
|
namespace skiagm {
|
|
|
|
static inline void read_into_pixmap(SkPixmap* dst, SkImageInfo dstInfo, void* dstPixels,
|
|
sk_sp<SkImage> src) {
|
|
dst->reset(dstInfo, dstPixels, dstInfo.minRowBytes());
|
|
src->readPixels(*dst, 0, 0, SkImage::CachingHint::kDisallow_CachingHint);
|
|
}
|
|
|
|
static inline sk_sp<SkImage> encode_pixmap_and_make_image(const SkPixmap& src,
|
|
SkJpegEncoder::AlphaOption alphaOption) {
|
|
SkDynamicMemoryWStream dst;
|
|
SkJpegEncoder::Options options;
|
|
options.fAlphaOption = alphaOption;
|
|
SkJpegEncoder::Encode(&dst, src, options);
|
|
return SkImage::MakeFromEncoded(dst.detachAsData());
|
|
}
|
|
|
|
class EncodeJpegAlphaOptsGM : public GM {
|
|
public:
|
|
EncodeJpegAlphaOptsGM() {}
|
|
|
|
protected:
|
|
SkString onShortName() override {
|
|
return SkString("encode-alpha-jpeg");
|
|
}
|
|
|
|
SkISize onISize() override {
|
|
return SkISize::Make(400, 200);
|
|
}
|
|
|
|
void onDraw(SkCanvas* canvas) override {
|
|
sk_sp<SkImage> srcImg = GetResourceAsImage("images/rainbow-gradient.png");
|
|
if (!srcImg) {
|
|
DrawFailureMessage(canvas, "Could not load images/rainbow-gradient.png. "
|
|
"Did you forget to set the resourcePath?");
|
|
return;
|
|
}
|
|
fStorage.reset(srcImg->width() * srcImg->height() *
|
|
SkColorTypeBytesPerPixel(kRGBA_F16_SkColorType));
|
|
|
|
SkPixmap src;
|
|
SkImageInfo info = SkImageInfo::MakeN32Premul(srcImg->width(), srcImg->height(),
|
|
canvas->imageInfo().colorSpace() ? SkColorSpace::MakeSRGB() : nullptr);
|
|
read_into_pixmap(&src, info, fStorage.get(), srcImg);
|
|
|
|
// Encode 8888 premul.
|
|
auto img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
|
|
auto img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
|
|
canvas->drawImage(img0, 0.0f, 0.0f);
|
|
canvas->drawImage(img1, 0.0f, 100.0f);
|
|
|
|
// Encode 8888 unpremul
|
|
info = info.makeAlphaType(kUnpremul_SkAlphaType);
|
|
read_into_pixmap(&src, info, fStorage.get(), srcImg);
|
|
img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
|
|
img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
|
|
canvas->drawImage(img0, 100.0f, 0.0f);
|
|
canvas->drawImage(img1, 100.0f, 100.0f);
|
|
|
|
if (canvas->imageInfo().colorSpace()) {
|
|
// Encode F16 premul
|
|
info = SkImageInfo::Make(srcImg->width(), srcImg->height(), kRGBA_F16_SkColorType,
|
|
kPremul_SkAlphaType, SkColorSpace::MakeSRGBLinear());
|
|
read_into_pixmap(&src, info, fStorage.get(), srcImg);
|
|
img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
|
|
img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
|
|
canvas->drawImage(img0, 200.0f, 0.0f);
|
|
canvas->drawImage(img1, 200.0f, 100.0f);
|
|
|
|
// Encode F16 unpremul
|
|
info = info.makeAlphaType(kUnpremul_SkAlphaType);
|
|
read_into_pixmap(&src, info, fStorage.get(), srcImg);
|
|
img0 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kIgnore);
|
|
img1 = encode_pixmap_and_make_image(src, SkJpegEncoder::AlphaOption::kBlendOnBlack);
|
|
canvas->drawImage(img0, 300.0f, 0.0f);
|
|
canvas->drawImage(img1, 300.0f, 100.0f);
|
|
}
|
|
}
|
|
|
|
private:
|
|
SkAutoTMalloc<uint8_t> fStorage;
|
|
|
|
typedef GM INHERITED;
|
|
};
|
|
|
|
DEF_GM( return new EncodeJpegAlphaOptsGM; )
|
|
|
|
};
|