diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/images/SkImageEncoderFns.h | 38 | ||||
-rw-r--r-- | src/images/SkJpegEncoder.cpp | 48 | ||||
-rw-r--r-- | src/images/SkJpegEncoder.h | 22 |
3 files changed, 96 insertions, 12 deletions
diff --git a/src/images/SkImageEncoderFns.h b/src/images/SkImageEncoderFns.h index 5120570c48..af75ac9fce 100644 --- a/src/images/SkImageEncoderFns.h +++ b/src/images/SkImageEncoderFns.h @@ -16,6 +16,7 @@ #include "SkColor.h" #include "SkColorPriv.h" #include "SkICC.h" +#include "SkOpts.h" #include "SkPreConfig.h" #include "SkRasterPipeline.h" #include "SkUnPreMultiply.h" @@ -165,6 +166,30 @@ static inline void transform_scanline_unpremultiply_sRGB(void* dst, const void* } /** + * Premultiply RGBA to rgbA. + */ +static inline void transform_scanline_to_premul_legacy(char* SK_RESTRICT dst, + const char* SK_RESTRICT src, + int width, int, const SkPMColor*) { + SkOpts::RGBA_to_rgbA((uint32_t*)dst, (const uint32_t*)src, width); +} + +/** + * Premultiply RGBA to rgbA linearly. + */ +static inline void transform_scanline_to_premul_linear(char* SK_RESTRICT dst, + const char* SK_RESTRICT src, + int width, int, const SkPMColor*) { + SkRasterPipeline p; + p.append(SkRasterPipeline::load_8888, (const void**) &src); + p.append_from_srgb(kUnpremul_SkAlphaType); + p.append(SkRasterPipeline::premul); + p.append(SkRasterPipeline::to_srgb); + p.append(SkRasterPipeline::store_8888, (void**) &dst); + p.run(0, width); +} + +/** * Transform from kPremul, kRGBA_8888_SkColorType to 4-bytes-per-pixel unpremultiplied RGBA. */ static inline void transform_scanline_srgbA(char* SK_RESTRICT dst, const char* SK_RESTRICT src, @@ -276,6 +301,19 @@ static inline void transform_scanline_F16_premul_to_8888(char* SK_RESTRICT dst, p.run(0, width); } +/** + * Transform from kUnpremul, kRGBA_F16 to premultiplied rgbA 8888. + */ +static inline void transform_scanline_F16_to_premul_8888(char* SK_RESTRICT dst, + const char* SK_RESTRICT src, int width, int, const SkPMColor*) { + SkRasterPipeline p; + p.append(SkRasterPipeline::load_f16, (const void**) &src); + p.append(SkRasterPipeline::premul); + p.append(SkRasterPipeline::to_srgb); + p.append(SkRasterPipeline::store_8888, (void**) &dst); + p.run(0, width); +} + static inline sk_sp<SkData> icc_from_color_space(const SkColorSpace& cs) { SkColorSpaceTransferFn fn; SkMatrix44 toXYZD50(SkMatrix44::kUninitialized_Constructor); diff --git a/src/images/SkJpegEncoder.cpp b/src/images/SkJpegEncoder.cpp index fd88a5c8f6..d87fed81bc 100644 --- a/src/images/SkJpegEncoder.cpp +++ b/src/images/SkJpegEncoder.cpp @@ -35,7 +35,7 @@ public: return std::unique_ptr<SkJpegEncoderMgr>(new SkJpegEncoderMgr(stream)); } - bool setParams(const SkImageInfo& srcInfo); + bool setParams(const SkImageInfo& srcInfo, const SkJpegEncoder::Options& options); jpeg_compress_struct* cinfo() { return &fCInfo; } @@ -65,15 +65,36 @@ private: transform_scanline_proc fProc; }; -bool SkJpegEncoderMgr::setParams(const SkImageInfo& srcInfo) { +bool SkJpegEncoderMgr::setParams(const SkImageInfo& srcInfo, const SkJpegEncoder::Options& options) +{ + auto chooseProc8888 = [&]() { + if (kUnpremul_SkAlphaType != srcInfo.alphaType() || + SkJpegEncoder::AlphaOption::kIgnore == options.fAlphaOption) + { + return (transform_scanline_proc) nullptr; + } + + // Note that kRespect mode is only supported with sRGB or linear transfer functions. + // The legacy code path is incidentally correct when the transfer function is linear. + const bool isSRGBTransferFn = srcInfo.gammaCloseToSRGB() && + (SkTransferFunctionBehavior::kRespect == options.fBlendBehavior); + if (isSRGBTransferFn) { + return transform_scanline_to_premul_linear; + } else { + return transform_scanline_to_premul_legacy; + } + }; + J_COLOR_SPACE jpegColorType = JCS_EXT_RGBA; int numComponents = 0; switch (srcInfo.colorType()) { case kRGBA_8888_SkColorType: + fProc = chooseProc8888(); jpegColorType = JCS_EXT_RGBA; numComponents = 4; break; case kBGRA_8888_SkColorType: + fProc = chooseProc8888(); jpegColorType = JCS_EXT_BGRA; numComponents = 4; break; @@ -83,11 +104,19 @@ bool SkJpegEncoderMgr::setParams(const SkImageInfo& srcInfo) { numComponents = 3; break; case kARGB_4444_SkColorType: + if (SkJpegEncoder::AlphaOption::kBlendOnBlack == options.fAlphaOption) { + return false; + } + fProc = transform_scanline_444; jpegColorType = JCS_RGB; numComponents = 3; break; case kIndex_8_SkColorType: + if (SkJpegEncoder::AlphaOption::kBlendOnBlack == options.fAlphaOption) { + return false; + } + fProc = transform_scanline_index8_opaque; jpegColorType = JCS_RGB; numComponents = 3; @@ -98,11 +127,18 @@ bool SkJpegEncoderMgr::setParams(const SkImageInfo& srcInfo) { numComponents = 1; break; case kRGBA_F16_SkColorType: - if (!srcInfo.colorSpace() || !srcInfo.colorSpace()->gammaIsLinear()) { + if (!srcInfo.colorSpace() || !srcInfo.colorSpace()->gammaIsLinear() || + SkTransferFunctionBehavior::kRespect != options.fBlendBehavior) { return false; } - fProc = transform_scanline_F16_to_8888; + if (kUnpremul_SkAlphaType != srcInfo.alphaType() || + SkJpegEncoder::AlphaOption::kIgnore == options.fAlphaOption) + { + fProc = transform_scanline_F16_to_8888; + } else { + fProc = transform_scanline_F16_to_premul_8888; + } jpegColorType = JCS_EXT_RGBA; numComponents = 4; break; @@ -125,7 +161,7 @@ bool SkJpegEncoderMgr::setParams(const SkImageInfo& srcInfo) { std::unique_ptr<SkJpegEncoder> SkJpegEncoder::Make(SkWStream* dst, const SkPixmap& src, const Options& options) { - if (!SkPixmapIsValid(src, SkTransferFunctionBehavior::kIgnore)) { + if (!SkPixmapIsValid(src, options.fBlendBehavior)) { return nullptr; } @@ -134,7 +170,7 @@ std::unique_ptr<SkJpegEncoder> SkJpegEncoder::Make(SkWStream* dst, const SkPixma return nullptr; } - if (!encoderMgr->setParams(src.info())) { + if (!encoderMgr->setParams(src.info(), options)) { return nullptr; } diff --git a/src/images/SkJpegEncoder.h b/src/images/SkJpegEncoder.h index 49b49786c9..07290aeb9f 100644 --- a/src/images/SkJpegEncoder.h +++ b/src/images/SkJpegEncoder.h @@ -16,17 +16,27 @@ class SkWStream; class SkJpegEncoder : public SkEncoder { public: - // TODO (skbug.com/1501): - // Since jpegs are always opaque, this encoder ignores the alpha channel and treats the - // pixels as opaque. - // Another possible behavior is to blend the pixels onto opaque black. We'll need to add - // an option for this - and an SkTransferFunctionBehavior. + enum class AlphaOption { + kIgnore, + kBlendOnBlack, + }; struct Options { /** - * |fQuality| must be in [0, 100] where 0 corresponds to the lowest quality. + * |fQuality| must be in [0, 100] where 0 corresponds to the lowest quality. */ int fQuality = 100; + + /** + * Jpegs must be opaque. This instructs the encoder on how to handle input + * images with alpha. + * + * The default is to ignore the alpha channel and treat the image as opaque. + * Another option is to blend the pixels onto a black background before encoding. + * In the second case, the encoder supports linear or legacy blending. + */ + AlphaOption fAlphaOption = AlphaOption::kIgnore; + SkTransferFunctionBehavior fBlendBehavior = SkTransferFunctionBehavior::kRespect; }; /** |