src/images/SkJpegEncoder.cpp - skia - Git at Google

 /*
  * Copyright 2007 The Android Open Source Project
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */

 #include "SkImageEncoderPriv.h"

 #ifdef SK_HAS_JPEG_LIBRARY

 #include "SkColorData.h"
 #include "SkImageEncoderFns.h"
 #include "SkImageInfoPriv.h"
 #include "SkJpegEncoder.h"
 #include "SkJPEGWriteUtility.h"
 #include "SkStream.h"
 #include "SkTemplates.h"

 #include <stdio.h>

 extern "C" {
     #include "jpeglib.h"
     #include "jerror.h"
 }

 class SkJpegEncoderMgr final : SkNoncopyable {
 public:

     /*
      * Create the decode manager
      * Does not take ownership of stream
      */
     static std::unique_ptr<SkJpegEncoderMgr> Make(SkWStream* stream) {
         return std::unique_ptr<SkJpegEncoderMgr>(new SkJpegEncoderMgr(stream));
     }

     bool setParams(const SkImageInfo& srcInfo, const SkJpegEncoder::Options& options);

     jpeg_compress_struct* cinfo() { return &fCInfo; }

     skjpeg_error_mgr* errorMgr() { return &fErrMgr; }

     transform_scanline_proc proc() const { return fProc; }

     ~SkJpegEncoderMgr() {
         jpeg_destroy_compress(&fCInfo);
     }

 private:

     SkJpegEncoderMgr(SkWStream* stream)
         : fDstMgr(stream)
         , fProc(nullptr)
     {
         fCInfo.err = jpeg_std_error(&fErrMgr);
         fErrMgr.error_exit = skjpeg_error_exit;
         jpeg_create_compress(&fCInfo);
         fCInfo.dest = &fDstMgr;
     }

     jpeg_compress_struct    fCInfo;
     skjpeg_error_mgr        fErrMgr;
     skjpeg_destination_mgr  fDstMgr;
     transform_scanline_proc fProc;
 };

 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;
         case kRGB_565_SkColorType:
             fProc = transform_scanline_565;
             jpegColorType = JCS_RGB;
             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 kGray_8_SkColorType:
             SkASSERT(srcInfo.isOpaque());
             jpegColorType = JCS_GRAYSCALE;
             numComponents = 1;
             break;
         case kRGBA_F16_SkColorType:
             if (!srcInfo.colorSpace() || !srcInfo.colorSpace()->gammaIsLinear() ||
                     SkTransferFunctionBehavior::kRespect != options.fBlendBehavior) {
                 return false;
             }

             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;
         default:
             return false;
     }

     fCInfo.image_width = srcInfo.width();
     fCInfo.image_height = srcInfo.height();
     fCInfo.in_color_space = jpegColorType;
     fCInfo.input_components = numComponents;
     jpeg_set_defaults(&fCInfo);

     if (kGray_8_SkColorType != srcInfo.colorType()) {
         switch (options.fDownsample) {
             case SkJpegEncoder::Downsample::k420:
                 SkASSERT(2 == fCInfo.comp_info[0].h_samp_factor);
                 SkASSERT(2 == fCInfo.comp_info[0].v_samp_factor);
                 SkASSERT(1 == fCInfo.comp_info[1].h_samp_factor);
                 SkASSERT(1 == fCInfo.comp_info[1].v_samp_factor);
                 SkASSERT(1 == fCInfo.comp_info[2].h_samp_factor);
                 SkASSERT(1 == fCInfo.comp_info[2].v_samp_factor);
                 break;
             case SkJpegEncoder::Downsample::k422:
                 fCInfo.comp_info[0].h_samp_factor = 2;
                 fCInfo.comp_info[0].v_samp_factor = 1;
                 fCInfo.comp_info[1].h_samp_factor = 1;
                 fCInfo.comp_info[1].v_samp_factor = 1;
                 fCInfo.comp_info[2].h_samp_factor = 1;
                 fCInfo.comp_info[2].v_samp_factor = 1;
                 break;
             case SkJpegEncoder::Downsample::k444:
                 fCInfo.comp_info[0].h_samp_factor = 1;
                 fCInfo.comp_info[0].v_samp_factor = 1;
                 fCInfo.comp_info[1].h_samp_factor = 1;
                 fCInfo.comp_info[1].v_samp_factor = 1;
                 fCInfo.comp_info[2].h_samp_factor = 1;
                 fCInfo.comp_info[2].v_samp_factor = 1;
                 break;
         }
     }

     // Tells libjpeg-turbo to compute optimal Huffman coding tables
     // for the image.  This improves compression at the cost of
     // slower encode performance.
     fCInfo.optimize_coding = TRUE;
     return true;
 }

 std::unique_ptr<SkEncoder> SkJpegEncoder::Make(SkWStream* dst, const SkPixmap& src,
                                                const Options& options) {
     if (!SkPixmapIsValid(src, options.fBlendBehavior)) {
         return nullptr;
     }

     std::unique_ptr<SkJpegEncoderMgr> encoderMgr = SkJpegEncoderMgr::Make(dst);

     skjpeg_error_mgr::AutoPushJmpBuf jmp(encoderMgr->errorMgr());
     if (setjmp(jmp)) {
         return nullptr;
     }

     if (!encoderMgr->setParams(src.info(), options)) {
         return nullptr;
     }

     jpeg_set_quality(encoderMgr->cinfo(), options.fQuality, TRUE);
     jpeg_start_compress(encoderMgr->cinfo(), TRUE);

     sk_sp<SkData> icc = icc_from_color_space(src.info());
     if (icc) {
         // Create a contiguous block of memory with the icc signature followed by the profile.
         sk_sp<SkData> markerData =
                 SkData::MakeUninitialized(kICCMarkerHeaderSize + icc->size());
         uint8_t* ptr = (uint8_t*) markerData->writable_data();
         memcpy(ptr, kICCSig, sizeof(kICCSig));
         ptr += sizeof(kICCSig);
         *ptr++ = 1; // This is the first marker.
         *ptr++ = 1; // Out of one total markers.
         memcpy(ptr, icc->data(), icc->size());

         jpeg_write_marker(encoderMgr->cinfo(), kICCMarker, markerData->bytes(), markerData->size());
     }

     return std::unique_ptr<SkJpegEncoder>(new SkJpegEncoder(std::move(encoderMgr), src));
 }

 SkJpegEncoder::SkJpegEncoder(std::unique_ptr<SkJpegEncoderMgr> encoderMgr, const SkPixmap& src)
     : INHERITED(src, encoderMgr->proc() ? encoderMgr->cinfo()->input_components*src.width() : 0)
     , fEncoderMgr(std::move(encoderMgr))
 {}

 SkJpegEncoder::~SkJpegEncoder() {}

 bool SkJpegEncoder::onEncodeRows(int numRows) {
     skjpeg_error_mgr::AutoPushJmpBuf jmp(fEncoderMgr->errorMgr());
     if (setjmp(jmp)) {
         return false;
     }

     const void* srcRow = fSrc.addr(0, fCurrRow);
     for (int i = 0; i < numRows; i++) {
         JSAMPLE* jpegSrcRow = (JSAMPLE*) srcRow;
         if (fEncoderMgr->proc()) {
             fEncoderMgr->proc()((char*)fStorage.get(), (const char*)srcRow, fSrc.width(),
                                 fEncoderMgr->cinfo()->input_components, nullptr);
             jpegSrcRow = fStorage.get();
         }

         jpeg_write_scanlines(fEncoderMgr->cinfo(), &jpegSrcRow, 1);
         srcRow = SkTAddOffset<const void>(srcRow, fSrc.rowBytes());
     }

     fCurrRow += numRows;
     if (fCurrRow == fSrc.height()) {
         jpeg_finish_compress(fEncoderMgr->cinfo());
     }

     return true;
 }

 bool SkJpegEncoder::Encode(SkWStream* dst, const SkPixmap& src, const Options& options) {
     auto encoder = SkJpegEncoder::Make(dst, src, options);
     return encoder.get() && encoder->encodeRows(src.height());
 }

 #endif
	/*
	* Copyright 2007 The Android Open Source Project
	*
	* Use of this source code is governed by a BSD-style license that can be
	* found in the LICENSE file.
	*/

	#include "SkImageEncoderPriv.h"

	#ifdef SK_HAS_JPEG_LIBRARY

	#include "SkColorData.h"
	#include "SkImageEncoderFns.h"
	#include "SkImageInfoPriv.h"
	#include "SkJpegEncoder.h"
	#include "SkJPEGWriteUtility.h"
	#include "SkStream.h"
	#include "SkTemplates.h"

	#include <stdio.h>

	extern "C" {
	#include "jpeglib.h"
	#include "jerror.h"
	}

	class SkJpegEncoderMgr final : SkNoncopyable {
	public:

	/*
	* Create the decode manager
	* Does not take ownership of stream
	*/
	static std::unique_ptr<SkJpegEncoderMgr> Make(SkWStream* stream) {
	return std::unique_ptr<SkJpegEncoderMgr>(new SkJpegEncoderMgr(stream));
	}

	bool setParams(const SkImageInfo& srcInfo, const SkJpegEncoder::Options& options);

	jpeg_compress_struct* cinfo() { return &fCInfo; }

	skjpeg_error_mgr* errorMgr() { return &fErrMgr; }

	transform_scanline_proc proc() const { return fProc; }

	~SkJpegEncoderMgr() {
	jpeg_destroy_compress(&fCInfo);
	}

	private:

	SkJpegEncoderMgr(SkWStream* stream)
	: fDstMgr(stream)
	, fProc(nullptr)
	{
	fCInfo.err = jpeg_std_error(&fErrMgr);
	fErrMgr.error_exit = skjpeg_error_exit;
	jpeg_create_compress(&fCInfo);
	fCInfo.dest = &fDstMgr;
	}

	jpeg_compress_struct fCInfo;
	skjpeg_error_mgr fErrMgr;
	skjpeg_destination_mgr fDstMgr;
	transform_scanline_proc fProc;
	};

	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;
	case kRGB_565_SkColorType:
	fProc = transform_scanline_565;
	jpegColorType = JCS_RGB;
	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 kGray_8_SkColorType:
	SkASSERT(srcInfo.isOpaque());
	jpegColorType = JCS_GRAYSCALE;
	numComponents = 1;
	break;
	case kRGBA_F16_SkColorType:
	if (!srcInfo.colorSpace() \|\| !srcInfo.colorSpace()->gammaIsLinear() \|\|
	SkTransferFunctionBehavior::kRespect != options.fBlendBehavior) {
	return false;
	}

	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;
	default:
	return false;
	}

	fCInfo.image_width = srcInfo.width();
	fCInfo.image_height = srcInfo.height();
	fCInfo.in_color_space = jpegColorType;
	fCInfo.input_components = numComponents;
	jpeg_set_defaults(&fCInfo);

	if (kGray_8_SkColorType != srcInfo.colorType()) {
	switch (options.fDownsample) {
	case SkJpegEncoder::Downsample::k420:
	SkASSERT(2 == fCInfo.comp_info[0].h_samp_factor);
	SkASSERT(2 == fCInfo.comp_info[0].v_samp_factor);
	SkASSERT(1 == fCInfo.comp_info[1].h_samp_factor);
	SkASSERT(1 == fCInfo.comp_info[1].v_samp_factor);
	SkASSERT(1 == fCInfo.comp_info[2].h_samp_factor);
	SkASSERT(1 == fCInfo.comp_info[2].v_samp_factor);
	break;
	case SkJpegEncoder::Downsample::k422:
	fCInfo.comp_info[0].h_samp_factor = 2;
	fCInfo.comp_info[0].v_samp_factor = 1;
	fCInfo.comp_info[1].h_samp_factor = 1;
	fCInfo.comp_info[1].v_samp_factor = 1;
	fCInfo.comp_info[2].h_samp_factor = 1;
	fCInfo.comp_info[2].v_samp_factor = 1;
	break;
	case SkJpegEncoder::Downsample::k444:
	fCInfo.comp_info[0].h_samp_factor = 1;
	fCInfo.comp_info[0].v_samp_factor = 1;
	fCInfo.comp_info[1].h_samp_factor = 1;
	fCInfo.comp_info[1].v_samp_factor = 1;
	fCInfo.comp_info[2].h_samp_factor = 1;
	fCInfo.comp_info[2].v_samp_factor = 1;
	break;
	}
	}

	// Tells libjpeg-turbo to compute optimal Huffman coding tables
	// for the image. This improves compression at the cost of
	// slower encode performance.
	fCInfo.optimize_coding = TRUE;
	return true;
	}

	std::unique_ptr<SkEncoder> SkJpegEncoder::Make(SkWStream* dst, const SkPixmap& src,
	const Options& options) {
	if (!SkPixmapIsValid(src, options.fBlendBehavior)) {
	return nullptr;
	}

	std::unique_ptr<SkJpegEncoderMgr> encoderMgr = SkJpegEncoderMgr::Make(dst);

	skjpeg_error_mgr::AutoPushJmpBuf jmp(encoderMgr->errorMgr());
	if (setjmp(jmp)) {
	return nullptr;
	}

	if (!encoderMgr->setParams(src.info(), options)) {
	return nullptr;
	}

	jpeg_set_quality(encoderMgr->cinfo(), options.fQuality, TRUE);
	jpeg_start_compress(encoderMgr->cinfo(), TRUE);

	sk_sp<SkData> icc = icc_from_color_space(src.info());
	if (icc) {
	// Create a contiguous block of memory with the icc signature followed by the profile.
	sk_sp<SkData> markerData =
	SkData::MakeUninitialized(kICCMarkerHeaderSize + icc->size());
	uint8_t* ptr = (uint8_t*) markerData->writable_data();
	memcpy(ptr, kICCSig, sizeof(kICCSig));
	ptr += sizeof(kICCSig);
	*ptr++ = 1; // This is the first marker.
	*ptr++ = 1; // Out of one total markers.
	memcpy(ptr, icc->data(), icc->size());

	jpeg_write_marker(encoderMgr->cinfo(), kICCMarker, markerData->bytes(), markerData->size());
	}

	return std::unique_ptr<SkJpegEncoder>(new SkJpegEncoder(std::move(encoderMgr), src));
	}

	SkJpegEncoder::SkJpegEncoder(std::unique_ptr<SkJpegEncoderMgr> encoderMgr, const SkPixmap& src)
	: INHERITED(src, encoderMgr->proc() ? encoderMgr->cinfo()->input_components*src.width() : 0)
	, fEncoderMgr(std::move(encoderMgr))
	{}

	SkJpegEncoder::~SkJpegEncoder() {}

	bool SkJpegEncoder::onEncodeRows(int numRows) {
	skjpeg_error_mgr::AutoPushJmpBuf jmp(fEncoderMgr->errorMgr());
	if (setjmp(jmp)) {
	return false;
	}

	const void* srcRow = fSrc.addr(0, fCurrRow);
	for (int i = 0; i < numRows; i++) {
	JSAMPLE* jpegSrcRow = (JSAMPLE*) srcRow;
	if (fEncoderMgr->proc()) {
	fEncoderMgr->proc()((char)fStorage.get(), (const char)srcRow, fSrc.width(),
	fEncoderMgr->cinfo()->input_components, nullptr);
	jpegSrcRow = fStorage.get();
	}

	jpeg_write_scanlines(fEncoderMgr->cinfo(), &jpegSrcRow, 1);
	srcRow = SkTAddOffset<const void>(srcRow, fSrc.rowBytes());
	}

	fCurrRow += numRows;
	if (fCurrRow == fSrc.height()) {
	jpeg_finish_compress(fEncoderMgr->cinfo());
	}

	return true;
	}

	bool SkJpegEncoder::Encode(SkWStream* dst, const SkPixmap& src, const Options& options) {
	auto encoder = SkJpegEncoder::Make(dst, src, options);
	return encoder.get() && encoder->encodeRows(src.height());
	}

	#endif