112 lines
4.1 KiB
C++
112 lines
4.1 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/gm.h"
|
|
#include "include/core/SkCanvas.h"
|
|
#include "include/core/SkColorSpace.h"
|
|
#include "include/core/SkData.h"
|
|
#include "include/core/SkImage.h"
|
|
#include "include/core/SkImageInfo.h"
|
|
#include "include/core/SkPixmap.h"
|
|
#include "include/core/SkRefCnt.h"
|
|
#include "include/core/SkSize.h"
|
|
#include "include/core/SkStream.h"
|
|
#include "include/core/SkString.h"
|
|
#include "include/core/SkTypes.h"
|
|
#include "include/encode/SkJpegEncoder.h"
|
|
#include "include/private/SkTemplates.h"
|
|
#include "tools/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(nullptr, *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);
|
|
}
|
|
|
|
DrawResult onDraw(SkCanvas* canvas, SkString* errorMsg) override {
|
|
sk_sp<SkImage> srcImg = GetResourceAsImage("images/rainbow-gradient.png");
|
|
if (!srcImg) {
|
|
*errorMsg = "Could not load images/rainbow-gradient.png. "
|
|
"Did you forget to set the resourcePath?";
|
|
return DrawResult::kFail;
|
|
}
|
|
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);
|
|
|
|
// Encode F16 premul
|
|
info = SkImageInfo::Make(srcImg->width(), srcImg->height(), kRGBA_F16_SkColorType,
|
|
kPremul_SkAlphaType, SkColorSpace::MakeSRGB());
|
|
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);
|
|
|
|
return DrawResult::kOk;
|
|
}
|
|
|
|
private:
|
|
SkAutoTMalloc<uint8_t> fStorage;
|
|
|
|
using INHERITED = GM;
|
|
};
|
|
|
|
DEF_GM( return new EncodeJpegAlphaOptsGM; )
|
|
|
|
} // namespace skiagm
|